mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-04 04:38:04 +08:00
126 lines
3.8 KiB
Go
126 lines
3.8 KiB
Go
// Copyright (C) 2014-2018 Goodrain Co., Ltd.
|
|
// RAINBOND, Application Management Platform
|
|
|
|
// 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 monitor
|
|
|
|
import (
|
|
"github.com/goodrain/rainbond/eventlog/cluster"
|
|
"github.com/goodrain/rainbond/eventlog/store"
|
|
"time"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
// Metric name parts.
|
|
const (
|
|
// Namespace for all metrics.
|
|
namespace = "event_log"
|
|
// Subsystem(s).
|
|
exporter = "exporter"
|
|
)
|
|
|
|
// Metric descriptors.
|
|
var (
|
|
scrapeDurationDesc = prometheus.NewDesc(
|
|
prometheus.BuildFQName(namespace, exporter, "collector_duration_seconds"),
|
|
"Collector time duration.",
|
|
[]string{"collector"}, nil,
|
|
)
|
|
)
|
|
|
|
//Exporter collects entrance metrics. It implements prometheus.Collector.
|
|
type Exporter struct {
|
|
error prometheus.Gauge
|
|
totalScrapes prometheus.Counter
|
|
scrapeErrors *prometheus.CounterVec
|
|
storeManager store.Manager
|
|
cluster cluster.Cluster
|
|
}
|
|
|
|
//NewExporter new a exporter
|
|
func NewExporter(storeManager store.Manager, cluster cluster.Cluster) *Exporter {
|
|
return &Exporter{
|
|
totalScrapes: prometheus.NewCounter(prometheus.CounterOpts{
|
|
Namespace: namespace,
|
|
Subsystem: exporter,
|
|
Name: "scrapes_total",
|
|
Help: "Total number of times Entrance was scraped for metrics.",
|
|
}),
|
|
scrapeErrors: prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Namespace: namespace,
|
|
Subsystem: exporter,
|
|
Name: "scrape_errors_total",
|
|
Help: "Total number of times an error occurred scraping a Entrance.",
|
|
}, []string{"collector"}),
|
|
error: prometheus.NewGauge(prometheus.GaugeOpts{
|
|
Namespace: namespace,
|
|
Subsystem: exporter,
|
|
Name: "last_scrape_error",
|
|
Help: "Whether the last scrape of metrics from Entrance resulted in an error (1 for error, 0 for success).",
|
|
}),
|
|
storeManager: storeManager,
|
|
cluster: cluster,
|
|
}
|
|
}
|
|
|
|
//Describe implements prometheus.Collector.
|
|
func (e *Exporter) Describe(ch chan<- *prometheus.Desc) {
|
|
metricCh := make(chan prometheus.Metric)
|
|
doneCh := make(chan struct{})
|
|
|
|
go func() {
|
|
for m := range metricCh {
|
|
ch <- m.Desc()
|
|
}
|
|
close(doneCh)
|
|
}()
|
|
|
|
e.Collect(metricCh)
|
|
close(metricCh)
|
|
<-doneCh
|
|
}
|
|
|
|
// Collect implements prometheus.Collector.
|
|
func (e *Exporter) Collect(ch chan<- prometheus.Metric) {
|
|
e.scrape(ch)
|
|
ch <- e.totalScrapes
|
|
ch <- e.error
|
|
e.scrapeErrors.Collect(ch)
|
|
}
|
|
|
|
func (e *Exporter) scrape(ch chan<- prometheus.Metric) {
|
|
scrapeTime := time.Now()
|
|
e.totalScrapes.Inc()
|
|
if err := e.storeManager.Scrape(ch, namespace, exporter, e.cluster.GetInstanceHost()); err != nil {
|
|
logrus.Error("core manager scrape for prometheus error.", err.Error())
|
|
e.error.Set(1)
|
|
}
|
|
if err := e.cluster.Scrape(ch, namespace, exporter); err != nil {
|
|
logrus.Error("core manager scrape for prometheus error.", err.Error())
|
|
e.error.Set(1)
|
|
}
|
|
//step last: scrape time
|
|
scrapeDurationDesc := prometheus.NewDesc(
|
|
prometheus.BuildFQName(namespace, exporter, "collector_duration_seconds"),
|
|
"Collector time duration.",
|
|
nil, nil,
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(scrapeDurationDesc, prometheus.GaugeValue, time.Since(scrapeTime).Seconds())
|
|
}
|