mirror of
https://github.com/prometheus-community/windows_exporter.git
synced 2026-02-24 05:36:36 +00:00
textfile: set windows_exporter_collector_success to 0, if an errors occurs (#1775)
Signed-off-by: Jan-Otto Kröpke <mail@jkroepke.de>
This commit is contained in:
@@ -304,8 +304,7 @@ func (c *Collector) collect(ch chan<- prometheus.Metric) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for nicName, nicData := range data {
|
for nicName, nicData := range data {
|
||||||
if c.config.NicExclude.MatchString(nicName) ||
|
if c.config.NicExclude.MatchString(nicName) || !c.config.NicInclude.MatchString(nicName) {
|
||||||
!c.config.NicInclude.MatchString(nicName) {
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -54,7 +54,7 @@ type Collector struct {
|
|||||||
// Only set for testing to get predictable output.
|
// Only set for testing to get predictable output.
|
||||||
mTime *float64
|
mTime *float64
|
||||||
|
|
||||||
mTimeDesc *prometheus.Desc
|
modTimeDesc *prometheus.Desc
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(config *Config) *Collector {
|
func New(config *Config) *Collector {
|
||||||
@@ -105,9 +105,9 @@ func (c *Collector) Close() error {
|
|||||||
func (c *Collector) Build(logger *slog.Logger, _ *mi.Session) error {
|
func (c *Collector) Build(logger *slog.Logger, _ *mi.Session) error {
|
||||||
c.logger = logger.With(slog.String("collector", Name))
|
c.logger = logger.With(slog.String("collector", Name))
|
||||||
|
|
||||||
c.logger.Info("textfile Collector directories: " + strings.Join(c.config.TextFileDirectories, ","))
|
c.logger.Info("textfile directories: " + strings.Join(c.config.TextFileDirectories, ","))
|
||||||
|
|
||||||
c.mTimeDesc = prometheus.NewDesc(
|
c.modTimeDesc = prometheus.NewDesc(
|
||||||
prometheus.BuildFQName(types.Namespace, "textfile", "mtime_seconds"),
|
prometheus.BuildFQName(types.Namespace, "textfile", "mtime_seconds"),
|
||||||
"Unixtime mtime of textfiles successfully read.",
|
"Unixtime mtime of textfiles successfully read.",
|
||||||
[]string{"file"},
|
[]string{"file"},
|
||||||
@@ -165,7 +165,7 @@ func (c *Collector) convertMetricFamily(logger *slog.Logger, metricFamily *dto.M
|
|||||||
|
|
||||||
for _, metric := range metricFamily.GetMetric() {
|
for _, metric := range metricFamily.GetMetric() {
|
||||||
if metric.TimestampMs != nil {
|
if metric.TimestampMs != nil {
|
||||||
logger.Warn(fmt.Sprintf("Ignoring unsupported custom timestamp on textfile Collector metric %v", metric))
|
logger.Warn(fmt.Sprintf("Ignoring unsupported custom timestamp on textfile metric %v", metric))
|
||||||
}
|
}
|
||||||
|
|
||||||
labels := metric.GetLabel()
|
labels := metric.GetLabel()
|
||||||
@@ -259,23 +259,24 @@ func (c *Collector) convertMetricFamily(logger *slog.Logger, metricFamily *dto.M
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Collector) exportMTimes(mTimes map[string]time.Time, ch chan<- prometheus.Metric) {
|
func (c *Collector) exportMTimes(modTimes map[string]time.Time, ch chan<- prometheus.Metric) {
|
||||||
// Export the mtimes of the successful files.
|
// Export the mtimes of the successful files.
|
||||||
if len(mTimes) > 0 {
|
if len(modTimes) > 0 {
|
||||||
// Sorting is needed for predictable output comparison in tests.
|
// Sorting is needed for predictable output comparison in tests.
|
||||||
filenames := make([]string, 0, len(mTimes))
|
filenames := make([]string, 0, len(modTimes))
|
||||||
for filename := range mTimes {
|
for filename := range modTimes {
|
||||||
filenames = append(filenames, filename)
|
filenames = append(filenames, filename)
|
||||||
}
|
}
|
||||||
|
|
||||||
sort.Strings(filenames)
|
sort.Strings(filenames)
|
||||||
|
|
||||||
for _, filename := range filenames {
|
for _, filename := range filenames {
|
||||||
mtime := float64(mTimes[filename].UnixNano() / 1e9)
|
modTime := float64(modTimes[filename].UnixNano() / 1e9)
|
||||||
if c.mTime != nil {
|
if c.mTime != nil {
|
||||||
mtime = *c.mTime
|
modTime = *c.mTime
|
||||||
}
|
}
|
||||||
ch <- prometheus.MustNewConstMetric(c.mTimeDesc, prometheus.GaugeValue, mtime, filename)
|
|
||||||
|
ch <- prometheus.MustNewConstMetric(c.modTimeDesc, prometheus.GaugeValue, modTime, filename)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -314,6 +315,8 @@ func (c *Collector) Collect(ch chan<- prometheus.Metric) error {
|
|||||||
// This will ensure that duplicate metrics are correctly detected between multiple .prom files.
|
// This will ensure that duplicate metrics are correctly detected between multiple .prom files.
|
||||||
var metricFamilies []*dto.MetricFamily
|
var metricFamilies []*dto.MetricFamily
|
||||||
|
|
||||||
|
errs := make([]error, 0)
|
||||||
|
|
||||||
// Iterate over files and accumulate their metrics.
|
// Iterate over files and accumulate their metrics.
|
||||||
for _, directory := range c.config.TextFileDirectories {
|
for _, directory := range c.config.TextFileDirectories {
|
||||||
err := filepath.WalkDir(directory, func(path string, dirEntry os.DirEntry, err error) error {
|
err := filepath.WalkDir(directory, func(path string, dirEntry os.DirEntry, err error) error {
|
||||||
@@ -326,24 +329,20 @@ func (c *Collector) Collect(ch chan<- prometheus.Metric) error {
|
|||||||
|
|
||||||
families_array, err := scrapeFile(path, c.logger)
|
families_array, err := scrapeFile(path, c.logger)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.logger.Error(fmt.Sprintf("Error scraping file: %q. Skip File.", path),
|
errs = append(errs, fmt.Errorf("error scraping file %q: %w", path, err))
|
||||||
slog.Any("err", err),
|
|
||||||
)
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
fileInfo, err := os.Stat(path)
|
fileInfo, err := os.Stat(path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.logger.Error(fmt.Sprintf("Error reading file info: %q. Skip File.", path),
|
errs = append(errs, fmt.Errorf("error reading file info %q: %w", path, err))
|
||||||
slog.Any("err", err),
|
|
||||||
)
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, hasName := mTimes[fileInfo.Name()]; hasName {
|
if _, hasName := mTimes[fileInfo.Name()]; hasName {
|
||||||
c.logger.Error(fmt.Sprintf("Duplicate filename detected: %q. Skip File.", path))
|
errs = append(errs, fmt.Errorf("duplicate filename detected: %q", path))
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -355,25 +354,24 @@ func (c *Collector) Collect(ch chan<- prometheus.Metric) error {
|
|||||||
|
|
||||||
return nil
|
return nil
|
||||||
})
|
})
|
||||||
|
|
||||||
if err != nil && directory != "" {
|
if err != nil && directory != "" {
|
||||||
c.logger.Error("Error reading textfile Collector directory: "+directory,
|
errs = append(errs, fmt.Errorf("error reading textfile directory %q: %w", directory, err))
|
||||||
slog.Any("err", err),
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
c.exportMTimes(mTimes, ch)
|
||||||
|
|
||||||
// If duplicates are detected across *multiple* files, return error.
|
// If duplicates are detected across *multiple* files, return error.
|
||||||
if duplicateMetricEntry(metricFamilies) {
|
if duplicateMetricEntry(metricFamilies) {
|
||||||
c.logger.Error("Duplicate metrics detected across multiple files")
|
c.logger.Warn("duplicate metrics detected across multiple files")
|
||||||
} else {
|
} else {
|
||||||
for _, mf := range metricFamilies {
|
for _, mf := range metricFamilies {
|
||||||
c.convertMetricFamily(c.logger, mf, ch)
|
c.convertMetricFamily(c.logger, mf, ch)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
c.exportMTimes(mTimes, ch)
|
return errors.Join(errs...)
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func scrapeFile(path string, logger *slog.Logger) ([]*dto.MetricFamily, error) {
|
func scrapeFile(path string, logger *slog.Logger) ([]*dto.MetricFamily, error) {
|
||||||
|
|||||||
@@ -26,6 +26,7 @@ import (
|
|||||||
"github.com/prometheus-community/windows_exporter/pkg/collector"
|
"github.com/prometheus-community/windows_exporter/pkg/collector"
|
||||||
"github.com/prometheus/client_golang/prometheus"
|
"github.com/prometheus/client_golang/prometheus"
|
||||||
dto "github.com/prometheus/client_model/go"
|
dto "github.com/prometheus/client_model/go"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -48,30 +49,26 @@ func TestMultipleDirectories(t *testing.T) {
|
|||||||
metrics := make(chan prometheus.Metric)
|
metrics := make(chan prometheus.Metric)
|
||||||
got := ""
|
got := ""
|
||||||
|
|
||||||
|
errCh := make(chan error, 1)
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
errCh <- textFileCollector.Collect(metrics)
|
||||||
var metric dto.Metric
|
|
||||||
|
|
||||||
val := <-metrics
|
close(metrics)
|
||||||
|
|
||||||
err := val.Write(&metric)
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("Unexpected error %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
got += metric.String()
|
|
||||||
}
|
|
||||||
}()
|
}()
|
||||||
|
|
||||||
err := textFileCollector.Collect(metrics)
|
for val := range metrics {
|
||||||
if err != nil {
|
var metric dto.Metric
|
||||||
t.Errorf("Unexpected error %s", err)
|
|
||||||
|
err := val.Write(&metric)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
got += metric.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
require.NoError(t, <-errCh)
|
||||||
|
|
||||||
for _, f := range []string{"dir1", "dir2", "dir3", "dir3sub"} {
|
for _, f := range []string{"dir1", "dir2", "dir3", "dir3sub"} {
|
||||||
if !strings.Contains(got, f) {
|
assert.Contains(t, got, f)
|
||||||
t.Errorf("Unexpected output %s: %q", f, got)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -89,31 +86,24 @@ func TestDuplicateFileName(t *testing.T) {
|
|||||||
metrics := make(chan prometheus.Metric)
|
metrics := make(chan prometheus.Metric)
|
||||||
got := ""
|
got := ""
|
||||||
|
|
||||||
|
errCh := make(chan error, 1)
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
errCh <- textFileCollector.Collect(metrics)
|
||||||
var metric dto.Metric
|
|
||||||
|
|
||||||
val := <-metrics
|
close(metrics)
|
||||||
|
|
||||||
err := val.Write(&metric)
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("Unexpected error %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
got += metric.String()
|
|
||||||
}
|
|
||||||
}()
|
}()
|
||||||
|
|
||||||
err := textFileCollector.Collect(metrics)
|
for val := range metrics {
|
||||||
if err != nil {
|
var metric dto.Metric
|
||||||
t.Errorf("Unexpected error %s", err)
|
|
||||||
|
err := val.Write(&metric)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
got += metric.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
if !strings.Contains(got, "file") {
|
require.ErrorContains(t, <-errCh, "duplicate filename detected")
|
||||||
t.Errorf("Unexpected output %q", got)
|
|
||||||
}
|
|
||||||
|
|
||||||
if strings.Contains(got, "sub_file") {
|
assert.Contains(t, got, "file")
|
||||||
t.Errorf("Unexpected output %q", got)
|
assert.NotContains(t, got, "sub_file")
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user