def onMsg(self):
pack = stackless.getcurrent()._fttask.pack
ftlog.debug('A2AProtocol->', FTTasklet.concurrent_task_count,
'[' + pack + ']')
src, dst, queryid, userheader1, userheader2, message = agentmsg.unpack(
pack)
if src == None or dst == None:
ftlog.info("ERROR, recive a error format message")
return
if self.peer_id == 0:
self.peer_id = src
ftcon.serverid_protocol_map[self.peer_id] = self
ftlog.info("Receive other agent register, agentid=", self.peer_id)
return
# 处理agent服务自身的命令
if dst == ftcon.global_config[
"server_id"] and A2AProtocol.onCommand != None:
# ftlog.debug('A2AProtocol-> it self command !!')
A2AProtocol.onCommand(self, src, queryid, userheader1, userheader2,
message)
return
try:
# 从其他agent过来的数据，转给连接的dst service
# ftlog.debug('A2AProtocol->send to target->', dst)
protocol = ftcon.serverid_protocol_map[dst]
if performance.PERFORMANCE_NET:
pack = performance.linkMsgTime('LW', pack)
protocol.transport.write(pack + '\r\n')
except:
ftlog.error('msg route error, dst_server_id=', dst)
def onMsg(self):
pack = stackless.getcurrent()._fttask.pack
ftlog.debug('A2AProtocol->', FTTasklet.concurrent_task_count, '[' + pack + ']')
src, dst, queryid, userheader1, userheader2, message = agentmsg.unpack(pack)
if src == None or dst == None:
ftlog.info("ERROR, recive a error format message")
return
if self.peer_id == 0:
self.peer_id = src
ftcon.serverid_protocol_map[self.peer_id] = self
ftlog.info("Receive other agent register, agentid=", self.peer_id)
return
# 处理agent服务自身的命令
if dst == ftcon.global_config["server_id"] and A2AProtocol.onCommand != None:
# ftlog.debug('A2AProtocol-> it self command !!')
A2AProtocol.onCommand(self, src, queryid, userheader1, userheader2, message)
return
try:
# 从其他agent过来的数据，转给连接的dst service
# ftlog.debug('A2AProtocol->send to target->', dst)
protocol = ftcon.serverid_protocol_map[dst]
if performance.PERFORMANCE_NET:
pack = performance.linkMsgTime('LW', pack)
protocol.transport.write(pack + '\r\n')
except:
ftlog.error('msg route error, dst_server_id=', dst)
def lineReceived(self, data):
if performance.PERFORMANCE_NET:
data = performance.linkMsgTime('LR', data)
ftlog.debug('S2AProtocol->lineReceived', FTTasklet.concurrent_task_count, time(), data)
src, dst, query_id, userheader1, userheader2, msg = agentmsg.unpack(data)
if ide_debug():
ide_print_pack("S2AP Recv [%5s]" % src.split("-")[0], json.loads(msg))
if src is None or dst is None:
ftlog.info("ERROR, recive a error format message")
return
if self.peer_id == 0:
self.peer_id = src
ftcon.serverid_protocol_map[self.peer_id] = self
ftlog.info('receive register, agentid=', self.peer_id)
return
_countProtocolPack(1, self)
# send过来的数据
if query_id == '':
self._runTasklet(data=msg, src=src, dst=dst, userheader1=userheader1, userheader2=userheader2,
time_recv=time())
else:
querysrc, _ = query_id.split('.')
server_id = ftcon.global_config["server_id"]
# query本服务的请求
if querysrc != server_id:
self._runTasklet(data=msg, src=src, dst=dst, query_id=query_id, userheader1=userheader1,
userheader2=userheader2, time_recv=time())
# response回来的请求
else:
if userheader1 == 'RQ': # 本进程内, 异步查询本进程的其他消息接口
self._runTasklet(data=msg, src=src, dst=dst, query_id=query_id, userheader1=userheader1,
userheader2=userheader2, time_recv=time())
else:
d, c, t = None, None, 0
# ftlog.debug('lineReceived', query_id, id(_LIVE_MESSAGES), id(self))
if query_id in _LIVE_MESSAGES:
d, c, t = _LIVE_MESSAGES[query_id]
del _LIVE_MESSAGES[query_id]
else:
if query_id in _FAILED_MESSAGES:
del _FAILED_MESSAGES[query_id]
ftlog.warn('QUERY TOO SLOW !!', query_id, msg)
if len(_FAILED_MESSAGES) > 100:
_FAILED_MESSAGES.clear()
else:
ftlog.warn('NOT KNOW of query_id->', query_id, msg)
if d and c:
try:
c.cancel()
d.callback((msg, t, time()))
except:
ftlog.error(msg)
def onMsg(self):
pack = stackless.getcurrent()._fttask.pack
ftlog.debug('A2SProtocol->', FTTasklet.concurrent_task_count, '[',
pack, ']')
src, dst, queryid, userheader1, userheader2, message = agentmsg.unpack(
pack)
#zxdebug:测试query时，AG直接返回的性能
#self.transport.write("GA01|CO01|"+queryid+"|||"+message+"\r\n")
#return
#zxdebug
if src == None or dst == None:
ftlog.info("ERROR, recive a error format message")
return
if self.peer_id == 0:
self.peer_id = src
ftcon.serverid_protocol_map[self.peer_id] = self
ftlog.info("Receive service register, serverid=", self.peer_id,
self)
return
# 处理agent服务自身的命令
if dst == ftcon.global_config[
"server_id"] and A2SProtocol.onCommand != None:
# ftlog.debug('A2SProtocol-> it self command !!')
A2SProtocol.onCommand(self, src, queryid, userheader1, userheader2,
message)
return
if dst.find('AG') == 0: # 如果是直接发送给另外的一个AGENT, 那器agent就是其本身
agent_id = dst
else:
server_conf = ftcon.getServerConf(dst)
agent_id = server_conf['agent']
try:
# 从src发过来的数据，转给dst所在的agent
# 如果是agent所属内部进程过来的数据, 直接转发至dst进程
if agent_id == ftcon.global_config["server_id"]:
agent_id = dst
# ftlog.debug('A2SProtocol->send to agentid->', agent_id, 'dst=', dst)
protocol = ftcon.serverid_protocol_map[agent_id]
if performance.PERFORMANCE_NET:
pack = performance.linkMsgTime('LW', pack)
protocol.transport.write(pack + '\r\n')
except:
ftlog.error(
'msg route error, not found agent protocol, dst_server_id=',
dst, 'agend=', agent_id)
def onMsg(self):
pack = stackless.getcurrent()._fttask.pack
ftlog.debug('A2SProtocol->', FTTasklet.concurrent_task_count, '[', pack, ']')
src, dst, queryid, userheader1, userheader2, message = agentmsg.unpack(pack)
# zxdebug:测试query时，AG直接返回的性能
# self.transport.write("GA01|CO01|"+queryid+"|||"+message+"\r\n")
# return
# zxdebug
if src == None or dst == None:
ftlog.info("ERROR, recive a error format message")
return
if self.peer_id == 0:
self.peer_id = src
ftcon.serverid_protocol_map[self.peer_id] = self
ftlog.info("Receive service register, serverid=", self.peer_id, self)
return
# 处理agent服务自身的命令
if dst == ftcon.global_config["server_id"] and A2SProtocol.onCommand != None:
# ftlog.debug('A2SProtocol-> it self command !!')
A2SProtocol.onCommand(self, src, queryid, userheader1, userheader2, message)
return
if dst.find('AG') == 0: # 如果是直接发送给另外的一个AGENT, 那器agent就是其本身
agent_id = dst
else:
server_conf = ftcon.getServerConf(dst)
agent_id = server_conf['agent']
try:
# 从src发过来的数据，转给dst所在的agent
# 如果是agent所属内部进程过来的数据, 直接转发至dst进程
if agent_id == ftcon.global_config["server_id"]:
agent_id = dst
# ftlog.debug('A2SProtocol->send to agentid->', agent_id, 'dst=', dst)
protocol = ftcon.serverid_protocol_map[agent_id]
if performance.PERFORMANCE_NET:
pack = performance.linkMsgTime('LW', pack)
protocol.transport.write(pack + '\r\n')
except:
ftlog.error('msg route error, not found agent protocol, dst_server_id=', dst, 'agend=', agent_id)
def lineReceived(self, data):
if performance.PERFORMANCE_NET:
data = performance.linkMsgTime('LR', data)
ftlog.debug('S2AProtocol->lineReceived',
FTTasklet.concurrent_task_count, time(), data)
src, dst, query_id, userheader1, userheader2, msg = agentmsg.unpack(
data)
if src == None or dst == None:
ftlog.info("ERROR, recive a error format message")
return
if self.peer_id == 0:
self.peer_id = src
ftcon.serverid_protocol_map[self.peer_id] = self
ftlog.info('receive register, agentid=', self.peer_id)
return
_countProtocolPack(1, self)
# send过来的数据
if query_id == '':
self._runTasklet(data=msg,
src=src,
dst=dst,
userheader1=userheader1,
userheader2=userheader2,
time_recv=time())
else:
querysrc, _ = query_id.split('.')
server_id = ftcon.global_config["server_id"]
# query本服务的请求
if querysrc != server_id:
self._runTasklet(data=msg,
src=src,
dst=dst,
query_id=query_id,
userheader1=userheader1,
userheader2=userheader2,
time_recv=time())
# response回来的请求
else:
if userheader1 == 'RQ': # 本进程内, 异步查询本进程的其他消息接口
self._runTasklet(data=msg,
src=src,
dst=dst,
query_id=query_id,
userheader1=userheader1,
userheader2=userheader2,
time_recv=time())
else:
d, c, t = None, None, 0
# ftlog.debug('lineReceived', query_id, id(_LIVE_MESSAGES), id(self))
if query_id in _LIVE_MESSAGES:
d, c, t = _LIVE_MESSAGES[query_id]
del _LIVE_MESSAGES[query_id]
else:
if query_id in _FAILED_MESSAGES:
del _FAILED_MESSAGES[query_id]
ftlog.warn('QUERY TOO SLOW !!', query_id, msg)
if len(_FAILED_MESSAGES) > 100:
_FAILED_MESSAGES.clear()
else:
ftlog.warn('NOT KNOW of query_id->', query_id, msg)
if d and c:
try:
c.cancel()
d.callback((msg, t, time()))
except:
ftlog.error(msg)