Commit dae4c828 authored by Cristiano Urban's avatar Cristiano Urban
Browse files

Added basic logging functionalities.

parent ac9c04fc
Loading
Loading
Loading
Loading
+15 −1
Original line number Diff line number Diff line
@@ -7,6 +7,7 @@
#

import json
import logging
import os
import sys

@@ -36,6 +37,19 @@ class DataRPCServer(RedisRPCServer):
        self.storageStorePath = self.params["store_path"]
        self.params = config.loadSection("scheduling")
        self.maxPendingJobs = self.params.getint("max_pending_jobs")
        params = config.loadSection("logging")
        self.logger = logging.getLogger("DataRPCServer")
        logLevel = "logging." + params["log_level"]
        logDir = params["log_dir"]
        logFile = logDir + '/' + "data_rpc_server.log"
        self.logger.setLevel(eval(logLevel))
        logFileHandler = logging.FileHandler(logFile)
        logStreamHandler = logging.StreamHandler()
        logFormatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
        logFileHandler.setFormatter(logFormatter)
        logStreamHandler.setFormatter(logFormatter)
        self.logger.addHandler(logFileHandler)
        self.logger.addHandler(logStreamHandler)
        self.pendingQueueWrite = JobQueue("write_pending")
        self.systemUtils = SystemUtils()
        super(DataRPCServer, self).__init__(host, port, db, rpcQueue)
@@ -133,6 +147,6 @@ class DataRPCServer(RedisRPCServer):
                os.chmod(os.path.join(folder, f), 0o555)

    def run(self):
        print(f"Starting RPC server of type {self.type}...")
        self.logger.info(f"Starting RPC server of type {self.type}...")
        super(DataRPCServer, self).run()