38 lines
1.0 KiB
C
38 lines
1.0 KiB
C
|
// Copyright (c) 2014, Facebook, Inc. All rights reserved.
|
||
|
|
// This source code is licensed under the BSD-style license found in the
|
||
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
||
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
#include <string>
|
||
|
|
#include "util/dynamic_bloom.h"
|
||
|
|
|
||
|
|
namespace rocksdb {
|
||
|
|
class Logger;
|
||
|
|
|
||
|
|
class BloomBlockBuilder {
|
||
|
|
public:
|
||
|
|
static const std::string kBloomBlock;
|
||
|
|
|
||
|
|
explicit BloomBlockBuilder(uint32_t num_probes = 6)
|
||
|
|
: bloom_(num_probes, nullptr) {}
|
||
|
|
|
||
|
|
void SetTotalBits(Arena* arena, uint32_t total_bits, uint32_t locality,
|
||
|
|
size_t huge_page_tlb_size, Logger* logger) {
|
||
|
|
bloom_.SetTotalBits(arena, total_bits, locality, huge_page_tlb_size,
|
||
|
|
logger);
|
||
|
|
}
|
||
|
|
|
||
|
|
uint32_t GetNumBlocks() const { return bloom_.GetNumBlocks(); }
|
||
|
|
|
||
|
|
void AddKeysHashes(const std::vector<uint32_t> keys_hashes);
|
||
|
|
|
||
|
|
Slice Finish();
|
||
|
|
|
||
|
|
private:
|
||
|
|
DynamicBloom bloom_;
|
||
|
|
};
|
||
|
|
|
||
|
|
}; // namespace rocksdb
|