-
Notifications
You must be signed in to change notification settings - Fork 248
/
client.go
42 lines (38 loc) · 1.92 KB
/
client.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
/*
* Tencent is pleased to support the open source community by making Blueking Container Service available.
* Copyright (C) 2019 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 federation Factory
package federation
import (
"context"
bkmonitor_client "github.com/Tencent/bk-bcs/bcs-services/bcs-monitor/pkg/component/bk_monitor"
"github.com/Tencent/bk-bcs/bcs-services/bcs-monitor/pkg/storegw/bcs_system/source/base"
bkmonitor "github.com/Tencent/bk-bcs/bcs-services/bcs-monitor/pkg/storegw/bcs_system/source/bk_monitor"
"github.com/Tencent/bk-bcs/bcs-services/bcs-monitor/pkg/storegw/bcs_system/source/compute"
"github.com/Tencent/bk-bcs/bcs-services/bcs-monitor/pkg/storegw/bcs_system/source/computev2"
"github.com/Tencent/bk-bcs/bcs-services/bcs-monitor/pkg/storegw/bcs_system/source/prometheus"
"github.com/Tencent/bk-bcs/bcs-services/bcs-monitor/pkg/storegw/clientutil"
)
// ClientFactory 自动切换Prometheus/蓝鲸监控
func ClientFactory(ctx context.Context, clusterID string, source clientutil.MonitorSourceType, metricsPrefix string) (
base.MetricHandler, error) {
switch source {
case clientutil.MonitorSourceCompute:
return compute.NewCompute(), nil
case clientutil.MonitorSourceComputeV2:
return computev2.NewCompute(metricsPrefix), nil
default:
if bkmonitor_client.IsBKMonitorEnabled(clusterID) {
return bkmonitor.NewBKMonitor(), nil
}
return prometheus.NewPrometheus(), nil
}
}