def save_data(self, agent_id, data):
# registering the agent as alive
hostname = data.get('hostname', '?')
agent_pid = agent_id.split('-')[-1]
self.register_agent({
'pid': agent_pid,
'hostname': hostname,
'agent_id': agent_id
})
if agent_id in self._runs:
data['run_id'], data['started'] = self._runs[agent_id]
else:
# this means we are receiving data from an agent that's
# no longer associated with the run, so
# we want to associate it back
self._associate(data.get('run_id'), [agent_id])
if data.get('data_type') == 'batch':
for data_type, message in unbatch(data):
message['data_type'] = data_type
callback = functools.partial(self._db.add, message)
self.loop.add_callback(callback)
else:
self._db.add(data)
def save_data(self, agent_id, data):
if agent_id in self._runs:
data['run_id'], data['started'] = self._runs[agent_id]
if data.get('data_type') == 'batch':
for data_type, message in unbatch(data):
message['data_type'] = data_type
callback = functools.partial(self._db.add, message)
self.loop.add_callback(callback)
else:
self._db.add(data)
def save_data(self, agent_id, data):
if agent_id in self._runs:
data['run_id'], data['started'] = self._runs[agent_id]
if data.get('data_type') == 'batch':
for data_type, message in unbatch(data):
message['data_type'] = data_type
callback = functools.partial(self._db.add, message)
self.loop.add_callback(callback)
else:
self._db.add(data)
def save_data(self, agent_id, data):
# registering the agent as alive
hostname = data.get('hostname', '?')
agent_pid = agent_id.split('-')[-1]
self.register_agent({'pid': agent_pid, 'hostname': hostname,
'agent_id': agent_id})
if agent_id in self._runs:
data['run_id'], data['started'] = self._runs[agent_id]
else:
# this means we are receiving data from an agent that's
# no longer associated with the run, so
# we want to associate it back
self._associate(data.get('run_id'), [agent_id])
if data.get('data_type') == 'batch':
for data_type, message in unbatch(data):
message['data_type'] = data_type
callback = functools.partial(self._db.add, message)
self.loop.add_callback(callback)
else:
self._db.add(data)
def batch(self, **args):
for field, message in unbatch(args):
if hasattr(self, field):
getattr(self, field)(**message)
def batch(self, **args):
for field, message in unbatch(args):
if hasattr(self, field):
getattr(self, field)(**message)
