我要做的是基本上使用线程模拟客户端 - 服务器 - 客户端聊天。这是迄今为止的代码:
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
答案 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()