sesssion middleware

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/cloudwego/base64x v0.1.4 // indirect
 	github.com/cloudwego/iasm v0.2.0 // indirect
 	github.com/gabriel-vasile/mimetype v1.4.5 // indirect
+	github.com/garyburd/redigo v1.6.4 // indirect
 	github.com/gin-contrib/sse v0.1.0 // indirect
 	github.com/gin-gonic/gin v1.10.0 // indirect
 	github.com/go-playground/locales v0.14.1 // indirect

go.sum

@@ -11,6 +11,8 @@ github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSs
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/gabriel-vasile/mimetype v1.4.5 h1:J7wGKdGu33ocBOhGy0z653k/lFKLFDPJMG8Gql0kxn4=
 github.com/gabriel-vasile/mimetype v1.4.5/go.mod h1:ibHel+/kbxn9x2407k1izTA1S81ku1z/DlgOW2QE0M4=
+github.com/garyburd/redigo v1.6.4 h1:LFu2R3+ZOPgSMWMOL+saa/zXRjw0ID2G8FepO53BGlg=
+github.com/garyburd/redigo v1.6.4/go.mod h1:rTb6epsqigu3kYKBnaF028A7Tf/Aw5s0cqA47doKKqw=
 github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
 github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
 github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=

session/init.go

@@ -0,0 +1,23 @@
+package session
+
+import "fmt"
+
+var (
+	sessionMgr SessionMgr
+)
+
+//中间件让用户选择使用哪个版本
+
+func Init(provider string, addr string, options ...string) (err error) {
+	switch provider {
+	case "memory":
+		sessionMgr = NewMemorySessionMgr()
+	case "redis":
+		sessionMgr = NewRedisSessionMgr()
+	default:
+		fmt.Errorf("不支持")
+		return
+	}
+	err = sessionMgr.Init(addr, options...)
+	return
+}

session/memorysession_mgr.go

@@ -1,6 +1,7 @@
 package session
 
 import (
+	"errors"
 	uuid "github.com/satori/go.uuid"
 	"sync"
 )
@@ -25,6 +26,8 @@ func (s *MemorySessionMgr) Init(addr string, options ...string) (err error) {
 	return
 }
 
