79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package repositories
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"strconv"
|
|
"time"
|
|
|
|
"git.kocoder.xyz/kocoded/vt/fx/interfaces/stores"
|
|
glide "github.com/valkey-io/valkey-glide/go/v2"
|
|
)
|
|
|
|
type SessionRepository struct {
|
|
store *glide.Client
|
|
logger *slog.Logger
|
|
}
|
|
|
|
func NewSessionRepository(store *glide.Client, logger *slog.Logger) stores.SessionStore {
|
|
return &SessionRepository{store: store, logger: logger}
|
|
}
|
|
|
|
func (sr *SessionRepository) AddSession(s *stores.Session) {
|
|
// options.HSetExOptions{Expiry: options.NewExpiryIn(time.Hour * 2)}
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
|
|
defer cancel()
|
|
|
|
_, err := sr.store.HSet(ctx, s.Token, s.Serialize())
|
|
if err != nil {
|
|
sr.logger.Error("Error adding session to store", "error", err)
|
|
panic(err)
|
|
}
|
|
|
|
_, err = sr.store.Expire(context.Background(), s.Token, time.Hour*2)
|
|
if err != nil {
|
|
sr.logger.Error("Error setting session expiration", "error", err)
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (sr *SessionRepository) GetSessionFromToken(token string) (*stores.Session, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
|
|
defer cancel()
|
|
|
|
s, err := sr.store.HGetAll(ctx, token)
|
|
if err != nil {
|
|
sr.logger.Error("Error getting session from store", "error", err)
|
|
fmt.Println("PANIC - HGETALL")
|
|
panic(err)
|
|
}
|
|
|
|
_, err = sr.store.Expire(ctx, token, time.Hour*2)
|
|
if err != nil {
|
|
sr.logger.Error("Error setting session expiration", "error", err)
|
|
panic(err)
|
|
}
|
|
|
|
return (&stores.Session{}).Deserialize(token, s)
|
|
}
|
|
func (sr *SessionRepository) SetMandantInSession(token string, mandantId uint) error {
|
|
_, err := sr.store.HSet(context.Background(), token, map[string]string{"mandantid": strconv.Itoa(int(mandantId))})
|
|
|
|
if err != nil {
|
|
sr.logger.Error("Error setting mandant in session", "error", err)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (sr *SessionRepository) RemoveSession(token string) {
|
|
_, err := sr.store.HDel(context.Background(), token, []string{"userid", "mandantid"})
|
|
|
|
if err != nil {
|
|
sr.logger.Error("Error removing session from store", "error", err)
|
|
panic(err)
|
|
}
|
|
}
|