Question

我在GitHub上找到了一个非常简单的开源project：同一Python脚本可以用作服务器或客户端。

import socket
import threading
import queue
import sys
import random
import os


#Client Code
def ReceiveData(sock):
    while True:
        try:
            data,addr = sock.recvfrom(1024)
            print(data.decode('utf-8'))
        except:
            pass

def RunClient(serverIP):
    host = socket.gethostbyname(socket.gethostname())
    port = random.randint(6000,10000)
    print('Client IP->'+str(host)+' Port->'+str(port))
    server = (str(serverIP),5000)
    s = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    s.bind((host,port))

    name = input('Please write your name here: ')
    if name == '':
        name = 'Guest'+str(random.randint(1000,9999))
        print('Your name is:'+name)
    s.sendto(name.encode('utf-8'),server)
    threading.Thread(target=ReceiveData,args=(s,)).start()
    while True:
        data = input()
        if data == 'qqq':
            break
        elif data=='':
            continue
        data = '['+name+']' + '->'+ data
        s.sendto(data.encode('utf-8'),server)
    s.sendto(data.encode('utf-8'),server)
    s.close()
    os._exit(1)
#Client Code Ends Here


#Server Code
def RecvData(sock,recvPackets):
    while True:
        data,addr = sock.recvfrom(1024)
        recvPackets.put((data,addr))

def RunServer():
    host = socket.gethostbyname(socket.gethostname())
    port = 5000
    print('Server hosting on IP-> '+str(host))
    s = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    s.bind((host,port))
    clients = set()
    recvPackets = queue.Queue()

    print('Server Running...')

    threading.Thread(target=RecvData,args=(s,recvPackets)).start()

    while True:
        while not recvPackets.empty():
            data,addr = recvPackets.get()
            if addr not in clients:
                clients.add(addr)
                continue
            clients.add(addr)
            data = data.decode('utf-8')
            if data.endswith('qqq'):
                clients.remove(addr)
                continue
            print(str(addr)+data)
            for c in clients:
                if c!=addr:
                    s.sendto(data.encode('utf-8'),c)

    s.close()
#Serevr Code Ends Here

if __name__ == '__main__':
    if len(sys.argv)==1:
        RunServer()
    elif len(sys.argv)==2:
        RunClient(sys.argv[1])
    else:
        print('Run Serevr:-> python Chat.py')
        print('Run Client:-> python Chat.py <ServerIP>')

您可以从图片中看到它确实运行良好：

但是我的目标是使服务器也加入聊天。

UDP聊天室在作为服务器工作时是否也可以让服务器充当客户端？

Answer 1

要让服务器也运行客户端，可以在代码的RunServer()部分中对__main__的调用之前插入以下行：

    threading.Thread(target=RunClient,args=('127.0.0.1',)).start()

这将启动与服务器的事件循环并行运行的RunClient('127.0.0.1')线程。

但是，在正常运行之前，您还需要修复代码的其他一些问题。特别是：

在RunServer()的顶部，将host设置为服务器计算机上网络接口卡之一的IP地址。这意味着服务器只能在该网络接口上接受传入的TCP连接，这意味着（除非该网络接口恰好是内部/环回接口），服务器将不接受与127.0.0.1的连接从其自己的客户端线程。简便的解决方法是将host设置为一个空字符串，这样服务器将接受所有本地网络接口上的传入TCP连接，并且问题消除了。

def RunServer():
    host = '' # was:  socket.gethostbyname(socket.gethostname())
    [...]

第二个问题是服务器的事件循环正在等待，这导致服务器进程在整个运行过程中不必要地消耗了100％的CPU核心。效率极低。该问题的原因在RunServer()内部：

while True:
    while not recvPackets.empty():
        data,addr = recvPackets.get()
        [...]

请注意，在Python Queue类中，对get()的调用将阻塞，直到有更多数据要接收为止，从而防止CPU旋转。但是在以上代码中，除非/直到get()为非空，否则recvPackets永远不会被调用，因此线程不会阻塞。

修复很简单，只需完全删除while not recvPackets.empty()行，这样即使队列为空，也将调用get()。 get()在有数据要返回之前不会返回。

while True:
   data,addr = recvPackets.get()
   [...]

UDP聊天，服务器也可以聊天

1 个答案: