| 1234567891011121314151617181920212223242526272829303132333435363738 |
- """Helper methods to tune a TCP connection"""
- import asyncio
- import socket
- from contextlib import suppress
- from typing import Optional # noqa
- __all__ = ("tcp_keepalive", "tcp_nodelay")
- if hasattr(socket, "SO_KEEPALIVE"):
- def tcp_keepalive(transport: asyncio.Transport) -> None:
- sock = transport.get_extra_info("socket")
- if sock is not None:
- sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
- else:
- def tcp_keepalive(transport: asyncio.Transport) -> None: # pragma: no cover
- pass
- def tcp_nodelay(transport: asyncio.Transport, value: bool) -> None:
- sock = transport.get_extra_info("socket")
- if sock is None:
- return
- if sock.family not in (socket.AF_INET, socket.AF_INET6):
- return
- value = bool(value)
- # socket may be closed already, on windows OSError get raised
- with suppress(OSError):
- sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, value)
|
