Многопоточный TCP-сервер в Python

Я создал простой многопоточный TCP-сервер, используя модуль threding Python. Этот сервер создает новый поток каждый раз, когда подключается новый клиент.

#!/usr/bin/env python

import socket, threading

class ClientThread(threading.Thread):

    def __init__(self,ip,port):
        threading.Thread.__init__(self)
        self.ip = ip
        self.port = port
        print "[+] New thread started for "+ip+":"+str(port)


    def run(self):    
        print "Connection from : "+ip+":"+str(port)

        clientsock.send("\nWelcome to the server\n\n")

        data = "dummydata"

        while len(data):
            data = clientsock.recv(2048)
            print "Client sent : "+data
            clientsock.send("You sent me : "+data)

        print "Client disconnected..."

host = "0.0.0.0"
port = 9999

tcpsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
tcpsock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

tcpsock.bind((host,port))
threads = []


while True:
    tcpsock.listen(4)
    print "\nListening for incoming connections..."
    (clientsock, (ip, port)) = tcpsock.accept()
    newthread = ClientThread(ip, port)
    newthread.start()
    threads.append(newthread)

for t in threads:
    t.join()

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

Есть ли способ, кроме использования массива с ограниченным количеством переменных сокета и использования каждой переменной для каждого соединения?


python sockets tcpserver
person Deepal    schedule 03.07.2013    source источник
comment
Не волнуйтесь, это хороший вопрос. Добро пожаловать в StackOverflow!   -  person Jordan    schedule 03.07.2013
comment
Кажется, что вы никогда не дойдете до последних двух строк. Это намеренно?   -  person Gijs van Oort    schedule 08.12.2014
comment
@deepal, ты нашел решение этой проблемы?   -  person rajat maan    schedule 28.02.2020

Ответы (3)


arrow_upward
16
arrow_downward

Вы должны передать клиентский носок в поток, как вы это делаете с IP-адресом и портом:

class ClientThread(threading.Thread):

    def __init__(self, ip, port, socket):
        threading.Thread.__init__(self)
        self.ip = ip
        self.port = port
        self.socket = socket
        print "[+] New thread started for "+ip+":"+str(port)

    def run(self):
        # use self.socket to send/receive

...
(clientsock, (ip, port)) = tcpsock.accept()
newthread = ClientThread(ip, port, clientsock)
...
person mata    schedule 03.07.2013
comment
Я использую питон 2.7. Я внес рекомендуемые изменения, но после подключения второго клиента netcat первый клиент зависает. Я не совсем понимаю, что происходит. У кого-нибудь еще есть эта проблема? - person John; 25.08.2016

arrow_upward
4
arrow_downward

Я создал этот хороший класс, который вы можете переопределить

import socket
import thread

class SocketServer(socket.socket):
    clients = []

    def __init__(self):
        socket.socket.__init__(self)
        #To silence- address occupied!!
        self.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.bind(('0.0.0.0', 8080))
        self.listen(5)

    def run(self):
        print "Server started"
        try:
            self.accept_clients()
        except Exception as ex:
            print ex
        finally:
            print "Server closed"
            for client in self.clients:
                client.close()
            self.close()

    def accept_clients(self):
        while 1:
            (clientsocket, address) = self.accept()
            #Adding client to clients list
            self.clients.append(clientsocket)
            #Client Connected
            self.onopen(clientsocket)
            #Receiving data from client
            thread.start_new_thread(self.recieve, (clientsocket,))

    def recieve(self, client):
        while 1:
            data = client.recv(1024)
            if data == '':
                break
            #Message Received
            self.onmessage(client, data)
        #Removing client from clients list
        self.clients.remove(client)
        #Client Disconnected
        self.onclose(client)
        #Closing connection with client
        client.close()
        #Closing thread
        thread.exit()
        print self.clients

    def broadcast(self, message):
        #Sending message to all clients
        for client in self.clients:
            client.send(message)

    def onopen(self, client):
        pass

    def onmessage(self, client, message):
        pass

    def onclose(self, client):
        pass

И вот пример:

class BasicChatServer(SocketServer):

    def __init__(self):
        SocketServer.__init__(self)

    def onmessage(self, client, message):
        print "Client Sent Message"
        #Sending message to all clients
        self.broadcast(message)

    def onopen(self, client):
        print "Client Connected"

    def onclose(self, client):
        print "Client Disconnected"

def main():
    server = BasicChatServer()
    server.run()

if __name__ == "__main__":
    main()
person arbel03    schedule 12.05.2017

arrow_upward
1
arrow_downward

В Python3 есть ThreadingTCPServer (и другие), который может сделать это за вас.

Обратите внимание, что потоковая обработка здесь выполняется для каждого запроса, а не для клиента. Недостаток этого заключается в малом времени завершения потока для каждого запроса, но преимущество в том, что он лучше масштабируется, поскольку увеличивается количество клиентов.

person c z    schedule 02.03.2021