Python 扭曲 - 需要遍历所有连接并找到客户端

Question

我正在尝试创建一个简单的优惠券程序。

客户端连接到服务器并询问凭证是否还有剩余时间，如果是，服务器会响应多少时间。

我控制服务器和客户端，客户端也由我编码。

现在这就是我的服务器端的内容，客户端是不言自明的。

其中最大的缺陷是，如果 2 个客户端使用相同的优惠券代码连接，它们都将具有访问权限，因为服务器不会检查是否存在具有该代码的活动客户端。

任何人都可以解释或提供有关这如何可能的文档吗？

#!/usr/bin/env python

from twisted.internet import reactor, protocol

class Responder(protocol.Protocol):

    def dataReceived(self, data):
        # check the voucher code, and return disabled if its out of time or not there. Otherwise return time left.
        if data.startswith("check="):
            param, vcode = data.split("=")
            checkcode = SQLConnect("check", vcode, vcode)
            if checkcode == "disabled":
                self.transport.write("disabled")
            else:
                self.transport.write(str(checkcode))
        # Update time left.
        if data.startswith("update="):
            param, vcode, vtime = data.split("=")
            SQLConnect("update", vcode, vtime)

def main():
    factory = protocol.ServerFactory()
    factory.protocol = Responder
    reactor.listenTCP(6500,factory)
    reactor.run()

if __name__ == '__main__':
    main()

Answer 1

如果凭证在客户端检查时变为“正在使用”，然后在客户端断开连接时变为未使用，听起来您只需要保留一组凭证，在检查完成时添加到其中，并在客户端断开连接时将其删除。您可以将其保留在工厂中，以便在所有客户端连接之间共享。例如：

#!/usr/bin/env python

from twisted.internet import reactor, protocol

class Responder(protocol.Protocol):
    def connectionMade(self):
        self.vcode = None

    def dataReceived(self, data):
        # check the voucher code, and return disabled if its out of time or not there. Otherwise return time left.
        if data.startswith("check="):
            param, vcode = data.split("=")
            if vcode in self.factory.activeVouchers:
                self.transport.write("in use")
                return
            self.factory.activeVouchers.add(vcode)
            self.vcode = vcode

            checkcode = SQLConnect("check", vcode, vcode)
            if checkcode == "disabled":
                self.transport.write("disabled")
            else:
                self.transport.write(str(checkcode))
        # Update time left.
        if data.startswith("update="):
            param, vcode, vtime = data.split("=")
            SQLConnect("update", vcode, vtime)

    def connectionLost(self, reason):
        if self.vcode is not None:
            self.factory.activeVouchers.remove(self.vcode)

def main():
    factory = protocol.ServerFactory()
    factory.activeVouchers = set()
    factory.protocol = Responder
    reactor.listenTCP(6500,factory)
    reactor.run()

if __name__ == '__main__':
    main()

Responder 上的新 vcode 属性允许在客户端断开连接时更新 activeVouchers 集（这会触发 Responder.connectionLost 调用）。 Responder.dataReceived 开头附近的附加检查会在凭证被使用时将其添加到该集合中，并防止任何正在使用的凭证被认领。

除此之外，您可能还需要考虑其他一些事情。首先，您可能应该使用 twisted.protocols.basic.LineOnlyReceiver 或 twisted.protocols.basic 中的其他协议之一，而不仅仅是 Protocol 。每当通过网络接收到任何字节时都会调用 dataReceived 。由于 TCP 是面向流的传输而不是面向消息的传输，因此您最终可能会得到类似 dataReceived("chec") 的调用，紧接着是 dataReceived("k=somecode" ）。由于您的 dataReceived 现已实现，因此不会处理这种情况，客户端将不会收到任何响应。 LineOnlyReceiver 添加了基于行的帧，因此“check=somecode\r\n”之类的字节可以解释为完整的消息，并且“check=somecode”可以在单个 lineReceived 呼叫。

其次，SQLConnect 看起来可能会执行一些阻塞 I/O。如果这是真的，则意味着您的服务器无法很好地处理并发客户端请求，因为任何阻塞都会阻止处理所有新事件，包括来自不同客户端的事件。您可能想查看一下 twisted.enterprise.adbapi 的非阻塞 SQL API。