82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package session
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
|
|
"github.com/go-redis/redis/v8"
|
|
"github.com/yauthdev/yauth/server/constants"
|
|
)
|
|
|
|
type SessionStore struct {
|
|
inMemoryStoreObj *InMemoryStore
|
|
redisMemoryStoreObj *RedisStore
|
|
}
|
|
|
|
var SessionStoreObj SessionStore
|
|
|
|
func SetToken(userId, token string) {
|
|
if SessionStoreObj.redisMemoryStoreObj != nil {
|
|
SessionStoreObj.redisMemoryStoreObj.AddToken(userId, token)
|
|
}
|
|
if SessionStoreObj.inMemoryStoreObj != nil {
|
|
SessionStoreObj.inMemoryStoreObj.AddToken(userId, token)
|
|
}
|
|
}
|
|
|
|
func DeleteToken(userId string) {
|
|
if SessionStoreObj.redisMemoryStoreObj != nil {
|
|
SessionStoreObj.redisMemoryStoreObj.DeleteToken(userId)
|
|
}
|
|
if SessionStoreObj.inMemoryStoreObj != nil {
|
|
SessionStoreObj.inMemoryStoreObj.DeleteToken(userId)
|
|
}
|
|
}
|
|
|
|
func GetToken(userId string) string {
|
|
if SessionStoreObj.redisMemoryStoreObj != nil {
|
|
return SessionStoreObj.redisMemoryStoreObj.GetToken(userId)
|
|
}
|
|
if SessionStoreObj.inMemoryStoreObj != nil {
|
|
return SessionStoreObj.inMemoryStoreObj.GetToken(userId)
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
func ClearStore() {
|
|
if SessionStoreObj.redisMemoryStoreObj != nil {
|
|
SessionStoreObj.redisMemoryStoreObj.ClearStore()
|
|
}
|
|
if SessionStoreObj.inMemoryStoreObj != nil {
|
|
SessionStoreObj.inMemoryStoreObj.ClearStore()
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
if constants.REDIS_URL != "" {
|
|
log.Println("Using redis store to save sessions")
|
|
opt, err := redis.ParseURL(constants.REDIS_URL)
|
|
if err != nil {
|
|
log.Fatalln("Error parsing redis url:", err)
|
|
}
|
|
rdb := redis.NewClient(opt)
|
|
ctx := context.Background()
|
|
_, err = rdb.Ping(ctx).Result()
|
|
|
|
if err != nil {
|
|
log.Fatalln("Error connecting to redis server", err)
|
|
}
|
|
SessionStoreObj.redisMemoryStoreObj = &RedisStore{
|
|
ctx: ctx,
|
|
store: rdb,
|
|
}
|
|
|
|
} else {
|
|
log.Println("Using in memory store to save sessions")
|
|
SessionStoreObj.inMemoryStoreObj = &InMemoryStore{
|
|
store: make(map[string]string),
|
|
}
|
|
}
|
|
}
|