Function NewWithStore

middleware/session/middleware.go:82–123

NewWithStore creates session middleware with an optional custom store. Parameters: - config: Variadic parameter to override default config. Returns: - fiber.Handler: The Fiber handler for the session middleware. - *Store: The session store. Usage: handler, store := session.NewWithStore()

(config ...Config)

//
//	handler, store := session.NewWithStore()
func NewWithStore(config ...Config) (fiber.Handler, *Store) {
	registerLogContextTagsOnce.Do(registerLogContextTags)

	cfg := configDefault(config...)

	if cfg.Store == nil {
		cfg.Store = NewStore(cfg)
	}

	handler := func(c fiber.Ctx) error {
		if cfg.Next != nil && cfg.Next(c) {
			return c.Next()
		}

		// Acquire session middleware
		m := acquireMiddleware()
		if err := m.initialize(c, &cfg); err != nil {
			releaseMiddleware(m)
			handleSessionError(c, cfg.ErrorHandler, err)
			return nil
		}

		stackErr := c.Next()

		m.mu.RLock()
		isDestroyed := m.isDestroyed
		m.mu.RUnlock()

		if !isDestroyed {
			m.saveSession()
		} else {
			// saveSession is skipped for destroyed sessions, so the session must
			// be returned to the pool here.
			releaseSession(m.Session)
		}

		releaseMiddleware(m)
		return stackErr
	}

	return handler, cfg.Store
}

var registerLogContextTagsOnce sync.Once