windows_exporter/internal/perfdata/v1/collector.go

package v1

import (
	"fmt"
	"strings"

	"github.com/prometheus-community/windows_exporter/internal/perfdata/perftypes"
	"github.com/prometheus/client_golang/prometheus"
)

type Collector struct {
	object string
	query  string
}

type Counter struct {
	Name      string
	Desc      string
	Instances map[string]uint32
	Type      uint32
	Frequency float64
}

func NewCollector(object string, _ []string, _ []string) (*Collector, error) {
	collector := &Collector{
		object: object,
		query:  MapCounterToIndex(object),
	}

	if _, err := collector.Collect(); err != nil {
		return nil, fmt.Errorf("failed to collect initial data: %w", err)
	}

	return collector, nil
}

func (c *Collector) Describe() map[string]string {
	return map[string]string{}
}

func (c *Collector) Collect() (map[string]map[string]perftypes.CounterValues, error) {
	perfObjects, err := QueryPerformanceData(c.query)
	if err != nil {
		return nil, fmt.Errorf("QueryPerformanceData: %w", err)
	}

	if len(perfObjects) == 0 {
		return map[string]map[string]perftypes.CounterValues{}, nil
	}

	data := make(map[string]map[string]perftypes.CounterValues, len(perfObjects[0].Instances))

	for _, perfObject := range perfObjects {
		if perfObject.Name != c.object {
			continue
		}

		for _, perfInstance := range perfObject.Instances {
			instanceName := perfInstance.Name
			if strings.HasSuffix(instanceName, "_Total") {
				continue
			}

			if instanceName == "" || instanceName == "*" {
				instanceName = perftypes.EmptyInstance
			}

			if _, ok := data[instanceName]; !ok {
				data[instanceName] = make(map[string]perftypes.CounterValues, len(perfInstance.Counters))
			}

			for _, perfCounter := range perfInstance.Counters {
				if perfCounter.Def.IsBaseValue && !perfCounter.Def.IsNanosecondCounter {
					continue
				}

				if _, ok := data[instanceName][perfCounter.Def.Name]; !ok {
					data[instanceName][perfCounter.Def.Name] = perftypes.CounterValues{
						Type: prometheus.GaugeValue,
					}
				}

				var metricType prometheus.ValueType
				if val, ok := perftypes.SupportedCounterTypes[perfCounter.Def.CounterType]; ok {
					metricType = val
				} else {
					metricType = prometheus.GaugeValue
				}

				values := perftypes.CounterValues{
					Type: metricType,
				}

				switch perfCounter.Def.CounterType {
				case perftypes.PERF_ELAPSED_TIME:
					values.FirstValue = float64(perfCounter.Value-perftypes.WindowsEpoch) / float64(perfObject.Frequency)
					values.SecondValue = float64(perfCounter.SecondValue-perftypes.WindowsEpoch) / float64(perfObject.Frequency)
				case perftypes.PERF_100NSEC_TIMER, perftypes.PERF_PRECISION_100NS_TIMER:
					values.FirstValue = float64(perfCounter.Value) * perftypes.TicksToSecondScaleFactor
					values.SecondValue = float64(perfCounter.SecondValue) * perftypes.TicksToSecondScaleFactor
				default:
					values.FirstValue = float64(perfCounter.Value)
					values.SecondValue = float64(perfCounter.SecondValue)
				}

				data[instanceName][perfCounter.Def.Name] = values
			}
		}
	}

	return data, nil
}

func (c *Collector) Close() {
}