1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
# -*- coding: utf-8 -
#
# This file is part of restkit released under the MIT license.
# See the NOTICE for more information.
import logging
import random
import select
import socket
import ssl
import time
from socketpool import Connector
from socketpool.util import is_connected
CHUNK_SIZE = 16 * 1024
MAX_BODY = 1024 * 112
DNS_TIMEOUT = 60
class Connection(Connector):
def __init__(self, host, port, backend_mod=None, pool=None,
is_ssl=False, extra_headers=[], **ssl_args):
# connect the socket, if we are using an SSL connection, we wrap
# the socket.
self._s = backend_mod.Socket(socket.AF_INET, socket.SOCK_STREAM)
self._s.connect((host, port))
if is_ssl:
self._s = ssl.wrap_socket(self._s, **ssl_args)
self.extra_headers = extra_headers
self.is_ssl = is_ssl
self.backend_mod = backend_mod
self.host = host
self.port = port
self._connected = True
self._life = time.time() - random.randint(0, 10)
self._pool = pool
self._released = False
def matches(self, **match_options):
target_host = match_options.get('host')
target_port = match_options.get('port')
return target_host == self.host and target_port == self.port
def is_connected(self):
if self._connected:
return is_connected(self._s)
return False
def handle_exception(self, exception):
raise
def get_lifetime(self):
return self._life
def invalidate(self):
self.close()
self._connected = False
self._life = -1
def release(self, should_close=False):
if self._pool is not None:
if self._connected:
if should_close:
self.invalidate()
self._pool.release_connection(self)
else:
self._pool = None
elif self._connected:
self.invalidate()
def close(self):
if not self._s or not hasattr(self._s, "close"):
return
try:
self._s.close()
except:
pass
def socket(self):
return self._s
def send_chunk(self, data):
chunk = "".join(("%X\r\n" % len(data), data, "\r\n"))
self._s.sendall(chunk)
def send(self, data, chunked=False):
if chunked:
return self.send_chunk(data)
return self._s.sendall(data)
def sendlines(self, lines, chunked=False):
for line in list(lines):
self.send(line, chunked=chunked)
# TODO: add support for sendfile api
def sendfile(self, data, chunked=False):
""" send a data from a FileObject """
if hasattr(data, 'seek'):
data.seek(0)
while True:
binarydata = data.read(CHUNK_SIZE)
if binarydata == '':
break
self.send(binarydata, chunked=chunked)
def recv(self, size=1024):
return self._s.recv(size)
|
