def change_env_variables(self):
for server in self.cluster.servers:
remote_client = RemoteMachineShellConnection(server)
vb_on_node, _ = remote_client.execute_command("grep ^COUCHBASE_NUM_VBUCKETS \
/opt/couchbase/bin/couchbase-server | cut -d \"=\" -f 2",)
self.log.debug("Current vBuckets on node %s: %s" % (server, vb_on_node))
if vb_on_node:
vb_on_node = int(vb_on_node[0])
else:
vb_on_node = 1024
if self.vbuckets != vb_on_node or self.upr is not None:
env_dict = {}
if self.vbuckets:
env_dict["COUCHBASE_NUM_VBUCKETS"] = self.vbuckets
if self.upr is not None:
if self.upr:
env_dict["COUCHBASE_REPL_TYPE"] = "upr"
else:
env_dict["COUCHBASE_REPL_TYPE"] = "tap"
if len(env_dict) >= 1:
remote_client.change_env_variables(env_dict)
remote_client.disconnect()
self.log.debug("========= CHANGED ENVIRONMENT SETTING ===========")
self.log.debug("Wait for all the services to come up after "
"change_env_vars update")
sleep(10, log_type="infra")
def change_env_variables(self):
if self.vbuckets != 1024 or self.upr != None:
for server in self.cluster.servers:
dict = {}
if self.vbuckets:
dict["COUCHBASE_NUM_VBUCKETS"] = self.vbuckets
if self.upr != None:
if self.upr:
dict["COUCHBASE_REPL_TYPE"] = "upr"
else:
dict["COUCHBASE_REPL_TYPE"] = "tap"
if len(dict) >= 1:
remote_client = RemoteMachineShellConnection(server)
remote_client.change_env_variables(dict)
remote_client.disconnect()
self.log.info("========= CHANGED ENVIRONMENT SETTING ===========")
