Merge pull request #54 from HarrisChu/support_http2_lts

support http2
vesoft-inc · Jan 4, 2024 · 93f145f · 93f145f
2 parents 509a486 + 5db262b
commit 93f145f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pkg/common/types.go b/pkg/common/types.go
@@ -68,6 +68,7 @@ type (
 		Username   string `json:"username"`
 		Password   string `json:"password"`
 		Space      string `json:"space"`
+		UseHttp    bool   `json:"use_http"`
 	}
 
 	OutputOption struct {

diff --git a/pkg/nebulagraph/client.go b/pkg/nebulagraph/client.go
@@ -181,6 +181,10 @@ func (gp *GraphPool) initConnectionPool() error {
 			return err
 		}
 	}
+	if gp.graphOption.UseHttp {
+		conf.UseHTTP2 = true
+	}
+
 	pool, err := graph.NewSslConnectionPool(hosts, conf, sslConfig, graph.DefaultLogger{})
 	if err != nil {
 		return err
@@ -217,6 +221,7 @@ func (gp *GraphPool) initSessionPool() error {
 		graph.WithMaxSize(gp.graphOption.MaxSize),
 		graph.WithMinSize(gp.graphOption.MinSize),
 		graph.WithSSLConfig(sslConfig),
+		graph.WithHTTP2(gp.graphOption.UseHttp),
 	)
 	if err != nil {
 		return err