feat: add a CE whitelist for remote logging

martynia · Jan 24, 2024 · b341099 · b341099
1 parent b6b9961
commit b341099
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Pilot/pilotTools.py b/Pilot/pilotTools.py
@@ -1221,6 +1221,16 @@ def __initJSON2(self):
         self.loggerTimerInterval = int(pilotOptions.get("RemoteLoggerTimerInterval", self.loggerTimerInterval))
         # logger buffer size in lines:
         self.loggerBufsize = max(1, int(pilotOptions.get("RemoteLoggerBufsize", self.loggerBufsize)))
+        # logger CE white list
+        loggerCEsWhiteList = pilotOptions.get("RemoteLoggerCEsWhiteList")
+        # restrict remote logging to whitelisted CEs ([] or None => no restriction)
+        self.log.debug("JSON: Remote logging CE white list: %s" % loggerCEsWhiteList)
+        if loggerCEsWhiteList is not None:
+            if not isinstance(loggerCEsWhiteList, list):
+                loggerCEsWhiteList = [elem.strip() for elem in loggerCEsWhiteList.split(",")]
+            if self.ceName not in loggerCEsWhiteList:
+                self.pilotLogging = False
+                self.log.debug("JSON: Remote logging disabled for this CE: %s" % self.ceName)
         pilotLogLevel = pilotOptions.get("PilotLogLevel", "INFO")
         if pilotLogLevel.lower() == "debug":
             self.debugFlag = True