New warehouse

cmd/server.go

@@ -0,0 +1,149 @@
+package cmd
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"networkDev/database"
+	"networkDev/middleware"
+	"networkDev/server"
+	"networkDev/utils"
+	"networkDev/utils/logger"
+
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+// serverCmd 代表服务器命令
+var serverCmd = &cobra.Command{
+	Use:   "server",
+	Short: "启动HTTP服务器",
+	Long:  `启动一个简单的HTTP服务器，监听配置文件中指定的端口。`,
+	Run:   runServer,
+}
+
+func init() {
+	// 将服务器命令添加到根命令
+	rootCmd.AddCommand(serverCmd)
+
+	// 添加服务器特定的标志
+	serverCmd.Flags().StringP("host", "H", "", "服务器监听地址 (覆盖配置文件)")
+	serverCmd.Flags().IntP("port", "p", 0, "服务器监听端口 (覆盖配置文件)")
+}
+
+// runServer 运行HTTP服务器
+func runServer(cmd *cobra.Command, args []string) {
+	// 获取配置
+	host := getServerHost(cmd)
+	port := getServerPort(cmd)
+	addr := fmt.Sprintf("%s:%d", host, port)
+
+	// 获取全局日志实例
+	logger := logger.GetLogger()
+	logger.LogServerStart(host, port)
+
+	// 初始化Redis（如果配置存在，失败不致命）
+	utils.InitRedis()
+
+	// 初始化数据库（根据 viper 配置选择 SQLite 或 MySQL）
+	// 如果初始化失败则回退并退出
+	if _, err := database.Init(); err != nil {
+		logrus.WithError(err).Fatal("数据库初始化失败")
+	}
+	// 执行自动迁移（确保表结构存在）
+	if err := database.AutoMigrate(); err != nil {
+		logrus.WithError(err).Fatal("数据库自动迁移失败")
+	}
+	// 初始化默认管理员账号（admin/admin123）
+	if err := database.SeedDefaultAdmin(); err != nil {
+		logrus.WithError(err).Fatal("默认管理员初始化失败")
+	}
+	// 初始化默认系统设置
+	if err := database.SeedDefaultSettings(); err != nil {
+		logrus.WithError(err).Fatal("默认系统设置初始化失败")
+	}
+
+	// 创建HTTP服务器
+	server := createHTTPServer(addr)
+
+	// 启动服务器
+	startServer(server)
+}
+
+// getServerHost 获取服务器监听地址
+func getServerHost(cmd *cobra.Command) string {
+	if host, _ := cmd.Flags().GetString("host"); host != "" {
+		return host
+	}
+	return viper.GetString("server.host")
+}
+
+// getServerPort 获取服务器监听端口
+func getServerPort(cmd *cobra.Command) int {
+	if port, _ := cmd.Flags().GetInt("port"); port != 0 {
+		return port
+	}
+	return viper.GetInt("server.port")
+}
+
+// createHTTPServer 创建HTTP服务器
+func createHTTPServer(addr string) *http.Server {
+	mux := http.NewServeMux()
+
+	// 注册路由
+	registerRoutes(mux)
+
+	// 使用日志中间件包装处理器
+	handler := middleware.WrapHandler(mux)
+
+	return &http.Server{
+		Addr:    addr,
+		Handler: handler,
+	}
+}
+
+// registerRoutes 注册HTTP路由
+func registerRoutes(mux *http.ServeMux) {
+	// 使用server包中的路由注册函数
+	server.RegisterRoutes(mux)
+}
+
+// startServer 启动服务器并处理优雅关闭
+func startServer(server *http.Server) {
+	// 获取全局日志实例
+	logger := logger.GetLogger()
+
+	// 创建一个通道来接收操作系统信号
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
+
+	// 在goroutine中启动服务器
+	go func() {
+		logger.WithField("addr", server.Addr).Info("HTTP服务器已启动")
+		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			logger.LogError(err, "服务器启动失败")
+			os.Exit(1)
+		}
+	}()
+
+	// 等待中断信号
+	<-sigChan
+	logger.Info("收到关闭信号，正在优雅关闭服务器...")
+
+	// 创建一个带超时的上下文
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	// 优雅关闭服务器
+	if err := server.Shutdown(ctx); err != nil {
+		logger.LogError(err, "服务器关闭时出错")
+	} else {
+		logger.LogServerStop()
+	}
+}