query_db.go

//go:build ignore
// +build ignore

package main

import (
	"fmt"
	"log"
	"net/http"
	"bytes"
	"encoding/json"
	"io"
)

type gremlinRequest struct {
	Gremlin  string            `json:"gremlin"`
	Bindings map[string]any    `json:"bindings"`
	Language string            `json:"language"`
	Aliases  map[string]string `json:"aliases"`
}

func main() {
	// Let's run a query to insert repo:repo-test
	query := "g.V(m0_id).fold().coalesce(unfold(), addV(m0_lbl).property(id, m0_id)).property('name', 'repo-test')"
	bindings := map[string]any{
		"m0_id": "repo:repo-test",
		"m0_lbl": "Repository",
	}

	payload := gremlinRequest{
		Gremlin:  query,
		Bindings: bindings,
		Language: "gremlin-groovy",
		Aliases: map[string]string{
			"graph": "hugegraph",
			"g":     "__g_hugegraph",
		},
	}
	reqBody, _ := json.Marshal(payload)
	resp, err := http.Post(gremlinURL, "application/json", bytes.NewReader(reqBody))
	if err != nil {
		log.Fatalf("POST failed: %v", err)
	}
	defer resp.Body.Close()
	body, _ := io.ReadAll(resp.Body)
	fmt.Printf("Insert Response: %s\n", string(body))

	// Now check if it exists
	checkPayload := gremlinRequest{
		Gremlin:  "g.V(vid)",
		Bindings: map[string]any{"vid": "repo:repo-test"},
		Language: "gremlin-groovy",
		Aliases: map[string]string{
			"graph": "hugegraph",
			"g":     "__g_hugegraph",
		},
	}
	reqBody2, _ := json.Marshal(checkPayload)
	resp2, err := http.Post(gremlinURL, "application/json", bytes.NewReader(reqBody2))
	if err != nil {
		log.Fatalf("POST failed: %v", err)
	}
	defer resp2.Body.Close()
	body2, _ := io.ReadAll(resp2.Body)
	fmt.Printf("Check Response: %s\n", string(body2))
}