Refactor: 统一配置类

backend/main.go

@@ -1,4 +1,4 @@
-package neo4j
+package main
 
 import (
 	"log"
@@ -7,18 +7,18 @@ import (
 	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
 
+	"knowledge-graph-backend/config"
 	"knowledge-graph-backend/handlers"
 	"knowledge-graph-backend/services"
 )
 
-const (
-	PORT      = "3001"
-	DATA_FILE = "data.json"
-)
+func main() {
+	cfg, err := config.Load()
+	if err != nil {
+		log.Fatalf("Failed to load config: %v", err)
+	}
 
-func test() {
-
-	dataService, err := services.NewDataService(DATA_FILE)
+	dataService, err := services.NewDataService(cfg.Data.FilePath)
 	if err != nil {
 		log.Fatalf("Failed to initialize data service: %v", err)
 	}
@@ -29,14 +29,7 @@ func test() {
 
 	router := gin.Default()
 
-	router.Use(cors.New(cors.Config{
-		AllowOrigins:     []string{"http://localhost:5173", "http://localhost:3000"},
-		AllowMethods:     []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
-		AllowHeaders:     []string{"Origin", "Content-Type", "Authorization", "Accept"},
-		ExposeHeaders:    []string{"Content-Length", "Content-Type"},
-		AllowCredentials: true,
-		MaxAge:           12 * 3600,
-	}))
+	router.Use(cors.New(cfg.CORS.ToGinConfig()))
 
 	router.GET("/health", func(c *gin.Context) {
 		c.JSON(http.StatusOK, gin.H{
@@ -57,8 +50,8 @@ func test() {
 		api.GET("/nodes/:id/neighbors", nodeHandler.GetNeighbors)
 	}
 
-	log.Println("Knowledge Graph API Server running on http://localhost:" + PORT)
-	log.Println("Data file loaded from: " + DATA_FILE)
+	log.Println("Knowledge Graph API Server running on http://localhost:" + cfg.Server.Port)
+	log.Println("Data file loaded from: " + cfg.Data.FilePath)
 	log.Println("Available endpoints:")
 	log.Println("  GET /health - Health check")
 	log.Println("  GET /api/graph - Get all graph data")
@@ -67,7 +60,7 @@ func test() {
 	log.Println("  GET /api/nodes/:id - Get node by ID")
 	log.Println("  GET /api/nodes/:id/neighbors - Get node neighbors")
 
-	if err := router.Run(":" + PORT); err != nil {
+	if err := router.Run(":" + cfg.Server.Port); err != nil {
 		log.Fatalf("Failed to start server: %v", err)
 	}
 }