Use Rucio's site-closeness algorithm for replica ordering

ssl-hep · Sep 25, 2024 · 21ab948 · 21ab948
1 parent f617d64
commit 21ab948
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/did_finder_rucio/src/rucio_did_finder/rucio_adapter.py b/did_finder_rucio/src/rucio_did_finder/rucio_adapter.py
@@ -46,6 +46,9 @@ def __init__(self, did_client, replica_client, report_logical_files=False):
 
     def client_location(self):
         client_location = {}
+        # setting the site actually seems to work
+        if 'SITE_NAME' in os.environ:
+            client_location['site'] = os.environ['SITE_NAME']
         latitude = os.environ.get('RUCIO_LATITUDE')
         longitude = os.environ.get('RUCIO_LONGITUDE')
         if latitude and longitude:

diff --git a/helm/servicex/templates/did-finder-rucio/deployment.yaml b/helm/servicex/templates/did-finder-rucio/deployment.yaml
@@ -42,6 +42,10 @@ spec:
             value: "{{ .Values.didFinder.rucio.servicex_latitude }}"
           - name: RUCIO_LONGITUDE
             value: "{{ .Values.didFinder.rucio.servicex_longitude }}"
+          {{- if .Values.didFinder.rucio.site }}
+          - name: SITE_NAME
+            value: "{{ .Values.didFinder.rucio.site }}"
+          {{- end }}
           - name: INSTANCE_NAME
             value: {{ .Release.Name }}
         volumeMounts: