98 lines
2.5 KiB
C++
98 lines
2.5 KiB
C++
|
// 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 "db/builder.h"
|
||
|
|
|
||
|
|
#include "db/filename.h"
|
||
|
|
#include "db/dbformat.h"
|
||
|
|
#include "db/table_cache.h"
|
||
|
|
#include "db/version_edit.h"
|
||
|
|
#include "include/db.h"
|
||
|
|
#include "include/env.h"
|
||
|
|
#include "include/iterator.h"
|
||
|
|
|
||
|
|
namespace leveldb {
|
||
|
|
|
||
|
|
Status BuildTable(const std::string& dbname,
|
||
|
|
Env* env,
|
||
|
|
const Options& options,
|
||
|
|
TableCache* table_cache,
|
||
|
|
Iterator* iter,
|
||
|
|
FileMetaData* meta,
|
||
|
|
VersionEdit* edit) {
|
||
|
|
Status s;
|
||
|
|
meta->file_size = 0;
|
||
|
|
iter->SeekToFirst();
|
||
|
|
|
||
|
|
std::string fname = TableFileName(dbname, meta->number);
|
||
|
|
if (iter->Valid()) {
|
||
|
|
WritableFile* file;
|
||
|
|
s = env->NewWritableFile(fname, &file);
|
||
|
|
if (!s.ok()) {
|
||
|
|
return s;
|
||
|
|
}
|
||
|
|
|
||
|
|
TableBuilder* builder = new TableBuilder(options, file);
|
||
|
|
meta->smallest.DecodeFrom(iter->key());
|
||
|
|
for (; iter->Valid(); iter->Next()) {
|
||
|
|
Slice key = iter->key();
|
||
|
|
meta->largest.DecodeFrom(key);
|
||
|
|
if (ExtractValueType(key) == kTypeLargeValueRef) {
|
||
|
|
if (iter->value().size() != LargeValueRef::ByteSize()) {
|
||
|
|
s = Status::Corruption("invalid indirect reference hash value (L0)");
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
edit->AddLargeValueRef(LargeValueRef::FromRef(iter->value()),
|
||
|
|
meta->number,
|
||
|
|
iter->key());
|
||
|
|
}
|
||
|
|
builder->Add(key, iter->value());
|
||
|
|
}
|
||
|
|
|
||
|
|
// Finish and check for builder errors
|
||
|
|
if (s.ok()) {
|
||
|
|
s = builder->Finish();
|
||
|
|
if (s.ok()) {
|
||
|
|
meta->file_size = builder->FileSize();
|
||
|
|
assert(meta->file_size > 0);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
builder->Abandon();
|
||
|
|
}
|
||
|
|
delete builder;
|
||
|
|
|
||
|
|
// Finish and check for file errors
|
||
|
|
if (s.ok()) {
|
||
|
|
s = file->Sync();
|
||
|
|
}
|
||
|
|
if (s.ok()) {
|
||
|
|
s = file->Close();
|
||
|
|
}
|
||
|
|
delete file;
|
||
|
|
file = NULL;
|
||
|
|
|
||
|
|
if (s.ok()) {
|
||
|
|
// Verify that the table is usable
|
||
|
|
Iterator* it = table_cache->NewIterator(ReadOptions(), meta->number);
|
||
|
|
s = it->status();
|
||
|
|
delete it;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check for input iterator errors
|
||
|
|
if (!iter->status().ok()) {
|
||
|
|
s = iter->status();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (s.ok() && meta->file_size > 0) {
|
||
|
|
edit->AddFile(0, meta->number, meta->file_size,
|
||
|
|
meta->smallest, meta->largest);
|
||
|
|
} else {
|
||
|
|
env->DeleteFile(fname);
|
||
|
|
}
|
||
|
|
return s;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|