Add timezone conversion.

CMakeLists.txt

@@ -141,6 +141,19 @@ install(DIRECTORY ${nebula_common_source_dir}/src/common/graph
         DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/common
         FILES_MATCHING PATTERN "*.h")
 
+# reuse the time utils in common
+install(DIRECTORY ${nebula_common_source_dir}/src/common/time
+        DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/common
+        FILES_MATCHING PATTERN "TimeUtils.h")
+
+# reuse the status in common
+install(DIRECTORY ${nebula_common_source_dir}/src/common/base
+        DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/common
+        FILES_MATCHING PATTERN "Status.h")
+install(DIRECTORY ${nebula_common_source_dir}/src/common/base
+        DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/common
+        FILES_MATCHING PATTERN "StatusOr.h")
+
 # reuse the thrift types in common
 install(
     FILES

include/nebula/client/Session.h

@@ -6,6 +6,8 @@
 
 #pragma once
 
+#include <common/datatypes/DataSet.h>
+
 #include "nebula/client/Connection.h"
 
 namespace nebula {
@@ -22,20 +24,27 @@ class Session {
             Connection &&conn,
             ConnectionPool *pool,
             const std::string &username,
-            const std::string &password)
+            const std::string &password,
+            const std::string &timezoneName,
+            int32_t offsetSecs)
         : sessionId_(sessionId),
           conn_(std::move(conn)),
           pool_(pool),
           username_(username),
-          password_(password) {}
+          password_(password),
+          timezoneName_(timezoneName),
+          offsetSecs_(offsetSecs) {}
     Session(Session &&session)
         : sessionId_(session.sessionId_),
           conn_(std::move(session.conn_)),
           pool_(session.pool_),
           username_(std::move(session.username_)),
-          password_(std::move(session.password_)) {
+          password_(std::move(session.password_)),
+          timezoneName_(std::move(session.timezoneName_)),
+          offsetSecs_(session.offsetSecs_) {
         session.sessionId_ = -1;
         session.pool_ = nullptr;
+        session.offsetSecs_ = 0;
     }
     ~Session() {
         release();
@@ -59,12 +68,23 @@ class Session {
         return sessionId_ > 0;
     }
 
+    // convert the time to server time zone
+    void toLocal(DataSet &data) {
+        return toLocal(data, offsetSecs_);
+    }
+
+    // convert the time to specific time zone
+    static void toLocal(DataSet &data, int32_t offsetSecs);
+
 private:
     int64_t sessionId_{-1};
     Connection conn_;
     ConnectionPool *pool_{nullptr};
     std::string username_;
     std::string password_;
+    // empty means not a named timezone
+    std::string timezoneName_;
+    int32_t offsetSecs_;
 };
 
 }   // namespace nebula

src/CMakeLists.txt

@@ -9,6 +9,7 @@ set(NEBULA_CLIENT_OBJS
     $<TARGET_OBJECTS:common_graph_thrift_obj>
     $<TARGET_OBJECTS:common_datatypes_obj>
     $<TARGET_OBJECTS:common_graph_obj>
+    $<TARGET_OBJECTS:common_time_utils_obj>
 )
 
 set(NEBULA_CLIENT_SOURCES

src/client/ConnectionPool.cpp

@@ -47,7 +47,8 @@ Session ConnectionPool::getSession(const std::string &username,
     if (resp.errorCode != ErrorCode::SUCCEEDED || resp.sessionId == nullptr) {
         return Session();
     }
-    return Session(*resp.sessionId, std::move(conn), this, username, password);
+    return Session(*resp.sessionId, std::move(conn), this, username, password,
+        *resp.timeZoneName, *resp.timeZoneOffsetSeconds);
 }
 
 Connection ConnectionPool::getConnection() {

src/client/Session.cpp

@@ -4,6 +4,8 @@
  * attached with Common Clause Condition 1.0, found in the LICENSES directory.
  */
 
+#include <common/time/TimeUtils.h>
+
 #include "nebula/client/Session.h"
 #include "nebula/client/ConnectionPool.h"
 
@@ -48,4 +50,17 @@ void Session::release() {
     }
 }
 
+void Session::toLocal(DataSet &data, int32_t offsetSecs) {
+    DataSet ds(std::move(data.colNames));
+    for (auto &row : data.rows) {
+        for (auto &col : row.values) {
+            if (col.isTime()) {
+                col.setTime(time::TimeUtils::timeShift(col.getTime(), offsetSecs));
+            } else if (col.isDateTime()) {
+                col.setDateTime(time::TimeUtils::dateTimeShift(col.getDateTime(), offsetSecs));
+            }
+        }
+    }
+}
+
 }   // namespace nebula