Update Go examples integration

- Add VERTEX_LABEL environment variable
- Use configurable vertex labels in examples
- Update root-level connection labels

Author: kirill-stepanishin

glv-examples/gremlin-go/connections.go

@@ -46,13 +46,9 @@ func withRemote() {
     // Creating the graph traversal
 	g := gremlingo.Traversal_().WithRemote(driverRemoteConnection)
 
-    // Drop existing vertices
-    prom := g.V().Drop().Iterate()
-    <-prom
-
     // Simple query to verify connection
-    g.AddV().Iterate()
-    count, _ := g.V().Count().Next()
+    g.AddV("connection").Iterate()
+    count, _ := g.V().HasLabel("connection").Count().Next()
     fmt.Println("Vertex count:", *count)
 }
 
@@ -75,7 +71,7 @@ func withConfigs() {
 	defer driverRemoteConnection.Close()
 	g := gremlingo.Traversal_().WithRemote(driverRemoteConnection)
 
-    g.AddV().Iterate()
-    count, _ := g.V().Count().Next()
+    g.AddV("connection").Iterate()
+    count, _ := g.V().HasLabel("connection").Count().Next()
     fmt.Println("Vertex count:", *count)
 }

gremlin-go/docker-compose.yml

@@ -60,6 +60,7 @@ services:
       - RUN_BASIC_AUTH_INTEGRATION_TESTS=true
       - GREMLIN_SOCKET_SERVER_URL=ws://gremlin-socket-server-go
       - GREMLIN_SOCKET_SERVER_CONFIG_PATH=/go_app/gremlin-socket-server/conf/test-ws-gremlin.yaml
+      - VERTEX_LABEL=go-example
     working_dir: /go_app
     command: >
       bash -c "go install github.com/gotesttools/gotestfmt/v2/cmd/gotestfmt@latest

gremlin-go/examples/basic_gremlin.go

@@ -36,6 +36,7 @@ func getEnvOrDefaultString(key string, defaultValue string) string {
 
 func main() {
 	serverURL := getEnvOrDefaultString("GREMLIN_SERVER_URL", "ws://localhost:8182/gremlin")
+	vertexLabel := getEnvOrDefaultString("VERTEX_LABEL", "person")
 	driverRemoteConnection, err := gremlingo.NewDriverRemoteConnection(serverURL)
 	if err != nil {
 		fmt.Println(err)
@@ -45,9 +46,9 @@ func main() {
 	g := gremlingo.Traversal_().WithRemote(driverRemoteConnection)
 
 	// Basic Gremlin: adding and retrieving data
-	v1, err := g.AddV("person-go-ex").Property("name", "marko").Next()
-	v2, err := g.AddV("person-go-ex").Property("name", "stephen").Next()
-	v3, err := g.AddV("person-go-ex").Property("name", "vadas").Next()
+	v1, err := g.AddV(vertexLabel).Property("name", "marko").Next()
+	v2, err := g.AddV(vertexLabel).Property("name", "stephen").Next()
+	v3, err := g.AddV(vertexLabel).Property("name", "vadas").Next()
 	v1Vertex, err := v1.GetVertex()
 	v2Vertex, err := v2.GetVertex()
 	v3Vertex, err := v3.GetVertex()
@@ -68,17 +69,17 @@ func main() {
 	}
 
 	// Retrieve the data from the "marko" vertex
-	marko, err := g.V().Has("person-go-ex", "name", "marko").Values("name").Next()
+	marko, err := g.V().Has(vertexLabel, "name", "marko").Values("name").Next()
 	fmt.Println("name:", marko.GetString())
 
 	// Find the "marko" vertex and then traverse to the people he "knows" and return their data
-	peopleMarkoKnows, err := g.V().Has("person-go-ex", "name", "marko").Out("knows").Values("name").ToList()
+	peopleMarkoKnows, err := g.V().Has(vertexLabel, "name", "marko").Out("knows").Values("name").ToList()
 	for _, person := range peopleMarkoKnows {
 		fmt.Println("marko knows", person.GetString())
 	}
 
 	// clean added data
-	promise = g.V().HasLabel("person-go-ex").Drop().Iterate()
+	promise = g.V().HasLabel(vertexLabel).Drop().Iterate()
 	err = <-promise
 	if err != nil {
 		fmt.Println(err)

gremlin-go/examples/connections.go

@@ -41,6 +41,7 @@ func main() {
 
 func withRemote() {
 	serverURL := getEnvOrDefaultString("GREMLIN_SERVER_URL", "ws://localhost:8182/gremlin")
+	vertexLabel := getEnvOrDefaultString("VERTEX_LABEL", "connection")
 
 	// Creating the connection to the server
 	driverRemoteConnection, err := gremlingo.NewDriverRemoteConnection(serverURL)
@@ -58,16 +59,17 @@ func withRemote() {
 	g := gremlingo.Traversal_().WithRemote(driverRemoteConnection)
 
 	// Simple query to verify connection
-	g.AddV("go-conn-ex").Iterate()
+	g.AddV(vertexLabel).Iterate()
 	count, _ := g.V().Count().Next()
 	fmt.Println("Vertex count:", *count)
 
 	// clean added data
-	g.V().HasLabel("go-conn-ex").Drop().Iterate()
+	g.V().HasLabel(vertexLabel).Drop().Iterate()
 }
 
 func withConfigs() {
 	serverURL := getEnvOrDefaultString("GREMLIN_SERVER_URL", "ws://localhost:8182/gremlin")
+	vertexLabel := getEnvOrDefaultString("VERTEX_LABEL", "connection")
 
 	// Connecting to the server with customized configurations
 	driverRemoteConnection, err := gremlingo.NewDriverRemoteConnection(serverURL,
@@ -87,10 +89,10 @@ func withConfigs() {
 	defer driverRemoteConnection.Close()
 	g := gremlingo.Traversal_().WithRemote(driverRemoteConnection)
 
-	g.AddV("go-conn-ex").Iterate()
+	g.AddV(vertexLabel).Iterate()
 	count, _ := g.V().Count().Next()
 	fmt.Println("Vertex count:", *count)
 
 	// clean added data
-	g.V().HasLabel("go-conn-ex").Drop().Iterate()
+	g.V().HasLabel(vertexLabel).Drop().Iterate()
 }