Add a method to the Logger class to enable local time displaying (#240)

Author: an-tao

trantor/utils/Logger.cc

@@ -88,24 +88,51 @@ void Logger::formatTime()
     if (now != lastSecond_)
     {
         lastSecond_ = now;
+        if (displayLocalTime_())
+        {
 #ifndef _MSC_VER
-        strncpy(lastTimeString_,
-                date_.toFormattedString(false).c_str(),
+            strncpy(lastTimeString_,
+                    date_.toFormattedStringLocal(false).c_str(),
+                    sizeof(lastTimeString_) - 1);
+#else
+            strncpy_s<sizeof lastTimeString_>(
+                lastTimeString_,
+                date_.toFormattedStringLocal(false).c_str(),
                 sizeof(lastTimeString_) - 1);
+#endif
+        }
+        else
+        {
+#ifndef _MSC_VER
+            strncpy(lastTimeString_,
+                    date_.toFormattedString(false).c_str(),
+                    sizeof(lastTimeString_) - 1);
 #else
-        strncpy_s<sizeof lastTimeString_>(
-            lastTimeString_,
-            date_.toFormattedString(false).c_str(),
-            sizeof(lastTimeString_) - 1);
+            strncpy_s<sizeof lastTimeString_>(
+                lastTimeString_,
+                date_.toFormattedString(false).c_str(),
+                sizeof(lastTimeString_) - 1);
 #endif
+        }
     }
     logStream_ << T(lastTimeString_, 17);
     char tmp[32];
-    snprintf(tmp,
-             sizeof(tmp),
-             ".%06llu UTC ",
-             static_cast<long long unsigned int>(microSec));
-    logStream_ << T(tmp, 12);
+    if (displayLocalTime_())
+    {
+        snprintf(tmp,
+                 sizeof(tmp),
+                 ".%06llu ",
+                 static_cast<long long unsigned int>(microSec));
+        logStream_ << T(tmp, 8);
+    }
+    else
+    {
+        snprintf(tmp,
+                 sizeof(tmp),
+                 ".%06llu UTC ",
+                 static_cast<long long unsigned int>(microSec));
+        logStream_ << T(tmp, 12);
+    }
 #ifdef __linux__
     if (threadId_ == 0)
         threadId_ = static_cast<pid_t>(::syscall(SYS_gettid));

trantor/utils/Logger.h

@@ -141,6 +141,22 @@ class TRANTOR_EXPORT Logger : public NonCopyable
         return logLevel_();
     }
 
+    /**
+     * @brief Check whether it shows local time or UTC time.
+     */
+    static bool displayLocalTime()
+    {
+        return displayLocalTime_();
+    }
+
+    /**
+     * @brief Set whether it shows local time or UTC time. the default is UTC.
+     */
+    static void setDisplayLocalTime(bool showLocalTime)
+    {
+        displayLocalTime_() = showLocalTime;
+    }
+
   protected:
     static void defaultOutputFunction(const char *msg, const uint64_t len)
     {
@@ -151,6 +167,12 @@ class TRANTOR_EXPORT Logger : public NonCopyable
         fflush(stdout);
     }
     void formatTime();
+    static bool &displayLocalTime_()
+    {
+        static bool showLocalTime = false;
+        return showLocalTime;
+    }
+
     static LogLevel &logLevel_()
     {
 #ifdef RELEASE