Commit 54ae70e8 authored by Cristiano Urban's avatar Cristiano Urban
Browse files

Added subclassess of AMQPServer. 



Signed-off-by: default avatarCristiano Urban <cristiano.urban.slack@gmail.com>
parent 11423a8f

docker/transfer_service/abort_job_amqp_server.py

0 → 100644
+15 −0
Original line number Diff line number Diff line 
from amqp_server import AMQPServer





class AbortJobAMQPServer(AMQPServer):

  

    def __init__(self, host, queue):

        self.type = "abort"

        super(AbortJobAMQPServer, self).__init__(host, queue)      



    def execute_callback(self, requestBody):

        return 42

      

    def run(self):

        print(f"Starting AMQP server of type {self.type}...")

        super(AbortJobAMQPServer, self).run()

docker/transfer_service/get_job_amqp_server.py

0 → 100644
+15 −0
Original line number Diff line number Diff line 
from amqp_server import AMQPServer





class GetJobAMQPServer(AMQPServer):

  

    def __init__(self, host, queue):

        self.type = "poll"

        super(GetJobAMQPServer, self).__init__(host, queue)      



    def execute_callback(self, requestBody):

        return 42

      

    def run(self):

        print(f"Starting AMQP server of type {self.type}...")

        super(GetJobAMQPServer, self).run()

docker/transfer_service/start_job_amqp_server.py

0 → 100644
+21 −0
Original line number Diff line number Diff line 
from amqp_server import AMQPServer

from job import Job



class StartJobAMQPServer(AMQPServer):

  

    def __init__(self, host, queue):

        self.type = "start"

        super(StartJobAMQPServer, self).__init__(host, queue)      



    def execute_callback(self, requestBody):

        job = Job()

        job.setInfo(requestBody)

        job.setPhase("RUN")

        self.jobCache.set(job)

        redis_res = self.jobCache.get(job.jobID)

        print(f"Redis response: {redis_res}")

        return redis_res

      

    def run(self):

        print(f"Starting AMQP server of type {self.type}...")

        super(StartJobAMQPServer, self).run()
 
 No newline at end of file