def _load_buckets(self): """ 1. Remove existing buckets 2. Create 2 buckets and load documents 3. Create full_doc_list for both buckets :return: """ rest = RestConnection(self.master) json_generator = JsonGenerator() self.standard_gens_load = json_generator.generate_doc_for_aggregate_pushdown(docs_per_day=self.docs_per_day, start=0) self.standard_full_docs_list = self.generate_full_docs_list(self.standard_gens_load) self.default_gens_load = json_generator.generate_doc_for_aggregate_pushdown(docs_per_day=self.docs_per_day, start=0) self.default_full_docs_list = self.generate_full_docs_list(self.default_gens_load) for bucket in self.buckets: rest.flush_bucket(bucket.name) count = 0 while rest.get_bucket_status(bucket.name) != "healthy" and count < 10: log.info( "Bucket {0} Status is {1}. Sleeping...".format(bucket.name, rest.get_bucket_status(bucket.name))) count += 1 self.sleep(15) if bucket.name.startswith("standard"): self.load(self.standard_gens_load, flag=self.item_flag, buckets=[bucket], verify_data=False) if bucket.name.startswith("default"): self.load(self.default_gens_load, flag=self.item_flag, buckets=[bucket], verify_data=False)
def _load_buckets(self): """ 1. Remove existing buckets 2. Create 2 buckets and load documents 3. Create full_doc_list for both buckets :return: """ rest = RestConnection(self.master) json_generator = JsonGenerator() self.standard_gens_load = json_generator.generate_doc_for_aggregate_pushdown(docs_per_day=self.docs_per_day, start=0) self.standard_full_docs_list = self.generate_full_docs_list(self.standard_gens_load) self.default_gens_load = json_generator.generate_doc_for_aggregate_pushdown(docs_per_day=self.docs_per_day, start=0) self.default_full_docs_list = self.generate_full_docs_list(self.default_gens_load) for bucket in self.buckets: rest.flush_bucket(bucket.name) count = 0 while rest.get_bucket_status(bucket.name) != "healthy" and count < 10: log.info("Bucket {0} Status is {1}. Sleeping...".format(bucket.name, rest.get_bucket_status(bucket.name))) count += 1 self.sleep(15) if bucket.name.startswith("standard"): self.load(self.standard_gens_load, flag=self.item_flag, buckets=[bucket], verify_data=False) if bucket.name.startswith("default"): self.load(self.default_gens_load, flag=self.item_flag, buckets=[bucket], verify_data=False)