45c956c489
* feat(backend): try a new approach for logs formatting (http) * feat(backend): added logger module * feat(backend): added project/session info to /i endpoint * feat(backend): found a solution for correct caller information * feat(backend): finished logs for http handlers * feat(backend): finished logs for mobile http handlers * feat(backend): finished ender * feat(backend): finished assets * feat(backend): finished heuristics * feat(backend): finished image-storage * feat(backend): finished sink * feat(backend): finished storage * feat(backend): formatted logs in all services * feat(backend): finished foss part * feat(backend): added missed foss part * feat(backend): fixed panic in memory manager and sink service * feat(backend): connectors
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package sessions
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"openreplay/backend/pkg/cache"
|
|
"openreplay/backend/pkg/logger"
|
|
)
|
|
|
|
type Cache interface {
|
|
Set(session *Session) error
|
|
Get(sessionID uint64) (*Session, error)
|
|
SetCache(sessID uint64, data map[string]string) error
|
|
GetCache(sessID uint64) (map[string]string, error)
|
|
}
|
|
|
|
var ErrSessionNotFound = errors.New("session not found")
|
|
|
|
type inMemoryCacheImpl struct {
|
|
log logger.Logger
|
|
sessions cache.Cache
|
|
redis Cache
|
|
}
|
|
|
|
func NewInMemoryCache(log logger.Logger, redisCache Cache) Cache {
|
|
return &inMemoryCacheImpl{
|
|
log: log,
|
|
sessions: cache.New(time.Minute*3, time.Minute*10),
|
|
redis: redisCache,
|
|
}
|
|
}
|
|
|
|
func (i *inMemoryCacheImpl) SetCache(sessID uint64, data map[string]string) error {
|
|
if err := i.redis.SetCache(sessID, data); err != nil && !errors.Is(err, ErrDisabledCache) {
|
|
ctx := context.WithValue(context.Background(), "sessionID", sessID)
|
|
i.log.Warn(ctx, "failed to cache session: %s", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (i *inMemoryCacheImpl) GetCache(sessID uint64) (map[string]string, error) {
|
|
session, err := i.redis.GetCache(sessID)
|
|
if err == nil {
|
|
return session, nil
|
|
}
|
|
if !errors.Is(err, ErrDisabledCache) && err.Error() != "redis: nil" {
|
|
ctx := context.WithValue(context.Background(), "sessionID", sessID)
|
|
i.log.Warn(ctx, "failed to get session from cache: %s", err)
|
|
}
|
|
return nil, ErrSessionNotFound
|
|
}
|
|
|
|
func (i *inMemoryCacheImpl) Set(session *Session) error {
|
|
i.sessions.Set(session.SessionID, session)
|
|
if err := i.redis.Set(session); err != nil && !errors.Is(err, ErrDisabledCache) {
|
|
ctx := context.WithValue(context.Background(), "sessionID", session.SessionID)
|
|
i.log.Warn(ctx, "failed to cache session: %s", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (i *inMemoryCacheImpl) Get(sessionID uint64) (*Session, error) {
|
|
if session, ok := i.sessions.Get(sessionID); ok {
|
|
return session.(*Session), nil
|
|
}
|
|
session, err := i.redis.Get(sessionID)
|
|
if err == nil {
|
|
i.sessions.Set(sessionID, session)
|
|
return session, nil
|
|
}
|
|
if !errors.Is(err, ErrDisabledCache) && err.Error() != "redis: nil" {
|
|
ctx := context.WithValue(context.Background(), "sessionID", sessionID)
|
|
i.log.Warn(ctx, "failed to get session from cache: %s", err)
|
|
}
|
|
return nil, ErrSessionNotFound
|
|
}
|