refactor: Refactor import statements and error handling in multiple files

- Fix import statements in main.go
- Improve error handling and logging in model/main.go
This commit is contained in:
Laisky.Cai
2023-12-12 06:34:08 +00:00
parent c503a87c74
commit 8d8fdaa2af
2 changed files with 26 additions and 23 deletions

13
main.go
View File

@@ -3,9 +3,6 @@ package main
import (
"embed"
"fmt"
"github.com/gin-contrib/sessions"
"github.com/gin-contrib/sessions/cookie"
"github.com/gin-gonic/gin"
"one-api/common"
"one-api/controller"
"one-api/middleware"
@@ -13,6 +10,10 @@ import (
"one-api/router"
"os"
"strconv"
"github.com/gin-contrib/sessions"
"github.com/gin-contrib/sessions/cookie"
"github.com/gin-gonic/gin"
)
//go:embed web/build
@@ -33,19 +34,19 @@ func main() {
// Initialize SQL Database
err := model.InitDB()
if err != nil {
common.FatalLog("failed to initialize database: " + err.Error())
common.FatalLog(fmt.Sprintf("failed to initialize database: %+v", err))
}
defer func() {
err := model.CloseDB()
if err != nil {
common.FatalLog("failed to close database: " + err.Error())
common.FatalLog(fmt.Sprintf("failed to close database: %+v", err))
}
}()
// Initialize Redis
err = common.InitRedisClient()
if err != nil {
common.FatalLog("failed to initialize Redis: " + err.Error())
common.FatalLog(fmt.Sprintf("failed to initialize Redis: %+v", err))
}
// Initialize options