90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"super-frpc/modules"
|
|
"syscall"
|
|
"time"
|
|
)
|
|
|
|
func main() {
|
|
configPath := flag.String("config", "./config.json", "path to config file")
|
|
dbPath := flag.String("db", "./database.db", "path to database file")
|
|
flag.Parse()
|
|
|
|
if _, err := os.Stat(*configPath); os.IsNotExist(err) {
|
|
defaultConfig := `{
|
|
listenAddr": " "0.0.0.0",
|
|
"listenPort": "8080",
|
|
"configDir": "./configs"
|
|
}`
|
|
if err := os.WriteFile(*configPath, []byte(defaultConfig), 0644); err != nil {
|
|
postLog.Fatal(fmt.Sprintf("failed to create default config file: %v", err))
|
|
}
|
|
postLog.Info(fmt.Sprintf("created default config file at %s", *configPath))
|
|
}
|
|
|
|
config, err := LoadConfig(*configPath)
|
|
if err != nil {
|
|
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 {
|
|
postLog.Fatal(fmt.Sprintf("failed to initialize database: %v", err))
|
|
}
|
|
postLog.Info("database initialized successfully")
|
|
|
|
if err := InitFrpcDatabase(*dbPath); err != nil {
|
|
postLog.Warning(fmt.Sprintf("failed to initialize frpc database: %v", err))
|
|
}
|
|
|
|
_, err = GetConfig()
|
|
if err != nil {
|
|
postLog.Fatal(fmt.Sprintf("failed to get config: %v", err))
|
|
}
|
|
|
|
setupRoutes()
|
|
|
|
addr := fmt.Sprintf("%s:%s", config.ListenAddr, config.ListenPort)
|
|
server := &http.Server{
|
|
Addr: addr,
|
|
ReadTimeout: 15 * time.Second,
|
|
WriteTimeout: 15 * time.Second,
|
|
IdleTimeout: 60 * time.Second,
|
|
}
|
|
|
|
go func() {
|
|
postLog.Info(fmt.Sprintf("server starting on %s", addr))
|
|
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
postLog.Fatal(fmt.Sprintf("failed to start server: %v", err))
|
|
}
|
|
}()
|
|
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
<-quit
|
|
|
|
postLog.Info("shutting down server...")
|
|
|
|
if err := server.Close(); err != nil {
|
|
postLog.Error(fmt.Sprintf("server closed with error: %v", err))
|
|
}
|
|
|
|
if err := CloseDatabase(); err != nil {
|
|
postLog.Error(fmt.Sprintf("error closing database: %v", err))
|
|
}
|
|
|
|
if err := CloseFrpcDatabase(); err != nil {
|
|
postLog.Error(fmt.Sprintf("error closing frpc database: %v", err))
|
|
}
|
|
|
|
postLog.Info("server stopped")
|
|
}
|