Merge pull request #47 from agungwk/master

Add redis session timeout
aichaos · Jun 23, 2023 · e38f316 · e38f316
2 parents e56ad19 + 1553313
commit e38f316
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/sessions/redis/extra.go b/sessions/redis/extra.go
@@ -79,7 +79,7 @@ func (s *Session) putRedisFrozen(username string, data *sessions.UserData, froze
 		return err
 	}
 
-	err = s.client.Set(key, string(encoded), 0).Err()
+	err = s.client.Set(key, string(encoded), s.sessionTimeout).Err()
 	if err != nil {
 		return err
 	}

diff --git a/sessions/redis/redis.go b/sessions/redis/redis.go
@@ -6,6 +6,7 @@ package redis
 import (
 	"fmt"
 	"strings"
+	"time"
 
 	"github.com/aichaos/rivescript-go/sessions"
 	redis "gopkg.in/redis.v5"
@@ -25,13 +26,17 @@ type Config struct {
 
 	// Settings for the Redis client.
 	Redis *redis.Options
+
+	// Session timeout using time.duration
+	SessionTimeout time.Duration
 }
 
 // Session wraps a Redis client connection.
 type Session struct {
-	prefix       string
-	frozenPrefix string
-	client       *redis.Client
+	prefix         string
+	frozenPrefix   string
+	client         *redis.Client
+	sessionTimeout time.Duration
 }
 
 // New creates a new Redis session instance.
@@ -58,9 +63,10 @@ func New(options *Config) *Session {
 	}
 
 	return &Session{
-		prefix:       options.Prefix,
-		frozenPrefix: options.FrozenPrefix,
-		client:       redis.NewClient(options.Redis),
+		prefix:         options.Prefix,
+		frozenPrefix:   options.FrozenPrefix,
+		client:         redis.NewClient(options.Redis),
+		sessionTimeout: options.SessionTimeout,
 	}
 }