mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-02 19:57:42 +08:00
86 lines
2.6 KiB
Go
86 lines
2.6 KiB
Go
// RAINBOND, Application Management Platform
|
|
// Copyright (C) 2014-2017 Goodrain Co., Ltd.
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version. For any non-GPL usage of Rainbond,
|
|
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
|
|
// must be obtained first.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package metric
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"k8s.io/apimachinery/pkg/util/sets"
|
|
|
|
"github.com/goodrain/rainbond/gateway/metric/collectors"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
// Collector defines the interface for a metric collector
|
|
type Collector interface {
|
|
Start()
|
|
Stop()
|
|
SetHosts(sets.String)
|
|
SetServerNum(httpNum, tcpNum int)
|
|
RemoveHostMetric([]string)
|
|
}
|
|
|
|
type collector struct {
|
|
registry *prometheus.Registry
|
|
socket *collectors.SocketCollector
|
|
gatewayController *collectors.Controller
|
|
nginxCmd *collectors.NginxCmdMetric
|
|
}
|
|
|
|
// NewCollector creates a new metric collector the for ingress controller
|
|
func NewCollector(gatewayHost string, registry *prometheus.Registry) (Collector, error) {
|
|
ic := collectors.NewController()
|
|
socketCollector, err := collectors.NewSocketCollector(gatewayHost, true)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create socket collector failure %s", err.Error())
|
|
}
|
|
return Collector(&collector{
|
|
gatewayController: ic,
|
|
socket: socketCollector,
|
|
registry: registry,
|
|
nginxCmd: &collectors.NginxCmdMetric{},
|
|
}), nil
|
|
}
|
|
|
|
func (c *collector) Start() {
|
|
c.registry.MustRegister(c.gatewayController)
|
|
c.registry.MustRegister(c.socket)
|
|
c.registry.MustRegister(c.nginxCmd)
|
|
go c.socket.Start()
|
|
}
|
|
|
|
func (c *collector) Stop() {
|
|
c.registry.Unregister(c.gatewayController)
|
|
c.registry.Unregister(c.socket)
|
|
c.registry.Unregister(c.nginxCmd)
|
|
}
|
|
|
|
func (c *collector) SetServerNum(httpNum, tcpNum int) {
|
|
c.gatewayController.SetServerNum(httpNum, tcpNum)
|
|
}
|
|
|
|
func (c *collector) SetHosts(hosts sets.String) {
|
|
c.socket.SetHosts(hosts)
|
|
}
|
|
|
|
//RemoveHostMetric -
|
|
func (c *collector) RemoveHostMetric(hosts []string) {
|
|
c.socket.RemoveMetrics(hosts, c.registry)
|
|
}
|