unistack-org/metrics
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

feat: Support configure request method

Browse Source
This commit is contained in:
aierui 2024-02-19 22:19:06 +08:00
parent da211e52b9
commit 9032bb9e48
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
push.go
View File

@ -32,6 +32,10 @@ type PushOptions struct {
// By default the compression is enabled. // By default the compression is enabled.
DisableCompression bool DisableCompression bool
// Method is an optional of HTTP request method.
// By default the Method is GET.
Method string
// Optional WaitGroup for waiting until all the push workers created with this WaitGroup are stopped. // Optional WaitGroup for waiting until all the push workers created with this WaitGroup are stopped.
WaitGroup *sync.WaitGroup WaitGroup *sync.WaitGroup
} }
@ -264,6 +268,7 @@ func PushMetricsExt(ctx context.Context, pushURL string, writeMetrics func(w io.
type pushContext struct { type pushContext struct {
pushURL *url.URL pushURL *url.URL
method string
pushURLRedacted string pushURLRedacted string
extraLabels string extraLabels string
headers http.Header headers http.Header
@ -295,6 +300,11 @@ func newPushContext(pushURL string, opts *PushOptions) (*pushContext, error) {
return nil, fmt.Errorf("missing host in pushURL=%q", pushURL) return nil, fmt.Errorf("missing host in pushURL=%q", pushURL)
} }
method := opts.Method
if len(method) == 0 {
method = http.MethodGet
}
// validate ExtraLabels // validate ExtraLabels
extraLabels := opts.ExtraLabels extraLabels := opts.ExtraLabels
if err := validateTags(extraLabels); err != nil { if err := validateTags(extraLabels); err != nil {
@ -317,6 +327,7 @@ func newPushContext(pushURL string, opts *PushOptions) (*pushContext, error) {
client := &http.Client{} client := &http.Client{}
return &pushContext{ return &pushContext{
pushURL: pu, pushURL: pu,
method: method,
pushURLRedacted: pushURLRedacted, pushURLRedacted: pushURLRedacted,
extraLabels: extraLabels, extraLabels: extraLabels,
headers: headers, headers: headers,
@ -367,18 +378,18 @@ func (pc *pushContext) pushMetrics(ctx context.Context, writeMetrics func(w io.W
// Prepare the request to sent to pc.pushURL // Prepare the request to sent to pc.pushURL
reqBody := bytes.NewReader(bb.B) reqBody := bytes.NewReader(bb.B)
req, err := http.NewRequestWithContext(ctx, "GET", pc.pushURL.String(), reqBody) req, err := http.NewRequestWithContext(ctx, pc.method, pc.pushURL.String(), reqBody)
if err != nil { if err != nil {
panic(fmt.Errorf("BUG: metrics.push: cannot initialize request for metrics push to %q: %w", pc.pushURLRedacted, err)) panic(fmt.Errorf("BUG: metrics.push: cannot initialize request for metrics push to %q: %w", pc.pushURLRedacted, err))
} }
// Set the needed headers req.Header.Set("Content-Type", "text/plain")
// Set the needed headers, and `Content-Type` allowed be overwrited.
for name, values := range pc.headers { for name, values := range pc.headers {
for _, value := range values { for _, value := range values {
req.Header.Add(name, value) req.Header.Add(name, value)
} }
} }
req.Header.Set("Content-Type", "text/plain")
if !pc.disableCompression { if !pc.disableCompression {
req.Header.Set("Content-Encoding", "gzip") req.Header.Set("Content-Encoding", "gzip")
} }