gf/g/frame/gmvc/controller.go

// MVC控制器基类
package gmvc

import (
    "gitee.com/johng/gf/g/net/ghttp"
    "gitee.com/johng/gf/g/frame/gbase"
    "gitee.com/johng/gf/g/net/gsession"
)

const (
    gDEFAULT_SESSION_ID_NAME = "gfsessionid"
)

// 控制器基类
type Controller struct {
    gbase.Base
    Server   *ghttp.Server         // Web Server对象
    Request  *ghttp.ClientRequest  // 请求数据对象
    Response *ghttp.ServerResponse // 返回数据对象
    Cookie   *ghttp.Cookie         // COOKIE操作对象
    Session  *gsession.Session     // SESSION操作对象
    View     *View                 // 视图对象
}

// 控制器初始化
func (c *Controller) Init(s *ghttp.Server, r *ghttp.ClientRequest, w *ghttp.ServerResponse) {
    c.Base.Init()
    c.Server   = s
    c.Request  = r
    c.Response = w
    c.Cookie   = ghttp.NewCookie(c.Request, c.Response)
    c.View     = NewView(c)
    if r := c.Cookie.Get(gDEFAULT_SESSION_ID_NAME); r != "" {
        c.Session = gsession.Get(r)
    } else {
        c.Session = gsession.Get(gsession.Id())
    }
}

// 控制器结束请求
func (c *Controller) Shut() {
    if c.Cookie.Get(gDEFAULT_SESSION_ID_NAME) == "" {
        c.Cookie.Set(gDEFAULT_SESSION_ID_NAME, c.Session.Id())
    }
    c.Cookie.Output()
    c.Response.Output()
}