Golang简单高效的HTTP会话管理与识别插件库sessionup的使用

简介

sessionup是一个简单但高效的HTTP会话管理和识别包，提供以下特性：

• 轻松的会话管理：
  • 初始化
  • 请求认证
  • 检索所有会话
  • 撤销当前会话
  • 撤销所有其他会话
  • 撤销所有会话
• 可选的会话识别（IP地址、操作系统、浏览器）
• 通过中间件进行认证
• 完全可定制，但有合理的默认值
• 轻量级
• 直观的API
• 允许自定义会话存储

安装

go get github.com/jellydator/sessionup

基本使用示例

首先需要创建一个Manager实例：

package main

import (
	"net/http"
	"time"
	
	"github.com/jellydator/sessionup"
	"github.com/jellydator/sessionup/memstore"
)

func main() {
	// 创建内存存储，会话5分钟后过期
	store := memstore.New(time.Minute * 5)
	
	// 创建会话管理器
	manager := sessionup.NewManager(store)
	
	// 可以自定义配置
	// manager := sessionup.NewManager(store, 
	//     sessionup.Secure(false), 
	//     sessionup.ExpiresIn(time.Hour * 24))
	
	// 设置路由
	http.Handle("/login", loginHandler(manager))
	http.Handle("/", manager.Public(homeHandler()))
	http.Handle("/private", manager.Auth(privateHandler()))
	http.Handle("/logout", manager.Auth(logoutHandler(manager)))
	
	http.ListenAndServe(":8080", nil)
}

func loginHandler(manager *sessionup.Manager) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// 假设用户认证通过，获取用户ID
		userID := "12345"
		
		// 创建新会话，可以添加元数据
		err := manager.Init(w, r, userID, 
			sessionup.MetaEntry("permission", "write"),
			sessionup.MetaEntry("age", "30"))
		if err != nil {
			http.Error(w, "Failed to create session", http.StatusInternalServerError)
			return
		}
		
		w.Write([]byte("Login successful!"))
	}
}

func homeHandler() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("Public home page"))
	}
}

func privateHandler() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// 从上下文中获取会话
		session, ok := sessionup.FromContext(r.Context())
		if !ok {
			http.Error(w, "Unauthorized", http.StatusUnauthorized)
			return
		}
		
		w.Write([]byte("Private page for user: " + session.UserKey))
	}
}

func logoutHandler(manager *sessionup.Manager) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// 撤销当前会话
		if err := manager.Revoke(r.Context(), w); err != nil {
			http.Error(w, "Failed to logout", http.StatusInternalServerError)
			return
		}
		
		w.Write([]byte("Logged out successfully"))
	}
}

高级功能示例

// 获取当前用户的所有会话
func getAllSessionsHandler(manager *sessionup.Manager) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		sessions, err := manager.FetchAll(r.Context())
		if err != nil {
			http.Error(w, "Failed to fetch sessions", http.StatusInternalServerError)
			return
		}
		
		// 处理会话列表
		for _, session := range sessions {
			// 打印每个会话的信息
			fmt.Printf("Session ID: %s, CreatedAt: %v\n", session.ID, session.CreatedAt)
		}
	}
}

// 撤销所有其他会话（保留当前会话）
func revokeOtherSessionsHandler(manager *sessionup.Manager) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if err := manager.RevokeOther(r.Context()); err != nil {
			http.Error(w, "Failed to revoke other sessions", http.StatusInternalServerError)
			return
		}
		
		w.Write([]byte("Revoked all other sessions"))
	}
}

// 撤销所有会话（包括当前会话）
func revokeAllSessionsHandler(manager *sessionup.Manager) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if err := manager.RevokeAll(r.Context(), w); err != nil {
			http.Error(w, "Failed to revoke all sessions", http.StatusInternalServerError)
			return
		}
		
		w.Write([]byte("Revoked all sessions"))
	}
}

存储实现

sessionup支持多种存储后端：

• memstore - 内存存储实现（已包含在包中）
• Redis存储实现
• PostgreSQL存储实现
• SQLite存储实现
• Bolt存储实现

自定义存储需要实现Store接口。

限制

sessionup仅提供服务器端会话存储和管理，因为无法通过cookie存储撤销/检索不在传入请求中的会话的功能。

总结

sessionup是一个简单高效的Golang HTTP会话管理库，提供了完整的会话生命周期管理功能，包括创建、验证、检索和撤销会话等操作。它设计轻量，API直观，并支持多种存储后端，非常适合需要会话管理的Web应用程序。