101 lines
2.3 KiB
Python
101 lines
2.3 KiB
Python
|
from __future__ import absolute_import
|
||
|
|
from __future__ import division
|
||
|
|
from __future__ import print_function
|
||
|
|
from __future__ import unicode_literals
|
||
|
|
rocksdb_target_header = """REPO_PATH = "internal_repo_rocksdb/repo/"
|
||
|
|
BUCK_BINS = "buck-out/gen/" + REPO_PATH
|
||
|
|
TEST_RUNNER = REPO_PATH + "buckifier/rocks_test_runner.sh"
|
||
|
|
rocksdb_compiler_flags = [
|
||
|
|
"-msse",
|
||
|
|
"-msse4.2",
|
||
|
|
"-fno-builtin-memcmp",
|
||
|
|
"-DROCKSDB_PLATFORM_POSIX",
|
||
|
|
"-DROCKSDB_LIB_IO_POSIX",
|
||
|
|
"-DROCKSDB_FALLOCATE_PRESENT",
|
||
|
|
"-DROCKSDB_MALLOC_USABLE_SIZE",
|
||
|
|
"-DOS_LINUX",
|
||
|
|
# Flags to enable libs we include
|
||
|
|
"-DSNAPPY",
|
||
|
|
"-DZLIB",
|
||
|
|
"-DBZIP2",
|
||
|
|
"-DLZ4",
|
||
|
|
"-DZSTD",
|
||
|
|
"-DGFLAGS=gflags",
|
||
|
|
"-DNUMA",
|
||
|
|
"-DTBB",
|
||
|
|
# Needed to compile in fbcode
|
||
|
|
"-Wno-expansion-to-defined",
|
||
|
|
]
|
||
|
|
|
||
|
|
rocksdb_external_deps = [
|
||
|
|
('bzip2', None, 'bz2'),
|
||
|
|
('snappy', None, "snappy"),
|
||
|
|
('zlib', None, 'z'),
|
||
|
|
('gflags', None, 'gflags'),
|
||
|
|
('lz4', None, 'lz4'),
|
||
|
|
('zstd', None),
|
||
|
|
('tbb', None),
|
||
|
|
("numa", "2.0.8", "numa"),
|
||
|
|
("googletest", None, "gtest"),
|
||
|
|
]
|
||
|
|
|
||
|
|
rocksdb_preprocessor_flags = [
|
||
|
|
# Directories with files for #include
|
||
|
|
"-I" + REPO_PATH + "include/",
|
||
|
|
"-I" + REPO_PATH,
|
||
|
|
]
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
library_template = """
|
||
|
|
cpp_library(
|
||
|
|
name = "%s",
|
||
|
|
headers = %s,
|
||
|
|
srcs = [%s],
|
||
|
|
deps = [%s],
|
||
|
|
preprocessor_flags = rocksdb_preprocessor_flags,
|
||
|
|
compiler_flags = rocksdb_compiler_flags,
|
||
|
|
external_deps = rocksdb_external_deps,
|
||
|
|
)
|
||
|
|
"""
|
||
|
|
|
||
|
|
binary_template = """
|
||
|
|
cpp_binary(
|
||
|
|
name = "%s",
|
||
|
|
srcs = [%s],
|
||
|
|
deps = [%s],
|
||
|
|
preprocessor_flags = rocksdb_preprocessor_flags,
|
||
|
|
compiler_flags = rocksdb_compiler_flags,
|
||
|
|
external_deps = rocksdb_external_deps,
|
||
|
|
)
|
||
|
|
"""
|
||
|
|
|
||
|
|
unittests_template = """
|
||
|
|
# [test_name, test_src, test_type]
|
||
|
|
ROCKS_TESTS = %s
|
||
|
|
|
||
|
|
|
||
|
|
# Generate a test rule for each entry in ROCKS_TESTS
|
||
|
|
for test_cfg in ROCKS_TESTS:
|
||
|
|
test_name = test_cfg[0]
|
||
|
|
test_cc = test_cfg[1]
|
||
|
|
ttype = "gtest" if test_cfg[2] == "parallel" else "simple"
|
||
|
|
test_bin = test_name + "_bin"
|
||
|
|
|
||
|
|
cpp_binary (
|
||
|
|
name = test_bin,
|
||
|
|
srcs = [test_cc],
|
||
|
|
deps = [":rocksdb_test_lib"],
|
||
|
|
preprocessor_flags = rocksdb_preprocessor_flags,
|
||
|
|
compiler_flags = rocksdb_compiler_flags,
|
||
|
|
external_deps = rocksdb_external_deps,
|
||
|
|
)
|
||
|
|
|
||
|
|
custom_unittest(
|
||
|
|
name = test_name,
|
||
|
|
type = ttype,
|
||
|
|
deps = [":" + test_bin],
|
||
|
|
command = [TEST_RUNNER, BUCK_BINS + test_bin]
|
||
|
|
)
|
||
|
|
"""
|