70 lines
2.2 KiB
Go
70 lines
2.2 KiB
Go
|
package sessions
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/boj/redistore"
|
||
|
|
"github.com/garyburd/redigo/redis"
|
||
|
|
"github.com/gorilla/sessions"
|
||
|
|
)
|
||
|
|
|
||
|
|
type RedisStore interface {
|
||
|
|
Store
|
||
|
|
}
|
||
|
|
|
||
|
|
// size: maximum number of idle connections.
|
||
|
|
// network: tcp or udp
|
||
|
|
// address: host:port
|
||
|
|
// password: redis-password
|
||
|
|
// Keys are defined in pairs to allow key rotation, but the common case is to set a single
|
||
|
|
// authentication key and optionally an encryption key.
|
||
|
|
//
|
||
|
|
// The first key in a pair is used for authentication and the second for encryption. The
|
||
|
|
// encryption key can be set to nil or omitted in the last pair, but the authentication key
|
||
|
|
// is required in all pairs.
|
||
|
|
//
|
||
|
|
// It is recommended to use an authentication key with 32 or 64 bytes. The encryption key,
|
||
|
|
// if set, must be either 16, 24, or 32 bytes to select AES-128, AES-192, or AES-256 modes.
|
||
|
|
func NewRedisStore(size int, network, address, password string, keyPairs ...[]byte) (RedisStore, error) {
|
||
|
|
store, err := redistore.NewRediStore(size, network, address, password, keyPairs...)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &redisStore{store}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewRedisStoreWithDB - like NewRedisStore but accepts `DB` parameter to select
|
||
|
|
// redis DB instead of using the default one ("0")
|
||
|
|
//
|
||
|
|
// Ref: https://godoc.org/github.com/boj/redistore#NewRediStoreWithDB
|
||
|
|
func NewRedisStoreWithDB(size int, network, address, password, DB string, keyPairs ...[]byte) (RedisStore, error) {
|
||
|
|
store, err := redistore.NewRediStoreWithDB(size, network, address, password, DB, keyPairs...)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &redisStore{store}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewRedisStoreWithPool instantiates a RediStore with a *redis.Pool passed in.
|
||
|
|
//
|
||
|
|
// Ref: https://godoc.org/github.com/boj/redistore#NewRediStoreWithPool
|
||
|
|
func NewRedisStoreWithPool(pool *redis.Pool, keyPairs ...[]byte) (RedisStore, error) {
|
||
|
|
store, err := redistore.NewRediStoreWithPool(pool, keyPairs...)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &redisStore{store}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type redisStore struct {
|
||
|
|
*redistore.RediStore
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *redisStore) Options(options Options) {
|
||
|
|
c.RediStore.Options = &sessions.Options{
|
||
|
|
Path: options.Path,
|
||
|
|
Domain: options.Domain,
|
||
|
|
MaxAge: options.MaxAge,
|
||
|
|
Secure: options.Secure,
|
||
|
|
HttpOnly: options.HttpOnly,
|
||
|
|
}
|
||
|
|
}
|