golang封装cookie

package lib

import (
    "net/http"
    "time"
)

//用来做cookie 处理
type CookieHandle struct {
    Http_writer  http.ResponseWriter //主要用来写入头部
    Http_request *http.Request       //主要用来获取头部信息
    Expires      time.Duration       //过期时间 纳秒
}

//cookie 初始化
func CookieInit(w http.ResponseWriter, req *http.Request, expire string) *CookieHandle {
    //cookie 初始化
    cookie := new(CookieHandle)
    cookie.Http_writer = w
    cookie.Http_request = req

    cookie.Expires = cookie.expireDuration(expire)
    return cookie
}

//获取cookie
func (this *CookieHandle) GetCookie(key string) string {
    cookie, cookie_err := this.Http_request.Cookie(key)
    if cookie_err != nil {
        return ""
    }

    return cookie.Value
}
func (this *CookieHandle) expireDuration(expire string) time.Duration {
    expire_time, err := time.ParseDuration(expire)
    //如果解析有效时时间失败 不设置有效时间
    if err != nil {

        return 0
    }
    return expire_time
}

//设置cookie
func (this *CookieHandle) SetCookie(key, val string, expire ...string) {
    //创建 http.cookie 指针
    cookie := new(http.Cookie)
    cookie.Name = key
    cookie.Value = val
    //如果有传递超时时间 重新设置超时时间
    if len(expire) > 0 {
        this.Expires = this.expireDuration(expire[0])
    }
    //如果不设置 过期时间 不赋值
    if this.Expires != 0 {
        cookie.Expires = time.Now().Add(this.Expires)
    }
    //调用http 包 进行设置cookie  大致是 讲cookie 构造体的数据 生成 string  header().set()
    http.SetCookie(this.Http_writer, cookie)

}

//删除cookie
func (this *CookieHandle) DelCookie(key string) {
    cookie := new(http.Cookie)
    cookie.Name = key
    //讲过期时间改为 -1 s 就可以清空对应的cookie
    cookie.Expires = time.Now().Add(this.expireDuration("-1s"))
    http.SetCookie(this.Http_writer, cookie)
}



调用代码


package main

import (
    "bbs/lib"

    "net/http"
)

func main() {
    http.HandleFunc("/", test)
    http.ListenAndServe(":8080", nil)
}

func test(w http.ResponseWriter, req *http.Request) {
     cookie := lib.CookieInit(w, req, "1m")
    cookie.SetCookie("test", "xiaochuan")

    w.Write([]byte(cookie.GetCookie("test")))
}

写个这玩意也是头痛万分啊。还是太渣