tqcq/leveldb
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Align windows_logger with posix_logger.

Browse Source 
Fixes GitHub issue #657.

This CL also makes the Windows CI green.

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=237255887
This commit is contained in:
costan 2019-03-07 08:52:24 -08:00 committed by Victor Costan
parent 808e59ec6a
commit ed76289b25
3 changed files with 39 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
util/env_windows.cc
View File

@ -621,16 +621,16 @@ class WindowsEnv : public Env {
return Status::OK(); return Status::OK();
} }
Status NewLogger(const std::string& fname, Logger** result) override { Status NewLogger(const std::string& filename, Logger** result) override {
ScopedHandle handle = std::FILE* fp = std::fopen(filename.c_str(), "w");
::CreateFileA(fname.c_str(), GENERIC_WRITE, FILE_SHARE_READ, nullptr, if (fp == nullptr) {
CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, nullptr); *result = nullptr;
if (!handle.is_valid()) {
return WindowsError("NewLogger", ::GetLastError()); return WindowsError("NewLogger", ::GetLastError());
} } else {
*result = new WindowsLogger(handle.Release()); *result = new WindowsLogger(fp);
return Status::OK(); return Status::OK();
} }
}
uint64_t NowMicros() override { uint64_t NowMicros() override {
// GetSystemTimeAsFileTime typically has a resolution of 10-20 msec. // GetSystemTimeAsFileTime typically has a resolution of 10-20 msec.

35
util/windows_logger.h
View File

@ -7,10 +7,9 @@
#ifndef STORAGE_LEVELDB_UTIL_WINDOWS_LOGGER_H_ #ifndef STORAGE_LEVELDB_UTIL_WINDOWS_LOGGER_H_
#define STORAGE_LEVELDB_UTIL_WINDOWS_LOGGER_H_ #define STORAGE_LEVELDB_UTIL_WINDOWS_LOGGER_H_
#include <stdio.h>
#include <cassert> #include <cassert>
#include <cstdarg> #include <cstdarg>
#include <cstdio>
#include <ctime> #include <ctime>
#include <sstream> #include <sstream>
#include <thread> #include <thread>
@ -21,11 +20,16 @@ namespace leveldb {
class WindowsLogger final : public Logger { class WindowsLogger final : public Logger {
public: public:
WindowsLogger(HANDLE handle) : handle_(handle) { // Creates a logger that writes to the given file.
assert(handle != INVALID_HANDLE_VALUE); //
// The PosixLogger instance takes ownership of the file handle.
explicit WindowsLogger(std::FILE* fp) : fp_(fp) {
assert(fp != nullptr);
} }
~WindowsLogger() override { ::CloseHandle(handle_); } ~WindowsLogger() override {
std::fclose(fp_);
}
void Logv(const char* format, va_list arguments) override { void Logv(const char* format, va_list arguments) override {
// Record the time as close to the Logv() call as possible. // Record the time as close to the Logv() call as possible.
@ -56,16 +60,20 @@ class WindowsLogger final : public Logger {
(iteration == 0) ? stack_buffer : new char[dynamic_buffer_size]; (iteration == 0) ? stack_buffer : new char[dynamic_buffer_size];
// Print the header into the buffer. // Print the header into the buffer.
// TODO(costan): Sync this logger with another logger.
int buffer_offset = snprintf( int buffer_offset = snprintf(
buffer, buffer_size, "%04d/%02d/%02d-%02d:%02d:%02d.%06d %s ", buffer, buffer_size,
now_components.wYear, now_components.wMonth, now_components.wDay, "%04d/%02d/%02d-%02d:%02d:%02d.%06d %s ",
now_components.wHour, now_components.wMinute, now_components.wSecond, now_components.wYear,
now_components.wMonth,
now_components.wDay,
now_components.wHour,
now_components.wMinute,
now_components.wSecond,
static_cast<int>(now_components.wMilliseconds * 1000), static_cast<int>(now_components.wMilliseconds * 1000),
std::stoull(thread_id)); thread_id.c_str());
// The header can be at most 28 characters (10 date + 15 time + // The header can be at most 28 characters (10 date + 15 time +
// 3 spacing) plus the thread ID, which should fit comfortably into the // 3 delimiters) plus the thread ID, which should fit comfortably into the
// static buffer. // static buffer.
assert(buffer_offset <= 28 + kMaxThreadIdSize); assert(buffer_offset <= 28 + kMaxThreadIdSize);
static_assert(28 + kMaxThreadIdSize < kStackBufferSize, static_assert(28 + kMaxThreadIdSize < kStackBufferSize,
@ -106,7 +114,8 @@ class WindowsLogger final : public Logger {
} }
assert(buffer_offset <= buffer_size); assert(buffer_offset <= buffer_size);
::WriteFile(handle_, buffer, buffer_offset, nullptr, nullptr); std::fwrite(buffer, 1, buffer_offset, fp_);
std::fflush(fp_);
if (iteration != 0) { if (iteration != 0) {
delete[] buffer; delete[] buffer;
@ -116,7 +125,7 @@ class WindowsLogger final : public Logger {
} }
private: private:
HANDLE handle_; std::FILE* const fp_;
}; };
} // namespace leveldb } // namespace leveldb