Commit 4227e293 authored by Cristiano Urban's avatar Cristiano Urban
Browse files

Added 'JobScheduler' class. 



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

transfer_service/job_scheduler.py

0 → 100644
+16 −0
Original line number Diff line number Diff line 
from retrieve_preprocessor import RetrievePreprocessor

from store_preprocessor import StorePreprocessor

from store_executor import StoreExecutor





class JobScheduler(object):



    def __init__(self):

        self.retrievePreprocessor = RetrievePreprocessor()

        self.storePreprocessor = StorePreprocessor()

        self.storeExecutor = StoreExecutor()



    def start(self):

        self.retrievePreprocessor.start()

        self.storePreprocessor.start()

        self.storeExecutor.start()

transfer_service/transfer_service.py

+5 −8
Original line number Diff line number Diff line 
import time

import os



from job_handler import JobHandler

from store_preprocessor import StorePreprocessor

from store_executor import StoreExecutor

from config import Config

from job_handler import JobHandler

from job_scheduler import JobScheduler





class TransferService(object):
@@ -13,10 +12,9 @@ class TransferService(object): 
        config = Config("vos_ts.conf")

        self.params = config.loadSection("amqp")

        self.jobHandler = JobHandler(self.params["host"], self.params.getint("port"))

        self.storePreprocessor = StorePreprocessor()

        self.storeExecutor = StoreExecutor()

        self.jobScheduler = JobScheduler()



        # PullFromVOSpace (via REST API)

        # PullToVOSpace (via REST API)

        self.jobHandler.addAMQPServer('start', 'start_job_queue')

        self.jobHandler.addAMQPServer('poll', 'poll_job_queue')

        self.jobHandler.addAMQPServer('abort', 'abort_job_queue')
@@ -25,8 +23,7 @@ class TransferService(object): 
        self.jobHandler.addAMQPServer('store', 'store_job_queue')        



    def start(self):

        self.storePreprocessor.start()

        self.storeExecutor.start()

        self.jobScheduler.start()

        self.jobHandler.start()



ts = TransferService()