xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-04 15:53:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: do not use redis to store session

Browse Source
This commit is contained in:
JustSong
2023-06-26 16:10:59 +08:00
parent f0dc7f3f06
commit 431d505f79
1 changed files with 2 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
main.go
View File

@@ -4,7 +4,6 @@ import (
"embed" "embed"
"github.com/gin-contrib/sessions" "github.com/gin-contrib/sessions"
"github.com/gin-contrib/sessions/cookie" "github.com/gin-contrib/sessions/cookie"
"github.com/gin-contrib/sessions/redis"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"one-api/common" "one-api/common"
"one-api/controller" "one-api/controller"
@@ -82,14 +81,8 @@ func main() {
server.Use(middleware.CORS()) server.Use(middleware.CORS())
// Initialize session store // Initialize session store
if common.RedisEnabled { store := cookie.NewStore([]byte(common.SessionSecret))
opt := common.ParseRedisOption() server.Use(sessions.Sessions("session", store))
store, _ := redis.NewStore(opt.MinIdleConns, opt.Network, opt.Addr, opt.Password, []byte(common.SessionSecret))
server.Use(sessions.Sessions("session", store))
} else {
store := cookie.NewStore([]byte(common.SessionSecret))
server.Use(sessions.Sessions("session", store))
}
router.SetRouter(server, buildFS, indexPage) router.SetRouter(server, buildFS, indexPage)
var port = os.Getenv("PORT") var port = os.Getenv("PORT")