2010-09-08 73 views
9

那么,我正在试图建立一个带有一个SocketServer的小型python prgram,它应该将它收到的消息发送给所有连接的客户端。我卡住了,我不知道如何在服务器端存储客户端,我不知道如何发送给多个客户端。哦,我的计划失败,每次多则1个客户端连接,并每次客户端发送一个以上的消息......Python的SocketServer:发送给多个客户端?

这里是我的代码至今:

 print str(self.client_address[0])+' connected.' 
    def handle(self): 
     new=1 
     for client in clients: 
      if client==self.request: 
       new=0 
     if new==1: 
      clients.append(self.request) 
     for client in clients: 
      data=self.request.recv(1024) 
      client.send(data) 

class Host: 
    def __init__(self): 
     self.address = ('localhost', 0) 
     self.server = SocketServer.TCPServer(self.address, EchoRequestHandler) 
     ip, port = self.server.server_address 
     self.t = threading.Thread(target=self.server.serve_forever) 
     self.t.setDaemon(True) 
     self.t.start() 
     print '' 
     print 'Hosted with IP: '+ip+' and port: '+str(port)+'. Clients can now connect.' 
     print '' 
    def close(self): 
     self.server.socket.close() 

class Client: 
    name='' 
    ip='' 
    port=0 
    def __init__(self,ip,port,name): 
     self.name=name 
     self.hostIp=ip 
     self.hostPort=port 
     self.s=socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
     self.s.connect((self.hostIp, self.hostPort)) 
    def reco(self): 
     self.s=socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
     self.s.connect((self.hostIp, self.hostPort)) 
    def nick(self,newName): 
     self.name=newName 
    def send(self,message): 
     message=self.name+' : '+message 
     len_sent=self.s.send(message) 
     response=self.s.recv(len_sent) 
     print response 
     self.reco() 
    def close(self): 
     self.s.close() 

很显然,我不知道我在做,所以任何帮助都会很棒。
在此先感谢!

编辑:我在Windows Vista上使用Python 2.7。

+0

