Files
knowledge-graph-agent/backend/neo4j/client.go
T

49 lines
921 B
Go
Raw Normal View History

2026-04-12 15:47:48 +08:00
package neo4j
import (
"context"
"fmt"
"log"
"os"
"github.com/joho/godotenv"
"github.com/neo4j/neo4j-go-driver/v5/neo4j"
)
var (
NEO4J_URI string
NEO4J_USERNAME string
NEO4J_PASSWORD string
)
func init() {
err := godotenv.Load()
if err != nil {
log.Fatal("Error loading .env file")
}
NEO4J_URI = os.Getenv("NEO4J_URI")
NEO4J_USERNAME = os.Getenv("NEO4J_USERNAME")
NEO4J_PASSWORD = os.Getenv("NEO4J_PASSWORD")
}
func doConnect() {
ctx := context.Background()
// dbUri := "<database-uri>"
// dbUser := "<username>"
// dbPassword := "<password>"
driver, err := neo4j.NewDriverWithContext(
NEO4J_URI,
neo4j.BasicAuth(NEO4J_USERNAME, NEO4J_PASSWORD, ""))
if err != nil {
panic(err)
}
defer driver.Close(ctx)
err = driver.VerifyConnectivity(ctx)
if err != nil {
panic(err)
}
fmt.Println("Connection established.")
}