Быстрее сокетов в Python
У меня есть клиент, написанный на Python для сервера, который работает через локальную сеть. Некоторая часть алгоритма интенсивно использует чтение сокетов и работает примерно в 3-6 раз медленнее, чемпочти такой же написанный на C ++. Какие существуют решения для ускорения чтения сокетов Python?
У меня реализована простая буферизация, и мой класс для работы с сокетами выглядит так:
import socket
import struct
class Sock():
def __init__(self):
self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.recv_buf = b''
self.send_buf = b''
def connect(self):
self.s.connect(('127.0.0.1', 6666))
def close(self):
self.s.close()
def recv(self, lngth):
while len(self.recv_buf) < lngth:
self.recv_buf += self.s.recv(lngth - len(self.recv_buf))
res = self.recv_buf[-lngth:]
self.recv_buf = self.recv_buf[:-lngth]
return res
def next_int(self):
return struct.unpack("i", self.recv(4))[0]
def next_float(self):
return struct.unpack("f", self.recv(4))[0]
def write_int(self, i):
self.send_buf += struct.pack('i', i)
def write_float(self, f):
self.send_buf += struct.pack('f', f)
def flush(self):
self.s.sendall(self.send_buf)
self.send_buf = b''
П.С .: Профилирование также показывает, что большую часть времени проводит чтение сокетов.
Edit: Поскольку данные принимаются в блоках с известным размером, я могу прочитать весь блок сразу. Поэтому я изменил свой код следующим образом:
class Sock():
def __init__(self):
self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.send_buf = b''
def connect(self):
self.s.connect(('127.0.0.1', 6666))
def close(self):
self.s.close()
def recv_prepare(self, cnt):
self.recv_buf = bytearray()
while len(self.recv_buf) < cnt:
self.recv_buf.extend(self.s.recv(cnt - len(self.recv_buf)))
self.recv_buf_i = 0
def skip_read(self, cnt):
self.recv_buf_i += cnt
def next_int(self):
self.recv_buf_i += 4
return struct.unpack("i", self.recv_buf[self.recv_buf_i - 4:self.recv_buf_i])[0]
def next_float(self):
self.recv_buf_i += 4
return struct.unpack("f", self.recv_buf[self.recv_buf_i - 4:self.recv_buf_i])[0]
def write_int(self, i):
self.send_buf += struct.pack('i', i)
def write_float(self, f):
self.send_buf += struct.pack('f', f)
def flush(self):
self.s.sendall(self.send_buf)
self.send_buf = b''
recv
в этом коде выглядит оптимально. Но сейчасnext_int
а такжеnext_float
стало вторым узким местом, для распаковки им требуется около 1 мс (3000 циклов ЦП) на вызов. Можно ли сделать их быстрее, как в C ++?