firefly的rpc。。
firefly使用了twisted的pb 来实现rpc:
http://twistedmatrix.com/documents/current/core/howto/pb-usage.html
服务端 #!/usr/bin/env python # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. from twisted.spread import pb from twisted.internet import reactor class One(pb.Root): def remote_takeTwo(self, two): print "received a Two called", two print "telling it to print(12)" two.callRemote("print", 12) reactor.listenTCP(8800, pb.PBServerFactory(One())) reactor.run() 客户端 #!/usr/bin/env python # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. from twisted.spread import pb from twisted.internet import reactor class Two(pb.Referenceable): def remote_print(self, arg): print "Two.print() called with", arg def main(): two = Two() factory = pb.PBClientFactory() reactor.connectTCP("localhost", 8800, factory) def1 = factory.getRootObject() def1.addCallback(got_obj, two) # hands our 'two' to the callback reactor.run() def got_obj(obj, two): print "got One:", obj print "giving it our two" obj.callRemote("takeTwo", two) main()
1.客户端通过factory.getRootObject() 获取def,并增加回调函数。其第一个参数为pb.Root(未证实),然后借助callRemote方法条用服务端的东西
2.可以在客户端首次连接的时候,将客户端的Referenceable传给服务端。。服务端进行保存。。之后需要像客户端发送消息的时候。。只要使用客户端的callRemote就OK了
root.py 实现PB的server功能
node.py 实现PB的client功能。
child.py 每个client连接上root都会初始化一个child来保存该client的相关信息,并且将这些child通过manager来管理。
manager.py 管理root的child,通过一个字典self._childs = {},实现一个增删改的小管理器。
reference.py 如果你看了前面twisted官网的介绍就会知道,node只要实例化一个 pb.Referenceable 类,并把它传递给root,那么root就能够把这个pb.Referenceble当成句柄来远程调用client的函数。
下面我们看看firefly的实现。
1.实例化客户端node,实例了客户端工厂,_reference和name
2.客户端建立连接:node.connect ,并条用了服务端的takeProxy。建立了通道
def callRemote(obj,funcName,*args,**kw): '''远程调用 @param funcName: str 远程方法 ''' return obj.callRemote(funcName, *args,**kw) 。。。。。。 。。。。。。 def takeProxy(self): '''像远程服务端发送代理通道对象 ''' deferedRemote = self._factory.getRootObject() deferedRemote.addCallback(callRemote,'takeProxy',self._name,self._reference)
2.服务端的takeProxy。根据第一次传过来的客户端名称和reference实例化了child,并保存到childManage中
3.执行了首次连接处理。doChildConnect。。请看rootapp.py.如果不看这里你一定会很奇怪,gate的root怎么和其他的server连接起来的。
def _doChildConnect(name,transport): """当server节点连接到master的处理 """ server_config = GlobalObject().json_config.get('servers',{}).get(name,{}) remoteport = server_config.get('remoteport',[]) child_host = transport.broker.transport.client[0] root_list = [rootport.get('rootname') for rootport in remoteport] GlobalObject().remote_map[name] = {"host":child_host,"root_list":root_list} #通知有需要连的node节点连接到此root节点 for servername,remote_list in GlobalObject().remote_map.items(): remote_host = remote_list.get("host","") remote_name_host = remote_list.get("root_list","") if name in remote_name_host: GlobalObject().root.callChild(servername,"remote_connect",name,remote_host) #查看当前是否有可供连接的root节点 master_node_list = GlobalObject().remote_map.keys() for root_name in root_list: if root_name in master_node_list: root_host = GlobalObject().remote_map[root_name]['host'] GlobalObject().root.callChild(name,"remote_connect",root_name,root_host)
4.服务端调用客户端。root.callChildByName.会在childManager中找到child并使用其中的reference,最终条用代码为:ProxyReference(pb.Referenceable):中的
def remote_callChild(self, command,*arg,**kw): '''代理发送数据 ''' return self._service.callTarget(command,*arg,**kw)
5.客户端条用,服务端使用,node.callRemote..其中最终使用的方法为:root.py中的。
def remote_callTarget(self,command,*args,**kw): '''远程调用方法 @param commandId: int 指令号 @param data: str 调用参数 ''' data = self.service.callTarget(command,*args,**kw) return data
6.从上可以看到。。使用service的意义何在了。。。