-
Notifications
You must be signed in to change notification settings - Fork 0
/
sessions.go
93 lines (86 loc) · 2.21 KB
/
sessions.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package http
import (
"crypto/rand"
"encoding/base64"
"fmt"
"io"
"net/http"
"net/url"
"github.com/pjmd89/goutils/systemutils"
"github.com/rs/xid"
)
func (o *SessionManager) Init(sessionName string, sessionLifetime int, w http.ResponseWriter, r *http.Request, sessionData interface{}) (id string, err error) {
o.lock.Lock()
cookie, err := r.Cookie(sessionName)
secureCookie := false
if err == nil {
id = cookie.Value
}
if r.TLS != nil {
secureCookie = true
}
if o.sessions == nil {
o.sessions = make(map[string]interface{})
}
if err != nil || cookie == nil || o.sessions[cookie.Value] == nil {
id = o.sessionID()
cookie = &http.Cookie{
Name: sessionName,
Value: id,
}
cookie.Path = "/"
cookie.HttpOnly = true
cookie.MaxAge = sessionLifetime
cookie.Secure = secureCookie
cookie.SameSite = http.SameSiteNoneMode
o.sessions[cookie.Value] = sessionData
http.SetCookie(w, cookie)
}
o.lock.Unlock()
return
}
func (o *SessionManager) Get() (r interface{}, err error) {
goID := systemutils.GetRoutineID()
if sessionIndex != nil && o.sessions[sessionIndex[goID]] != nil {
r = o.sessions[sessionIndex[goID]]
} else {
err = fmt.Errorf("session is no seted.")
}
return
}
func (o *SessionManager) GetByID(sessionID string) (r interface{}, err error) {
if sessionIndex != nil && o.sessions[sessionID] != nil {
r = o.sessions[sessionID]
} else {
err = fmt.Errorf("session is no seted.")
}
return
}
func (o *SessionManager) Set(sessionData interface{}) {
goID := systemutils.GetRoutineID()
if o.sessions == nil {
o.sessions = map[string]any{}
}
if sessionIndex != nil {
if _, ok := sessionIndex[goID]; ok {
o.sessions[sessionIndex[goID]] = sessionData
} else {
sessionIndex[goID] = fmt.Sprintf("%s", xid.New())
o.sessions[sessionIndex[goID]] = sessionData
}
}
}
func (o *SessionManager) Destroy() {
goID := systemutils.GetRoutineID()
if sessionIndex != nil && o.sessions[sessionIndex[goID]] != nil {
o.sessions[sessionIndex[goID]] = nil
}
delete(sessionIndex, goID)
}
func (o *SessionManager) sessionID() string {
id := make([]byte, 32)
if _, err := io.ReadFull(rand.Reader, id); err != nil {
return ""
}
return url.QueryEscape(base64.URLEncoding.EncodeToString(id))
}