skip scan

pkg/common/env.go

@@ -4,6 +4,7 @@ import "github.com/kelseyhightower/envconfig"
 
 type Environment struct {
 	NebulaStmtPrefix string `envconfig:"STMT_PREFIX" `
+	NebulaSkipScan   bool   `envconfig:"SKIP_SCAN" default:"false"`
 }
 
 var nebulaEnv *Environment
@@ -27,3 +28,7 @@ func ProcessStmt(stmt string) string {
 	}
 	return stmt
 }
+
+func ShouldSkipScan() bool {
+	return nebulaEnv.NebulaSkipScan
+}

pkg/common/env_test.go

@@ -18,3 +18,11 @@ func TestProcessStmt(t *testing.T) {
 	getEnv()
 	assert.Equal(t, "nebula stmt", ProcessStmt("stmt"))
 }
+
+func TestSkipScan(t *testing.T) {
+	getEnv()
+	assert.False(t, ShouldSkipScan())
+	os.Setenv("NEBULA_SKIP_SCAN", "true")
+	getEnv()
+	assert.True(t, ShouldSkipScan())
+}

pkg/nebulagraph5/client.go

@@ -354,11 +354,12 @@ func (gc *GraphClient) Execute(stmt string) (common.IGraphResponse, error) {
 			}
 		}
 	}
-	//TODO could add a flag to just decode the first row
-	if rows != 0 {
-		for resp.HasNext() {
-			if err := resp.Scan(anyValues...); err != nil {
-				return nil, err
+	if !common.ShouldSkipScan() {
+		if rows != 0 {
+			for resp.HasNext() {
+				if err := resp.Scan(anyValues...); err != nil {
+					return nil, err
+				}
 			}
 		}
 	}