1// Copyright (c) 2012 The LevelDB Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file. See the AUTHORS file for names of contributors.
4
5#include <cstdio>
6
7#include "leveldb/dumpfile.h"
8#include "leveldb/env.h"
9#include "leveldb/status.h"
10
11namespace leveldb {
12namespace {
13
14class StdoutPrinter : public WritableFile {
15 public:
16 Status Append(const Slice& data) override {
17 fwrite(data.data(), 1, data.size(), stdout);
18 return Status::OK();
19 }
20 Status Close() override { return Status::OK(); }
21 Status Flush() override { return Status::OK(); }
22 Status Sync() override { return Status::OK(); }
23};
24
25bool HandleDumpCommand(Env* env, char** files, int num) {
26 StdoutPrinter printer;
27 bool ok = true;
28 for (int i = 0; i < num; i++) {
29 Status s = DumpFile(env, files[i], &printer);
30 if (!s.ok()) {
31 std::fprintf(stderr, "%s\n", s.ToString().c_str());
32 ok = false;
33 }
34 }
35 return ok;
36}
37
38} // namespace
39} // namespace leveldb
40
41static void Usage() {
42 std::fprintf(
43 stderr,
44 "Usage: leveldbutil command...\n"
45 " dump files... -- dump contents of specified files\n");
46}
47
48int main(int argc, char** argv) {
49 leveldb::Env* env = leveldb::Env::Default();
50 bool ok = true;
51 if (argc < 2) {
52 Usage();
53 ok = false;
54 } else {
55 std::string command = argv[1];
56 if (command == "dump") {
57 ok = leveldb::HandleDumpCommand(env, argv + 2, argc - 2);
58 } else {
59 Usage();
60 ok = false;
61 }
62 }
63 return (ok ? 0 : 1);
64}
65