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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
|
"""TCP Server."""
import log
import select
import socket
import threading
import time
from json_package import JSONPackage
from json_package import JSONPackageError
from request_handler import RequestHandler
FREQUENCY = 8
TIMEOUT = 1
class TCPServer(threading.Thread):
"""A thread to be the tcp server.
Attributes:
_port: Port number.
_sock: Socket fd.
_users_text_manager: An instance of UsersTextManager.
_stop_flag: Flag for stopping.
_connection_handler_threads: List of connction handler threads.
"""
def __init__(self, port, users_text_manager):
"""Constructor.
Args:
port: Port number.
users_text_manager: An instance of UsersTextManager.
"""
super(TCPServer, self).__init__()
self._port = port
self._sock = None
self._users_text_manager = users_text_manager
self._stop_flag = False
self._connection_handler_threads = []
@property
def port(self):
"""Gets the port of this server. None for unconnected case."""
return self._port if self._sock else None
def run(self):
"""Runs the thread."""
self._build()
self._accept()
def stop(self):
"""Stops the thread."""
self._stop_flag = True
for thr in self._connection_handler_threads:
thr.stop()
thr.join()
def _build(self):
"""Creates the socket."""
timeout = 1
while not self._stop_flag and not self._sock:
try:
self._sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self._sock.bind(('', self._port))
self._sock.listen(1024)
except socket.error as e:
self._sock = None
log.error(str(e) + '\n')
log.info('Try it %d second(s) later.\n' % timeout)
for _ in range(timeout * FREQUENCY):
if self._stop_flag:
break
time.sleep(float(1) / FREQUENCY)
timeout *= 2
if self._sock:
log.info('Successfully built the tcp server.\n')
def _accept(self):
"""Accepts the connection and calls the handler."""
while not self._stop_flag:
readable, _, _ = select.select([self._sock], [], [],
float(1) / FREQUENCY)
if readable:
sock, addr = self._sock.accept()
log.info('Client %r connect to server.\n' % str(addr))
thr = _TCPConnectionHandler(sock, self._users_text_manager)
thr.start()
self._connection_handler_threads += [thr]
class _TCPConnectionHandler(threading.Thread):
"""A thread to handle a connection.
Attributes:
_sock: The connection socket.
_users_text_manager: An instance of UsersTextManager.
_stop_flag: Stopping flag.
"""
def __init__(self, conn, users_text_manager):
"""Constructor.
Args:
conn: The connection.
users_text_manager: An instance of UsersTextManager.
"""
super(_TCPConnectionHandler, self).__init__()
self._conn = TCPConnection(conn)
self._users_text_manager = users_text_manager
self._stop_flag = False
self._request_handler = RequestHandler(self._users_text_manager)
def run(self):
"""Runs the thread."""
try:
while not self._stop_flag:
try:
request = JSONPackage(recv_func=self._conn.recv_all).content
response = self._request_handler.handle(request)
JSONPackage(response).send(self._conn.send_all)
except JSONPackageError as e:
log.error(str(e))
except socket.error as e:
log.error(str(e))
self._conn.close()
def stop(self):
"""Stops the thread."""
self._stop_flag = True
self._conn.stop()
class TCPConnection(object):
"""My custom tcp connection.
Args:
_conn: The TCP-connection.
_stop_flag: Stopping flag.
"""
def __init__(self, conn):
"""Constructor.
Args:
conn: TCP-connection.
"""
self._conn = conn
self._conn.settimeout(TIMEOUT)
self._stop_flag = False
def send_all(self, data):
"""Sends the data until timeout or the socket closed.
Args:
data: Data to be sent.
"""
recvd_byte, total_byte = 0, len(data)
while recvd_byte < total_byte and not self._stop_flag:
try:
recvd_byte += self._conn.send(data[recvd_byte : ])
except socket.timeout:
continue
def recv_all(self, nbyte):
"""Receives the data until timeout or the socket closed.
Args:
nbyte: Bytes of data to receive.
Return:
Bytes of data.
"""
ret = b''
while nbyte > 0 and not self._stop_flag:
try:
recv = self._conn.recv(nbyte)
except socket.timeout:
continue
if not recv:
raise socket.error('Connection die.')
ret += recv
nbyte -= len(recv)
return ret
def close(self):
"""Closes the connection."""
self._conn.close()
def stop(self):
"""Stops."""
self._stop_flag = True
|