e8d332d97e
Summary: In the current implementation, sst file checksum is calculated by a shared checksum function object, which may make some checksum function hard to be applied here such as SHA1. In this implementation, each sst file will have its own checksum generator obejct, created by FileChecksumGenFactory. User needs to implement its own FilechecksumGenerator and Factory to plugin the in checksum calculation method. Pull Request resolved: https://github.com/facebook/rocksdb/pull/6600 Test Plan: tested with make asan_check Reviewed By: riversand963 Differential Revision: D20717670 Pulled By: zhichao-cao fbshipit-source-id: 2a74c1c280ac11a07a1980185b43b671acaa71c6
81 lines
2.6 KiB
C++
81 lines
2.6 KiB
C++
// 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).
|
|
//
|
|
// Copyright (c) 2011 The LevelDB Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file. See the AUTHORS file for names of contributors.
|
|
|
|
#include "util/file_checksum_helper.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
void FileChecksumListImpl::reset() { checksum_map_.clear(); }
|
|
|
|
size_t FileChecksumListImpl::size() const { return checksum_map_.size(); }
|
|
|
|
Status FileChecksumListImpl::GetAllFileChecksums(
|
|
std::vector<uint64_t>* file_numbers, std::vector<std::string>* checksums,
|
|
std::vector<std::string>* checksum_func_names) {
|
|
if (file_numbers == nullptr || checksums == nullptr ||
|
|
checksum_func_names == nullptr) {
|
|
return Status::InvalidArgument("Pointer has not been initiated");
|
|
}
|
|
|
|
for (auto i : checksum_map_) {
|
|
file_numbers->push_back(i.first);
|
|
checksums->push_back(i.second.first);
|
|
checksum_func_names->push_back(i.second.second);
|
|
}
|
|
return Status::OK();
|
|
}
|
|
|
|
Status FileChecksumListImpl::SearchOneFileChecksum(
|
|
uint64_t file_number, std::string* checksum,
|
|
std::string* checksum_func_name) {
|
|
if (checksum == nullptr || checksum_func_name == nullptr) {
|
|
return Status::InvalidArgument("Pointer has not been initiated");
|
|
}
|
|
|
|
auto it = checksum_map_.find(file_number);
|
|
if (it == checksum_map_.end()) {
|
|
return Status::NotFound();
|
|
} else {
|
|
*checksum = it->second.first;
|
|
*checksum_func_name = it->second.second;
|
|
}
|
|
return Status::OK();
|
|
}
|
|
|
|
Status FileChecksumListImpl::InsertOneFileChecksum(
|
|
uint64_t file_number, const std::string& checksum,
|
|
const std::string& checksum_func_name) {
|
|
auto it = checksum_map_.find(file_number);
|
|
if (it == checksum_map_.end()) {
|
|
checksum_map_.insert(std::make_pair(
|
|
file_number, std::make_pair(checksum, checksum_func_name)));
|
|
} else {
|
|
it->second.first = checksum;
|
|
it->second.second = checksum_func_name;
|
|
}
|
|
return Status::OK();
|
|
}
|
|
|
|
Status FileChecksumListImpl::RemoveOneFileChecksum(uint64_t file_number) {
|
|
auto it = checksum_map_.find(file_number);
|
|
if (it == checksum_map_.end()) {
|
|
return Status::NotFound();
|
|
} else {
|
|
checksum_map_.erase(it);
|
|
}
|
|
return Status::OK();
|
|
}
|
|
|
|
FileChecksumList* NewFileChecksumList() {
|
|
FileChecksumListImpl* checksum_list = new FileChecksumListImpl();
|
|
return checksum_list;
|
|
}
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|