def __init__(self,
index,
type,
condition,
mapping,
from_key,
func_set,
process_count=1,
query_interval=3):
"""
初始化方法
:param str index: 事件在es中的index名
:param str type: 事件在es中的type名
:param dict condition: 查询条件
:param dict mapping: 策略名和执行方法的映射
:param str from_key: 关注的事件中的字段名
:param BaseExecFuncSet func_set: 执行方法类
:param int process_count: 进程数量
:param int query_interval: 查询事件的事件间隔
"""
sen = EsCallbackSensor(index, type, condition, query_interval)
dec = decision.KeyMappingDecisionMaker(mapping, from_key)
exe = executor.CallbackExecutor(func_set, process_count)
listener_list = [sen, dec, exe]
for listener in listener_list:
self.add_listener(listener)
def test_callback_executor(self, mock_send):
"""
:return:
"""
mock_send.return_value = None
exe = executor.CallbackExecutor(DemoExecFuncSet())
message = framework.OperationMessage("DECIDED_MESSAGE", "id1",
{".inner_executor_key": "demo"})
ret = exe.execute_message(message)
self.assertDictEqual(ret, {"code": 0})
def __init__(self,
subscribe_condition,
mapping,
from_key,
func_set,
process_count=1):
"""
初始化方法
:param str subscribe_condition: 订阅条件
:param dict mapping: 策略名与执行方法的映射
:param str from_key: 关注的事件中的字段名
:param BaseExecFuncSet func_set: 执行方法类
:param int process_count: 进程数量
"""
sen = MqPushCallbackSensor(subscribe_condition)
dec = decision.KeyMappingDecisionMaker(mapping, from_key)
exe = executor.CallbackExecutor(func_set, process_count)
listener_list = [sen, dec, exe]
for listener in listener_list:
self.add_listener(listener)
def __init__(self,
event_file,
mapping,
from_key,
func_set,
process_count=1,
query_interval=3):
"""
初始化方法
:param str event_file: 存放事件的文件路径
:param dict mapping: 策略名和执行方法的映射
:param str from_key: 关注的事件中的字段名
:param BaseExecFuncSet func_set: 执行方法类
:param int process_count: 进程数
:param int query_interval: 查询事件的时间间隔
"""
sen = LocalCallbackSensor(event_file, query_interval)
dec = decision.KeyMappingDecisionMaker(mapping, from_key)
exe = executor.CallbackExecutor(func_set, process_count)
listener_list = [sen, dec, exe]
for listener in listener_list:
self.add_listener(listener)