feat: add redisEndpoint

casbin · Jun 3, 2024 · 1d4179b · 1d4179b
1 parent b0ca44b
commit 1d4179b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/conf/app.conf b/conf/app.conf
@@ -6,6 +6,7 @@ copyrequestbody = true
 driverName = mysql
 dataSourceName = root:123@tcp(localhost:3306)/
 dbName = caswaf
+redisEndpoint =
 gatewayEnabled = true
 gatewayHttpPort = 80
 gatewayHttpsPort = 443

diff --git a/main.go b/main.go
@@ -17,6 +17,7 @@ package main
 import (
 	"github.com/beego/beego"
 	"github.com/beego/beego/plugins/cors"
+	_ "github.com/beego/beego/session/redis"
 	"github.com/casbin/caswaf/casdoor"
 	"github.com/casbin/caswaf/object"
 	"github.com/casbin/caswaf/proxy"
@@ -51,8 +52,13 @@ func main() {
 	beego.InsertFilter("/*", beego.BeforeRouter, routers.TransparentStatic)
 	beego.InsertFilter("/api/*", beego.BeforeRouter, routers.ApiFilter)
 
-	beego.BConfig.WebConfig.Session.SessionProvider = "file"
-	beego.BConfig.WebConfig.Session.SessionProviderConfig = "./tmp"
+	if beego.AppConfig.String("redisEndpoint") == "" {
+		beego.BConfig.WebConfig.Session.SessionProvider = "file"
+		beego.BConfig.WebConfig.Session.SessionProviderConfig = "./tmp"
+	} else {
+		beego.BConfig.WebConfig.Session.SessionProvider = "redis"
+		beego.BConfig.WebConfig.Session.SessionProviderConfig = beego.AppConfig.String("redisEndpoint")
+	}
 	beego.BConfig.WebConfig.Session.SessionGCMaxLifetime = 3600 * 24 * 365
 
 	service.Start()