我如何在与主程序不同的线程中编写套接字服务器(使用gevent)



我正在开发一个Flask/gevent WSGIserver web服务器,它需要使用XML通过两个套接字与硬件设备进行通信(在后台)。

一个套接字由客户端(我的应用程序)发起,我可以向设备发送XML命令。设备在不同的端口上应答,并发送回我的应用程序必须确认的信息。所以我的应用程序必须监听第二个端口。

到目前为止,我已经发出了一个命令,作为服务器打开了第二个端口,等待设备的响应并关闭了第二个端口。

问题是,设备可能会发送多个我必须确认的响应。因此,我的解决方案是保持端口打开并继续响应传入的请求。然而,最终设备完成了发送请求,我的应用程序仍在监听(我不知道设备何时完成),从而阻塞了其他一切。

这似乎是一个线程的完美用例,因此我的应用程序在一个单独的线程中启动侦听服务器。因为我已经使用gevent作为Flask的WSGI服务器,所以我可以使用greenlet。

问题是,我已经寻找了这样一个事情的一个很好的例子,但我所能找到的是一个套接字服务器多线程处理程序的例子。我不需要在套接字服务器上处理很多连接,但我需要它在一个单独的线程中启动,这样它就可以监听和处理传入的消息,而我的主程序可以继续发送消息。我遇到的第二个问题是,在服务器中,我需要使用一些方法从我的"主"类。作为Python的新手,我不确定如何以一种可能的方式构建它。

class Device(object):
def __init__(self, ...):
    self.clientsocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    self.serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
def _connect_to_device(self):
    print "OPEN CONNECTION TO DEVICE"
    try:
        self.clientsocket.connect((self.ip, 5100))
    except socket.error as e:
        pass
def _disconnect_from_device(self):
    print "CLOSE CONNECTION TO DEVICE"
    self.clientsocket.close()
def deviceaction1(self, ...):
    # the data that is sent is an XML document that depends on the parameters of this method.
    self._connect_to_device()
    self._send_data(XMLdoc)
    self._wait_for_response()
    return True
def _send_data(self, data):
    print "SEND:"
    print(data)
    self.clientsocket.send(data)
def _wait_for_response(self):
    print "WAITING FOR REQUESTS FROM DEVICE (CHANNEL 1)"
    self.serversocket.bind(('10.0.0.16', 5102))
    self.serversocket.listen(5)                         # listen for answer, maximum 5 connections
    connection, address = self.serversocket.accept()
    # the data is of a specific length I can calculate
    if len(data) > 0:
        self._process_response(data)
        self.serversocket.close()
def _process_response(self, data):
    print "RECEIVED:"
    print(data)
    # here is some code that processes the incoming data and
    # responds to the device
    # this may or may not result in more incoming data

if __name__ == '__main__':
    machine = Device(ip="10.0.0.240")
    Device.deviceaction1(...)

这是我现在正在做的(全局而言,我遗漏了敏感信息)。正如你所看到的,一切都是连续的。如果有人能提供一个单独线程中侦听服务器的例子(最好使用greenlet),并提供一种从侦听服务器通信回产卵线程的方法,这将是很大的帮助。

谢谢。编辑:

在尝试了几种方法之后,我决定使用python默认的select()方法来解决这个问题。这是有效的,所以我关于线程使用的问题不再相关。感谢那些花费时间和精力提供意见的人。

希望它能提供一些帮助,在示例类中,如果我们将调用tenMessageSender函数,那么它将启动一个异步线程,而不会阻塞主循环,然后_zmqBasedListener将开始在单独的端口上侦听,直到该线程是活的。tenMessageSender函数发送的任何消息都将被客户端接收并响应给zmqBasedListener

服务器端

import threading
import zmq
import sys
class Example:
    def __init__(self):
        self.context = zmq.Context()
        self.publisher = self.context.socket(zmq.PUB)
        self.publisher.bind('tcp://127.0.0.1:9997')
        self.subscriber = self.context.socket(zmq.SUB)
        self.thread = threading.Thread(target=self._zmqBasedListener)
    def _zmqBasedListener(self):
        self.subscriber.connect('tcp://127.0.0.1:9998')
        self.subscriber.setsockopt(zmq.SUBSCRIBE, "some_key")
        while True:
            message = self.subscriber.recv()
            print message
        sys.exit()
    def tenMessageSender(self):
        self._decideListener()
        for message in range(10):
            self.publisher.send("testid : %d: I am a task" %message)
    def _decideListener(self):
        if not self.thread.is_alive():
            print "STARTING THREAD"
            self.thread.start()

import zmq
context = zmq.Context()
subscriber = context.socket(zmq.SUB)
subscriber.connect('tcp://127.0.0.1:9997')
publisher = context.socket(zmq.PUB)
publisher.bind('tcp://127.0.0.1:9998')
subscriber.setsockopt(zmq.SUBSCRIBE, "testid")
count = 0
print "Listener"
while True:
    message = subscriber.recv()
    print message
    publisher.send('some_key : Message received %d' %count)
    count+=1

可以用greenlet等代替thread

相关内容

  • 没有找到相关文章

最新更新