go-gin中session实现redis前缀和db库选择+单点登录

分别实现了redigo中自动加前缀和session中自动加前缀

等有空了整理一个demo放到github上,到时候求个小星星

  1. 在gin-contrib/sessions/redis库中redis的前缀是被封装起来了,所以自定义前缀没有内部方法
  2. 在这里我们自己实现一下NewStoreWithDBPrefix方法
  3. 配置文件可以看到引用的redis配置
  4. 这样方面多个项目在同一个redis中方便管理
  5. 也方便做单点登录

项目目录

app/
├── common/
│   ├── global/
│   ├────  global.go
├── config/
│   ├── application.yaml
├── initialize/
│   ├── config.go
│   ├── initialize.go
│   ├── redis.go
│   ├── store.go
├── middleware/
│   ├── middleware.go
├── router/
│   ├── router.go
├── main.go

下面是代码实现

配置文件:config/application.yaml

redis: &redisConfigaddr: 127.0.0.1port: 6379password: 123456db: 3size: 20max-idle: 24active: 10auth: trueprefix: demotimeout: 180
store:size: 10redis: *redisConfigkey-pairs: sessionKey

全局变量:common/global/global.go

package globalimport ("demo/config""github.com/gin-contrib/sessions""github.com/robfig/cron/v3""github.com/garyburd/redigo/redis"ut "github.com/go-playground/universal-translator""github.com/jordan-wright/email""go.uber.org/zap""gorm.io/gorm"
)var (CONFIG    config.ConfigDB        *gorm.DBTrans     *ut.TranslatorREDISPoll *redis.PoolRootDir   stringEmailPool *email.PoolStore     *sessions.StoreLogger    *zap.SugaredLoggerCron      *cron.Cron
)

配置文件初始化:initialize/config.go

package initializeimport ("fmt""io/ioutil""log""os""path/filepath""strings""demo/common/global""demo/config""github.com/joho/godotenv""gopkg.in/yaml.v2"
)/*** 读取配置文件,最先被初始化*/func InitilizeConfig() {// 获取当前工作目录rootDir, err := os.Getwd()if err != nil {log.Fatalf("Failed to get current working directory: %s", err)}// 这是单元测试if strings.HasSuffix(rootDir, "service") || strings.HasSuffix(rootDir, "command") || strings.HasSuffix(rootDir, "util") || strings.HasSuffix(rootDir, "model") {rootDir = filepath.Dir(rootDir)}if strings.HasSuffix(rootDir, "oss") {rootDir = filepath.Dir(filepath.Dir(rootDir))}// rootDir = "/Users/dupeisheng.vendor/go/src/gitlab.bj.sensetime.com/entry_manage"// 设置工作目录if err := os.Chdir(rootDir); err != nil {log.Fatalf("Failed to change working directory: %s", err)}fmt.Println("工作目录:", rootDir)// 加载.envgodotenv.Load(".env")// 设置当前环境mode := os.Getenv("Mode")fmt.Println("当前环境:", mode)if mode != "" {mode = "-" + mode}// 加载配置文件configPath := rootDir + "/config/application" + mode + ".yaml"yamlFile, err := ioutil.ReadFile(configPath)if err != nil {log.Panicf("Failed to read file , cause is %s", err.Error())}config := config.Config{}err = yaml.Unmarshal(yamlFile, &config)global.CONFIG = configif err != nil {fmt.Println(err.Error())}
}

session初始化:initialize/store.go

