Python HerokuWorkerの例

コード例 #1

ファイル: test_worker.py プロジェクト: olingerc/rq

    def test_handle_shutdown_request_no_horse(self):
        """Mutate HerokuWorker so _horse_pid refers to non existent process
        and test handle_warm_shutdown_request"""
        w = HerokuWorker('foo')

        w._horse_pid = 19999
        w.handle_warm_shutdown_request()

コード例 #2

ファイル: test_worker.py プロジェクト: olingerc/rq

    def test_handle_shutdown_request(self):
        """Mutate HerokuWorker so _horse_pid refers to an artificial process
        and test handle_warm_shutdown_request"""
        w = HerokuWorker('foo')

        path = os.path.join(self.sandbox, 'shouldnt_exist')
        p = Process(target=create_file_after_timeout, args=(path, 2))
        p.start()
        self.assertEqual(p.exitcode, None)

        w._horse_pid = p.pid
        w.handle_warm_shutdown_request()
        p.join(2)
        self.assertEqual(p.exitcode, -34)
        self.assertFalse(os.path.exists(path))

コード例 #3

ファイル: test_worker.py プロジェクト: thomasmatecki/rq

    def test_handle_shutdown_request(self, mock_logger_info):
        """Mutate HerokuWorker so _horse_pid refers to an artificial process
        and test handle_warm_shutdown_request"""
        w = HerokuWorker('foo')

        path = os.path.join(self.sandbox, 'shouldnt_exist')
        p = Process(target=create_file_after_timeout, args=(path, 2))
        p.start()
        self.assertEqual(p.exitcode, None)

        w._horse_pid = p.pid
        w.handle_warm_shutdown_request()
        p.join(2)
        # would expect p.exitcode to be -34
        self.assertEqual(p.exitcode, -34)
        self.assertFalse(os.path.exists(path))
        mock_logger_info.assert_called_with('Killed horse pid %s', p.pid)

コード例 #4

import time
from redis import Redis
from rq import Queue, Connection
from rq.worker import HerokuWorker as Worker

listen = ['high', 'default', 'low']

conn = Redis(host='redis', port=6379, db=0)

if __name__ == '__main__':
    with Connection(conn):
        worker = Worker(map(Queue, listen))
        worker.work()

コード例 #5

def connect_worker():
    with Connection(conn):
        worker = Worker(map(Q, listen))
        worker.work()

コード例 #6

ファイル: run-worker.py プロジェクト: gamgi/powerline

                      port=config.REDIS_PORT,
                      db=0,
                      password=config.REDIS_PASSWORD)

        logger.info('attempting to connect to database at {}'.format(
            config.SQLALCHEMY_DATABASE_URI))
        db = create_engine(config.SQLALCHEMY_DATABASE_URI)
        Session = sessionmaker(bind=db)

        # Note: no additional config required due to webhook.py
        # sending required values to telegram
        bot = Bot(config.TELEGRAM_TOKEN)

        # Bind worker to db and redis
        worker.bind(bot, Session, redis, State)

        with Connection(redis):
            rq_worker = Worker(map(Queue, listen))

            # Set final logging things
            logging.getLogger('rq.worker').setLevel(logging.WARNING)
            logging.getLogger('telegram*').setLevel(logging.WARNING)
            # if not config.DEVELOPMENT:
            #    logging.getLogger('transitions').setLevel(logging.WARNING)

            logger.info('worker started')
            rq_worker.work()
    except redis_exceptions.ConnectionError:
        logger.error('Unable to connect to redis')
        sleep(10)

コード例 #7

ファイル: updater.py プロジェクト: produvia/kryptos

def start_worker():
    with Connection(CONN):
        log.info("Starting update worker")
        worker = Worker("updates")
        worker.work()

コード例 #8

ファイル: python_rq_backend.py プロジェクト: ograycode/python-rpc

 def start(self):
     log('Starting worker')
     with Connection(self._redis_client):
         worker = Worker(map(Queue, listen))
         worker.work()