Rename GC worker package to gc

workers/gc/gcworker.py

@@ -0,0 +1,19 @@
+from app import app
+from data.database import UseThenDisconnect
+from workers.gc.models_pre_oci import pre_oci_model as model
+from workers.worker import Worker
+
+class GarbageCollectionWorker(Worker):
+  def __init__(self):
+    super(GarbageCollectionWorker, self).__init__()
+    self.add_operation(self._garbage_collection_repos,
+                       app.config.get('GARBAGE_COLLECTION_FREQUENCY', 30))
+
+  def _garbage_collection_repos(self):
+    """ Performs garbage collection on repositories. """
+    with UseThenDisconnect(app.config):
+      model.perform_garbage_collection()
+
+if __name__ == "__main__":
+  worker = GarbageCollectionWorker()
+  worker.start()

workers/gc/models_interface.py

@@ -0,0 +1,13 @@
+from abc import ABCMeta, abstractmethod
+
+from six import add_metaclass
+
+@add_metaclass(ABCMeta)
+class GCWorkerDataInterface(object):
+  """
+  Interface that represents all data store interactions required by the GC worker.
+  """
+  @abstractmethod
+  def perform_garbage_collection(self):
+    """ Performs garbage collection on a single repository, if any can be found with garbage. """
+    pass

workers/gc/models_pre_oci.py

@@ -0,0 +1,21 @@
+import logging
+
+from data.model.repository import (find_repository_with_garbage, garbage_collect_repo,
+                                   get_random_gc_policy)
+
+from workers.gc.models_interface import GCWorkerDataInterface
+
+logger = logging.getLogger(__name__)
+
+class PreOCIModel(GCWorkerDataInterface):
+  def perform_garbage_collection(self):
+    repository = find_repository_with_garbage(get_random_gc_policy())
+    if repository is None:
+      logger.debug('No repository with garbage found')
+      return
+
+    logger.debug('Starting GC of repository #%s (%s)', repository.id, repository.name)
+    garbage_collect_repo(repository)
+    logger.debug('Finished GC of repository #%s (%s)', repository.id, repository.name)
+
+pre_oci_model = PreOCIModel()

workers/gc/test/test_gcworker.py

@@ -0,0 +1,7 @@
+from workers.gc.gcworker import GarbageCollectionWorker
+
+from test.fixtures import *
+
+def test_gc(initialized_db):
+  worker = GarbageCollectionWorker()
+  worker._garbage_collection_repos()