Golang web 开发实战之 session 缓存:如何使用 redigo 将一个结构体数据保存到 redis?

defonds · · 2865 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

自定义 session 结构体:
type Session struct {
	SessionID  string        `json:"sessionId" bson:"sessionId"`
	User       *User         `json:"-" bson:"user"`
	UserType   string        `json:"userType" bson:"userType"`
	NickName   string        `json:"nickName" bson:"nickName"`
	CreateTime time.Time     `json:"-" bson:"createTime"`
	UpdateTime time.Time     `json:"-" bson:"updateTime"`
	Expires    time.Time     `json:"-" bson:"expires"`
	Locale     string        `json:"-" bson:"locale"` // default is zh_CN
	Menus      []wmodel.Menu `json:"menus" bson:"menus"`
}

1. session 的保存

使用 json.Marshal 将结构体 json 化之后保存到 redis:

/*
	【增】
	描述:插入一个 session 对象
	session 顶级 key,顶级 key 可以设置过期时间
	<[session]: 要插入的 session 对象
	>[error]:插入失败相关信息
*/
func (s *sessionService) SetSession(session *model.Session) error {
	// 从池里获取连接
	conn := pool.Get()
	if conn == nil {
		log.Errorf("redis connnection is nil")
		return errors.New("redis connnection is nil")
	}
	// 用完后将连接放回连接池
	defer conn.Close()
	// 将session转换成json数据,注意:转换后的value是一个byte数组
	value, err := json.Marshal(session)
	if err != nil {
		log.Errorf("json marshal err,%s", err)
		return err
	}
	log.Infof("send data[%s]", session.SessionID, value)
	_, err = conn.Do("SET", session.SessionID, value, "EX", sessionTimeOutInSeconds)
	if err != nil {
		return err
	}
	return nil
}

Golang 测试验证:

func TestSetSession(t *testing.T) {
	s := &model.Session{
		SessionID: "20150421120000",
		UserType:  "admin",
		NickName:  "df",
	}
	err := SessionService.SetSession(s)
	if err != nil {
		t.Errorf("fail to add one session(%+v): %s", s, err)
		t.FailNow()
	}
}

redis 客户端查看该 session 保存情况:

redis保存情况.png


2. session 的删除

直接通过 redigo 驱动远程调用 DEL 命令:

/*
	【删】
	描述: 删除一个 session 对象
	session 顶级 key,一般情况下 session 会在用户无操作 30 分钟后自行过期删除
	但用户登出操作可以提前对 session 进行删除,这就是本方法被调用的地方
	<[sessionID]: 要删除的 session 对象的 id
	>[error]:删除失败相关信息
*/
func (s *sessionService) DelSession(sessionID string) (err error) {
	// 从池里获取连接
	conn := pool.Get()
	if conn == nil {
		log.Errorf("redis connnection is nil")
		return errors.New("redis connnection is nil")
	}
	// 用完后将连接放回连接池
	defer conn.Close()
	log.Infof("move data[%s]", sessionID)
	_, err = conn.Do("DEL", sessionID)
	if err != nil {
		return err
	}
	return nil
}
Golang 测试验证:

func TestDelSession(t *testing.T) {
	err := SessionService.DelSession("20150421120000")
	if err != nil {
		t.Errorf("fail to delete one session(%s): %s", "20150421120000", err)
		t.FailNow()
	}
}

3. session 的获取

使用 json.Unmarshal 将序列化之后的结构体还原:

/*
	【查】
	描述: 查看并返回一个 session 实体
	session 顶级 key
	<[sessionID]: 要查看的 session 对象的 id
	>[error]:查看失败相关信息
*/
func (s *sessionService) GetSession(sessionID string) (session *model.Session, err error) {
	// 从池里获取连接
	conn := pool.Get()
	if conn == nil {
		log.Errorf("redis connnection is nil")
		return nil, errors.New("redis connnection is nil")
	}
	// 用完后将连接放回连接池
	defer conn.Close()
	log.Infof("exists data[%s]", sessionID)
	// 先查看该session是否存在
	var ifExists bool
	ifExists, err = SessionService.ExistsSession(sessionID)
	if err != nil {
		log.Errorf("fail to exists one session(%s): %s", sessionID, err)
		return nil, errors.New("session not exists, sessionID: " + sessionID)
	}
	if ifExists {
		// json数据在go中是[]byte类型,所以此处用redis.Bytes转换
		valueBytes, err2 := redis.Bytes(conn.Do("GET", sessionID))
		if err2 != nil {
			return nil, err2
		}
		//log.Infof("receive data[%s]:%s", sessionID, string(valueBytes))
		session = &model.Session{}
		err = json.Unmarshal(valueBytes, session)
		if err != nil {
			return nil, err
		}
		return session, nil
	} else {
		return nil, errors.New("session not exists, sessionID: " + sessionID)
	}

}
Golang 测试验证:

func TestGetSession(t *testing.T) {
	s, err := SessionService.GetSession("20150421120000")
	if err != nil {
		t.Errorf("fail to exists one session(%s): %s", "20150421120000", err)
		t.FailNow()
	}
	log.Debug("session exists, session nickname is: %s", s.NickName)
}


后记

redis 的作者为了保持简单的架构只允许我们对 top level 的 key 设置超时时间,次级 key(比如 Hash 里边的每个子 key)是不能设置超时时间的,所以我们单独使用了一个 index 为 3 的 redis 库专门存放 session,就是方便管理。另外,我们将每个 session 的 key(即 top level 的 key)有效期设置为半小时,有效期断定及处理托管 redis,避开了程序里对 session 超时机制管理的复杂性——特别是分布式环境。

另外,本文只提供了 session 的增、删、查操作,对于 session 的修改以及有效期推延操作笔者建议可以先删除再增加。

参考资料

GO: How to save and retrieve a struct to redis using redigo

有疑问加站长微信联系(非本文作者)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

2865 次点击  
加入收藏 微博
1 回复  |  直到 2018-07-04 14:43:05
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传