| import six |
| import requests.adapters |
| |
| from .. import constants |
| from .npipesocket import NpipeSocket |
| |
| if six.PY3: |
| import http.client as httplib |
| else: |
| import httplib |
| |
| try: |
| import requests.packages.urllib3 as urllib3 |
| except ImportError: |
| import urllib3 |
| |
| RecentlyUsedContainer = urllib3._collections.RecentlyUsedContainer |
| |
| |
| class NpipeHTTPConnection(httplib.HTTPConnection, object): |
| def __init__(self, npipe_path, timeout=60): |
| super(NpipeHTTPConnection, self).__init__( |
| 'localhost', timeout=timeout |
| ) |
| self.npipe_path = npipe_path |
| self.timeout = timeout |
| |
| def connect(self): |
| sock = NpipeSocket() |
| sock.settimeout(self.timeout) |
| sock.connect(self.npipe_path) |
| self.sock = sock |
| |
| |
| class NpipeHTTPConnectionPool(urllib3.connectionpool.HTTPConnectionPool): |
| def __init__(self, npipe_path, timeout=60, maxsize=10): |
| super(NpipeHTTPConnectionPool, self).__init__( |
| 'localhost', timeout=timeout, maxsize=maxsize |
| ) |
| self.npipe_path = npipe_path |
| self.timeout = timeout |
| |
| def _new_conn(self): |
| return NpipeHTTPConnection( |
| self.npipe_path, self.timeout |
| ) |
| |
| # When re-using connections, urllib3 tries to call select() on our |
| # NpipeSocket instance, causing a crash. To circumvent this, we override |
| # _get_conn, where that check happens. |
| def _get_conn(self, timeout): |
| conn = None |
| try: |
| conn = self.pool.get(block=self.block, timeout=timeout) |
| |
| except AttributeError: # self.pool is None |
| raise urllib3.exceptions.ClosedPoolError(self, "Pool is closed.") |
| |
| except six.moves.queue.Empty: |
| if self.block: |
| raise urllib3.exceptions.EmptyPoolError( |
| self, |
| "Pool reached maximum size and no more " |
| "connections are allowed." |
| ) |
| pass # Oh well, we'll create a new connection then |
| |
| return conn or self._new_conn() |
| |
| |
| class NpipeAdapter(requests.adapters.HTTPAdapter): |
| def __init__(self, base_url, timeout=60, |
| num_pools=constants.DEFAULT_NUM_POOLS): |
| self.npipe_path = base_url.replace('npipe://', '') |
| self.timeout = timeout |
| self.pools = RecentlyUsedContainer( |
| num_pools, dispose_func=lambda p: p.close() |
| ) |
| super(NpipeAdapter, self).__init__() |
| |
| def get_connection(self, url, proxies=None): |
| with self.pools.lock: |
| pool = self.pools.get(url) |
| if pool: |
| return pool |
| |
| pool = NpipeHTTPConnectionPool( |
| self.npipe_path, self.timeout |
| ) |
| self.pools[url] = pool |
| |
| return pool |
| |
| def request_url(self, request, proxies): |
| # The select_proxy utility in requests errors out when the provided URL |
| # doesn't have a hostname, like is the case when using a UNIX socket. |
| # Since proxies are an irrelevant notion in the case of UNIX sockets |
| # anyway, we simply return the path URL directly. |
| # See also: https://github.com/docker/docker-py/issues/811 |
| return request.path_url |
| |
| def close(self): |
| self.pools.clear() |
