15fb21ed7ac9e91010d473ac77e132ae68f348d6agabor@google.com// Copyright (c) 2011 The LevelDB Authors. All rights reserved. 25fb21ed7ac9e91010d473ac77e132ae68f348d6agabor@google.com// Use of this source code is governed by a BSD-style license that can be 35fb21ed7ac9e91010d473ac77e132ae68f348d6agabor@google.com// found in the LICENSE file. See the AUTHORS file for names of contributors. 4f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com// 5f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com// Logger implementation that can be shared by all environments 6f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com// where enough posix functionality is available. 7f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 8f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com#ifndef STORAGE_LEVELDB_UTIL_POSIX_LOGGER_H_ 9f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com#define STORAGE_LEVELDB_UTIL_POSIX_LOGGER_H_ 10f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 11f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com#include <algorithm> 12f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com#include <stdio.h> 13f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com#include <sys/time.h> 14f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com#include <time.h> 15f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com#include "leveldb/env.h" 16f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 17f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.comnamespace leveldb { 18f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 19f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.comclass PosixLogger : public Logger { 20f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com private: 21f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com FILE* file_; 22f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com uint64_t (*gettid_)(); // Return the thread id for the current thread 23f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com public: 24f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com PosixLogger(FILE* f, uint64_t (*gettid)()) : file_(f), gettid_(gettid) { } 25f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com virtual ~PosixLogger() { 26f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com fclose(file_); 27f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } 28f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com virtual void Logv(const char* format, va_list ap) { 29f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com const uint64_t thread_id = (*gettid_)(); 30f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 31f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com // We try twice: the first time with a fixed-size stack allocated buffer, 32f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com // and the second time with a much larger dynamically allocated buffer. 33f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com char buffer[500]; 34f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com for (int iter = 0; iter < 2; iter++) { 35f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com char* base; 36f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com int bufsize; 37f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com if (iter == 0) { 38f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com bufsize = sizeof(buffer); 39f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com base = buffer; 40f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } else { 41f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com bufsize = 30000; 42f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com base = new char[bufsize]; 43f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } 44f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com char* p = base; 45f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com char* limit = base + bufsize; 46f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 47f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com struct timeval now_tv; 48f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com gettimeofday(&now_tv, NULL); 49f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com const time_t seconds = now_tv.tv_sec; 50f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com struct tm t; 51f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com localtime_r(&seconds, &t); 52f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com p += snprintf(p, limit - p, 53f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com "%04d/%02d/%02d-%02d:%02d:%02d.%06d %llx ", 54f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com t.tm_year + 1900, 55f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com t.tm_mon + 1, 56f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com t.tm_mday, 57f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com t.tm_hour, 58f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com t.tm_min, 59f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com t.tm_sec, 60f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com static_cast<int>(now_tv.tv_usec), 61f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com static_cast<long long unsigned int>(thread_id)); 62f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 63f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com // Print the message 64f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com if (p < limit) { 65f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com va_list backup_ap; 66f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com va_copy(backup_ap, ap); 67f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com p += vsnprintf(p, limit - p, format, backup_ap); 68f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com va_end(backup_ap); 69f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } 70f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 71f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com // Truncate to available space if necessary 72f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com if (p >= limit) { 73f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com if (iter == 0) { 74f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com continue; // Try again with larger buffer 75f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } else { 76f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com p = limit - 1; 77f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } 78f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } 79f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 80f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com // Add newline if necessary 81f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com if (p == base || p[-1] != '\n') { 82f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com *p++ = '\n'; 83f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } 84f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 85f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com assert(p <= limit); 86f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com fwrite(base, 1, p - base, file_); 87f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com fflush(file_); 88f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com if (base != buffer) { 89f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com delete[] base; 90f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } 91f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com break; 92f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } 93f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com } 94f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com}; 95f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 9645b9940be332834440bd5299419f396e38085ebehans@chromium.org} // namespace leveldb 97f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com 98f65a55c8d0744b95be29a65d06b59b22b012f37bgabor@google.com#endif // STORAGE_LEVELDB_UTIL_POSIX_LOGGER_H_ 99