def __init__(self, meta_configs, appname, collection_name="talib_states"): """ :meta_configs: dict like and contains checkpoint_dir, session_key, server_uri etc :app_name: the name of the app :collection_name: the collection name to be used. Don"t use other method to visit the collection if you are using StateStore to visit it. """ super(StateStore, self).__init__(meta_configs, appname) # State cache is a dict from _key to value self._states_cache = {} self._kv_client = None self._collection = collection_name self._kv_client = kvc.KVClient(meta_configs["server_uri"], meta_configs["session_key"]) kvc.create_collection(self._kv_client, self._collection, self._appname) self._load_states_cache()
def __init__(self, meta_configs, appname, collection_name, http=None): """ :meta_configs: dict like and contains checkpoint_dir, session_key, server_uri etc :app_name: the name of the app :collection_name: the collection name to be used. Don"t use other method to visit the collection if you are using StateStore to visit it. """ super(StateStore, self).__init__(meta_configs, appname) self._kv_client = None if not collection_name: self._collection = appname else: self._collection = collection_name self._kv_client = kvc.KVClient(meta_configs["server_uri"], meta_configs["session_key"], http=http) kvc.create_collection(self._kv_client, self._collection, self._appname)