Go 尝试实现session管理

package session

import (
	"fmt"
	"github.com/satori/go.uuid"
	"log"
	"net/http"
	"net/url"
	"sync"
	"time"
)

const (
	CookieName = "goodsCool"
	Timeout    = 60 * 30
)

var Cmgr *ConversationManager

func init() {
	Cmgr = &ConversationManager{
		CookieName: CookieName,
		MaxAge:     Timeout,
		Storage:    MemStorage{DataStorage: make(map[string]interface{}, 0)},
	}
	go Cmgr.GC()
}

type MemStorage struct {
	DataStorage   map[string]interface{}
	Lock          sync.Mutex
	ReadWriteLock sync.RWMutex
}

func (ms *MemStorage) Store(key string, value interface{}) bool {
	ms.Lock.Lock()
	storage := ms.DataStorage
	storage[key] = value
	ms.Lock.Unlock()
	return true
}
func (ms *MemStorage) Load(key string) (value interface{}, ok bool) {
	ms.ReadWriteLock.RLock()
	storage := ms.DataStorage
	value = storage[key]
	if value != nil {
		ok = true
	} else {
		ok = false
	}

	ms.ReadWriteLock.RUnlock()
	return
}
func (ms *MemStorage) Delete(key string) (value interface{}) {
	ms.Lock.Lock()
	storage := ms.DataStorage
	if v, ok := storage[key]; ok {
		value = v
	}
	delete(storage, key)
	ms.Lock.Unlock()
	return
}

func (ms *MemStorage) Range() {
	for key, value := range ms.DataStorage {
		fmt.Printf(" 遍历 %v,%v\n", key, value)
	}
}

type IConversationManager interface {
	Create(cid string) (conversation *Conversation, err error)
	Destroy(cid string) (b bool, err error)
	GC()
}

type IConversation interface {
	Get(key interface{}) (info interface{}, err error)
	Set(key interface{}, value interface{}) (b bool, err error)
}

type Conversation struct {
	LastAccessTime time.Time
	Cid            string
	CSData         MemStorage
	MaxAge         int64
}
type ConversationManager struct {
	CookieName string
	MaxAge     int64
	Storage    MemStorage
}

//实现 Create / Distroy / Get / Set 方法
//创建会话
func (cm *ConversationManager) Create(cid string) (*Conversation, error) {
	//根据cid 创建一个conversatioin
	storage := cm.Storage
	value, ok := storage.Load(cid)
	if ok {
		conversation := value.(*Conversation)
		return conversation, nil
	}

	m := MemStorage{DataStorage: make(map[string]interface{}, 0)}
	c2 := &Conversation{
		LastAccessTime: time.Now(),
		Cid:            cid,
		CSData:         m,
		MaxAge:         60 * 30,
	}

	cm.Storage.Store(cid, c2)
	return c2, nil
}

//getConversation
func (cm *ConversationManager) GetConversation(cid string) (conversation *Conversation) {
	storage := cm.Storage
	value, ok := storage.Load(cid)
	if ok {
		conversation = value.(*Conversation)
		return conversation
	}
	return conversation
}

// 销毁cid 对应的会话
func (cm *ConversationManager) Destroy(cid string) (b bool, err error) {
	if _, ok := cm.Storage.Load(cid); ok {
		cm.Storage.Delete(cid)
	}
	return true, nil
}

func (cm *ConversationManager) GetConversationData(r *http.Request) (mem *MemStorage) {
	cookie, _ := r.Cookie(CookieName)
	if cookie != nil {
		conversationId := cookie.Value
		conversation := Cmgr.GetConversation(conversationId)
		return &(conversation.CSData)
	} else {
		return nil
	}
}

// 会话域存储Get
func (c *Conversation) Get(key string) (interface{}, error) {
	//log.Println("从", c.Cid, "中取数据：key=", key)
	if load, ok := c.CSData.Load(key); ok {
		return load, nil
	} else {
		return nil, nil
	}
}

// 会话域存储Set
func (c *Conversation) Set(key string, value interface{}) (b bool, err error) {
	c.CSData.Store(key, value)
	return true, nil
}

// ConversationManager GC
func (cm *ConversationManager) GC() {
	for {
		time.Sleep(time.Second * 10)
		log.Println("GC ing ...")
		storage := cm.Storage.DataStorage
		duration := time.Duration(cm.MaxAge * 1000 * 1000 * 1000)
		now := time.Now()
		for cid, conversation := range storage {
			//log.Printf("GC 遍历：%v,%v\n", cid, conversation)
			if c, ok := conversation.(*Conversation); ok {
				expire := c.LastAccessTime.Add(duration)
				//log.Println("如果 expire 大于当前时间:", expire, " 当前时间：now=", now)
				if now.After(expire) {
					//将conversation销毁
					log.Println("销毁Conversation: Cid=", cid)
					cm.Destroy(cid)
				}
			}
		}
	}
}

