Python中的聊天模拟器无法按预期工作

时间:2014-01-10 10:08:17

标签: python multithreading sockets chat sendto

我要做的是基本上使用线程模拟客户端 - 服务器 - 客户端聊天。这是迄今为止的代码:

from socket import *
from threading import Thread
import threading
import time
from random import randint

clients = []
HOST = 'localhost'
PORT = 8000

class Server():
    def __init__(self):        
        self.addr = (HOST,PORT)
        global clients

        self.start()
        for i in range(5): Thread(target=self.clientHandler).start()

        self.s.close()

    def clientHandler(self): 
        conn, addr = self.s.accept() 
        clients.append(addr)

        print addr[1], "is Connected" 
        while 1: 
            time.sleep(5)

            # message=repr(data)
            # message=message[1:-1].split(':')
            message='test'                               #for debug
            conn.sendto(message, clients[1])             #for debug
            # conn.sendto(message[1], clients[int(message[0])])

    def start(self):    
        self.s = socket(AF_INET, SOCK_STREAM)
        self.s.bind((HOST, PORT))
        self.s.listen(5)

        print "Server is running......"


class Client(threading.Thread):
    global clients
    def sendMessage(self):
        if len(clients)>1:
                to=randint(0, len(clients)-1)
                message = str(to)+':hello from '+self.name
                print message
                self.ss.send(message)

    def receiveMessage(self):        
        while 1:
            reply=self.ss.recv(1024)
            if reply != '':
                print self.name+" received ", repr(reply)
                break

    def run(self):    
        self.ss = socket(AF_INET, SOCK_STREAM)
        self.ss.connect((HOST, PORT)) # client-side, connects to a host
        self.name=self.ss.getsockname()[1]

        while True: 
            # time.sleep(randint(1,5))
            # self.sendMessage()
            self.receiveMessage()

server=Server()
client1=Client()
client2=Client()
client3=Client()

client1.start()
client2.start()
client3.start()

这个想法是一个客户端应该反复向另一个随机发送消息(尚未考虑排除自己)。为此,邮件的格式为dest: message,其中dest是从clients列表中选择客户端的随机索引。

经过几个小时的“调试”后,我发现sendto()方法将消息发送给所有客户端,而不仅仅是具有指定地址的客户端(如上面的代码那样)。我使用它错了吗?

此外,当我让客户端发送消息时,他们只是立即收到消息。我做错了什么?

我知道它很乱,对此感到抱歉。我刚开始使用Python,我正在测试多种方法,看看它们是如何工作的。

谢谢。 - Python Noob

1 个答案:

答案 0 :(得分:1)

您的代码有效,因为TCP套接字忽略了sendto的第二个参数。

为了让你的想法发挥作用,你必须真正阅读服务器上的内容。但首先,让我们看看其他明显的问题:

您正在服务器上创建5个线程。这不是有害的,但可能在将来。相反,只需在需要时创建一个线程,如

class Server(threading.Thread):
    def __init__(self):
        super(Server, self).__init__()
        self.addr = (HOST,PORT)
        self.start()

    def run(self):
        self.s = socket(AF_INET, SOCK_STREAM)
        self.s.bind((HOST, PORT))
        self.s.listen(5)
        print "Server is running......"

        while True:
            conn, addr = self.s.accept()
            threading.Thread(target=self.clientHandler, args=(conn, addr)).start()

然后,不要存储地址,存储套接字:

def clientHandler(self, conn, addr):
    clients.append(conn)

此外,在clientHandler中,从套接字中读取:

def clientHandler(self, conn, addr):
    clients.append(conn)
    while True:
         msg = read_from_socket(conn)
         id_str, _, msg = msg.partition(b':')
         clients[int(id_str.decode('ascii'))].sendall(msg)

read_from_socket必须是一个读取您的邮件格式的函数。由于TCP是无包的(将其视为一个无限流,直到远程端关闭),您必须定义消息格式。我们现在就这样做,等待换行符b'\n'

def read_from_socket(conn):
    buf = bytearray(0)
    while True:
        b = conn.recv(1)
        buf.extend(b)
        if b == b'\n':
             return bytes(buf)

然后我们只需要调整客户端以添加\n字节:

class Client(threading.Thread):
    def sendMessage(self):
        if len(clients)>1:
                to = randint(0, len(clients) - 1)
                message = ('%s:hello from %s\n' % (to, self.name)).encode('utf-8')
                print(message)
                self.ss.send(message)

最后,确保实际发送一条消息(此刻,你没有):

server=Server()
client1=Client()
client2=Client()
client3=Client()

client1.start()
client2.start()
client3.start()

time.sleep(1)
client1.sendMessage()

Here is the entire code