Merge pull request DIRACGrid#7190 from simon-mazenoux/feat-make-tqdb-…

…vo-aware [8.1] Make TaskQueueDB VO aware for diracx transition
chaen · Sep 25, 2023 · d2a29a7 · d2a29a7
2 parents 4a3efe8 + 9ce9a86
commit d2a29a7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/DIRAC/WorkloadManagementSystem/DB/TaskQueueDB.py b/src/DIRAC/WorkloadManagementSystem/DB/TaskQueueDB.py
@@ -97,6 +97,7 @@ def __initializeDB(self):
                 "Owner": "VARCHAR(255) NOT NULL",
                 "OwnerDN": "VARCHAR(255)",
                 "OwnerGroup": "VARCHAR(32) NOT NULL",
+                "VO": "VARCHAR(32) NOT NULL",
                 "CPUTime": "BIGINT(20) UNSIGNED NOT NULL",
                 "Priority": "FLOAT NOT NULL",
                 "Enabled": "TINYINT(1) NOT NULL DEFAULT 0",
@@ -251,6 +252,8 @@ def __createTaskQueue(self, tqDefDict, priority=1, connObj=False):
         for field in singleValueDefFields:
             sqlSingleFields.append(field)
             sqlValues.append(tqDefDict[field])
+        sqlSingleFields.append("VO")
+        sqlValues.append(Registry.getVOForGroup(tqDefDict["OwnerGroup"]))
         # Insert the TQ Disabled
         sqlSingleFields.append("Enabled")
         sqlValues.append("0")