forked from TencentBlueKing/bk-cmdb
-
Notifications
You must be signed in to change notification settings - Fork 0
/
manager.go
112 lines (87 loc) · 3.05 KB
/
manager.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
/*
* Tencent is pleased to support the open source community by making 蓝鲸 available.
* Copyright (C) 2017-2018 THL A29 Limited, a Tencent company. All rights reserved.
* Licensed under the MIT License (the "License"); you may not use this file except
* in compliance with the License. You may obtain a copy of the License at
* http://opensource.org/licenses/MIT
* Unless required by applicable law or agreed to in writing, software distributed under
* the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
* either express or implied. See the License for the specific language governing permissions and
* limitations under the License.
*/
package manager
import (
"configcenter/src/framework/common"
"configcenter/src/framework/core/httpserver"
"configcenter/src/framework/core/types"
"configcenter/src/framework/core/input"
"configcenter/src/framework/core/output"
"configcenter/src/framework/core/log"
"github.com/emicklei/go-restful"
"encoding/json"
"io/ioutil"
"context"
)
// Manager contains the basic framework data and the publisher client used to publis events.
type Manager struct {
cancel context.CancelFunc
eventMgr *eventSubscription
ms []Action
OutputerMgr output.Manager
InputerMgr input.Manager
}
// Actions returns metricActions
func (m *Manager) Actions() []httpserver.Action {
var httpactions []httpserver.Action
for _, a := range m.ms {
httpactions = append(httpactions, httpserver.Action{Method: a.Method, Path: a.Path, Handler: func(req *restful.Request, resp *restful.Response) {
value, err := ioutil.ReadAll(req.Request.Body)
if err != nil {
log.Errorf("read http request body failed, error:%s", err.Error())
return
}
mData := types.MapStr{}
if err := json.Unmarshal(value, &mData); nil != err {
log.Errorf("failed to unmarshal the data, error %s", err.Error())
return
}
data, dataErr := a.HandlerFunc(mData)
if nil != dataErr {
log.Errorf("%s", dataErr.Error())
}
// TODO:需要处理返回值的情况
if nil != data {
_ = data
}
}})
}
return httpactions
}
// CreateFrameworkContext create a new framework context instance
func (cli *Manager) CreateFrameworkContext() FrameworkContext {
return cli
}
// RegisterEvent register cmdb 3.0 event
func (cli *Manager) RegisterEvent(key types.EventKey, eventType types.EventType, eventFunc types.EventCallbackFunc) types.EventKey {
return cli.eventMgr.register(key, eventType, eventFunc)
}
// UnRegisterEvent unregister cmdb 3.0 event
func (cli *Manager) UnRegisterEvent(eventKey types.EventKey) {
cli.eventMgr.unregister(eventKey)
}
// stop used to stop the business cycles.
func (cli *Manager) stop() error {
if nil != cli.cancel {
cli.cancel()
}
return cli.InputerMgr.Stop()
}
// Run start the business cycle until the stop method is called.
func (cli *Manager) Run(ctx context.Context, cancel context.CancelFunc) {
cli.cancel = cancel
cli.eventMgr.setOutputer(cli.OutputerMgr)
common.GoRun(func() {
cli.eventMgr.run(ctx)
}, nil)
cli.InputerMgr.Run(ctx, cli)
}