def _remove_key_from_group(self, key, group_id): """ Creates task for gatling gun on destination group, updates key in meta database """ if not self.dryrun: cache_task_manager.put_task( self._gatlinggun_task(key, group_id, [], 'remove')) key['cache_groups'].remove(group_id) if len(key['cache_groups']): self.keys_db.update({'id': key['id'], 'couple': key['couple']}, key) else: self.keys_db.remove({'id': key['id'], 'couple': key['couple']})
def _add_key_to_group(self, key, group_id, data_groups, tx_rate, size): """ Creates task for gatling gun on destination group, updates key in meta database """ task = self._gatlinggun_task(key, group_id, data_groups, 'add', tx_rate=tx_rate, size=size) if not self.dryrun: cache_task_manager.put_task(task) logger.debug('Key {}, task for gatlinggun created for cache ' 'group {}'.format(key['id'], group_id)) key['cache_groups'].append(group_id) key['expand_ts'] = int(time.time()) self.keys_db.update({'id': key['id'], 'couple': key['couple']}, key, upsert=True) return task