class SGPerfTest(PerfTest):
COLLECTORS = {
'disk': False,
'ns_server': False,
'ns_server_overview': False,
'active_tasks': False,
'syncgateway_stats': True
}
ALL_HOSTNAMES = True
LOCAL_DIR = "YCSB"
def __init__(self, cluster_spec: ClusterSpec, test_config: TestConfig,
verbose: bool):
self.cluster_spec = cluster_spec
self.test_config = test_config
self.memcached = MemcachedHelper(test_config)
self.remote = RemoteHelper(cluster_spec, test_config, verbose)
self.rest = RestHelper(cluster_spec)
# self.build = os.environ.get('SGBUILD') or "0.0.0-000"
self.master_node = next(cluster_spec.masters)
self.build = self.rest.get_sgversion(self.master_node)
self.metrics = MetricHelper(self)
self.reporter = ShowFastReporter(cluster_spec, test_config, self.build)
if self.test_config.test_case.use_workers:
self.worker_manager = WorkerManager(cluster_spec, test_config,
verbose)
self.settings = self.test_config.access_settings
self.settings.syncgateway_settings = self.test_config.syncgateway_settings
self.profiler = Profiler(cluster_spec, test_config)
self.cluster = ClusterManager(cluster_spec, test_config)
self.target_iterator = TargetIterator(cluster_spec, test_config)
self.monitor = Monitor(cluster_spec, test_config, verbose)
def download_ycsb(self):
if self.worker_manager.is_remote:
self.remote.clone_ycsb(
repo=self.test_config.syncgateway_settings.repo,
branch=self.test_config.syncgateway_settings.branch,
worker_home=self.worker_manager.WORKER_HOME,
ycsb_instances=int(self.test_config.syncgateway_settings.
instances_per_client))
else:
local.clone_ycsb(
repo=self.test_config.syncgateway_settings.repo,
branch=self.test_config.syncgateway_settings.branch)
def collect_execution_logs(self):
if self.worker_manager.is_remote:
if os.path.exists(self.LOCAL_DIR):
shutil.rmtree(self.LOCAL_DIR, ignore_errors=True)
os.makedirs(self.LOCAL_DIR)
self.remote.get_syncgateway_YCSB_logs(
self.worker_manager.WORKER_HOME,
self.test_config.syncgateway_settings, self.LOCAL_DIR)
def run_sg_phase(self,
phase: str,
task: Callable,
settings: PhaseSettings,
timer: int = None,
distribute: bool = False) -> None:
logger.info('Running {}: {}'.format(phase, pretty_dict(settings)))
self.worker_manager.run_sg_tasks(task, settings, timer, distribute,
phase)
self.worker_manager.wait_for_workers()
def start_memcached(self):
self.run_sg_phase("start memcached", syncgateway_task_start_memcached,
self.settings, self.settings.time, False)
def load_users(self):
self.run_sg_phase("load users", syncgateway_task_load_users,
self.settings, self.settings.time, False)
def init_users(self):
if self.test_config.syncgateway_settings.auth == 'true':
self.run_sg_phase("init users", syncgateway_task_init_users,
self.settings, self.settings.time, False)
def grant_access(self):
if self.test_config.syncgateway_settings.grant_access == 'true':
self.run_sg_phase("grant access to users",
syncgateway_task_grant_access, self.settings,
self.settings.time, False)
def load_docs(self):
self.run_sg_phase("load docs", syncgateway_task_load_docs,
self.settings, self.settings.time, False)
@with_stats
@with_profiles
def run_test(self):
self.run_sg_phase("run test", syncgateway_task_run_test, self.settings,
self.settings.time, True)
def compress_sg_logs(self):
self.remote.compress_sg_logs()
def get_sg_logs(self):
initial_nodes = int(self.test_config.syncgateway_settings.nodes)
ssh_user, ssh_pass = self.cluster_spec.ssh_credentials
for _server in range(initial_nodes):
server = self.cluster_spec.servers[_server]
local.get_sg_logs(host=server,
ssh_user=ssh_user,
ssh_pass=ssh_pass)
def run(self):
self.download_ycsb()
self.start_memcached()
self.load_users()
self.load_docs()
self.init_users()
self.grant_access()
self.run_test()
self.report_kpi()
def __exit__(self, exc_type, exc_val, exc_tb):
if self.test_config.test_case.use_workers:
self.worker_manager.download_celery_logs()
self.worker_manager.terminate()
if self.test_config.cluster.online_cores:
self.remote.enable_cpu()
if self.test_config.cluster.kernel_mem_limit:
self.remote.reset_memory_settings()
self.monitor.wait_for_servers()
