twisted-异步网络

最新推荐文章于 2024-05-14 18:31:09 发布

原创最新推荐文章于 2024-05-14 18:31:09 发布 · 205 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

C++/C/lisp 专栏收录该内容

273 篇文章

订阅专栏

本文介绍了一个使用Python Twisted库进行简单连接测试的例子。通过定义一个特殊的协议类CallbackAndDisconnectProtocol来实现连接建立后立即断开的功能，并通过Deferred对象管理异步操作。此测试可用于验证与目标主机端口的连接是否正常。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from twisted.internet import reactor, defer, protocol



class CallbackAndDisconnectProtocol(protocol.Protocol):

    def connectionMade(self):

        self.factory.deferred.callback("Connected!")

        self.transport.loseConnection( )



class ConnectionTestFactory(protocol.ClientFactory):

    protocol = CallbackAndDisconnectProtocol



    def _ _init_ _(self):

        self.deferred = defer.Deferred( )



    def clientConnectionFailed(self, connector, reason):

        self.deferred.errback(reason)



def testConnect(host, port):

    testFactory = ConnectionTestFactory( )

    reactor.connectTCP(host, port, testFactory)

    return testFactory.deferred



def handleSuccess(result, port):

    print "Connected to port %i" % port

    reactor.stop( )



def handleFailure(failure, port):

    print "Error connecting to port %i: %s" % (

        port, failure.getErrorMessage( ))

    reactor.stop( )



if __name__ == "_ _main_ _":

    import sys

    if not len(sys.argv) == 3:

        print "Usage: connectiontest.py host port"

        sys.exit(1)



    host = sys.argv[1]

    port = int(sys.argv[2])

    connecting = testConnect(host, port)

    connecting.addCallback(handleSuccess, port)

    connecting.addErrback(handleFailure, port)

    reactor.run( )

E:\test>python test.py www.hao123.com 80
Connected to port 80