Support execute timeout (#344)

* support execute timeout * update
vesoft-inc · Jun 24, 2024 · c23dd4f · c23dd4f
1 parent 4f1a4e6
commit c23dd4f
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{})
 }