forked from woshihaomei/pitaya
/
module.go
175 lines (147 loc) · 5.26 KB
/
module.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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
// Copyright (c) TFG Co. All Rights Reserved.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.
package pitaya
import (
"fmt"
"github.com/hnlxhzw/pitaya/interfaces"
"github.com/hnlxhzw/pitaya/logger"
)
var (
modulesMap = make(map[string]interfaces.Module)
modulesArr = []moduleWrapper{}
customerModulesArr = []moduleWrapper{} //add by shawn 用户自定义组件 加载流程 component > customerComp > module > customerModule 销毁流程 customerModule > module > component > customerComp
)
type moduleWrapper struct {
module interfaces.Module
name string
}
// RegisterModule registers a module, by default it register after registered modules
func RegisterModule(module interfaces.Module, name string) error {
return RegisterModuleAfter(module, name)
}
// RegisterModuleAfter registers a module after all registered modules
func RegisterModuleAfter(module interfaces.Module, name string) error {
if err := alreadyRegistered(name); err != nil {
return err
}
modulesMap[name] = module
modulesArr = append(modulesArr, moduleWrapper{
module: module,
name: name,
})
return nil
}
// RegisterModuleBefore registers a module before all registered modules
func RegisterModuleBefore(module interfaces.Module, name string) error {
if err := alreadyRegistered(name); err != nil {
return err
}
modulesMap[name] = module
modulesArr = append([]moduleWrapper{
{
module: module,
name: name,
},
}, modulesArr...)
return nil
}
// RegisterCustomerModule 用户自动义模块
func RegisterCustomerModule(module interfaces.Module, name string) error {
if err := alreadyRegistered(name); err != nil {
return err
}
modulesMap[name] = module
customerModulesArr = append(customerModulesArr, moduleWrapper{
module: module,
name: name,
})
return nil
}
// GetModule gets a module with a name
func GetModule(name string) (interfaces.Module, error) {
if m, ok := modulesMap[name]; ok {
return m, nil
}
return nil, fmt.Errorf("module with name %s not found", name)
}
func alreadyRegistered(name string) error {
if _, ok := modulesMap[name]; ok {
return fmt.Errorf("module with name %s already exists", name)
}
return nil
}
// startModules starts all modules in order
func startModules() {
logger.Log.Debug("initializing all modules")
for _, modWrapper := range modulesArr {
logger.Log.Debugf("initializing module: %s", modWrapper.name)
if err := modWrapper.module.Init(); err != nil {
logger.Log.Fatalf("error starting module %s, error: %s", modWrapper.name, err.Error())
}
}
for _, modWrapper := range modulesArr {
modWrapper.module.AfterInit()
logger.Log.Infof("module: %s successfully loaded", modWrapper.name)
}
}
// startCustomerModules add by shawn 启动所有自定义的模块
func startCustomerModules() {
logger.Log.Debug("initializing all customer modules")
for _, modWrapper := range customerModulesArr {
logger.Log.Debugf("initializing customer module: %s", modWrapper.name)
if err := modWrapper.module.Init(); err != nil {
logger.Log.Fatalf("error starting customer module %s, error: %s", modWrapper.name, err.Error())
}
}
for _, modWrapper := range customerModulesArr {
modWrapper.module.AfterInit()
logger.Log.Infof("customer module: %s successfully loaded", modWrapper.name)
}
}
// shutdownModules starts all modules in reverse order
func shutdownModules() {
for i := len(modulesArr) - 1; i >= 0; i-- {
modulesArr[i].module.BeforeShutdown()
}
for i := len(modulesArr) - 1; i >= 0; i-- {
name := modulesArr[i].name
mod := modulesArr[i].module
logger.Log.Debugf("stopping module: %s", name)
if err := mod.Shutdown(); err != nil {
logger.Log.Warnf("error stopping module: %s", name)
}
logger.Log.Infof("module: %s stopped!", name)
}
}
// shutdownCustomerModules add by shawn 关闭所有自定义的模块
func shutdownCustomerModules() {
for i := len(customerModulesArr) - 1; i >= 0; i-- {
customerModulesArr[i].module.BeforeShutdown()
}
for i := len(customerModulesArr) - 1; i >= 0; i-- {
name := customerModulesArr[i].name
mod := customerModulesArr[i].module
logger.Log.Debugf("stopping customer module: %s", name)
if err := mod.Shutdown(); err != nil {
logger.Log.Warnf("error stopping customer module: %s", name)
}
logger.Log.Infof("customer module: %s stopped!", name)
}
}