26adaad438
For ease of reuse and customization as a library without wrapping. WinEnvThreads is a class for replacement. WintEnvIO is a class for reuse and behavior override. Added private virtual functions for custom override of fallocate pread for io classes.
63 lines
1.5 KiB
C++
63 lines
1.5 KiB
C++
// Copyright (c) 2011-present, 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.
|
|
//
|
|
// 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.
|
|
//
|
|
// Logger implementation that can be shared by all environments
|
|
// where enough posix functionality is available.
|
|
|
|
#pragma once
|
|
|
|
#include <atomic>
|
|
|
|
#include "rocksdb/env.h"
|
|
|
|
#include <stdint.h>
|
|
#include <Windows.h>
|
|
|
|
namespace rocksdb {
|
|
|
|
class Env;
|
|
|
|
namespace port {
|
|
|
|
class WinLogger : public rocksdb::Logger {
|
|
public:
|
|
WinLogger(uint64_t (*gettid)(), Env* env, HANDLE file,
|
|
const InfoLogLevel log_level = InfoLogLevel::ERROR_LEVEL);
|
|
|
|
virtual ~WinLogger();
|
|
|
|
WinLogger(const WinLogger&) = delete;
|
|
|
|
WinLogger& operator=(const WinLogger&) = delete;
|
|
|
|
void close();
|
|
|
|
void Flush() override;
|
|
|
|
void Logv(const char* format, va_list ap) override;
|
|
|
|
size_t GetLogFileSize() const override;
|
|
|
|
void DebugWriter(const char* str, int len);
|
|
|
|
private:
|
|
HANDLE file_;
|
|
uint64_t (*gettid_)(); // Return the thread id for the current thread
|
|
std::atomic_size_t log_size_;
|
|
std::atomic_uint_fast64_t last_flush_micros_;
|
|
Env* env_;
|
|
bool flush_pending_;
|
|
|
|
const static uint64_t flush_every_seconds_ = 5;
|
|
};
|
|
|
|
}
|
|
|
|
} // namespace rocksdb
|