// 给普通链接处理会话，没有就创建同时处理客户端
//的cookie
func (cm *ConversationManager) ManagerNormalRequest(w http.ResponseWriter, r *http.Request) (conversation *Conversation, e error) {
	cookie, err := r.Cookie(CookieName)
	if err != nil {
		e = err
	}
	var conversationId string
	if cookie == nil {
		conversationId, err = url.QueryUnescape(uuid.Must(uuid.NewV4(), nil).String())
		if err != nil {
			e = err
			return
		}
	} else {
		conversationId = cookie.Value
	}
	//创建Conversation,如果有就返回
	cs, err := cm.Create(conversationId)
	if err != nil {
		e = err
		return
	}
	cs.LastAccessTime = time.Now()
	expire := cs.LastAccessTime.Add(time.Duration(cm.MaxAge * 1000 * 1000 * 1000))
	//log.Println("expire: ", expire)
	//设置cookie
	c := http.Cookie{
		Name:  CookieName,
		Value: conversationId,
		Path:  "/",
		//Domain:     "",
		Expires:    expire,
		RawExpires: "",
		MaxAge:     int(cm.MaxAge),
		Secure:     false,
		HttpOnly:   false,
		SameSite:   0,
		Raw:        "",
		Unparsed:   nil,
	}
	//log.Println(c)
	http.SetCookie(w, &c)
	return cs, e
}

func UUID() (string, error) {
	conversationId, err := url.QueryUnescape(uuid.Must(uuid.NewV4(), nil).String())
	return conversationId, err
}

func (cm *ConversationManager) ManagerLogin(loginUser interface{}, w http.ResponseWriter, r *http.Request) {
	cookie, _ := r.Cookie(CookieName)
	//获取当前时间
	now := time.Now()
	var conversationId string
	if cookie != nil {
		//log.Println("已经有了cookie ==>", cookie)
		conversationId = cookie.Value
	} else {
		conversationId, _ = UUID()
		//log.Println("没有cookie,创建会话 cid=", conversationId)
	}
	expire := now.Add(time.Duration(Cmgr.MaxAge * 1000 * 1000 * 1000))
	//设置cookie
	cookie2 := http.Cookie{
		Name:  CookieName,
		Value: conversationId,
		Path:  "/",
		//	Domain:     "",
		Expires:    expire,
		RawExpires: "",
		MaxAge:     0,
		Secure:     false,
		HttpOnly:   false,
		SameSite:   0,
		Raw:        "",
		Unparsed:   nil,
	}
	//存放登录数据
	var conversation *Conversation
	conversation = Cmgr.GetConversation(conversationId)
	if conversation == nil {
		create, _ := Cmgr.Create(cookie2.Value)
		conversation = create
	}
	conversation.LastAccessTime = now
	conversation.Set("user", loginUser)
	log.Println("存放了User: conversationId:", conversation.Cid, "存放的用户：", loginUser)
	http.SetCookie(w, &cookie2)
}

　　使用非常简单，添加Gin 中间件

func ConversationManager() gin.HandlerFunc {
	return func(c *gin.Context) {
		url := c.Request.URL
		//	log.Println("当前访问链接：", url.String())
		if url.Path == "/login" {
			//如果是登录链接，就放行
			c.Next()
			session.Cmgr.ManagerNormalRequest(c.Writer, c.Request)
			return
		}
	 
		//判断有没有cookie ，若有cookie 就根据ConversationId寻找后台对应的Conversation
		//判断Conversation中有没有登录用户
		cookie, _ := c.Request.Cookie(session.CookieName)
		if cookie != nil {
			conversation := session.Cmgr.GetConversation(cookie.Value)
			if conversation != nil {
				login, _ := conversation.Get("user")
				if login == nil {
					session.Cmgr.ManagerNormalRequest(c.Writer, c.Request)
					c.HTML(200, "login.tmpl", nil)
				} else {
					//user := login.(*models.User)
					//log.Println("conversationId:", conversation.Cid, "  用户： ", user.ToString())
					c.Next()
				}
			} else {
				session.Cmgr.ManagerNormalRequest(c.Writer, c.Request)
				c.HTML(200, "login.tmpl", nil)
			}
		} else {
			session.Cmgr.ManagerNormalRequest(c.Writer, c.Request)
			c.HTML(200, "login.tmpl", nil)
		}

	}
}

　　

func main() {
 
	router := gin.Default()
 
	//router.Use(middleware.Cors())
	router.StaticFS("/static", http.Dir("src/static"))
	router.StaticFS("/image/user", http.Dir(imagePath))
	router.LoadHTMLGlob("src/views/admin/*")
	router.NoRoute(controllers.Handle404)

	router.Use(middleware.ConversationManager())
	router.GET("/", controllers.Index)
}

　　

 

func LoginAction(c *gin.Context) {
	userName := c.PostForm("username")
	passWord := c.PostForm("password")

	u := &models.User{UserName: userName, PassWord: passWord}
	login, err := u.Login(u)
	if err != nil {
		log.Println(err)
	}
	if login == nil {
		c.HTML(http.StatusOK, "login.tmpl", nil)
	} else {
		session.Cmgr.ManagerLogin(login, c.Writer, c.Request)
		data := make(map[string]interface{})
		data["loginUser"] = login
		c.HTML(http.StatusOK, "index.tmpl", data)
	}
}