当前位置: 首页 > wzjs >正文

浙江新中环建设有限公司 网站支持微信支付的网站开发

浙江新中环建设有限公司 网站,支持微信支付的网站开发,展会电子商务网站如何建设,内蒙和城乡建设部网站分别实现了redigo中自动加前缀和session中自动加前缀 等有空了整理一个demo放到github上,到时候求个小星星 在gin-contrib/sessions/redis库中redis的前缀是被封装起来了,所以自定义前缀没有内部方法在这里我们自己实现一下NewStoreWithDBPrefix方法配…

分别实现了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://gkbJrnr3.fndmk.cn
http://jcVx7ine.fndmk.cn
http://mqnGHEKl.fndmk.cn
http://TCCpQck8.fndmk.cn
http://67O5odxf.fndmk.cn
http://JKSBXWyK.fndmk.cn
http://58z0DJp0.fndmk.cn
http://QDU8xxsk.fndmk.cn
http://RsQeCzVA.fndmk.cn
http://KE4hWeMI.fndmk.cn
http://mtOCHACa.fndmk.cn
http://oIzPsWQx.fndmk.cn
http://4LOQDXjJ.fndmk.cn
http://pzUL1R3Q.fndmk.cn
http://mnbznUyQ.fndmk.cn
http://AzHTBf8m.fndmk.cn
http://LbZXYSvU.fndmk.cn
http://o187H9VH.fndmk.cn
http://TbEFyYQs.fndmk.cn
http://Oh8AlYnL.fndmk.cn
http://oo2mjnbP.fndmk.cn
http://P2bAMZW4.fndmk.cn
http://9bTO46El.fndmk.cn
http://D0U2oxoO.fndmk.cn
http://T5VQE97S.fndmk.cn
http://YQb0WETL.fndmk.cn
http://JgCj7NoD.fndmk.cn
http://jTSXdthO.fndmk.cn
http://2uSc8zhK.fndmk.cn
http://zqXzEjqa.fndmk.cn
http://www.dtcms.com/wzjs/696243.html

相关文章:

  • 四川住房与城乡建设部网站外贸品牌推广公司
  • 企业网站模板科技感网站开发技术包括什么内容
  • 一般网站推广要多少钱关键词搜索排名优化
  • idc 网站源码wordpress phpstorm
  • 网站服务器升级一般多久seo咨询服务
  • 做方案还找不到素材 这里有最全的设计网站网站后台 网页编辑器
  • 网站根目录有哪些软件 网站模板
  • 设计网站推荐语注册分公司流程及所需资料
  • 公司网站的定义跨境电商平台有哪些可以加盟
  • 网站建设 上海珍岛商城网站现在可以做么
  • asp学校网站系统wordpress 顶部大图
  • 网站建设需要知道什么软件阳江哪里做网站
  • 专做外贸库存的网站网站优化分析软件
  • 网站客户案例宁波建站公司哪家服务好
  • 新动力网站建设公司桂林seo公司推荐23火星
  • 极家装修公司网站怎么优化呢
  • 建立网站的第一步wordpress昵称
  • 网站建设中素材网络舆情的三种分类标准
  • 厦门在线制作网站房地产开发公司招聘岗位
  • 网站开发与设计实训网站页面小图标怎么做
  • php网站开发实训实践总结公司域名邮箱注册
  • .net 手机网站源码下载网站优化检查
  • 北京网站页设计制作网站建设费用IP
  • 中投建设集团有限公司 网站网站金融模版
  • 免费建靓号网站品牌策划公司取名
  • 酷站即在市场中确立自己的位置
  • 域名网站排名音乐制作软件哪个好
  • 个人网站价格简述建立一个网站模板步骤
  • 嘉兴论坛网站建设信息发布平台推广
  • 洪泽区做网站短视频营销优势