Initial Codebase (untested)

internal/session_store.go

@@ -0,0 +1,90 @@
+package internal
+
+import (
+	"time"
+
+	"github.com/patrickmn/go-cache"
+	log "github.com/sirupsen/logrus"
+
+	"git.mills.io/prologic/spyda/internal/session"
+)
+
+// SessionStore ...
+type SessionStore struct {
+	store  Store
+	cached *cache.Cache
+}
+
+func NewSessionStore(store Store, sessionCacheTTL time.Duration) *SessionStore {
+	return &SessionStore{
+		store:  store,
+		cached: cache.New(sessionCacheTTL, time.Minute*5),
+	}
+}
+
+func (s *SessionStore) Count() int {
+	return s.cached.ItemCount()
+}
+
+func (s *SessionStore) GetSession(sid string) (*session.Session, error) {
+	val, found := s.cached.Get(sid)
+	if found {
+		return val.(*session.Session), nil
+	}
+
+	return s.store.GetSession(sid)
+}
+
+func (s *SessionStore) SetSession(sid string, sess *session.Session) error {
+	s.cached.Set(sid, sess, cache.DefaultExpiration)
+	if persist, ok := sess.Get("persist"); !ok || persist != "1" {
+		return nil
+	}
+
+	return s.store.SetSession(sid, sess)
+}
+
+func (s *SessionStore) HasSession(sid string) bool {
+	_, ok := s.cached.Get(sid)
+	if ok {
+		return true
+	}
+
+	return s.store.HasSession(sid)
+}
+
+func (s *SessionStore) DelSession(sid string) error {
+	if s.store.HasSession(sid) {
+		if err := s.store.DelSession(sid); err != nil {
+			log.WithError(err).Errorf("error deleting persistent session %s", sid)
+			return err
+		}
+	}
+	s.cached.Delete(sid)
+	return nil
+}
+
+func (s *SessionStore) SyncSession(sess *session.Session) error {
+	if persist, ok := sess.Get("persist"); ok && persist == "1" {
+		if err := s.store.SetSession(sess.ID, sess); err != nil {
+			log.WithError(err).Errorf("error persisting session %s", sess.ID)
+			return err
+		}
+	}
+
+	return s.SetSession(sess.ID, sess)
+}
+
+func (s *SessionStore) GetAllSessions() ([]*session.Session, error) {
+	var sessions []*session.Session
+	for _, item := range s.cached.Items() {
+		sess := item.Object.(*session.Session)
+		sessions = append(sessions, sess)
+	}
+	persistedSessions, err := s.store.GetAllSessions()
+	if err != nil {
+		log.WithError(err).Error("error getting all persisted sessions")
+		return sessions, err
+	}
+	return append(sessions, persistedSessions...), nil
+}