1

cmd/api/server.go

@@ -3,13 +3,15 @@ package api
 import (
 	"context"
 	"fmt"
-	"github.com/go-admin-team/go-admin-core/sdk/runtime"
 	"log"
 	"net/http"
 	"os"
 	"os/signal"
 	"time"
 
+	"github.com/go-admin-team/go-admin-core/logger"
+	"github.com/go-admin-team/go-admin-core/sdk/runtime"
+
 	"github.com/gin-gonic/gin"
 	"github.com/go-admin-team/go-admin-core/config/source/file"
 	"github.com/go-admin-team/go-admin-core/sdk"
@@ -20,6 +22,7 @@ import (
 
 	"go-admin/app/admin/models"
 	"go-admin/app/admin/router"
+	"go-admin/app/admin/service"
 	"go-admin/app/jobs"
 	"go-admin/common/database"
 	"go-admin/common/global"
@@ -27,6 +30,7 @@ import (
 	"go-admin/common/middleware/handler"
 	"go-admin/common/storage"
 	ext "go-admin/config"
+	helper "go-admin/utils/redishelper"
 )
 
 var (
@@ -86,6 +90,16 @@ func run() error {
 		f()
 	}
 
+	//初始化redis
+	helper.InitDefaultRedis(config.CacheConfig.Redis.Addr, config.CacheConfig.Redis.Password, config.CacheConfig.Redis.DB)
+
+	if err := helper.DefaultRedis.Ping(); err != nil {
+		fmt.Println("初始化redis失败")
+		os.Exit(2)
+	}
+
+	initCache()
+
 	srv := &http.Server{
 		Addr:    fmt.Sprintf("%s:%d", config.ApplicationConfig.Host, config.ApplicationConfig.Port),
 		Handler: sdk.Runtime.GetEngine(),
@@ -184,3 +198,17 @@ func initRouter() {
 	common.InitMiddleware(r)
 
 }
+
+// 初始化缓存
+func initCache() {
+	netWorkService := service.WmNetwork{}
+	dbs := sdk.Runtime.GetDb()
+	netWorkService.Orm = dbs["*"]
+	netWorkService.Log = logger.NewHelper(logger.DefaultLogger)
+	tokenService := service.WmToken{}
+	tokenService.Orm = netWorkService.Orm
+	tokenService.Log = netWorkService.Log
+
+	netWorkService.ReCache()
+	tokenService.ReCache("")
+}