authorizer/server/session/session.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),
		}
	}
}