fix: faster queries for DISTINCT values

Author: fstagni

src/DIRAC/WorkloadManagementSystem/Agent/JobCleaningAgent.py

@@ -87,7 +87,7 @@ def initialize(self):
 
     def _getAllowedJobTypes(self):
         """Get valid jobTypes"""
-        result = self.jobDB.getDistinctJobAttributes("JobType")
+        result = self.jobDB._query("SELECT DISTINCT JobType FROM JobsHistorySummary")
         if not result["OK"]:
             return result
         cleanJobTypes = []

src/DIRAC/WorkloadManagementSystem/Agent/test/Test_Agent_JobCleaningAgent.py

@@ -26,9 +26,7 @@ def jca(mocker):
         create=True,
     )
     mocker.patch("DIRAC.WorkloadManagementSystem.Agent.JobCleaningAgent.AgentModule.am_getOption", return_value=mockAM)
-    mocker.patch(
-        "DIRAC.WorkloadManagementSystem.Agent.JobCleaningAgent.JobDB.getDistinctJobAttributes", side_effect=mockReply
-    )
+    mocker.patch("DIRAC.WorkloadManagementSystem.Agent.JobCleaningAgent.JobDB._query", side_effect=mockReply)
     mocker.patch("DIRAC.WorkloadManagementSystem.Agent.JobCleaningAgent.JobDB.selectJobs", side_effect=mockReply)
     mocker.patch("DIRAC.WorkloadManagementSystem.Agent.JobCleaningAgent.JobDB.__init__", side_effect=mockNone)
     mocker.patch("DIRAC.WorkloadManagementSystem.Agent.JobCleaningAgent.ReqClient", return_value=mockNone)

src/DIRAC/WorkloadManagementSystem/DB/JobDB.py

@@ -1197,7 +1197,7 @@ def getSiteSummary(self):
         waitingList = ['"Submitted"', '"Assigned"', '"Waiting"', '"Matched"']
         waitingString = ",".join(waitingList)
 
-        result = self.getDistinctJobAttributes("Site")
+        result = self._query("SELECT DISTINCT Site FROM JobsHistorySummary")
         if not result["OK"]:
             return result