Как сделать простой многопоточный сервер сокетов на Python, который запоминает клиентов

Как создать простой эхо-сервер Python, который запоминает клиентов и не создает новый сокет для каждого запроса? Должна быть возможность поддерживать одновременный доступ. Я хочу иметь возможность подключиться один раз и постоянно отправлять и получать данные с помощью этого клиента или аналогичного:

import socket

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
host = raw_input("Server hostname or ip? ")
port = input("Server port? ")
sock.connect((host,port))
while True:
    data = raw_input("message: ")
    sock.send(data)
    print "response: ", sock.recv(1024)

т.е. с сервером, работающим на порту 50000, используя вышеуказанный клиент, я хочу сделать это:

me@mine:~$ client.py
Server hostname or ip? localhost
Server Port? 50000
message: testa
response: testa
message: testb
response: testb
message: testc
response: testc

person nettux    schedule 23.05.2014    source источник


Ответы (1)


Вы можете использовать поток для каждого клиента, чтобы избежать блокировки client.recv(), а затем использовать основной поток только для прослушивания новых клиентов. Когда кто-то подключается, основной поток создает новый поток, который просто слушает нового клиента и завершается, когда он не разговаривает в течение 60 секунд.

import socket
import threading

class ThreadedServer(object):
    def __init__(self, host, port):
        self.host = host
        self.port = port
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.sock.bind((self.host, self.port))

    def listen(self):
        self.sock.listen(5)
        while True:
            client, address = self.sock.accept()
            client.settimeout(60)
            threading.Thread(target = self.listenToClient,args = (client,address)).start()

    def listenToClient(self, client, address):
        size = 1024
        while True:
            try:
                data = client.recv(size)
                if data:
                    # Set the response to echo back the recieved data 
                    response = data
                    client.send(response)
                else:
                    raise error('Client disconnected')
            except:
                client.close()
                return False

if __name__ == "__main__":
    while True:
        port_num = input("Port? ")
        try:
            port_num = int(port_num)
            break
        except ValueError:
            pass

    ThreadedServer('',port_num).listen()

Время ожидания клиентов истекает через 60 секунд бездействия, и они должны повторно подключиться. См. строку client.settimeout(60) в функции ThreadedServer.listen()

person nettux    schedule 23.05.2014
comment
Я нашел этот интересный короткий вопрос, отвечает ли строка self.sock.listen(5) за количество разрешенных подключений? Так что в принципе я могу принять 5 клиентов одновременно, но не больше, или для чего нужны 5? - person Kev1n91; 14.04.2017
comment
@ Kev1n91 Kev1n91 5 — это аргумент невыполненной работы, который указывает, сколько соединений может быть поставлено в очередь в ожидании принятия. Объяснено в документации здесь: docs.python.org/2/ библиотека/socket.html#socket.socket.listen - person nettux; 14.04.2017
comment
@cascading-style Я написал это на Python 2.7, и я предполагаю, что вы используете Python 3? input возвращает строку в python 3 — просто сделайте port_num int: port_num = int(port_num) - person nettux; 03.05.2017