14.2. Session 支持

第六章的时候我们介绍过如何在 Go 语言中使用 session，也实现了一个 sessionManger，beego 框架基于 sessionManager 实现了方便的 session 处理功能。

session 集成

beego 中主要有以下的全局变量来控制 session 处理：

// related to session 
SessionOn            bool   // 是否开启 session 模块，默认不开启
SessionProvider      string // session 后端提供处理模块，默认是 sessionManager 支持的 memory
SessionName          string // 客户端保存的 cookies 的名称
SessionGCMaxLifetime int64  // cookies有效期

GlobalSessions *session.Manager // 全局session控制器

当然上面这些变量需要初始化值，也可以按照下面的代码来配合配置文件以设置这些值：

if ar, err := AppConfig.Bool("sessionon"); err != nil {
    SessionOn = false
} else {
    SessionOn = ar
}
if ar := AppConfig.String("sessionprovider"); ar == "" {
    SessionProvider = "memory"
} else {
    SessionProvider = ar
}
if ar := AppConfig.String("sessionname"); ar == "" {
    SessionName = "beegosessionID"
} else {
    SessionName = ar
}
if ar, err := AppConfig.Int("sessiongcmaxlifetime"); err != nil && ar != 0 {
    int64val, _ := strconv.ParseInt(strconv.Itoa(ar), 10, 64)
    SessionGCMaxLifetime = int64val
} else {
    SessionGCMaxLifetime = 3600
}   

在 beego.Run 函数中增加如下代码：

if SessionOn {
    GlobalSessions, _ = session.NewManager(SessionProvider, SessionName, SessionGCMaxLifetime)
    go GlobalSessions.GC()
}

这样只要 SessionOn 设置为 true，那么就会默认开启 session 功能，独立开一个 goroutine 来处理 session。

为了方便我们在自定义 Controller 中快速使用 session，作者在 beego.Controller 中提供了如下方法：

func (c *Controller) StartSession() (sess session.Session) {
    sess = GlobalSessions.SessionStart(c.Ctx.ResponseWriter, c.Ctx.Request)
    return
}       

session 使用

通过上面的代码我们可以看到，beego 框架简单地继承了 session 功能，那么在项目中如何使用呢？

首先我们需要在应用的 main 入口处开启 session：

beego.SessionOn = true

然后我们就可以在控制器的相应方法中如下所示的使用 session了：

func (this *MainController) Get() {
    var intcount int
    sess := this.StartSession()
    count := sess.Get("count")
    if count == nil {
        intcount = 0
    } else {
        intcount = count.(int)
    }
    intcount = intcount + 1
    sess.Set("count", intcount)
    this.Data["Username"] = "astaxie"
    this.Data["Email"] = "astaxie@gmail.com"
    this.Data["Count"] = intcount
    this.TplNames = "index.tpl"
}

上面的代码展示了如何在控制逻辑中使用 session，主要分两个步骤：

1. 获取 session 对象

    // 获取对象, 类似 PHP 中的 session_start()
    sess := this.StartSession()

2. 使用 session 进行一般的 session 值操作

    // 获取 session 值，类似 PHP 中的 $_SESSION["count"]
    sess.Get("count")

    // 设置 session 值
    sess.Set("count", intcount)

从上面代码可以看出基于 beego 框架开发的应用中使用 session 相当方便，基本上和 PHP 中调用 session_start() 类似。