-
Notifications
You must be signed in to change notification settings - Fork 0
/
context.go
69 lines (57 loc) · 1.21 KB
/
context.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
package light
import (
"context"
"time"
)
type Context struct {
ctx context.Context
metaData map[string]string
internalMap map[string]interface{}
}
func NewCtx(ctx context.Context) *Context {
if ctx == nil {
ctx = context.Background()
}
return &Context{
ctx: ctx,
metaData: map[string]string{},
internalMap: map[string]interface{}{},
}
}
func DefaultCtx() *Context {
return &Context{
ctx: context.Background(),
metaData: map[string]string{},
internalMap: map[string]interface{}{},
}
}
func (c *Context) Value(key string) string {
return c.metaData[key]
}
func (c *Context) SetValue(key, val string) {
c.metaData[key] = val
}
func (c *Context) GetMetaData() map[string]string {
return c.metaData
}
func (c *Context) SetMetaData(r map[string]string) {
c.metaData = r
}
func (c *Context) SetTimeout(timeout time.Duration) {
if timeout > 0 {
c.internalMap["timeout"] = timeout
}
}
func (c *Context) GetTimeout() time.Duration {
i, ex := c.internalMap["timeout"]
if !ex {
return 0
}
return i.(time.Duration)
}
func (c *Context) GetPath() string {
return c.metaData["light_path"]
}
func (c *Context) SetPath(path string) {
c.metaData["light_path"] = path
}