-
Notifications
You must be signed in to change notification settings - Fork 0
/
graceful.go
99 lines (87 loc) · 1.63 KB
/
graceful.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
94
95
96
97
98
99
package graceful
import (
"context"
"errors"
"sync/atomic"
)
var (
ErrRunning = errors.New(`graceful is running`)
ErrStopped = errors.New(`graceful is stopped`)
g = New()
)
// Service 服务方法
type Service interface {
Service() error
Shutdown(ctx context.Context) error
}
type Graceful struct {
running int32
listen Listen
logger *logger
services map[string]*service
}
func (g *Graceful) log() *logger {
if g.logger != nil {
return g.logger
}
return defaultLog
}
// Add 添加服务
func (g *Graceful) Add(name string, svr Service) error {
if atomic.LoadInt32(&g.running) == 1 {
return ErrRunning
}
g.services[name] = &service{
g: g,
running: 0,
name: name,
svr: svr,
}
return nil
}
// Run 运行所有服务
func (g *Graceful) Run() error {
if atomic.SwapInt32(&g.running, 1) != 0 {
return ErrRunning
}
for _, svr := range g.services {
svr.start()
}
g.log().Infof("received signal: %s", g.listen())
return nil
}
// Stop 停止所有服务
func (g *Graceful) Stop() error {
if atomic.SwapInt32(&g.running, 0) != 1 {
return ErrStopped
}
for _, svr := range g.services {
svr.stop()
}
return nil
}
// New 创建服务维护对象
func New(opts ...Option) *Graceful {
graceful := &Graceful{
running: 0,
listen: DefaultListen,
logger: nil,
services: make(map[string]*service),
}
for _, opt := range opts {
opt(graceful)
}
return graceful
}
// Add 添加服务
func Add(name string, svr Service) error {
return g.Add(name, svr)
}
// Run 运行所有服务
func Run() error {
return g.Run()
}
// Stop 停止所有服务
func Stop() error {
return g.Stop()
}