147 lines
3.7 KiB
Go
147 lines
3.7 KiB
Go
// Copyright 2016 The Prometheus Authors
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// 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.
|
|
|
|
// +build !nodevstat
|
|
|
|
package collector
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
/*
|
|
#cgo LDFLAGS: -ldevstat
|
|
#include <devstat.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
typedef struct {
|
|
char device[DEVSTAT_NAME_LEN];
|
|
int unit;
|
|
uint64_t bytes;
|
|
uint64_t transfers;
|
|
uint64_t blocks;
|
|
} Stats;
|
|
|
|
int _get_ndevs() {
|
|
struct statinfo current;
|
|
int num_devices;
|
|
|
|
current.dinfo = (struct devinfo *)calloc(1, sizeof(struct devinfo));
|
|
if (current.dinfo == NULL)
|
|
return -2;
|
|
|
|
checkversion();
|
|
|
|
if (getdevs(¤t) == -1)
|
|
return -1;
|
|
|
|
return current.dinfo->numdevs;
|
|
}
|
|
|
|
Stats _get_stats(int i) {
|
|
struct statinfo current;
|
|
int num_devices;
|
|
|
|
current.dinfo = (struct devinfo *)calloc(1, sizeof(struct devinfo));
|
|
getdevs(¤t);
|
|
|
|
num_devices = current.dinfo->numdevs;
|
|
Stats stats;
|
|
|
|
uint64_t total_bytes, total_transfers, total_blocks;
|
|
long double kb_per_transfer, transfers_per_second, mb_per_second, blocks_per_second, ms_per_transaction;
|
|
|
|
strcpy(stats.device, current.dinfo->devices[i].device_name);
|
|
stats.unit = current.dinfo->devices[i].unit_number;
|
|
compute_stats(¤t.dinfo->devices[i],
|
|
NULL,
|
|
1.0,
|
|
&total_bytes,
|
|
&total_transfers,
|
|
&total_blocks,
|
|
&kb_per_transfer,
|
|
&transfers_per_second,
|
|
&mb_per_second,
|
|
&blocks_per_second,
|
|
&ms_per_transaction);
|
|
|
|
stats.bytes = total_bytes;
|
|
stats.transfers = total_transfers;
|
|
stats.blocks = total_blocks;
|
|
|
|
return stats;
|
|
}
|
|
*/
|
|
import "C"
|
|
|
|
const (
|
|
devstatSubsystem = "devstat"
|
|
)
|
|
|
|
type devstatCollector struct {
|
|
bytesDesc *prometheus.Desc
|
|
transfersDesc *prometheus.Desc
|
|
blocksDesc *prometheus.Desc
|
|
}
|
|
|
|
func init() {
|
|
Factories["devstat"] = NewDevstatCollector
|
|
}
|
|
|
|
// Takes a prometheus registry and returns a new Collector exposing
|
|
// Device stats.
|
|
func NewDevstatCollector() (Collector, error) {
|
|
return &devstatCollector{
|
|
bytesDesc: prometheus.NewDesc(
|
|
prometheus.BuildFQName(Namespace, devstatSubsystem, "bytes_total"),
|
|
"The total number of bytes transferred for reads and writes on the device.",
|
|
[]string{"device"}, nil,
|
|
),
|
|
transfersDesc: prometheus.NewDesc(
|
|
prometheus.BuildFQName(Namespace, devstatSubsystem, "transfers_total"),
|
|
"The total number of transactions completed.",
|
|
[]string{"device"}, nil,
|
|
),
|
|
blocksDesc: prometheus.NewDesc(
|
|
prometheus.BuildFQName(Namespace, devstatSubsystem, "blocks_total"),
|
|
"The total number of bytes given in terms of the devices blocksize.",
|
|
[]string{"device"}, nil,
|
|
),
|
|
}, nil
|
|
}
|
|
|
|
func (c *devstatCollector) Update(ch chan<- prometheus.Metric) error {
|
|
count := C._get_ndevs()
|
|
if count == -1 {
|
|
return errors.New("getdevs() failed")
|
|
}
|
|
if count == -2 {
|
|
return errors.New("calloc() failed")
|
|
}
|
|
|
|
for i := C.int(0); i < count; i++ {
|
|
stats := C._get_stats(i)
|
|
device := fmt.Sprintf("%s%d", C.GoString(&stats.device[0]), stats.unit)
|
|
|
|
ch <- prometheus.MustNewConstMetric(c.bytesDesc, prometheus.CounterValue, float64(stats.bytes), device)
|
|
ch <- prometheus.MustNewConstMetric(c.transfersDesc, prometheus.CounterValue, float64(stats.transfers), device)
|
|
ch <- prometheus.MustNewConstMetric(c.blocksDesc, prometheus.CounterValue, float64(stats.blocks), device)
|
|
}
|
|
|
|
return err
|
|
}
|