Refactor: 统一配置类

backend/neo4j/client.go

@@ -3,46 +3,25 @@ package neo4j
 import (
 	"context"
 	"fmt"
-	"log"
-	"os"
 
-	"github.com/joho/godotenv"
 	"github.com/neo4j/neo4j-go-driver/v5/neo4j"
+
+	"knowledge-graph-backend/config"
 )
 
-var (
-	NEO4J_URI string
-	NEO4J_USERNAME string
-	NEO4J_PASSWORD string
-)
+func DoConnect(cfg config.Neo4jConfig) {
+	ctx := context.Background()
+	driver, err := neo4j.NewDriverWithContext(
+		cfg.URI,
+		neo4j.BasicAuth(cfg.Username, cfg.Password, ""))
+	if err != nil {
+		panic(err)
+	}
+	defer driver.Close(ctx)
 
-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.")
+	err = driver.VerifyConnectivity(ctx)
+	if err != nil {
+		panic(err)
+	}
+	fmt.Println("Connection established.")
 }