Commit 44195d4a authored by Cristiano Urban's avatar Cristiano Urban
Browse files

Added code to handle wrong job ID. 



Signed-off-by: default avatarCristiano Urban <cristiano.urban@inaf.it>
parent 6ba32625

transfer_service/job_amqp_server.py

+16 −6
Original line number Diff line number Diff line
@@ -40,18 +40,28 @@ class JobAMQPServer(AMQPServer): 
        elif requestBody["requestType"] == "JOB_INFO":

            self.jobId = requestBody["jobId"]

            self.dbConn.connect()

            if self.dbConn.jobExists(self.jobId):

                result = self.dbConn.getJobInfo(self.jobId)

            self.dbConn.disconnect()

                response = { "responseType": "LST_INFO_DONE", "jobInfo": result }

            else:

                response = { "responseType": "ERROR",

                             "errorCode": 2,

                             "errorMsg": "Wrong job ID." }

            self.dbConn.disconnect()

        elif requestBody["requestType"] == "JOB_RESULTS":

            self.jobId = requestBody["jobId"]

            self.dbConn.connect()

            if self.dbConn.jobExists(self.jobId):

                result = self.dbConn.getJobResults(self.jobId)

            self.dbConn.disconnect()

                response = { "responseType": "LST_RESULTS_DONE", "jobResults": result }

            else:

                response = { "responseType": "ERROR",

                             "errorCode": 2,

                             "errorMsg": "Wrong job ID." }

            self.dbConn.disconnect()            

        else:

            response = { "responseType": "ERROR",

                         "errorCode": 3,

                         "errorMsg": "Unkown request type." }

        return response