refactor(logger): add individual log module

main.go

@@ -3,10 +3,10 @@ package main
 import (
 	"flag"
 	"fmt"
-	"log"
 	"net/http"
 	"os"
 	"os/signal"
+	"super-frpc/modules"
 	"syscall"
 	"time"
 )
@@ -23,28 +23,31 @@ func main() {
 	"configDir": "./configs"
 }`
 		if err := os.WriteFile(*configPath, []byte(defaultConfig), 0644); err != nil {
-			log.Fatalf("failed to create default config file: %v", err)
+			postLog.Fatal(fmt.Sprintf("failed to create default config file: %v", err))
 		}
-		log.Printf("created default config file at %s", *configPath)
+		postLog.Info(fmt.Sprintf("created default config file at %s", *configPath))
 	}
 
-	_, err := LoadConfig(*configPath)
+	config, err := LoadConfig(*configPath)
 	if err != nil {
-		log.Fatalf("failed to load config: %v", err)
+		postLog.Fatal(fmt.Sprintf("failed to load config: %v", err))
 	}
 
+	// Initialize logger with debug mode
+	postLog.SetDebugMode(config.Debug)
+
 	if err := InitDatabase(*dbPath); err != nil {
-		log.Fatalf("failed to initialize database: %v", err)
+		postLog.Fatal(fmt.Sprintf("failed to initialize database: %v", err))
 	}
-	log.Println("database initialized successfully")
+	postLog.Info("database initialized successfully")
 
 	if err := InitFrpcDatabase(*dbPath); err != nil {
-		log.Printf("warning: failed to initialize frpc database: %v", err)
+		postLog.Warning(fmt.Sprintf("failed to initialize frpc database: %v", err))
 	}
 
-	config, err := GetConfig()
+	_, err = GetConfig()
 	if err != nil {
-		log.Fatalf("failed to get config: %v", err)
+		postLog.Fatal(fmt.Sprintf("failed to get config: %v", err))
 	}
 
 	setupRoutes()
@@ -58,9 +61,9 @@ func main() {
 	}
 
 	go func() {
-		log.Printf("server starting on %s", addr)
+		postLog.Info(fmt.Sprintf("server starting on %s", addr))
 		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
-			log.Fatalf("failed to start server: %v", err)
+			postLog.Fatal(fmt.Sprintf("failed to start server: %v", err))
 		}
 	}()
 
@@ -68,20 +71,19 @@ func main() {
 	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
 	<-quit
 
-	log.Println("shutting down server...")
+	postLog.Info("shutting down server...")
 
 	if err := server.Close(); err != nil {
-		log.Printf("server closed with error: %v", err)
+		postLog.Error(fmt.Sprintf("server closed with error: %v", err))
 	}
 
 	if err := CloseDatabase(); err != nil {
-		log.Printf("error closing database: %v", err)
+		postLog.Error(fmt.Sprintf("error closing database: %v", err))
 	}
 
 	if err := CloseFrpcDatabase(); err != nil {
-		log.Printf("error closing frpc database: %v", err)
+		postLog.Error(fmt.Sprintf("error closing frpc database: %v", err))
 	}
 
-	log.Println("server stopped")
+	postLog.Info("server stopped")
 }
-