Skip to content

Commit dec5d82

Browse files
committed
Add addr to cookie name
1 parent f67da17 commit dec5d82

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

controller/base_controller.go

+4
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,7 @@ func (h *BaseHandler) CurrentUser(ctx *fasthttp.RequestCtx) (*model.User, error)
9797
}
9898

9999
func (h *BaseHandler) SetCookie(ctx *fasthttp.RequestCtx, name, value string, days int) error {
100+
name = name + h.App.Cf.Main.Addr
100101
encoded, err := h.App.Sc.Encode(name, value)
101102
if err != nil {
102103
return err
@@ -108,13 +109,15 @@ func (h *BaseHandler) SetCookie(ctx *fasthttp.RequestCtx, name, value string, da
108109
c.SetPath("/")
109110
c.SetSecure(false)
110111
c.SetHTTPOnly(true)
112+
c.SetSameSite(fasthttp.CookieSameSiteStrictMode)
111113
c.SetExpire(time.Now().UTC().AddDate(0, 0, days))
112114
ctx.Response.Header.SetCookie(&c)
113115

114116
return err
115117
}
116118

117119
func (h *BaseHandler) GetCookie(ctx *fasthttp.RequestCtx, name string) string {
120+
name = name + h.App.Cf.Main.Addr
118121
if cookieByte := ctx.Request.Header.Cookie(name); len(cookieByte) > 0 {
119122
var value string
120123
if err := h.App.Sc.Decode(name, sdb.B2s(cookieByte), &value); err == nil {
@@ -126,6 +129,7 @@ func (h *BaseHandler) GetCookie(ctx *fasthttp.RequestCtx, name string) string {
126129

127130
func (h *BaseHandler) DelCookie(ctx *fasthttp.RequestCtx, name string) {
128131
if len(name) > 0 {
132+
name = name + h.App.Cf.Main.Addr
129133
ctx.Response.Header.DelClientCookie(name)
130134
}
131135
}

0 commit comments

Comments
 (0)