blob: b4f3247e927a2faa7bcdfd1989797eb95509b7fe [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
#pragma once
#include <stdint.h>
#include <memory>
#include <string>
#include <unordered_map>
#include "statistics.h"
#include "test/bench_test/utils.h"
namespace rocksdb {
class Statistics;
} // namespace rocksdb
namespace pegasus {
class pegasus_client;
namespace test {
class benchmark;
struct thread_arg;
typedef void (benchmark::*bench_method)(thread_arg *);
struct thread_arg
{
int64_t seed;
statistics stats;
bench_method method;
benchmark *bm;
thread_arg(uint64_t seed_,
std::shared_ptr<rocksdb::Statistics> hist_stats_,
bench_method bench_method_,
benchmark *benchmark_)
: seed(seed_), stats(hist_stats_), method(bench_method_), bm(benchmark_)
{
}
};
class benchmark
{
public:
benchmark();
~benchmark() = default;
void run();
private:
/** thread main function */
static void thread_body(void *v);
/** benchmark operations **/
void run_benchmark(int thread_count, operation_type op_type);
void write_random(thread_arg *thread);
void read_random(thread_arg *thread);
void delete_random(thread_arg *thread);
void multi_set_random(thread_arg *thread);
void multi_get_random(thread_arg *thread);
/** generate hash/sort key and value */
void generate_kv_pair(std::string &hashkey, std::string &sortkey, std::string &value);
/** some auxiliary functions */
operation_type get_operation_type(const std::string &name);
void print_header();
void print_warnings();
private:
// the pegasus client to do read/write/delete operations
pegasus_client *_client;
// the map of operation type and the process method
std::unordered_map<operation_type, bench_method, std::hash<unsigned char>> _operation_method;
};
} // namespace test
} // namespace pegasus