Skip to content
Snippets Groups Projects
Commit 7908dfbcac1f authored by Christophe de Vienne's avatar Christophe de Vienne
Browse files

Fix prometheus generated code when using operation groups

parent 5091da796767
No related branches found
No related tags found
No related merge requests found
Pipeline #21704 passed
......@@ -22,7 +22,11 @@
file_name: "doc.go"
# custom templates
- name: logging
source: server/logging.gotmpl
target: "{{ if gt (len .Tags) 0 }}{{ joinFilePath .Target .ServerPackage .APIPackage .Package }}{{ else }}{{ joinFilePath .Target .ServerPackage .Package }}{{ end }}"
file_name: "logging.go"
- name: init_prometheus
source: server/initPrometheus.gotmpl
target: "{{ joinFilePath .Target .ServerPackage .Package }}"
file_name: "prometheus.go"
......@@ -25,11 +29,7 @@
- name: init_prometheus
source: server/initPrometheus.gotmpl
target: "{{ joinFilePath .Target .ServerPackage .Package }}"
file_name: "prometheus.go"
- name: logging
source: server/logging.gotmpl
target: "{{ if gt (len .Tags) 0 }}{{ joinFilePath .Target .ServerPackage .APIPackage .Package }}{{ else }}{{ joinFilePath .Target .ServerPackage .Package }}{{ end }}"
file_name: "logging.go"
- name: main
source: server/main.gotmpl
......@@ -103,3 +103,7 @@
file_name: "{{ (snakize (pascalize .Name)) }}_prometheus.go"
operation_groups:
- name: operation_init_prometheus
source: server/operationInitPrometheus.gotmpl
target: "{{ joinFilePath .Target .ServerPackage .APIPackage (snakize (pascalize .Name)) }}"
file_name: "prometheus.go"
......@@ -24,10 +24,7 @@
"api_version": "{{ .Info.Version }}",
}, prometheus.DefaultRegisterer)
{{range .Operations}}
reg.MustRegister(
{{if ne .Package $package}}{{.Package}}.{{end}}{{ camelize .Name }}HandlerDuration,
{{if ne .Package $package}}{{.Package}}.{{end}}{{ camelize .Name }}HandlerInFlight,
)
{{end}}
{{range .OperationGroups}}
{{ .Name }}.RegisterPrometheus(reg)
{{- end}}
}
// Code generated by go-swagger with a alpe-api template; DO NOT EDIT.
package {{.Name}}
{{ $package := .Name }}
import (
"github.com/prometheus/client_golang/prometheus"
{{ range $key, $value := .Imports }}{{ $key }} {{ printf "%q" $value }}
{{ end }}
)
func RegisterPrometheus(reg prometheus.Registerer) {
reg = prometheus.WrapRegistererWith(prometheus.Labels{
"api_operation_group": "{{ .Name }}",
}, reg)
{{ range .Operations}}
reg.MustRegister(
{{ camelize .Name }}HandlerDuration,
{{ camelize .Name }}HandlerInFlight,
)
{{- end}}
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment