def __init__(self, options): self.options = options riak_config = { 'bucket_prefix': options['bucket-prefix'], } self.manager = self.get_riak_manager(riak_config) self.mdb = MessageStore(self.manager, None)
def setup_middleware(self): store_prefix = self.config.store_prefix r_config = self.config.redis_manager self.redis = yield TxRedisManager.from_config(r_config) manager = TxRiakManager.from_config(self.config.riak_manager) self.store = MessageStore(manager, self.redis.sub_manager(store_prefix)) self.store_on_consume = self.config.store_on_consume
def setup_middleware(self): store_prefix = self.config.get('store_prefix', 'message_store') r_config = self.config.get('redis_manager', {}) self.redis = yield TxRedisManager.from_config(r_config) manager = TxRiakManager.from_config(self.config.get('riak_manager')) self.store = MessageStore(manager, self.redis.sub_manager(store_prefix)) self.store_on_consume = self.config.get('store_on_consume', True)
def setup_worker(self): config = self.get_static_config() riak = yield TxRiakManager.from_config(config.riak_manager) redis = yield TxRedisManager.from_config(config.redis_manager) self.store = MessageStore(riak, redis) site = build_web_site({ config.web_path: MessageStoreResource(self.store), config.health_path: httprpc.HttpRpcHealthResource(self), }) self.addService( StreamServerEndpointService(config.twisted_endpoint, site))
class TestMessageStoreBase(VumiTestCase): @inlineCallbacks def setUp(self): self.persistence_helper = self.add_helper( PersistenceHelper(use_riak=True)) try: from vumi.components.message_store import MessageStore except ImportError, e: import_skip(e, 'riak') self.redis = yield self.persistence_helper.get_redis_manager() self.manager = self.persistence_helper.get_riak_manager() self.store = MessageStore(self.manager, self.redis) self.msg_helper = self.add_helper(MessageHelper())
def startWorker(self): web_path = self.config['web_path'] web_port = int(self.config['web_port']) health_path = self.config['health_path'] riak = yield TxRiakManager.from_config(self.config['riak_manager']) redis = yield TxRedisManager.from_config(self.config['redis_manager']) self.store = MessageStore(riak, redis) self.webserver = self.start_web_resources([ (MessageStoreAPI(self.store), web_path), (httprpc.HttpRpcHealthResource(self), health_path), ], web_port)
def setUp(self): self.persistence_helper = self.add_helper( PersistenceHelper(use_riak=True, is_sync=False)) self.msg_helper = self.add_helper(MessageHelper()) # Since we're never loading the actual objects, we can't detect # tombstones. Therefore, each test needs its own bucket prefix. config = self.persistence_helper.mk_config({})["riak_manager"].copy() config["bucket_prefix"] = "%s-%s" % ( uuid4().hex, config["bucket_prefix"]) self.riak_manager = self.persistence_helper.get_riak_manager(config) self.redis_manager = yield self.persistence_helper.get_redis_manager() self.mdb = MessageStore(self.riak_manager, self.redis_manager) self.expected_bucket_prefix = "bucket" self.default_args = [ "-b", self.expected_bucket_prefix, ]
def setUp(self): self.persistence_helper = self.add_helper( PersistenceHelper(use_riak=True, is_sync=False)) self.msg_helper = self.add_helper(MessageHelper()) # Since we're never loading the actual objects, we can't detect # tombstones. Therefore, each test needs its own bucket prefix. self.expected_bucket_prefix = "bucket-%s" % (uuid4().hex,) self.riak_manager = self.persistence_helper.get_riak_manager({ "bucket_prefix": self.expected_bucket_prefix, }) self.add_cleanup(self.riak_manager.close_manager) self.redis_manager = yield self.persistence_helper.get_redis_manager() self.mdb = MessageStore(self.riak_manager, self.redis_manager) self.default_args = [ "-b", self.expected_bucket_prefix, ]
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
class MessageStoreCacheTestCase(VumiTestCase): start_batch = True @inlineCallbacks def setUp(self): self.persistence_helper = self.add_helper( PersistenceHelper(use_riak=True)) try: from vumi.components.message_store import MessageStore except ImportError, e: import_skip(e, 'riak') self.redis = yield self.persistence_helper.get_redis_manager() self.manager = yield self.persistence_helper.get_riak_manager() self.store = yield MessageStore(self.manager, self.redis) self.cache = self.store.cache self.batch_id = 'a-batch-id' if self.start_batch: yield self.cache.batch_start(self.batch_id) self.msg_helper = self.add_helper(MessageHelper())