publish v12.2.0-alpha6

sessions/provider.go

@@ -4,6 +4,8 @@ import (
 	"errors"
 	"sync"
 	"time"
+
+	"github.com/kataras/iris/v12/context"
 )
 
 type (
@@ -13,6 +15,7 @@ type (
 		mu               sync.RWMutex
 		sessions         map[string]*Session
 		db               Database
+		dbRequestHandler DatabaseRequestHandler
 		destroyListeners []DestroyListener
 	}
 )
@@ -35,6 +38,9 @@ func (p *provider) RegisterDatabase(db Database) {
 
 	p.mu.Lock() // for any case
 	p.db = db
+	if dbreq, ok := db.(DatabaseRequestHandler); ok {
+		p.dbRequestHandler = dbreq
+	}
 	p.mu.Unlock()
 }
 
@@ -84,6 +90,12 @@ func (p *provider) Init(man *Sessions, sid string, expires time.Duration) *Sessi
 	return newSession
 }
 
+func (p *provider) EndRequest(ctx *context.Context, session *Session) {
+	if p.dbRequestHandler != nil {
+		p.dbRequestHandler.EndRequest(ctx, session)
+	}
+}
+
 // ErrNotFound may be returned from `UpdateExpiration` of a non-existing or
 // invalid session entry from memory storage or databases.
 // Usage: