You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

44 lines
1.3 KiB

"""Low level HTTP server."""
import asyncio
from .web_protocol import RequestHandler
from .web_request import BaseRequest
__all__ = ('Server',)
class Server:
def __init__(self, handler, *, request_factory=None, loop=None, **kwargs):
if loop is None:
loop = asyncio.get_event_loop()
self._loop = loop
self._connections = {}
self._kwargs = kwargs
self.requests_count = 0
self.request_handler = handler
self.request_factory = request_factory or self._make_request
@property
def connections(self):
return list(self._connections.keys())
def connection_made(self, handler, transport):
self._connections[handler] = transport
def connection_lost(self, handler, exc=None):
if handler in self._connections:
del self._connections[handler]
def _make_request(self, message, payload, protocol, writer, task):
return BaseRequest(
message, payload, protocol, writer, task, self._loop)
async def shutdown(self, timeout=None):
coros = [conn.shutdown(timeout) for conn in self._connections]
await asyncio.gather(*coros, loop=self._loop)
self._connections.clear()
def __call__(self):
return RequestHandler(self, loop=self._loop, **self._kwargs)