prosody-hg: net/server_epoll.lua comparison

server_epoll: Add native support for per socket bandwith limits

comparison

equal deleted inserted replaced

-:1f45d316b222
+:dda1c7ccb209
 			self:destroy()
 			return;
 		end
 	end
 	if not self.conn then return; end
+	if self._limit and (data or partial) then
+		local cost = self._limit * #(data or partial);
+		if cost > cfg.min_wait then
+			self:setreadtimeout(false);
+			self:pausefor(cost);
+			return;
+		end
+	end
 	if self._wantread and self.conn:dirty() then
 		self:setreadtimeout(false);
 		self:pausefor(cfg.read_retry_delay);
 	else
 		self:setreadtimeout();
 	return self:set(true);
 end
 -- Pause connection for some time
 function interface:pausefor(t)
+	self:debug("Pause for %fs", t);
 	if self._pausefor then
 		self._pausefor:close();
 	end
 	if t == false then return; end
 	self:set(false);
 		self:set(true);
 		if self.conn and self.conn:dirty() then
 			self:onreadable();
 		end
 	end);
+end
+function interface:setlimit(Bps)
+	if Bps > 0 then
+		self._limit = 1/Bps;
+	else
+		self._limit = nil;
+	end
 end
 function interface:pause_writes()
 	self._write_lock = true;
 	self:setwritetimeout(false);

Mercurial > prosody-hg