我不得不解决在这里了类似的问题:服务器/客户端代码(http://stackoverflow.com/questions/41785969/python-tcp-server-accepting-connections - 广播命令/ 41786133#41786133) – 2017-01-22 19:22:36

回答

15

你想在这里看看asyncore。您在客户端调用的套接字操作是阻塞的(直到收到某些数据或发生超时时才会返回),这使得难以监听从主机发送的消息并让客户端实例将数据排入队列同一时间。 asyncore应该将基于超时的轮询循环从你身上抽象出来。

这里有一个代码 “样本” - 让我知道,如果有不清楚的地方:

from __future__ import print_function 

import asyncore 
import collections 
import logging 
import socket 


MAX_MESSAGE_LENGTH = 1024 


class RemoteClient(asyncore.dispatcher): 

    """Wraps a remote client socket.""" 

    def __init__(self, host, socket, address): 
     asyncore.dispatcher.__init__(self, socket) 
     self.host = host 
     self.outbox = collections.deque() 

    def say(self, message): 
     self.outbox.append(message) 

    def handle_read(self): 
     client_message = self.recv(MAX_MESSAGE_LENGTH) 
     self.host.broadcast(client_message) 

    def handle_write(self): 
     if not self.outbox: 
      return 
     message = self.outbox.popleft() 
     if len(message) > MAX_MESSAGE_LENGTH: 
      raise ValueError('Message too long') 
     self.send(message) 


class Host(asyncore.dispatcher): 

    log = logging.getLogger('Host') 

    def __init__(self, address=('localhost', 0)): 
     asyncore.dispatcher.__init__(self) 
     self.create_socket(socket.AF_INET, socket.SOCK_STREAM) 
     self.bind(address) 
     self.listen(1) 
     self.remote_clients = [] 

    def handle_accept(self): 
     socket, addr = self.accept() # For the remote client. 
     self.log.info('Accepted client at %s', addr) 
     self.remote_clients.append(RemoteClient(self, socket, addr)) 

    def handle_read(self): 
     self.log.info('Received message: %s', self.read()) 

    def broadcast(self, message): 
     self.log.info('Broadcasting message: %s', message) 
     for remote_client in self.remote_clients: 
      remote_client.say(message) 


class Client(asyncore.dispatcher): 

    def __init__(self, host_address, name): 
     asyncore.dispatcher.__init__(self) 
     self.log = logging.getLogger('Client (%7s)' % name) 
     self.create_socket(socket.AF_INET, socket.SOCK_STREAM) 
     self.name = name 
     self.log.info('Connecting to host at %s', host_address) 
     self.connect(host_address) 
     self.outbox = collections.deque() 

    def say(self, message): 
     self.outbox.append(message) 
     self.log.info('Enqueued message: %s', message) 

    def handle_write(self): 
     if not self.outbox: 
      return 
     message = self.outbox.popleft() 
     if len(message) > MAX_MESSAGE_LENGTH: 
      raise ValueError('Message too long') 
     self.send(message) 

    def handle_read(self): 
     message = self.recv(MAX_MESSAGE_LENGTH) 
     self.log.info('Received message: %s', message) 


if __name__ == '__main__': 
    logging.basicConfig(level=logging.INFO) 
    logging.info('Creating host') 
    host = Host() 
    logging.info('Creating clients') 
    alice = Client(host.getsockname(), 'Alice') 
    bob = Client(host.getsockname(), 'Bob') 
    alice.say('Hello, everybody!') 
    logging.info('Looping') 
    asyncore.loop() 

导致下面的输出:

INFO:root:Creating host 
INFO:root:Creating clients 
INFO:Client ( Alice):Connecting to host at ('127.0.0.1', 51117) 
INFO:Client ( Bob):Connecting to host at ('127.0.0.1', 51117) 
INFO:Client ( Alice):Enqueued message: Hello, everybody! 
INFO:root:Looping 
INFO:Host:Accepted client at ('127.0.0.1', 55628) 
INFO:Host:Accepted client at ('127.0.0.1', 55629) 
INFO:Host:Broadcasting message: Hello, everybody! 
INFO:Client ( Alice):Received message: Hello, everybody! 
INFO:Client ( Bob):Received message: Hello, everybody! 
+0

谢谢,看起来像我在找什么!不幸的是,我没有设法使它在__main__之外工作:我在Host .__ init __()的末尾添加了asyncore.loop(),并且我的主机对象接受客户端连接,但它对发送的消息没有反应。 – Alex 2010-09-09 08:06:23

+0

@Alex:asyncore.loop()永远运行!实际上,通过调用它,你会说:“我完成了程序的控制,将事情交给异步循环,以便在其余时间处理发送/接收。”请注意,在调用asyncore.loop()之前,我是如何设置的。你想通过移动它做什么? – cdleary 2010-09-09 09:30:52

+0

好吧,我不想只运行一些明确的连接,但让主机运行,并且客户端可以随时连接/发送消息。谢谢你的时间! – Alex 2010-09-09 11:11:15

0

为什么使用SocketServer的?一个简单的客户端不符合您的需求?

import socket 

HOST = '' 
PORT = 8000 
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
sock.bind((HOST, PORT)) 
sock.listen(5) 
while True: 
    conn, addr = sock.accept() 
    print 'connecting to', addr 
    while True: 
     data = conn.recv(1024) 
     if not data: 
      break 
     conn.send(data) 
+6

是否不会阻止其他客户端连接? – Codler 2010-12-05 23:39:05

+0

这不会工作。 – 2016-07-07 22:01:52

0

要同时承担多个客户端,你将不得不增加SocketServer.ForkingMixInThreadingMixIn

5

您可以使用socketserver向所有连接的客户端广播消息。然而,这个能力并没有嵌入到代码中,需要通过扩展已经提供的一些类来实现。在下面的例子中,这是通过使用ThreadingTCPServerStreamRequestHandler类来实现的。它们提供了构建的基础,但仍需要进行一些修改才能实现您正在尝试完成的任务。这些文档应该有助于解释为了完成工作,每个功能,类别和方法正在尝试做什么。

服务器

#! /usr/bin/env python3 
import argparse 
import pickle 
import queue 
import select 
import socket 
import socketserver 


def main(): 
    """Start a chat server and serve clients forever.""" 
    parser = argparse.ArgumentParser(description='Execute a chat server demo.') 
    parser.add_argument('port', type=int, help='location where server listens') 
    arguments = parser.parse_args() 
    server_address = socket.gethostbyname(socket.gethostname()), arguments.port 
    server = CustomServer(server_address, CustomHandler) 
    server.serve_forever() 


class CustomServer(socketserver.ThreadingTCPServer): 

    """Provide server support for the management of connected clients.""" 

    def __init__(self, server_address, request_handler_class): 
     """Initialize the server and keep a set of registered clients.""" 
     super().__init__(server_address, request_handler_class, True) 
     self.clients = set() 

    def add_client(self, client): 
     """Register a client with the internal store of clients.""" 
     self.clients.add(client) 

    def broadcast(self, source, data): 
     """Resend data to all clients except for the data's source.""" 
     for client in tuple(self.clients): 
      if client is not source: 
       client.schedule((source.name, data)) 

    def remove_client(self, client): 
     """Take a client off the register to disable broadcasts to it.""" 
     self.clients.remove(client) 


class CustomHandler(socketserver.StreamRequestHandler): 

    """Allow forwarding of data to all other registered clients.""" 

    def __init__(self, request, client_address, server): 
     """Initialize the handler with a store for future date streams.""" 
     self.buffer = queue.Queue() 
     super().__init__(request, client_address, server) 

    def setup(self): 
     """Register self with the clients the server has available.""" 
     super().setup() 
     self.server.add_client(self) 

    def handle(self): 
     """Run a continuous message pump to broadcast all client data.""" 
     try: 
      while True: 
       self.empty_buffers() 
     except (ConnectionResetError, EOFError): 
      pass 

    def empty_buffers(self): 
     """Transfer data to other clients and write out all waiting data.""" 
     if self.readable: 
      self.server.broadcast(self, pickle.load(self.rfile)) 
     while not self.buffer.empty(): 
      pickle.dump(self.buffer.get_nowait(), self.wfile) 

    @property 
    def readable(self): 
     """Check if the client's connection can be read without blocking.""" 
     return self.connection in select.select(
      (self.connection,),(),(), 0.1)[0] 

    @property 
    def name(self): 
     """Get the client's address to which the server is connected.""" 
     return self.connection.getpeername() 

    def schedule(self, data): 
     """Arrange for a data packet to be transmitted to the client.""" 
     self.buffer.put_nowait(data) 

    def finish(self): 
     """Remove the client's registration from the server before closing.""" 
     self.server.remove_client(self) 
     super().finish() 


if __name__ == '__main__': 
    main() 

当然,你还需要一个客户端,可以与服务器进行通信,并使用服务器讲同一种协议。由于这是Python,因此决定使用pickle模块来促进服务器和客户端之间的数据传输。其他数据传输方法可能已经被使用(例如JSON,XML等等),但是能够腌制和取消数据的方式足以满足该程序的需求。文档再次包含在内,因此找出发生的事情不应太难。请注意,服务器命令可以中断用户数据输入。

客户

#! /usr/bin/env python3 
import argparse 
import cmd 
import pickle 
import socket 
import threading 


def main(): 
    """Connect a chat client to a server and process incoming commands.""" 
    parser = argparse.ArgumentParser(description='Execute a chat client demo.') 
    parser.add_argument('host', type=str, help='name of server on the network') 
    parser.add_argument('port', type=int, help='location where server listens') 
    arguments = parser.parse_args() 
    client = User(socket.create_connection((arguments.host, arguments.port))) 
    client.start() 


class User(cmd.Cmd, threading.Thread): 

    """Provide a command interface for internal and external instructions.""" 

    prompt = '>>> ' 

    def __init__(self, connection): 
     """Initialize the user interface for communicating with the server.""" 
     cmd.Cmd.__init__(self) 
     threading.Thread.__init__(self) 
     self.connection = connection 
     self.reader = connection.makefile('rb', -1) 
     self.writer = connection.makefile('wb', 0) 
     self.handlers = dict(print=print, ping=self.ping) 

    def start(self): 
     """Begin execution of processor thread and user command loop.""" 
     super().start() 
     super().cmdloop() 
     self.cleanup() 

    def cleanup(self): 
     """Close the connection and wait for the thread to terminate.""" 
     self.writer.flush() 
     self.connection.shutdown(socket.SHUT_RDWR) 
     self.connection.close() 
     self.join() 

    def run(self): 
     """Execute an automated message pump for client communications.""" 
     try: 
      while True: 
       self.handle_server_command() 
     except (BrokenPipeError, ConnectionResetError): 
      pass 

    def handle_server_command(self): 
     """Get an instruction from the server and execute it.""" 
     source, (function, args, kwargs) = pickle.load(self.reader) 
     print('Host: {} Port: {}'.format(*source)) 
     self.handlers[function](*args, **kwargs) 

    def preloop(self): 
     """Announce to other clients that we are connecting.""" 
     self.call('print', socket.gethostname(), 'just entered.') 

    def call(self, function, *args, **kwargs): 
     """Arrange for a handler to be executed on all other clients.""" 
     assert function in self.handlers, 'You must create a handler first!' 
     pickle.dump((function, args, kwargs), self.writer) 

    def do_say(self, arg): 
     """Causes a message to appear to all other clients.""" 
     self.call('print', arg) 

    def do_ping(self, arg): 
     """Ask all clients to report their presence here.""" 
     self.call('ping') 

    def ping(self): 
     """Broadcast to all other clients that we are present.""" 
     self.call('print', socket.gethostname(), 'is here.') 

    def do_exit(self, arg): 
     """Disconnect from the server and close the client.""" 
     return True 

    def postloop(self): 
     """Make an announcement to other clients that we are leaving.""" 
     self.call('print', socket.gethostname(), 'just exited.') 


if __name__ == '__main__': 
    main() 
相关问题