commit
503ca0fd72
@ -0,0 +1,3 @@
|
||||
MIRTH_ENDPOINT=https://mirth-connect.yourcompane.com
|
||||
MIRTH_USERNAME=admin
|
||||
MIRTH_PASSWORD=admin
|
||||
@ -0,0 +1,9 @@
|
||||
/.idea
|
||||
/.idea/
|
||||
/.idea/*
|
||||
.idea
|
||||
.idea/
|
||||
.idea/*
|
||||
go.sum
|
||||
.env
|
||||
mirth_channel_exporter
|
||||
@ -0,0 +1,7 @@
|
||||
linux:
|
||||
CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build
|
||||
mac:
|
||||
CGO_ENABLED=0 GOOS=darwin GOARCH=amd64 go build
|
||||
arm:
|
||||
CGO_ENABLED=0 GOOS=linux GOARCH=arm64 go mod tidy
|
||||
CGO_ENABLED=0 GOOS=linux GOARCH=arm64 go build -o exporter main.go
|
||||
@ -0,0 +1,114 @@
|
||||
package collector
|
||||
|
||||
//Exporter.go中主要是通过New 以及NewMetrics 初始化exporter结构体以及Metrics 结构体 exporter结构体需要满足Describe 和Collect 方法
|
||||
import (
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"sync"
|
||||
)
|
||||
|
||||
var (
|
||||
name string = "test_exporter"
|
||||
namespace string = "test"
|
||||
exporter string = "exporter"
|
||||
)
|
||||
|
||||
func Name() string {
|
||||
return name
|
||||
}
|
||||
|
||||
type Exporter struct {
|
||||
metrics Metrics
|
||||
scrapers []Scraper
|
||||
}
|
||||
|
||||
type Metrics struct {
|
||||
//收集指标的总次数
|
||||
TotalScrapes prometheus.Counter
|
||||
//收集指标中发生错误的次数
|
||||
ScrapeErrors *prometheus.CounterVec
|
||||
//最后一次是否发生了错误
|
||||
Error prometheus.Gauge
|
||||
}
|
||||
|
||||
// fqname会把namespace subsystem name拼接起来
|
||||
// 传入动态以及静态标签 设置标签
|
||||
func NewDesc(subsystem, name, help string, movinglabel []string, label prometheus.Labels) *prometheus.Desc {
|
||||
return prometheus.NewDesc(
|
||||
prometheus.BuildFQName(namespace, subsystem, name),
|
||||
help, movinglabel, label,
|
||||
)
|
||||
}
|
||||
|
||||
// 判断*exporter是否实现了collector这个接口的所有方法
|
||||
var _ prometheus.Collector = (*Exporter)(nil)
|
||||
|
||||
func New(metrics Metrics, scrapers []Scraper) *Exporter {
|
||||
return &Exporter{
|
||||
metrics: metrics,
|
||||
scrapers: scrapers,
|
||||
}
|
||||
}
|
||||
|
||||
func NewMetrics() Metrics {
|
||||
subsystem := exporter
|
||||
return Metrics{
|
||||
TotalScrapes: prometheus.NewCounter(prometheus.CounterOpts{
|
||||
Namespace: namespace,
|
||||
Subsystem: subsystem,
|
||||
Name: "scrapes_total",
|
||||
Help: "Total number of times was scraped for metrics.",
|
||||
}),
|
||||
ScrapeErrors: prometheus.NewCounterVec(prometheus.CounterOpts{
|
||||
Namespace: namespace,
|
||||
Subsystem: subsystem,
|
||||
Name: "scrape_errors_total",
|
||||
Help: "Total number of times an error occurred scraping .",
|
||||
}, []string{"collector"}),
|
||||
Error: prometheus.NewGauge(prometheus.GaugeOpts{
|
||||
Namespace: namespace,
|
||||
Subsystem: subsystem,
|
||||
Name: "last_scrape_error",
|
||||
Help: "Whether the last scrape of metrics resulted in an error (1 for error, 0 for success).",
|
||||
}),
|
||||
}
|
||||
}
|
||||
|
||||
// 添加描述符
|
||||
func (e *Exporter) Describe(ch chan<- *prometheus.Desc) {
|
||||
ch <- e.metrics.TotalScrapes.Desc()
|
||||
ch <- e.metrics.Error.Desc()
|
||||
e.metrics.ScrapeErrors.Describe(ch)
|
||||
}
|
||||
|
||||
// 收集指标
|
||||
func (e *Exporter) Collect(ch chan<- prometheus.Metric) {
|
||||
e.scrape(ch)
|
||||
ch <- e.metrics.TotalScrapes
|
||||
ch <- e.metrics.Error
|
||||
e.metrics.ScrapeErrors.Collect(ch)
|
||||
}
|
||||
|
||||
// 通过例程并发的收集指标 需要加waitgroup
|
||||
func (e *Exporter) scrape(ch chan<- prometheus.Metric) {
|
||||
var (
|
||||
wg sync.WaitGroup
|
||||
err error
|
||||
)
|
||||
|
||||
defer wg.Wait()
|
||||
for _, scraper := range e.scrapers {
|
||||
wg.Add(1)
|
||||
//使用匿名函数 并且并发的收集指标
|
||||
go func(scraper Scraper) {
|
||||
defer wg.Done()
|
||||
label := scraper.Name()
|
||||
err = scraper.Scrape(ch)
|
||||
if err != nil {
|
||||
log.WithField("scraper", scraper.Name()).Error(err)
|
||||
e.metrics.ScrapeErrors.WithLabelValues(label).Inc()
|
||||
e.metrics.Error.Set(1)
|
||||
}
|
||||
}(scraper)
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,8 @@
|
||||
package collector
|
||||
|
||||
// Scrapers 这里就做了开关控制我们在代码中实现就是通过修改这个bool值来做 每当我们添加一个指标 需要在map中添加一个指标的name 以及bool 添加指标后续会介绍
|
||||
var (
|
||||
Scrapers = map[Scraper]bool{
|
||||
CpuLoad{}: true,
|
||||
}
|
||||
)
|
||||
@ -0,0 +1,24 @@
|
||||
module sreExporter
|
||||
|
||||
go 1.19
|
||||
|
||||
require (
|
||||
github.com/prometheus/client_golang v1.12.2
|
||||
github.com/sirupsen/logrus v1.6.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/beorn7/perks v1.0.1 // indirect
|
||||
github.com/cespare/xxhash/v2 v2.1.2 // indirect
|
||||
github.com/go-ole/go-ole v1.2.6 // indirect
|
||||
github.com/golang/protobuf v1.5.2 // indirect
|
||||
github.com/konsorten/go-windows-terminal-sequences v1.0.3 // indirect
|
||||
github.com/matttproud/golang_protobuf_extensions v1.0.1 // indirect
|
||||
github.com/prometheus/client_model v0.2.0 // indirect
|
||||
github.com/prometheus/common v0.32.1 // indirect
|
||||
github.com/prometheus/procfs v0.7.3 // indirect
|
||||
github.com/shirou/gopsutil v3.21.11+incompatible // indirect
|
||||
github.com/yusufpapurcu/wmi v1.2.2 // indirect
|
||||
golang.org/x/sys v0.0.0-20220114195835-da31bd327af9 // indirect
|
||||
google.golang.org/protobuf v1.26.0 // indirect
|
||||
)
|
||||
@ -0,0 +1,77 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
"github.com/prometheus/client_golang/prometheus/promhttp"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"net/http"
|
||||
"sreExporter/collector"
|
||||
"strings"
|
||||
)
|
||||
|
||||
var (
|
||||
metricsPath string = "https://github.com/strive-after/demo-exporter"
|
||||
version string = "v1.0"
|
||||
listenAddress string
|
||||
help bool
|
||||
disable string //命令行传入的需要关闭的指标
|
||||
disables []string //处理命令行传入的根据,分割为一个切片做处理
|
||||
)
|
||||
|
||||
func init() {
|
||||
flag.StringVar(&listenAddress, "addr", ":9101", "addr")
|
||||
flag.BoolVar(&help, "h", false, "help")
|
||||
flag.StringVar(&disable, "disable", "", "关闭的指标收集器")
|
||||
}
|
||||
|
||||
// main函数作为总入口 提供web url /metrices以及访问/的时候提供一些基本介绍
|
||||
func main() {
|
||||
flag.Parse()
|
||||
if help {
|
||||
flag.Usage()
|
||||
return
|
||||
}
|
||||
disables = strings.Split(disable, ",")
|
||||
//手动开关
|
||||
//通过用户输入的我们做关闭
|
||||
for scraper, _ := range collector.Scrapers {
|
||||
for _, v := range disables {
|
||||
if v == scraper.Name() {
|
||||
collector.Scrapers[scraper] = false
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//访问/的时候返回一些基础提示
|
||||
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Write([]byte(`<html>
|
||||
<head><title>` + collector.Name() + `</title></head>
|
||||
<body>
|
||||
<h1><a style="text-decoration:none" href=''>` + collector.Name() + `</a></h1>
|
||||
<p><a href='` + metricsPath + `'>Metrics</a></p>
|
||||
<h2>Build</h2>
|
||||
<pre>` + version + `</pre>
|
||||
</body>
|
||||
</html>`))
|
||||
})
|
||||
//根据开关来判断指标的是否需要收集 这里只有代码里面的判断 用户手动开关还未做
|
||||
enabledScrapers := []collector.Scraper{}
|
||||
for scraper, enabled := range collector.Scrapers {
|
||||
if enabled {
|
||||
log.Info("Scraper enabled ", scraper.Name())
|
||||
enabledScrapers = append(enabledScrapers, scraper)
|
||||
}
|
||||
}
|
||||
|
||||
//注册自身采集器
|
||||
exporter := collector.New(collector.NewMetrics(), enabledScrapers)
|
||||
prometheus.MustRegister(exporter)
|
||||
|
||||
http.Handle("/metrics", promhttp.Handler())
|
||||
//监听端口
|
||||
if err := http.ListenAndServe(listenAddress, nil); err != nil {
|
||||
log.Printf("Error occur when start server %v", err)
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,281 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"crypto/tls"
|
||||
"encoding/xml"
|
||||
"flag"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"strconv"
|
||||
|
||||
"github.com/joho/godotenv"
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
"github.com/prometheus/client_golang/prometheus/promhttp"
|
||||
)
|
||||
|
||||
/*
|
||||
<map>
|
||||
|
||||
<entry>
|
||||
<string>101af57f-f26c-40d3-86a3-309e74b93512</string>
|
||||
<string>Send-Email-Notification</string>
|
||||
</entry>
|
||||
|
||||
</map>
|
||||
*/
|
||||
type ChannelIdNameMap struct {
|
||||
XMLName xml.Name `xml:"map"`
|
||||
Entries []ChannelEntry `xml:"entry"`
|
||||
}
|
||||
type ChannelEntry struct {
|
||||
XMLName xml.Name `xml:"entry"`
|
||||
Values []string `xml:"string"`
|
||||
}
|
||||
|
||||
/*
|
||||
<list>
|
||||
|
||||
<channelStatistics>
|
||||
<serverId>c5e6a736-0e88-46a7-bf32-5b4908c4d859</serverId>
|
||||
<channelId>101af57f-f26c-40d3-86a3-309e74b93512</channelId>
|
||||
<received>0</received>
|
||||
<sent>0</sent>
|
||||
<error>0</error>
|
||||
<filtered>0</filtered>
|
||||
<queued>0</queued>
|
||||
</channelStatistics>
|
||||
|
||||
</list>
|
||||
*/
|
||||
type ChannelStatsList struct {
|
||||
XMLName xml.Name `xml:"list"`
|
||||
Channels []ChannelStats `xml:"channelStatistics"`
|
||||
}
|
||||
type ChannelStats struct {
|
||||
XMLName xml.Name `xml:"channelStatistics"`
|
||||
ServerId string `xml:"serverId"`
|
||||
ChannelId string `xml:"channelId"`
|
||||
Received string `xml:"received"`
|
||||
Sent string `xml:"sent"`
|
||||
Error string `xml:"error"`
|
||||
Filtered string `xml:"filtered"`
|
||||
Queued string `xml:"queued"`
|
||||
}
|
||||
|
||||
const namespace = "mirth"
|
||||
const channelIdNameApi = "/api/channels/idsAndNames"
|
||||
const channelStatsApi = "/api/channels/statistics"
|
||||
|
||||
var (
|
||||
tr = &http.Transport{
|
||||
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
||||
}
|
||||
client = &http.Client{Transport: tr}
|
||||
|
||||
listenAddress = flag.String("web.listen-address", ":9101",
|
||||
"Address to listen on for telemetry")
|
||||
metricsPath = flag.String("web.telemetry-path", "/metrics",
|
||||
"Path under which to expose metrics")
|
||||
|
||||
// Metrics
|
||||
up = prometheus.NewDesc(
|
||||
prometheus.BuildFQName(namespace, "", "up"),
|
||||
"Was the last Mirth query successful.",
|
||||
nil, nil,
|
||||
)
|
||||
messagesReceived = prometheus.NewDesc(
|
||||
prometheus.BuildFQName(namespace, "", "messages_received_total"),
|
||||
"How many messages have been received (per channel).",
|
||||
[]string{"channel"}, nil,
|
||||
)
|
||||
messagesFiltered = prometheus.NewDesc(
|
||||
prometheus.BuildFQName(namespace, "", "messages_filtered_total"),
|
||||
"How many messages have been filtered (per channel).",
|
||||
[]string{"channel"}, nil,
|
||||
)
|
||||
messagesQueued = prometheus.NewDesc(
|
||||
prometheus.BuildFQName(namespace, "", "messages_queued"),
|
||||
"How many messages are currently queued (per channel).",
|
||||
[]string{"channel"}, nil,
|
||||
)
|
||||
messagesSent = prometheus.NewDesc(
|
||||
prometheus.BuildFQName(namespace, "", "messages_sent_total"),
|
||||
"How many messages have been sent (per channel).",
|
||||
[]string{"channel"}, nil,
|
||||
)
|
||||
messagesErrored = prometheus.NewDesc(
|
||||
prometheus.BuildFQName(namespace, "", "messages_errored_total"),
|
||||
"How many messages have errored (per channel).",
|
||||
[]string{"channel"}, nil,
|
||||
)
|
||||
)
|
||||
|
||||
type Exporter struct {
|
||||
mirthEndpoint, mirthUsername, mirthPassword string
|
||||
}
|
||||
|
||||
func NewExporter(mirthEndpoint string, mirthUsername string, mirthPassword string) *Exporter {
|
||||
return &Exporter{
|
||||
mirthEndpoint: mirthEndpoint,
|
||||
mirthUsername: mirthUsername,
|
||||
mirthPassword: mirthPassword,
|
||||
}
|
||||
}
|
||||
|
||||
func (e *Exporter) Describe(ch chan<- *prometheus.Desc) {
|
||||
ch <- up
|
||||
ch <- messagesReceived
|
||||
ch <- messagesFiltered
|
||||
ch <- messagesQueued
|
||||
ch <- messagesSent
|
||||
ch <- messagesErrored
|
||||
}
|
||||
|
||||
func (e *Exporter) Collect(ch chan<- prometheus.Metric) {
|
||||
channelIdNameMap, err := e.LoadChannelIdNameMap()
|
||||
if err != nil {
|
||||
ch <- prometheus.MustNewConstMetric(
|
||||
up, prometheus.GaugeValue, 0,
|
||||
)
|
||||
log.Println(err)
|
||||
return
|
||||
}
|
||||
ch <- prometheus.MustNewConstMetric(
|
||||
up, prometheus.GaugeValue, 1,
|
||||
)
|
||||
|
||||
e.HitMirthRestApisAndUpdateMetrics(channelIdNameMap, ch)
|
||||
}
|
||||
|
||||
func (e *Exporter) LoadChannelIdNameMap() (map[string]string, error) {
|
||||
// Create the map of channel id to names
|
||||
channelIdNameMap := make(map[string]string)
|
||||
|
||||
req, err := http.NewRequest("GET", e.mirthEndpoint+channelIdNameApi, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// This one line implements the authentication required for the task.
|
||||
req.SetBasicAuth(e.mirthUsername, e.mirthPassword)
|
||||
// Make request and show output.
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// fmt.Println(string(body))
|
||||
|
||||
// we initialize our array
|
||||
var channelIdNameMapXML ChannelIdNameMap
|
||||
// we unmarshal our byteArray which contains our
|
||||
// xmlFiles content into 'users' which we defined above
|
||||
err = xml.Unmarshal(body, &channelIdNameMapXML)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for i := 0; i < len(channelIdNameMapXML.Entries); i++ {
|
||||
channelIdNameMap[channelIdNameMapXML.Entries[i].Values[0]] = channelIdNameMapXML.Entries[i].Values[1]
|
||||
}
|
||||
|
||||
return channelIdNameMap, nil
|
||||
}
|
||||
|
||||
func (e *Exporter) HitMirthRestApisAndUpdateMetrics(channelIdNameMap map[string]string, ch chan<- prometheus.Metric) {
|
||||
// Load channel stats
|
||||
req, err := http.NewRequest("GET", e.mirthEndpoint+channelStatsApi, nil)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
// This one line implements the authentication required for the task.
|
||||
req.SetBasicAuth(e.mirthUsername, e.mirthPassword)
|
||||
// Make request and show output.
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
// fmt.Println(string(body))
|
||||
|
||||
// we initialize our array
|
||||
var channelStatsList ChannelStatsList
|
||||
// we unmarshal our byteArray which contains our
|
||||
// xmlFiles content into 'users' which we defined above
|
||||
err = xml.Unmarshal(body, &channelStatsList)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
for i := 0; i < len(channelStatsList.Channels); i++ {
|
||||
channelName := channelIdNameMap[channelStatsList.Channels[i].ChannelId]
|
||||
|
||||
channelReceived, _ := strconv.ParseFloat(channelStatsList.Channels[i].Received, 64)
|
||||
ch <- prometheus.MustNewConstMetric(
|
||||
messagesReceived, prometheus.GaugeValue, channelReceived, channelName,
|
||||
)
|
||||
|
||||
channelSent, _ := strconv.ParseFloat(channelStatsList.Channels[i].Sent, 64)
|
||||
ch <- prometheus.MustNewConstMetric(
|
||||
messagesSent, prometheus.GaugeValue, channelSent, channelName,
|
||||
)
|
||||
|
||||
channelError, _ := strconv.ParseFloat(channelStatsList.Channels[i].Error, 64)
|
||||
ch <- prometheus.MustNewConstMetric(
|
||||
messagesErrored, prometheus.GaugeValue, channelError, channelName,
|
||||
)
|
||||
|
||||
channelFiltered, _ := strconv.ParseFloat(channelStatsList.Channels[i].Filtered, 64)
|
||||
ch <- prometheus.MustNewConstMetric(
|
||||
messagesFiltered, prometheus.GaugeValue, channelFiltered, channelName,
|
||||
)
|
||||
|
||||
channelQueued, _ := strconv.ParseFloat(channelStatsList.Channels[i].Queued, 64)
|
||||
ch <- prometheus.MustNewConstMetric(
|
||||
messagesQueued, prometheus.GaugeValue, channelQueued, channelName,
|
||||
)
|
||||
}
|
||||
|
||||
log.Println("Endpoint scraped")
|
||||
}
|
||||
|
||||
func main() {
|
||||
err := godotenv.Load()
|
||||
if err != nil {
|
||||
log.Println("Error loading .env file, assume env variables are set.")
|
||||
}
|
||||
|
||||
flag.Parse()
|
||||
|
||||
mirthEndpoint := os.Getenv("MIRTH_ENDPOINT")
|
||||
mirthUsername := os.Getenv("MIRTH_USERNAME")
|
||||
mirthPassword := os.Getenv("MIRTH_PASSWORD")
|
||||
|
||||
exporter := NewExporter(mirthEndpoint, mirthUsername, mirthPassword)
|
||||
prometheus.MustRegister(exporter)
|
||||
|
||||
http.Handle(*metricsPath, promhttp.Handler())
|
||||
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Write([]byte(`<html>
|
||||
<head><title>Mirth Channel Exporter</title></head>
|
||||
<body>
|
||||
<h1>Mirth Channel Exporter</h1>
|
||||
<p><a href='` + *metricsPath + `'>Metrics</a></p>
|
||||
</body>
|
||||
</html>`))
|
||||
})
|
||||
log.Fatal(http.ListenAndServe(*listenAddress, nil))
|
||||
}
|
||||
@ -0,0 +1,14 @@
|
||||
[Unit]
|
||||
Description=mirth channel exporter
|
||||
After=network.target
|
||||
StartLimitIntervalSec=0
|
||||
[Service]
|
||||
Type=simple
|
||||
Restart=always
|
||||
RestartSec=1
|
||||
WorkingDirectory=/mirth/mirthconnect
|
||||
EnvironmentFile=/etc/sysconfig/mirth_channel_exporter
|
||||
ExecStart=/mirth/mirthconnect/mirth_channel_exporter
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
Loading…
Reference in new issue