+//mgr创建一个session
+
 func (s *MemorySessionMgr) CreateSession() (session Session, err error) {
 	s.rwLock.Lock()
 	defer s.rwLock.Unlock()
@@ -33,7 +36,16 @@ func (s *MemorySessionMgr) CreateSession() (session Session, err error) {
 	//创建一个session
 	memorySession := NewMemorySession(sessionId)
 	s.sessionMap[sessionId] = memorySession
-
+	return
-	return memorySession, nil
+}
-
+
+func (s *MemorySessionMgr) GetSession(sessionId string) (session Session, err error) {
+	s.rwLock.Lock()
+	defer s.rwLock.Unlock()
+	session, ok := s.sessionMap[sessionId]
+	if !ok {
+		err = errors.New("session not found")
+		return
+	}
+	return
 }

session/redissession.go

@@ -0,0 +1,109 @@
+package session
+
+import (
+	"encoding/json"
+	"errors"
+	"github.com/garyburd/redigo/redis"
+	"sync"
+)
+
+type RedisSession struct {
+	sessionId string
+	pool      *redis.Pool
+	//设置session key value
+	sessionMap map[string]interface{}
+	rwLock     sync.RWMutex
+	flag       int
+}
+
+//用常量定义状态
+
+const (
+	//内存数据没变化
+	SessionFlagNone = iota
+
+	SessionFlagModify
+)
+
+//构造函数
+
+func NewRedisSession(pool *redis.Pool, id string) *RedisSession {
+	s := &RedisSession{
+		sessionId:  id,
+		pool:       pool,
+		sessionMap: make(map[string]interface{}, 16),
+		flag:       SessionFlagNone,
+	}
+	return s
+}
+
+func (r *RedisSession) Set(key string, value interface{}) (err error) {
+	r.rwLock.Lock()
+	defer r.rwLock.Unlock()
+	r.sessionMap[key] = value
+	//flag
+	r.flag = SessionFlagModify
+	return
+}
+
+func (r *RedisSession) Save() (err error) {
+	r.rwLock.RLock()
+	defer r.rwLock.RUnlock()
+	//如果数据没变，不需要存
+	if r.flag != SessionFlagModify {
+		return
+	}
+	//内存中的sessionMap进行序列化
+	data, err := json.Marshal(r.sessionMap)
+	if err != nil {
+		return
+	}
+	//获取redis链接
+	_, err = r.pool.Get().Do("SET", r.sessionId, string(data))
+	r.flag = SessionFlagNone
+	if err != nil {
+		return
+	}
+	return
+}
+
+func (r *RedisSession) Get(key string) (result interface{}, err error) {
+	r.rwLock.RLock()
+	defer r.rwLock.RUnlock()
+
+	//判断内存有没有数据
+	result, ok := r.sessionMap[key]
+	if !ok {
+		err = errors.New("key not found")
+		return
+	}
+	return
+}
+
+//从redis里再次加载
+
+func (r *RedisSession) LoadFormRedis() (err error) {
+	reply, err := r.pool.Get().Do("GET", r.sessionId)
+	if err != nil {
+		return
+	}
+	//转字符串
+	data, err := redis.String(reply, err)
+	if err != nil {
+		return
+	}
+	//取到的东西反序列化到内存的map中
+	err = json.Unmarshal([]byte(data), &r.sessionMap)
+	if err != nil {
+		return
+	}
+	return
+}
+
+func (r *RedisSession) Del(key string) (err error) {
+	r.rwLock.Lock()
+	defer r.rwLock.Unlock()
+	r.flag = SessionFlagModify
+	delete(r.sessionMap, key)
+	return
+}

session/redissession_mgr.go

@@ -0,0 +1,92 @@
+package session
+
+import (
+	"errors"
+	"github.com/garyburd/redigo/redis"
+	uuid "github.com/satori/go.uuid"
+	"sync"
+	"time"
+)
+
+type RedisSessionMgr struct {
+	//redis 地址
+	addr string
+	//密码
+	password string
+	//链接池
+	pool *redis.Pool
+	//锁
+	rwLock sync.RWMutex
+	//大map
+	sessionMap map[string]Session
+}
+
+//构造
+
+func NewRedisSessionMgr() SessionMgr {
+	sr := &RedisSessionMgr{
+		sessionMap: make(map[string]Session, 32),
+	}
+	return sr
+}
+
+func (r *RedisSessionMgr) Init(addr string, options ...string) (err error) {
+	//若有其他参数
+	if len(options) > 0 {
+		r.password = options[0]
+	}
+	//创建链接池
+	r.pool = myPool(addr, r.password)
+	r.addr = addr
+	return
+}
+
+func myPool(addr, password string) *redis.Pool {
+	return &redis.Pool{
+		MaxIdle:     64,
+		IdleTimeout: 240 * time.Second,
+		MaxActive:   1000,
+		Dial: func() (redis.Conn, error) {
+			conn, err := redis.Dial("tcp", addr)
+			if err != nil {
+				return nil, err
+			}
+			//如果有密码，判断
+			if _, err = conn.Do("AUTH", password); err != nil {
+				conn.Close()
+				return nil, err
+			}
+			return conn, nil
+
+		},
+		//链接测试,开发时写
+		TestOnBorrow: func(c redis.Conn, t time.Time) error {
+			_, err := c.Do("PING")
+			return err
+		},
+	}
+}
+
+//mgr创建一个session
+
+func (r *RedisSessionMgr) CreateSession() (session Session, err error) {
+	r.rwLock.Lock()
+	defer r.rwLock.Unlock()
+	//用uuid作为session id
+	sessionId := uuid.NewV4().String()
+	//创建一个session
+	redisSession := NewRedisSession(r.pool, sessionId)
+	r.sessionMap[sessionId] = redisSession
+	return
+}
+
+func (r *RedisSessionMgr) GetSession(sessionId string) (session Session, err error) {
+	r.rwLock.Lock()
+	defer r.rwLock.Unlock()
+	session, ok := r.sessionMap[sessionId]
+	if !ok {
+		err = errors.New("session not found")
+		return
+	}
+	return
+}