toutoukan/main.go

package main

import (
	"log"
	"net/http"
	"strconv"
	"time"
	"toutoukan/controllers/search"
	"toutoukan/init/bleveInit"
	"toutoukan/init/config"
	"toutoukan/init/databaseInit"
	"toutoukan/init/redisInit"
	"toutoukan/router"

	"github.com/sirupsen/logrus"
)

func main() {
	logrus.SetFormatter(&logrus.TextFormatter{
		ForceColors:      true,
		FullTimestamp:    true,                  // 启用完整时间戳
		TimestampFormat:  "2006-01-02 15:04:05", // 自定义绝对时间格式
		DisableTimestamp: false,                 // false=显示时间戳，true=去除时间戳
	})
	logrus.Info("------启动-----")
	bleveInit.InitIndex()
	databaseInit.DbInit()
	redisInit.RedisInit()
	if err := config.Goinit(); err != nil {
		panic(err)
	}
	r := router.SetupRouter()
	server := &http.Server{
		Addr:         ":" + strconv.Itoa(config.Conf.Port),
		Handler:      r,
		ReadTimeout:  5 * time.Second,  // 客户端读取超时
		WriteTimeout: 10 * time.Second, // 服务端写入超时
		IdleTimeout:  30 * time.Second, // 空闲连接超时
	}
	if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
		log.Fatalf("Server failed to start: %v", err)
	}
	defer func() {
		sqlDB, err := databaseInit.UserDB.DB()
		if err != nil {
			// 处理获取失败的情况
			log.Printf("获取底层数据库连接失败: %v", err)
			return
		}

		// 2. 调用 Close() 关闭连接池
		if err := sqlDB.Close(); err != nil {
			log.Printf("关闭数据库连接池失败: %v", err)
		} else {
			log.Println("数据库连接池已成功关闭")
		}
	}()
	defer redisInit.RedisClient.Close()
	defer search.CloseIndex()
}