fork of https://github.com/oxigraph/rocksdb and https://github.com/facebook/rocksdb for nextgraph and oxigraph
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
3.2 KiB
98 lines
3.2 KiB
# Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
# This source code is licensed under both the GPLv2 (found in the
|
|
# COPYING file in the root directory) and Apache 2.0 License
|
|
# (found in the LICENSE.Apache file in the root directory).
|
|
|
|
import argparse
|
|
|
|
from advisor.db_log_parser import DatabaseLogs, DataSource
|
|
from advisor.db_options_parser import DatabaseOptions
|
|
from advisor.db_stats_fetcher import LogStatsParser, OdsStatsFetcher
|
|
from advisor.rule_parser import RulesSpec
|
|
|
|
|
|
def main(args):
|
|
# initialise the RulesSpec parser
|
|
rule_spec_parser = RulesSpec(args.rules_spec)
|
|
rule_spec_parser.load_rules_from_spec()
|
|
rule_spec_parser.perform_section_checks()
|
|
# initialize the DatabaseOptions object
|
|
db_options = DatabaseOptions(args.rocksdb_options)
|
|
# Create DatabaseLogs object
|
|
db_logs = DatabaseLogs(args.log_files_path_prefix, db_options.get_column_families())
|
|
# Create the Log STATS object
|
|
db_log_stats = LogStatsParser(
|
|
args.log_files_path_prefix, args.stats_dump_period_sec
|
|
)
|
|
data_sources = {
|
|
DataSource.Type.DB_OPTIONS: [db_options],
|
|
DataSource.Type.LOG: [db_logs],
|
|
DataSource.Type.TIME_SERIES: [db_log_stats],
|
|
}
|
|
if args.ods_client:
|
|
data_sources[DataSource.Type.TIME_SERIES].append(
|
|
OdsStatsFetcher(
|
|
args.ods_client,
|
|
args.ods_entity,
|
|
args.ods_tstart,
|
|
args.ods_tend,
|
|
args.ods_key_prefix,
|
|
)
|
|
)
|
|
triggered_rules = rule_spec_parser.get_triggered_rules(
|
|
data_sources, db_options.get_column_families()
|
|
)
|
|
rule_spec_parser.print_rules(triggered_rules)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(
|
|
description="Use this script to get\
|
|
suggestions for improving Rocksdb performance."
|
|
)
|
|
parser.add_argument(
|
|
"--rules_spec",
|
|
required=True,
|
|
type=str,
|
|
help="path of the file containing the expert-specified Rules",
|
|
)
|
|
parser.add_argument(
|
|
"--rocksdb_options",
|
|
required=True,
|
|
type=str,
|
|
help="path of the starting Rocksdb OPTIONS file",
|
|
)
|
|
parser.add_argument(
|
|
"--log_files_path_prefix",
|
|
required=True,
|
|
type=str,
|
|
help="path prefix of the Rocksdb LOG files",
|
|
)
|
|
parser.add_argument(
|
|
"--stats_dump_period_sec",
|
|
required=True,
|
|
type=int,
|
|
help="the frequency (in seconds) at which STATISTICS are printed to "
|
|
+ "the Rocksdb LOG file",
|
|
)
|
|
# ODS arguments
|
|
parser.add_argument("--ods_client", type=str, help="the ODS client binary")
|
|
parser.add_argument(
|
|
"--ods_entity",
|
|
type=str,
|
|
help="the servers for which the ODS stats need to be fetched",
|
|
)
|
|
parser.add_argument(
|
|
"--ods_key_prefix",
|
|
type=str,
|
|
help="the prefix that needs to be attached to the keys of time "
|
|
+ "series to be fetched from ODS",
|
|
)
|
|
parser.add_argument(
|
|
"--ods_tstart", type=int, help="start time of timeseries to be fetched from ODS"
|
|
)
|
|
parser.add_argument(
|
|
"--ods_tend", type=int, help="end time of timeseries to be fetched from ODS"
|
|
)
|
|
args = parser.parse_args()
|
|
main(args)
|
|
|