def __init__(self, options):
self.options = options
self.config = yaml.safe_load(open(options['config'], "rb"))
self.pools = self.config.get('pools', {})
redis = RedisManager.from_config(self.config.get('redis_manager', {}))
self.tagpool = TagpoolManager(
redis.sub_manager(self.config.get('tagpool_prefix', 'vumi')))
def setup_backend(self, config):
self.redis = RedisManager.from_config(config['redis_manager'])
self.riak = RiakManager.from_config(config['riak_manager'])
# this prefix is hard coded in VumiApi
self.tagpool = TagpoolManager(
self.redis.sub_manager('tagpool_store'))
self.api = VumiApi(self.riak, self.redis)
def setUp(self):
self.persistence_helper = self.add_helper(PersistenceHelper())
self.redis = yield self.persistence_helper.get_redis_manager()
self.tagpool = TagpoolManager(self.redis)
site = Site(TagpoolApiServer(self.tagpool))
self.server = yield reactor.listenTCP(0, site, interface='127.0.0.1')
self.add_cleanup(self.server.loseConnection)
addr = self.server.getHost()
self.proxy = Proxy("http://%s:%d/" % (addr.host, addr.port))
yield self.setup_tags()
def setup_worker(self):
config = self.get_static_config()
self.redis_manager = yield TxRedisManager.from_config(
config.redis_manager)
tagpool = TagpoolManager(self.redis_manager)
rpc = TagpoolApiServer(tagpool)
addIntrospection(rpc)
site = build_web_site({
config.web_path:
rpc,
config.health_path:
httprpc.HttpRpcHealthResource(self),
})
self.addService(
StreamServerEndpointService(config.twisted_endpoint, site))
def __init__(self, manager, redis, sender=None, metric_publisher=None):
# local import to avoid circular import since
# go.api.go_api needs to access VumiApi
from go.api.go_api.session_manager import SessionManager
self.manager = manager
self.redis = redis
self.tpm = TagpoolManager(self.redis.sub_manager('tagpool_store'))
self.mdb = MessageStore(self.manager,
self.redis.sub_manager('message_store'))
self.account_store = AccountStore(self.manager)
self.token_manager = TokenManager(
self.redis.sub_manager('token_manager'))
self.session_manager = SessionManager(
self.redis.sub_manager('session_manager'))
self.mapi = sender
self.metric_publisher = metric_publisher
def setUp(self):
self.persistence_helper = self.add_helper(PersistenceHelper())
self.redis = yield self.persistence_helper.get_redis_manager()
yield self.redis._purge_all() # Just in case
self.tpm = TagpoolManager(self.redis)
