buildgrid 0.3.3__py3-none-any.whl → 0.3.4__py3-none-any.whl

buildgrid/server/scheduler/impl.py

@@ -2464,14 +2464,16 @@ class Scheduler:
         action_result.execution_metadata.worker_completed_timestamp.FromDatetime(job.worker_completed_timestamp or now)
         response = ExecuteResponse(result=action_result, cached_result=False, status=status)
 
-        job.result = digest_to_string(self.storage.put_message(response))
+        with instance_context(job.instance_name):
+            job.result = digest_to_string(self.storage.put_message(response))
 
         self._update_job_timestamps(session, job, action_result.execution_metadata)
 
         if self.action_cache and result and not job.do_not_cache:
             action_digest = string_to_digest(job.action_digest)
             try:
-                self.action_cache.update_action_result(action_digest, action_result)
+                with instance_context(job.instance_name):
+                    self.action_cache.update_action_result(action_digest, action_result)
                 LOGGER.debug(
                     "Stored action result in ActionCache.",
                     tags=dict(action_result=action_result, digest=action_digest),
@@ -2662,7 +2664,9 @@ class Scheduler:
         return metadata
 
     def _fetch_execution_stats(
-        self, auxiliary_metadata: RepeatedCompositeFieldContainer[ProtoAny]
+        self,
+        auxiliary_metadata: RepeatedCompositeFieldContainer[ProtoAny],
+        instance_name: str,
     ) -> ExecutionStatistics | None:
         """Fetch ExecutionStatistics from Storage
         ProtoAny[Digest] -> ProtoAny[ExecutionStatistics]
@@ -2675,7 +2679,8 @@ class Scheduler:
             try:
                 aux_metadata_any.Unpack(aux_metadata_digest)
                 # Get the blob from CAS
-                execution_stats_any = self.storage.get_message(aux_metadata_digest, ProtoAny)
+                with instance_context(instance_name):
+                    execution_stats_any = self.storage.get_message(aux_metadata_digest, ProtoAny)
                 # Get the wrapped ExecutionStatistics
                 if execution_stats_any and execution_stats_any.Is(ExecutionStatistics.DESCRIPTOR):
                     execution_stats = ExecutionStatistics()
@@ -2739,7 +2744,7 @@ class Scheduler:
         if self.metering_client is None or len(execution_metadata.auxiliary_metadata) == 0:
             return
 
-        execution_stats = self._fetch_execution_stats(execution_metadata.auxiliary_metadata)
+        execution_stats = self._fetch_execution_stats(execution_metadata.auxiliary_metadata, instance_name)
         if execution_stats is None:
             return
         usage = Usage(
@@ -2809,7 +2814,8 @@ class Scheduler:
         try:
             # BuildGrid doesn't store action_result in CAS, but if we push it as an asset
             # we need it to be accessible
-            digest = self.storage.put_message(action_result)
+            with instance_context(instance_name):
+                digest = self.storage.put_message(action_result)
 
             uri = DIGEST_URI_TEMPLATE.format(digest_hash=digest.hash)
             qualifier = {"resource_type": PROTOBUF_MEDIA_TYPE}

buildgrid/server/version.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 
 
-__version__ = "0.3.3"
+__version__ = "0.3.4"

{buildgrid-0.3.3.dist-info → buildgrid-0.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: buildgrid
-Version: 0.3.3
+Version: 0.3.4
 Summary: A remote execution service
 License: Apache License, Version 2.0
 Project-URL: Homepage, https://buildgrid.build

{buildgrid-0.3.3.dist-info → buildgrid-0.3.4.dist-info}/RECORD

@@ -208,7 +208,7 @@ buildgrid/server/servicer.py,sha256=oqU9MaSxxHTDmSxobFTo9YmJctaUCklE2Dj-vfYWKkc,
 buildgrid/server/settings.py,sha256=I1UK-g4_GwkX0nGC3hdGS3Cc0Rh1HTYDv2aHU2sq_eU,5604
 buildgrid/server/threading.py,sha256=4QKQYev2KoO2Q-S_OyaoR9qpWyDTVzGMWVe9o2a1yIU,4743
 buildgrid/server/types.py,sha256=xG3bx64pbWMuEwXLuI0o8c2unt2rU2C4zsmUfmMT12c,1323
-buildgrid/server/version.py,sha256=ZFnZSeeeNvUxJJyB2m9gdtWL-3UIBuVI7L2U-mwR2NU,603
+buildgrid/server/version.py,sha256=4k5OkV-7Atvya4h3lD2iv9J9jKDYco6vdVsLmuievhM,603
 buildgrid/server/actioncache/__init__.py,sha256=g9lb8Sn7NY5KOjkMr9GQoJovCVDEg_Fxz_EhdDbhP1I,579
 buildgrid/server/actioncache/instance.py,sha256=UCR7ZGkv4fJOXjeIILMAdTSFWcGgBSYlBg8fMaPJpaI,3139
 buildgrid/server/actioncache/service.py,sha256=WcikJAzFYOYX-tgiOfGGcOnPoubrCd4yP-EhKCHEW0c,2021
@@ -331,7 +331,7 @@ buildgrid/server/scheduler/__init__.py,sha256=arCg8LWFATeX1tj-s0keVYP8p3wwrrUlCV
 buildgrid/server/scheduler/assigner.py,sha256=wHPAhyiQxYABZJXaUc2g5yFzM78Z0U5nvGV3X9h5pCM,10512
 buildgrid/server/scheduler/cohorts.py,sha256=L_5YZRiVOwPPGStfqnnQXknO5Ja-SC0vq0xjw4XgP-I,1426
 buildgrid/server/scheduler/events.py,sha256=cM7Z7Htr2pYKhltJxfg1YRo0q524yZaGm8yXvRehivk,1453
-buildgrid/server/scheduler/impl.py,sha256=BT5JIdh7B6-pw1OHxnhJjZgQ1sM3T1UxakXoc6ppBxs,136236
+buildgrid/server/scheduler/impl.py,sha256=BOeR89gJfZ1L9BPzaUoxCUmWqvJGBYauIbphMprrs3U,136516
 buildgrid/server/scheduler/notifier.py,sha256=22ZsKwyf2oQirAjrwROkvgvr4C_TMUNyhOmtro4uM4I,7121
 buildgrid/server/scheduler/properties.py,sha256=2GydX8KUy9MFv1_JznIkGfWE_wOS0m_XapSv6Gp4pCM,11260
 buildgrid/server/sql/__init__.py,sha256=zbeeRP9BEeDzR-Mx2Ip6SUr49J8eeXsuREgljJTrHkk,579
@@ -366,9 +366,9 @@ buildgrid/server/utils/async_lru_cache.py,sha256=iLKeRPoZtZb1wC5AtcyQm8Wt0Bx-KZm
 buildgrid/server/utils/bots.py,sha256=c8hn7tbCecru-m2wicRmtKU5v5rSZPGlk97Yc6eUHgQ,1729
 buildgrid/server/utils/cancellation.py,sha256=pNETzKNoXg0AsXOXKCcLWlFl7SVKdkKinlqWl7MesRA,1703
 buildgrid/server/utils/digests.py,sha256=YNrWeHdbNp7OVTcsInjs30C33z_t9GQ_noMd14bpqPQ,2424
-buildgrid-0.3.3.dist-info/licenses/LICENSE,sha256=swa3Vs7GgALaG9p-e05M-WLkhd_U9QknacNkyVZ85xA,11338
-buildgrid-0.3.3.dist-info/METADATA,sha256=dshu6eNHOz_JPLV3tM3gGitvI9BaGiWpgTWvv2sqjb0,7086
-buildgrid-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-buildgrid-0.3.3.dist-info/entry_points.txt,sha256=uyFAXiR9d6EDfSA5vWT8xskz6xalt4PdTuRruT6Q8rk,49
-buildgrid-0.3.3.dist-info/top_level.txt,sha256=T6TYhI_k6NTm2871tIxGCyBIqzlKxylgF9KDLU0Hi7o,10
-buildgrid-0.3.3.dist-info/RECORD,,
+buildgrid-0.3.4.dist-info/licenses/LICENSE,sha256=swa3Vs7GgALaG9p-e05M-WLkhd_U9QknacNkyVZ85xA,11338
+buildgrid-0.3.4.dist-info/METADATA,sha256=YkM4cIQyoDCPVLbBqsCh5DdB6shR2WJ1863d2XZUtwQ,7086
+buildgrid-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+buildgrid-0.3.4.dist-info/entry_points.txt,sha256=uyFAXiR9d6EDfSA5vWT8xskz6xalt4PdTuRruT6Q8rk,49
+buildgrid-0.3.4.dist-info/top_level.txt,sha256=T6TYhI_k6NTm2871tIxGCyBIqzlKxylgF9KDLU0Hi7o,10
+buildgrid-0.3.4.dist-info/RECORD,,