package initializeimport ("fmt""demo/common/global""github.com/boj/redistore""github.com/gin-contrib/sessions""github.com/gin-contrib/sessions/redis""github.com/spf13/cast"
)const SessionMaxAge = 8 * 3600func InitilizeStore() {var Store sessions.StoreStore, err := NewStoreWithDBPrefix(global.CONFIG.Store.Size,"tcp",global.CONFIG.Store.Redis.Addr+":"+global.CONFIG.Store.Redis.Port,global.CONFIG.Store.Redis.Password,cast.ToString(global.CONFIG.Store.Redis.Db),// []byte(global.CONFIG.Store.Redis.Prefix),[]byte(global.CONFIG.Store.KeyPairs),)if err != nil {fmt.Println("创建Session-Redis存储失败" + err.Error())}Store.Options(sessions.Options{MaxAge: SessionMaxAge,})global.Store = &Store
}type store struct {*redistore.RediStore
}func NewStoreWithDBPrefix(size int, network, address, password, DB string, keyPairs ...[]byte) (redis.Store, error) {s, err := redistore.NewRediStoreWithDB(size, network, address, password, DB, keyPairs...)if err != nil {return nil, err}// 这里设置前缀s.SetKeyPrefix(global.CONFIG.Store.Redis.Prefix + ":session:")return &store{s}, nil
}func (c *store) Options(options sessions.Options) {c.RediStore.Options = options.ToGorillaOptions()
}

redis初始化:initialize/redis.go

package initializeimport ("fmt""log""strings""time""demo/common/global""demo/config""github.com/garyburd/redigo/redis""github.com/samber/lo""github.com/spf13/cast"
)/*** 初始化redis,并赋值给全局变量*/func InitilizeRedis() {// 创建redis连接池global.REDISPoll = GetRedisPool(global.CONFIG)
}func GetRedisPool(config config.Config) *redis.Pool {return &redis.Pool{MaxIdle:     config.Redis.MaxIdle, // 最大空闲连接数MaxActive:   config.Redis.Active,  // 最大连接数IdleTimeout: time.Duration(config.Redis.Timeout) * time.Second,Wait:        true, // 超过连接数后是否等待Dial: func() (redis.Conn, error) {redisUri := fmt.Sprintf("%s:%s", config.Redis.Addr, config.Redis.Port)var redisConn redis.Connvar err errorif config.Redis.Auth {redisConn, err = redis.Dial("tcp", redisUri, redis.DialPassword(config.Redis.Password))} else {redisConn, err = redis.Dial("tcp", redisUri)}if err != nil {log.Println("获取连接失败:" + err.Error())return nil, err}// 添加 Redis 前缀if config.Redis.Prefix != "" {redisConn = &PrefixedConn{Conn: redisConn, config: config.Redis}}return redisConn, nil},TestOnBorrow: func(c redis.Conn, t time.Time) error {if time.Since(t) < time.Minute {return nil}_, err := c.Do("PING")return err},}
}// PrefixedConn 是一个实现了 redis.Conn 接口的自定义结构体,它在键名前添加了前缀
type PrefixedConn struct {redis.Connconfig config.Redis
}// 允许适配前缀的命令
var commandsWithPrefix = []string{"GET", "SET", "EXISTS", "DEL", "TYPE","RPUSH", "LPOP", "RPOP", "LLEN", "LRANGE","SADD", "SREM", "SISMEMBER", "SMEMBERS", "SCARD","HSET", "HMSET", "HGET", "HGETALL","ZADD", "ZRANGE", "ZRANGEBYSCORE", "ZREVRANGEBYSCORE", "ZREM","INCR", "INCRBY","WATCH", "MULTI", "EXEC", "EXPIRE",
}// Do 实现了 redis.Conn 接口的 Do 方法
func (c *PrefixedConn) Do(command string, args ...any) (any, error) {// 执行命令前切换到数据库if _, err := c.Conn.Do("SELECT", c.config.Db); err != nil {return nil, err}command = strings.ToUpper(command)// 判断是不是单独实现了命令 不用反射吧-性能问题switch command {case "MGET":return c.MGET(args...)case "MSET":return c.MSET(args...)}// 加前缀if len(args) > 0 {key := args[0].(string)if lo.IndexOf[string](commandsWithPrefix, command) != -1 && key != "" {args[0] = c.config.Prefix + ":" + key}}return c.Conn.Do(command, args...)
}// MGET 实现了批量获取命令
func (c *PrefixedConn) MGET(args ...any) (interface{}, error) {for i := range args {args[i] = c.config.Prefix + ":" + cast.ToString(args[i])}return c.Conn.Do("MGET", args...)
}// MSET 实现了批量设置命令
func (c *PrefixedConn) MSET(args ...any) (interface{}, error) {for i := range args {if i%2 == 0 {args[i] = c.config.Prefix + ":" + cast.ToString(args[i])}}return c.Conn.Do("MSET", args...)
}

gin路由:router/router.go

package routerimport ("github.com/gin-contrib/cors""github.com/gin-contrib/sessions""github.com/gin-gonic/gin"
)
// InitRouters 初始化路由
func InitRouters() *gin.Engine {// 创建默认带中间件的路由 Logger、Recoveryr := gin.Default()// 允许跨域r.Use(cors.Default())r.Use(sessions.Sessions("sessionID", *global.Store))
}

下面是单点登录实现逻辑

middleware/middleware.go

package middlewareimport ("encoding/json""net/http"myerror "demo/common/error""demo/common/global""demo/model""demo/util""github.com/gin-contrib/sessions""github.com/gin-gonic/gin""github.com/gomodule/redigo/redis""github.com/spf13/cast"
)const UserSessionExpireDuration = 8 * 3600
const UserSessionMapPrefix = "session:user:"type SessionData struct {Member model.Member `json:"member"`
}// 保存登录信息
func GenerateSession(c *gin.Context, sessionData SessionData) error {c.Request.Header.Set("Cookie", "")redisConn := global.REDISPoll.Get()defer redisConn.Close()sessionKey := UserSessionMapPrefix + cast.ToString(sessionData.Member.ID)// 单点登录删除旧的登录态oldSessionID, err := redis.String(redisConn.Do("get", sessionKey))if err != nil && err != redis.ErrNil {return err}_, err = redisConn.Do("del", oldSessionID)if err != nil {return err}// 保存用户相关token := cast.ToString(sessionData.Member.ID) + sessionData.Member.Name + util.Md5(util.GetUUID()+util.RandString(20))session := sessions.Default(c)by, err := json.Marshal(sessionData)if err != nil {return err}session.Set("authenticated", true)session.Set("data", by)session.Set("member_id", sessionData.Member.ID)session.Set("member_name", sessionData.Member.Name)session.Set("token", token)session.Save()sessionVal := "session:" + session.ID()_, err = redisConn.Do("set", sessionKey, sessionVal, "EX", UserSessionExpireDuration)return err
}// 身份验证
func Authentication(ctx *gin.Context) {session := sessions.Default(ctx)if auth, ok := session.Get("authenticated").(bool); !ok || !auth {ctx.JSON(http.StatusUnauthorized, gin.H{"code": myerror.SERVER_UNAUTHORIZED_ERROR,"msg":  myerror.SERVER_UNAUTHORIZED_ERROR.String(),})ctx.Abort()return}session.Save()// 保存用户相关信息到ctxmemberId := session.Get("member_id")memberName := session.Get("member_name")data := session.Get("data")var sessionData SessionDataif data != nil {by := data.([]byte)err := json.Unmarshal(by, &sessionData)if err != nil {ctx.JSON(http.StatusUnauthorized, gin.H{"code": myerror.SERVER_SYSTEM_ERROR,"msg":  myerror.SERVER_SYSTEM_ERROR.String(),})ctx.Abort()return}}ctx.Set("member_id", memberId)ctx.Set("member_name", memberName)ctx.Set("member_info", sessionData.Member)ctx.Set("data", data)// 登录续时redisConn := global.REDISPoll.Get()defer redisConn.Close()sessionKey := UserSessionMapPrefix + cast.ToString(memberId)sessionVal := "session:" + session.ID()redisConn.Do("set", sessionKey, sessionVal, "EX", UserSessionExpireDuration)ctx.Next()
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/pingmian/18110.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

记录一次开源 MaxKey 安装部署

官方文档&#xff1a;https://www.maxkey.top/doc/docs/intro/ 开源代码&#xff1a;https://toscode.mulanos.cn/dromara/MaxKey 发行版&#xff1a;https://toscode.mulanos.cn/dromara/MaxKey/releases 一、准备工作 yum install -y yum-utils yum-config-manager --add-r…

SwiftUI初探

SwiftUI 虽然出现了好几年(1.0好像2019年出的&#xff0c;还有SPM也是同一年)&#xff0c;现在已经到从1.0到5.0&#xff0c;但受限于对系统的要求(最低iOS13.0,有的要求17.0及以上)&#xff0c;每个版本里面差异也很大&#xff0c;语法和Flutter 的Dart 比较像。空闲之余可以先…

谈谈什么是生成器,它与列表推导式有何区别?以及如何使用Python进行异常处理?

生成器&#xff08;Generator&#xff09;是一种特殊的函数&#xff0c;它一次生成一个值&#xff0c;而不是一次性生成所有值。可以将其视为可恢复函数&#xff0c;在函数执行过程中&#xff0c;yield语句会返回需要的值给调用生成器的地方&#xff0c;然后退出函数。下一次调…

linux定时删除历史日志

在Linux系统中&#xff0c;日志文件是记录系统、应用程序或服务的运行信息、错误消息和警告的重要工具。然而&#xff0c;随着时间的推移&#xff0c;这些日志文件会不断积累&#xff0c;占用大量的磁盘空间。如果不及时清理&#xff0c;可能会导致磁盘空间不足&#xff0c;从而…

5、sqlmap注入post类型+os-shell

题目&#xff1a;青少年&#xff1a;Easy_SQLi 1、打开网页&#xff0c;是一个登入表单 2、判断注入类型&#xff0c;是一个字符注入&#xff0c;使用or直接绕过密码进去了 3、上bp抓取数据包&#xff0c;sqlmmap用post注入走一遍&#xff0c;找到数据库&#xff0c;账号密码&…

如何制定一个有效的现货黄金投资策略(EEtrade)

制定一个有效的现货黄金投资策略涉及多方面的考量。以下是几个步骤和考虑因素&#xff0c;可以帮助您建立一个坚实的投资策略&#xff1a; 1. 设立清晰的投资目标 决定您投资现货黄金的主要目的。是否是为了短期利润&#xff0c;长期保值增值&#xff0c;还是为了投资组合的多…

N的阶乘(高精度)

目录 题目描述 输入格式 输出格式 样例输入 样例输出 思路 参考代码 题目描述 输入正整数n&#xff0c;输出n&#xff01; 输入格式 一个正整数n&#xff0c;n 3000 输出格式 输出n&#xff01; 样例输入 3 样例输出 9 思路 主要就是高精度乘法的模版&#x…

Python中Web开发-Django框架

大家好&#xff0c;本文将带领大家进入 Django 的世界&#xff0c;探索其强大的功能和灵活的开发模式。我们将从基础概念开始&#xff0c;逐步深入&#xff0c;了解 Django 如何帮助开发人员快速构建现代化的 Web 应用&#xff0c;并探讨一些最佳实践和高级技术。无论是初学者还…

Unity3D插件开发教程(二):制作批处理工具

Unity3D插件开发教程&#xff08;二&#xff09;&#xff1a;制作批处理工具 文章来源&#xff1a;Unity3D插件开发教程&#xff08;二&#xff09;&#xff1a;制作批处理工具 - 知乎 (zhihu.com) 声明&#xff1a; 题图来自于Gratisography | Free High Resolution Pictures…

QA测试开发工程师面试题满分问答25: JVM瓶颈分析,举例说明

回答思路 JVM 性能瓶颈概述: JVM (Java Virtual Machine)是 Java 程序运行的基础环境,其性能直接影响到应用程序的整体性能。在实际项目中,我们经常会遇到 JVM 性能瓶颈,导致应用程序出现响应缓慢、CPU 利用率高、内存溢出等问题。因此,分析和诊断 JVM 瓶颈是非常重要的工作,有…

SpringValidation

一、概述&#xff1a; ​ JSR 303中提出了Bean Validation&#xff0c;表示JavaBean的校验&#xff0c;Hibernate Validation是其具体实现&#xff0c;并对其进行了一些扩展&#xff0c;添加了一些实用的自定义校验注解。 ​ Spring中集成了这些内容&#xff0c;你可以在Spri…

小程序使用vant组件库

一:下载组件库 在小程序内npm下载的包 vant组件库官网:快速上手 - Vant Weapp (youzan.github.io) 1)首先有有package.json文件,没有的话则先初始化 即使通过package.json去下载包,也需要有,可以观察下载的包. 2)下载包 3)构建npm包 下载包之后存储在node_modules内,但是我们…

初学者必读:Midjourney AI创作工具的简易使用手册!

在数字化时代&#xff0c;AI的应用不断推动着各个领域的发展。在这些领域中&#xff0c;AI在艺术和设计方面的应用引起了广泛的关注。AI绘画软件作为今年的热门&#xff0c;Midjourney 通过其独特的原理和方便的使用方法&#xff0c;为创作者提供了一个全新的创作逼真绘画的平台…

大模型日报|今日必读的 5 篇大模型论文

大家好&#xff0c;今日必读的大模型论文来啦&#xff01; 1.Meta 领衔&#xff1a;一文读懂视觉语言建模&#xff08;VLM&#xff09; 人们正在尝试将大型语言模型&#xff08;LLMs&#xff09;扩展到视觉领域。从可以引导我们穿越陌生环境的视觉助手&#xff0c;到仅使用高…

富港银行 优惠链接 邀请码 兑换码 优惠码 分享

首次记得一定要点击链接注册&#xff0c;注册开户费50美金 限时&#xff01;优惠开通国际银行账户&#xff01; cbi帐户管理费&#xff1a;10美元/月&#xff0c;余额>500美元&#xff0c;1美元/月/&#xff0c;余额>2000美元&#xff0c;0美元/月。 一定要显示50的时候…

一分钟学习数据安全——数字身份的三种模式

微软首席身份架构师金卡梅隆曾说&#xff1a;互联网的构建缺少一个身份层。互联网的构建方式让你无法得知所连接的人和物是什么。这限制了我们对互联网的使用&#xff0c;并让我们面临越来越多的危险。如果我们坐视不管&#xff0c;将面临迅速激增的盗窃和欺诈事件&#xff0c;…

短视频内容创意方法有哪些?成都科成博通文化传媒公司

短视频内容创意方法有哪些&#xff1f; 随着移动互联网的迅猛发展&#xff0c;短视频平台已成为人们日常生活中不可或缺的一部分。短视频以其短平快的特点&#xff0c;迅速吸引了大量用户。然而&#xff0c;面对海量的短视频内容&#xff0c;如何让自己的作品脱颖而出&#xf…

web自动化-下拉框操作/键鼠操作/文件上传

在我们做UI自动化测试的时候&#xff0c;会有一些元素需要特殊操作&#xff0c;比如下拉框操作/键鼠操作/文件上传。 下拉框操作 在我们很多页面里有下拉框的选择&#xff0c;这种元素怎么定位呢&#xff1f;下拉框分为两种类型&#xff1a;我们分别针对这两种元素进行定位和…

Nocobase快速上手 -第一个collection

本文记录Nocobase中如何创建collection&#xff0c;以及如何将collection展示到页面中&#xff0c;并且配置CRUD相应的操作. Collection 在NocoBase中&#xff0c;collection&#xff08;集合&#xff09;是用来组织和存储各种数据的容器&#xff0c;如订单、产品、用户、评论…

spring6 文件上传

Spring 6 的主要变化 Jakarta EE 9 支持: Spring 6完全迁移到了Jakarta EE 9&#xff0c;这意味着所有的包名都从javax前缀改为了jakarta前缀。例如&#xff0c;javax.servlet 变为 jakarta.servlet。因此&#xff0c;如果你在Spring 6中使用文件上传&#xff0c;必须确保依赖库…