def build_index2(self, build_config='mygene_allspecies', last_build_idx=-1, use_parallel=False, es_host=None, es_index_name=None, noconfirm=False):
"""Build ES index from last successfully-merged mongodb collection.
optional "es_host" argument can be used to specified another ES host, otherwise default ES_HOST.
optional "es_index_name" argument can be used to pass an alternative index name, otherwise same as mongodb collection name
"""
self.load_build_config(build_config)
assert "build" in self._build_config, "Abort. No such build records for config %s" % build_config
last_build = self._build_config['build'][last_build_idx]
logging.info("Last build record:")
logging.info(pformat(last_build))
assert last_build['status'] == 'success', \
"Abort. Last build did not success."
assert last_build['target_backend'] == "mongodb", \
'Abort. Last build need to be built using "mongodb" backend.'
assert last_build.get('stats', None), \
'Abort. Last build stats are not available.'
self._stats = last_build['stats']
assert last_build.get('target', None), \
'Abort. Last build target_collection is not available.'
# Get the source collection to build the ES index
# IMPORTANT: the collection in last_build['target'] does not contain _timestamp field,
# only the "genedoc_*_current" collection does. When "timestamp" is enabled
# in mappings, last_build['target'] collection won't be indexed by ES correctly,
# therefore, we use "genedoc_*_current" collection as the source here:
#target_collection = last_build['target']
target_collection = "genedoc_{}_current".format(build_config)
_db = get_target_db()
target_collection = _db[target_collection]
logging.info("")
logging.info('Source: %s' % target_collection.name)
_mapping = self.get_mapping()
_meta = {}
src_version = self.get_src_version()
if src_version:
_meta['src_version'] = src_version
if getattr(self, '_stats', None):
_meta['stats'] = self._stats
if 'timestamp' in last_build:
_meta['timestamp'] = last_build['timestamp']
if _meta:
_mapping['_meta'] = _meta
es_index_name = es_index_name or target_collection.name
es_idxer = ESIndexer(mapping=_mapping,
es_index_name=es_index_name,
es_host=es_host,
step=5000)
if build_config == 'mygene_allspecies':
es_idxer.number_of_shards = 10 # default 5
es_idxer.check()
if noconfirm or ask("Continue to build ES index?") == 'Y':
es_idxer.use_parallel = use_parallel
#es_idxer.s = 609000
if es_idxer.exists_index(es_idxer.ES_INDEX_NAME):
if noconfirm or ask('Index "{}" exists. Delete?'.format(es_idxer.ES_INDEX_NAME)) == 'Y':
es_idxer.conn.indices.delete(es_idxer.ES_INDEX_NAME)
else:
logging.info("Abort.")
return
es_idxer.create_index()
#es_idxer.delete_index_type(es_idxer.ES_INDEX_TYPE, noconfirm=True)
es_idxer.build_index(target_collection, verbose=False)
def build_index2(self,
build_config='mygene_allspecies',
last_build_idx=-1,
use_parallel=False,
es_host=None,
es_index_name=None,
noconfirm=False):
"""Build ES index from last successfully-merged mongodb collection.
optional "es_host" argument can be used to specified another ES host, otherwise default ES_HOST.
optional "es_index_name" argument can be used to pass an alternative index name, otherwise same as mongodb collection name
"""
self.load_build_config(build_config)
assert "build" in self._build_config, "Abort. No such build records for config %s" % build_config
last_build = self._build_config['build'][last_build_idx]
logging.info("Last build record:")
logging.info(pformat(last_build))
assert last_build['status'] == 'success', \
"Abort. Last build did not success."
assert last_build['target_backend'] == "mongodb", \
'Abort. Last build need to be built using "mongodb" backend.'
assert last_build.get('stats', None), \
'Abort. Last build stats are not available.'
self._stats = last_build['stats']
assert last_build.get('target', None), \
'Abort. Last build target_collection is not available.'
# Get the source collection to build the ES index
# IMPORTANT: the collection in last_build['target'] does not contain _timestamp field,
# only the "genedoc_*_current" collection does. When "timestamp" is enabled
# in mappings, last_build['target'] collection won't be indexed by ES correctly,
# therefore, we use "genedoc_*_current" collection as the source here:
#target_collection = last_build['target']
target_collection = "genedoc_{}_current".format(build_config)
_db = get_target_db()
target_collection = _db[target_collection]
logging.info("")
logging.info('Source: %s' % target_collection.name)
_mapping = self.get_mapping()
_meta = {}
src_version = self.get_src_version()
if src_version:
_meta['src_version'] = src_version
if getattr(self, '_stats', None):
_meta['stats'] = self._stats
if 'timestamp' in last_build:
_meta['timestamp'] = last_build['timestamp']
if _meta:
_mapping['_meta'] = _meta
es_index_name = es_index_name or target_collection.name
es_idxer = ESIndexer(mapping=_mapping,
es_index_name=es_index_name,
es_host=es_host,
step=5000)
if build_config == 'mygene_allspecies':
es_idxer.number_of_shards = 10 # default 5
es_idxer.check()
if noconfirm or ask("Continue to build ES index?") == 'Y':
es_idxer.use_parallel = use_parallel
#es_idxer.s = 609000
if es_idxer.exists_index(es_idxer.ES_INDEX_NAME):
if noconfirm or ask('Index "{}" exists. Delete?'.format(
es_idxer.ES_INDEX_NAME)) == 'Y':
es_idxer.conn.indices.delete(es_idxer.ES_INDEX_NAME)
else:
logging.info("Abort.")
return
es_idxer.create_index()
#es_idxer.delete_index_type(es_idxer.ES_INDEX_TYPE, noconfirm=True)
es_idxer.build_index(target_collection, verbose=False)
