Support execute timeout (#344) (#345)

* support execute timeout * update
vesoft-inc · Jun 24, 2024 · 530bb43 · 530bb43
1 parent 4f1a4e6
commit 530bb43
Show file tree

Hide file tree

Showing 5 changed files with 1,086 additions and 68 deletions.
diff --git a/connection.go b/connection.go
@@ -187,6 +187,10 @@ func (cn *connection) executeWithParameter(sessionID int64, stmt string,
 	return resp, nil
 }
 
+func (cn *connection) executeWithParameterTimeout(sessionID int64, stmt string, params map[string]*nebula.Value, timeoutMs int64) (*graph.ExecutionResponse, error) {
+	return cn.graph.ExecuteWithTimeout(sessionID, []byte(stmt), params, timeoutMs)
+}
+
 func (cn *connection) executeJson(sessionID int64, stmt string) ([]byte, error) {
 	return cn.ExecuteJsonWithParameter(sessionID, stmt, map[string]*nebula.Value{})
 }