36a9f22931
Summary: Make blob_dump tool able to show uncompressed values if the blob file is compressed. Also show total compressed vs. raw size at the end if --show_summary is provided. Closes https://github.com/facebook/rocksdb/pull/3633 Differential Revision: D7348926 Pulled By: yiwu-arbug fbshipit-source-id: ca709cb4ed5cf6a550ff2987df8033df81516f8e
58 lines
1.7 KiB
C++
58 lines
1.7 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).
|
|
#pragma once
|
|
#ifndef ROCKSDB_LITE
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
#include "rocksdb/slice.h"
|
|
#include "rocksdb/status.h"
|
|
#include "util/file_reader_writer.h"
|
|
#include "utilities/blob_db/blob_log_format.h"
|
|
|
|
namespace rocksdb {
|
|
namespace blob_db {
|
|
|
|
class BlobDumpTool {
|
|
public:
|
|
enum class DisplayType {
|
|
kNone,
|
|
kRaw,
|
|
kHex,
|
|
kDetail,
|
|
};
|
|
|
|
BlobDumpTool();
|
|
|
|
Status Run(const std::string& filename, DisplayType show_key,
|
|
DisplayType show_blob, DisplayType show_uncompressed_blob,
|
|
bool show_summary);
|
|
|
|
private:
|
|
std::unique_ptr<RandomAccessFileReader> reader_;
|
|
std::unique_ptr<char> buffer_;
|
|
size_t buffer_size_;
|
|
|
|
Status Read(uint64_t offset, size_t size, Slice* result);
|
|
Status DumpBlobLogHeader(uint64_t* offset, CompressionType* compression);
|
|
Status DumpBlobLogFooter(uint64_t file_size, uint64_t* footer_offset);
|
|
Status DumpRecord(DisplayType show_key, DisplayType show_blob,
|
|
DisplayType show_uncompressed_blob, bool show_summary,
|
|
CompressionType compression, uint64_t* offset,
|
|
uint64_t* total_records, uint64_t* total_key_size,
|
|
uint64_t* total_blob_size,
|
|
uint64_t* total_uncompressed_blob_size);
|
|
void DumpSlice(const Slice s, DisplayType type);
|
|
|
|
template <class T>
|
|
std::string GetString(std::pair<T, T> p);
|
|
};
|
|
|
|
} // namespace blob_db
|
|
} // namespace rocksdb
|
|
|
|
#endif // ROCKSDB_LITE
|