feat: added foreign keys

Author: fstagni

src/DIRAC/WorkloadManagementSystem/DB/PilotAgentsDB.py

@@ -193,19 +193,13 @@ def deletePilots(self, pilotIDs, conn=False):
         if not isinstance(pilotIDs, list):
             return S_ERROR("Input argument is not a List")
 
-        failed = []
-
         result = self._escapeValues(pilotIDs)
         if not result["OK"]:
             return S_ERROR(f"Failed to remove pilot: {result['Value']}")
         stringIDs = ",".join(result["Value"])
-        for table in ["PilotOutput", "JobToPilotMapping", "PilotAgents"]:
-            result = self._update(f"DELETE FROM {table} WHERE PilotID in ({stringIDs})", conn=conn)
-            if not result["OK"]:
-                failed.append(table)
-
-        if failed:
-            return S_ERROR(f"Failed to remove pilot from {', '.join(failed)} tables")
+        result = self._update(f"DELETE FROM PilotAgents WHERE PilotID in ({stringIDs})", conn=conn)
+        if not result["OK"]:
+            return S_ERROR("Failed to remove pilots: ", result["Message"])
         return S_OK(pilotIDs)
 
     ##########################################################################################

src/DIRAC/WorkloadManagementSystem/DB/PilotAgentsDB.sql

@@ -54,8 +54,9 @@ CREATE TABLE `JobToPilotMapping` (
   `PilotID` INT(11) UNSIGNED NOT NULL,
   `JobID` INT(11) UNSIGNED NOT NULL,
   `StartTime` DATETIME NOT NULL,
-  KEY `JobID` (`JobID`),
-  KEY `PilotID` (`PilotID`)
+  PRIMARY KEY (`PilotID`, `JobID`),
+  FOREIGN KEY (`PilotID`) REFERENCES `PilotAgents`(`PilotID`) ON DELETE CASCADE,
+  FOREIGN KEY (`JobID`) REFERENCES `Jobs`(`JobID`) ON DELETE CASCADE
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
 
 DROP TABLE IF EXISTS `PilotOutput`;
@@ -64,4 +65,5 @@ CREATE TABLE `PilotOutput` (
   `StdOutput` MEDIUMTEXT,
   `StdError` MEDIUMTEXT,
   PRIMARY KEY (`PilotID`)
+  FOREIGN KEY (`PilotID`) REFERENCES `PilotAgents`(`PilotID`) ON DELETE CASCADE
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;