fiber/middleware/session/store.go

package session

import (
	"fmt"

	"github.com/gofiber/fiber/v2"
)

type Store struct {
	Config
}

func New(config ...Config) *Store {
	// Set default config
	cfg := configDefault(config...)

	return &Store{
		cfg,
	}
}

func (s *Store) Get(c *fiber.Ctx) *Session {
	var fresh bool

	// Get key from cookie
	id := c.Cookies(s.Cookie.Name)

	// If no key exist, create new one
	if len(id) == 0 {
		id = s.KeyGenerator()
		fresh = true
	}

	// Create session object
	sess := &Session{
		ctx:    c,
		config: s,
		fresh:  fresh,
		db:     acquireDB(),
		id:     id,
	}

	// Fetch existing data
	if !fresh {
		raw, err := s.Storage.Get(id)

		// Set data
		if err == nil {
			if _, err := sess.db.UnmarshalMsg(raw); err != nil {
				fmt.Println("[SESSION]", err.Error())
			}
		}
	}
	// Return session object
	return sess
}

func (s *Store) Reset() error {
	return s.Storage.Reset()
}