_scheduler: Migrate to Context.artifactcache

buildstream/_scheduler/jobs/cachesizejob.py

@@ -17,7 +17,6 @@
 #        Tristan Daniël Maat <tristan.maat@codethink.co.uk>
 #
 from .job import Job
-from ..._platform import Platform
 
 
 class CacheSizeJob(Job):
@@ -25,8 +24,8 @@ class CacheSizeJob(Job):
         super().__init__(*args, **kwargs)
         self._complete_cb = complete_cb
 
-        platform = Platform.get_platform()
-        self._artifacts = platform.artifactcache
+        context = self._scheduler.context
+        self._artifacts = context.artifactcache
 
     def child_process(self):
         return self._artifacts.compute_cache_size()

buildstream/_scheduler/jobs/cleanupjob.py

@@ -17,15 +17,14 @@
 #        Tristan Daniël Maat <tristan.maat@codethink.co.uk>
 #
 from .job import Job
-from ..._platform import Platform
 
 
 class CleanupJob(Job):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
-        platform = Platform.get_platform()
-        self._artifacts = platform.artifactcache
+        context = self._scheduler.context
+        self._artifacts = context.artifactcache
 
     def child_process(self):
         return self._artifacts.clean()

buildstream/_scheduler/queues/buildqueue.py

@@ -24,7 +24,6 @@ from . import Queue, QueueStatus
 from ..jobs import ElementJob
 from ..resources import ResourceType
 from ..._message import MessageType
-from ..._platform import Platform
 
 
 # A queue which assembles elements
@@ -94,8 +93,8 @@ class BuildQueue(Queue):
         # as returned from Element._assemble() to the estimated
         # artifact cache size
         #
-        platform = Platform.get_platform()
-        artifacts = platform.artifactcache
+        context = self._scheduler.context
+        artifacts = context.artifactcache
 
         artifacts.add_artifact_size(artifact_size)
 

buildstream/_scheduler/scheduler.py

@@ -29,7 +29,6 @@ from contextlib import contextmanager
 # Local imports
 from .resources import Resources, ResourceType
 from .jobs import CacheSizeJob, CleanupJob
-from .._platform import Platform
 
 
 # A decent return code for Scheduler.run()
@@ -348,8 +347,8 @@ class Scheduler():
     #       which will report the calculated cache size.
     #
     def _run_cleanup(self, cache_size):
-        platform = Platform.get_platform()
-        artifacts = platform.artifactcache
+        context = self.context
+        artifacts = context.artifactcache
 
         if not artifacts.has_quota_exceeded():
             return