Skip to content

Commit 7ed4523

Browse files
mmorel-35SuperQ
andauthored
Allow custom user-agent definition (#725)
* Allow custom user-agent definition --------- Signed-off-by: Matthieu MOREL <[email protected]> Co-authored-by: Ben Kochie <[email protected]>
1 parent e25286d commit 7ed4523

File tree

2 files changed

+34
-2
lines changed

2 files changed

+34
-2
lines changed

version/info.go

+6-2
Original file line numberDiff line numberDiff line change
@@ -90,8 +90,12 @@ func GetTags() string {
9090
return computedTags
9191
}
9292

93-
func UserAgent() string {
94-
return "Prometheus/" + Version
93+
func PrometheusUserAgent() string {
94+
return ComponentUserAgent("Prometheus")
95+
}
96+
97+
func ComponentUserAgent(component string) string {
98+
return component + "/" + Version
9599
}
96100

97101
func init() {

version/info_test.go

+28
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
// Copyright 2024 The Prometheus Authors
2+
// Licensed under the Apache License, Version 2.0 (the "License");
3+
// you may not use this file except in compliance with the License.
4+
// You may obtain a copy of the License at
5+
//
6+
// http://www.apache.org/licenses/LICENSE-2.0
7+
//
8+
// Unless required by applicable law or agreed to in writing, software
9+
// distributed under the License is distributed on an "AS IS" BASIS,
10+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11+
// See the License for the specific language governing permissions and
12+
// limitations under the License.
13+
14+
package version
15+
16+
import (
17+
"testing"
18+
19+
"github.com/stretchr/testify/require"
20+
)
21+
22+
func TestPrometheusUserAgent(t *testing.T) {
23+
require.Equal(t, "Prometheus/"+Version, PrometheusUserAgent())
24+
}
25+
26+
func TestComponentUserAgent(t *testing.T) {
27+
require.Equal(t, "Component/"+Version, ComponentUserAgent("Component"))
28+
}

0 commit comments

Comments
 (0)