fixup! fix: fallback to gauge for protofmt-based negotiations

Author: rexagod

pkg/customresourcestate/config.go

@@ -20,6 +20,8 @@ import (
 	"fmt"
 	"strings"
 
+	"k8s.io/kube-state-metrics/v2/pkg/metric"
+
 	"github.com/gobuffalo/flect"
 	"k8s.io/apimachinery/pkg/runtime/schema"
 	"k8s.io/klog/v2"
@@ -148,7 +150,7 @@ type Generator struct {
 type Metric struct {
 	// Type defines the type of the metric.
 	// +unionDiscriminator
-	Type MetricType `yaml:"type" json:"type"`
+	Type metric.Type `yaml:"type" json:"type"`
 
 	// Gauge defines a gauge metric.
 	// +optional
@@ -170,9 +172,16 @@ type ConfigDecoder interface {
 func FromConfig(decoder ConfigDecoder, discovererInstance *discovery.CRDiscoverer) (func() ([]customresource.RegistryFactory, error), error) {
 	var customResourceConfig Metrics
 	factoriesIndex := map[string]bool{}
+
+	// Decode the configuration.
 	if err := decoder.Decode(&customResourceConfig); err != nil {
 		return nil, fmt.Errorf("failed to parse Custom Resource State metrics: %w", err)
 	}
+
+	// Override the configuration with any custom overrides.
+	configOverrides(&customResourceConfig)
+
+	// Create a factory for each resource.
 	fn := func() (factories []customresource.RegistryFactory, err error) {
 		resources := customResourceConfig.Spec.Resources
 		// resolvedGVKPs will have the final list of GVKs, in addition to the resolved G** resources.
@@ -206,3 +215,14 @@ func FromConfig(decoder ConfigDecoder, discovererInstance *discovery.CRDiscovere
 	}
 	return fn, nil
 }
+
+// configOverrides applies overrides to the configuration.
+func configOverrides(config *Metrics) {
+	for i := range config.Spec.Resources {
+		for j := range config.Spec.Resources[i].Metrics {
+
+			// Override the metric type to lowercase, so the internals have a single source of truth for metric type definitions.
+			config.Spec.Resources[i].Metrics[j].Each.Type = metric.Type(strings.ToLower(string(config.Spec.Resources[i].Metrics[j].Each.Type)))
+		}
+	}
+}

pkg/customresourcestate/config_metrics_types.go

@@ -16,16 +16,6 @@ limitations under the License.
 
 package customresourcestate
 
-// MetricType is the type of a metric.
-type MetricType string
-
-// Supported metric types.
-const (
-	MetricTypeGauge    MetricType = "Gauge"
-	MetricTypeStateSet MetricType = "StateSet"
-	MetricTypeInfo     MetricType = "Info"
-)
-
 // MetricMeta are variables which may used for any metric type.
 type MetricMeta struct {
 	// LabelsFromPath adds additional labels where the value of the label is taken from a field under Path.

pkg/customresourcestate/custom_resource_metrics_test.go

@@ -21,6 +21,8 @@ import (
 	"reflect"
 	"testing"
 
+	"k8s.io/kube-state-metrics/v2/pkg/metric"
+
 	"k8s.io/apimachinery/pkg/runtime/schema"
 	"k8s.io/utils/ptr"
 )
@@ -55,7 +57,7 @@ func TestNewCustomResourceMetrics(t *testing.T) {
 						Name: "test_metrics",
 						Help: "metrics for testing",
 						Each: Metric{
-							Type: MetricTypeInfo,
+							Type: metric.Info,
 							Info: &MetricInfo{
 								MetricMeta: MetricMeta{
 									Path: []string{
@@ -117,7 +119,7 @@ func TestNewCustomResourceMetrics(t *testing.T) {
 						Name: "test_metrics",
 						Help: "metrics for testing",
 						Each: Metric{
-							Type: MetricTypeInfo,
+							Type: metric.Info,
 							Info: &MetricInfo{
 								MetricMeta: MetricMeta{
 									Path: []string{
@@ -180,7 +182,7 @@ func TestNewCustomResourceMetrics(t *testing.T) {
 						Name: "test_metrics",
 						Help: "metrics for testing",
 						Each: Metric{
-							Type: MetricTypeInfo,
+							Type: metric.Info,
 							Info: &MetricInfo{
 								MetricMeta: MetricMeta{
 									Path: []string{

pkg/customresourcestate/registry_factory.go

@@ -72,7 +72,7 @@ func compileCommon(c MetricMeta) (*compiledCommon, error) {
 func compileFamily(f Generator, resource Resource) (*compiledFamily, error) {
 	labels := resource.Labels.Merge(f.Labels)
 
-	if f.Each.Type == MetricTypeInfo && !strings.HasSuffix(f.Name, "_info") {
+	if f.Each.Type == metric.Info && !strings.HasSuffix(f.Name, "_info") {
 		klog.InfoS("Info metric does not have _info suffix", "gvk", resource.GroupVersionKind.String(), "name", f.Name)
 	}
 
@@ -153,7 +153,7 @@ type compiledMetric interface {
 // newCompiledMetric returns a compiledMetric depending on the given metric type.
 func newCompiledMetric(m Metric) (compiledMetric, error) {
 	switch m.Type {
-	case MetricTypeGauge:
+	case metric.Gauge:
 		if m.Gauge == nil {
 			return nil, errors.New("expected each.gauge to not be nil")
 		}
@@ -172,7 +172,7 @@ func newCompiledMetric(m Metric) (compiledMetric, error) {
 			NilIsZero:      m.Gauge.NilIsZero,
 			labelFromKey:   m.Gauge.LabelFromKey,
 		}, nil
-	case MetricTypeInfo:
+	case metric.Info:
 		if m.Info == nil {
 			return nil, errors.New("expected each.info to not be nil")
 		}
@@ -185,7 +185,7 @@ func newCompiledMetric(m Metric) (compiledMetric, error) {
 			compiledCommon: *cc,
 			labelFromKey:   m.Info.LabelFromKey,
 		}, nil
-	case MetricTypeStateSet:
+	case metric.StateSet:
 		if m.StateSet == nil {
 			return nil, errors.New("expected each.stateSet to not be nil")
 		}

pkg/metric/metric.go

@@ -37,21 +37,68 @@ var (
 	}
 )
 
-// Type represents the type of a metric e.g. a counter. See
-// https://github.com/OpenObservability/OpenMetrics/blob/main/specification/OpenMetrics.md#metric-types.
+// Type represents the type of the metric. See https://github.com/OpenObservability/OpenMetrics/blob/main/specification/OpenMetrics.md#metric-types.
 type Type string
 
-// Gauge defines a OpenMetrics gauge.
-var Gauge Type = "gauge"
+// TypeN represents the type of the metric as an integer.
+type TypeN int
 
-// Info defines an OpenMetrics info.
-var Info Type = "info"
+// String returns the string representation of the metric type.
+func (i TypeN) String() Type {
+	return MetricTypeNMap[i]
+}
+
+// NString returns the string representation of the metric type as an integer.
+func (i TypeN) NString() string {
+	return strconv.Itoa(int(i))
+}
+
+// Supported metric types.
+const (
+	// GaugeN defines a OpenMetrics gauge.
+	GaugeN TypeN = iota
+
+	// InfoN defines an OpenMetrics info.
+	InfoN
+
+	// StateSetN defines an OpenMetrics stateset.
+	StateSetN
+
+	// CounterN defines a OpenMetrics counter.
+	CounterN
+)
 
-// StateSet defines an OpenMetrics stateset.
-var StateSet Type = "stateset"
+// Supported metric types.
+var (
+
+	// Gauge defines a OpenMetrics gauge.
+	Gauge Type = "gauge"
+
+	// Info defines an OpenMetrics info.
+	Info Type = "info"
+
+	// StateSet defines an OpenMetrics stateset.
+	StateSet Type = "stateset"
+
+	// Counter defines a OpenMetrics counter.
+	Counter Type = "counter"
 
-// Counter defines a OpenMetrics counter.
-var Counter Type = "counter"
+	// MetricTypeNMap is a map of MetricTypeN to MetricType.
+	MetricTypeNMap = map[TypeN]Type{
+		GaugeN:    Gauge,
+		InfoN:     Info,
+		StateSetN: StateSet,
+		CounterN:  Counter,
+	}
+
+	// MetricTypeMap is a map of MetricType to MetricTypeN.
+	MetricTypeMap = map[Type]TypeN{
+		Gauge:    GaugeN,
+		Info:     InfoN,
+		StateSet: StateSetN,
+		Counter:  CounterN,
+	}
+)
 
 // Metric represents a single time series.
 type Metric struct {

pkg/metric_generator/generator.go

@@ -90,7 +90,7 @@ func (g *FamilyGenerator) generateHeader() string {
 	header.WriteString("# TYPE ")
 	header.WriteString(g.Name)
 	header.WriteByte(' ')
-	header.WriteString(string(g.Type))
+	header.WriteString(metric.MetricTypeMap[g.Type].NString())
 
 	return header.String()
 }

pkg/metrics_store/metrics_writer.go

@@ -93,20 +93,31 @@ func SanitizeHeaders(contentType string, writers MetricsWriterList) MetricsWrite
 	for _, writer := range writers {
 		if len(writer.stores) > 0 {
 			for i, header := range writer.stores[0].headers {
-				// If the requested content type was proto-based, replace the type with "gauge", as "info" and "statesets" are not recognized by Prometheus' protobuf machinery.
-				if strings.HasPrefix(contentType, expfmt.ProtoType) &&
-					strings.HasPrefix(header, "# HELP") &&
-					(strings.HasSuffix(header, " "+string(metric.Info)) || strings.HasSuffix(header, " "+string(metric.StateSet))) {
-					typeStringWithoutTypePaddedIndex := strings.LastIndex(header, " ")
-					typeStringWithoutType := header[:typeStringWithoutTypePaddedIndex]
-					writer.stores[0].headers[i] = typeStringWithoutType + " " + string(metric.Gauge)
+
+				// Nothing to sanitize.
+				if len(header) == 0 {
+					continue
 				}
+
 				// Removes duplicate headers from the given MetricsWriterList for the same family (generated through CRS).
 				// These are expected to be consecutive since G** resolution generates groups of similar metrics with same headers before moving onto the next G** spec in the CRS configuration.
 				if header == lastHeader {
 					writer.stores[0].headers[i] = ""
-				} else {
+				} else if strings.HasPrefix(header, "# HELP") {
 					lastHeader = header
+
+					// If the requested content type was proto-based, replace the type with "gauge", as "info" and "statesets" are not recognized by Prometheus' protobuf machinery,
+					// else replace them by their respective string representations.
+					if strings.HasPrefix(contentType, expfmt.ProtoType) &&
+						(strings.HasSuffix(header, metric.InfoN.NString()) || strings.HasSuffix(header, metric.StateSetN.NString())) {
+						writer.stores[0].headers[i] = header[:len(header)-1] + string(metric.Gauge)
+					}
+
+					// Replace all remaining type enums with their string representations.
+					n := int(header[len(header)-1]) - '0'
+					if n >= 0 && n < len(metric.MetricTypeNMap) {
+						writer.stores[0].headers[i] = header[:len(header)-1] + string(metric.MetricTypeNMap[metric.TypeN(n)])
+					}
 				}
 			}
 		}

pkg/metrics_store/metrics_writer_test.go

@@ -18,10 +18,11 @@ package metricsstore_test
 
 import (
 	"fmt"
-	"github.com/prometheus/common/expfmt"
 	"strings"
 	"testing"
 
+	"github.com/prometheus/common/expfmt"
+
 	v1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/meta"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -297,7 +298,7 @@ func BenchmarkSanitizeHeaders(b *testing.B) {
 		headers := []string{}
 		for j := 0; j < 10e4; j++ {
 			if benchmark.writersContainsDuplicates {
-				headers = append(headers, fmt.Sprintf("# HELP foo foo_help\n# TYPE foo info"))
+				headers = append(headers, "# HELP foo foo_help\n# TYPE foo info")
 			} else {
 				headers = append(headers, fmt.Sprintf("# HELP foo_%d foo_help\n# TYPE foo_%d info", j, j))
 			}