fix session error

vesoft-inc · Sep 4, 2023 · 5c500ca · 5c500ca
1 parent 1af0ff7
commit 5c500ca
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/include/nebula/client/Session.h b/include/nebula/client/Session.h
@@ -110,6 +110,12 @@ class Session {
     return toLocal(data, offsetSecs_);
   }
 
+  bool isSessionError(const ExecutionResponse &resp) {
+    return resp.errorCode == ErrorCode::E_SESSION_INVALID ||
+           resp.errorCode == ErrorCode::E_SESSION_NOT_FOUND ||
+           resp.errorCode == ErrorCode::E_SESSION_TIMEOUT;
+  }
+
   // convert the time to specific time zone
   static void toLocal(DataSet &data, int32_t offsetSecs);
 

diff --git a/src/client/Session.cpp b/src/client/Session.cpp
@@ -37,7 +37,7 @@ ExecutionResponse Session::executeWithParameter(
   auto resp = conn_.executeWithParameter(sessionId_, stmt, parameters);
   if (resp.errorCode == nebula::ErrorCode::SUCCEEDED) {
     return resp;
-  } else if (resp.errorCode == nebula::ErrorCode::E_FAIL_TO_CONNECT) {
+  } else if (resp.errorCode == nebula::ErrorCode::E_FAIL_TO_CONNECT || isSessionError(resp)) {
     connectionIsBroken_ = true;
     if (retryConnect_) {
       if (retryConnect() == nebula::ErrorCode::SUCCEEDED) {