diff --git a/swh/scheduler/backend.py b/swh/scheduler/backend.py --- a/swh/scheduler/backend.py +++ b/swh/scheduler/backend.py @@ -61,6 +61,10 @@ return self._db return BaseDb.from_pool(self._pool) + def put_db(self, db): + if db is not self._db: + db.put_conn() + task_type_keys = [ 'type', 'description', 'backend_name', 'default_interval', 'min_interval', 'max_interval', 'backoff_factor', 'max_queue_length', @@ -121,7 +125,7 @@ 'type', 'arguments', 'next_run', 'policy', 'status', 'retries_left', 'priority' ] - task_keys = task_create_keys + ['id', 'current_interval', 'status'] + task_keys = task_create_keys + ['id', 'current_interval'] @db_transaction() def create_tasks(self, tasks, policy='recurring', db=None, cur=None): diff --git a/swh/scheduler/updater/backend.py b/swh/scheduler/updater/backend.py --- a/swh/scheduler/updater/backend.py +++ b/swh/scheduler/updater/backend.py @@ -40,6 +40,10 @@ return self._db return BaseDb.from_pool(self._pool) + def put_db(self, db): + if db is not self._db: + db.put_conn() + cache_put_keys = ['url', 'cnt', 'last_seen', 'origin_type'] @db_transaction()