Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

buiding a new body for conflict reponses #1068

Merged
merged 3 commits into from
Apr 25, 2024
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 22 additions & 11 deletions cmd/prober/write.go
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,24 @@
return cert[0], nil
}

func makeRekorRequest(cert *x509.Certificate, priv *ecdsa.PrivateKey, hostPath string) (*http.Response, time.Time, error) {

Check failure on line 142 in cmd/prober/write.go

View workflow job for this annotation

GitHub Actions / lint

unnecessary leading newline (whitespace)

body, err := rekorEntryRequest(cert, priv)
t := time.Now()
if err != nil {
return nil, t, fmt.Errorf("rekor entry: %w", err)
}
req, err := retryablehttp.NewRequest(http.MethodPost, hostPath, bytes.NewBuffer(body))
if err != nil {
return nil, t, fmt.Errorf("new request: %w", err)
}
setHeaders(req, "")

t = time.Now()
resp, err := retryableClient.Do(req)
return resp, t, err
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'd suggest returning the latency calculation latency := time.Since(t).Milliseconds() rather than the time object.

}

// rekorWriteEndpoint tests the write endpoint for rekor, which is
// /api/v1/log/entries and adds an entry to the log
// if a certificate is provided, the Rekor entry will contain that certificate,
Expand All @@ -151,19 +169,12 @@
verificationCounter.With(prometheus.Labels{verifiedLabel: verified}).Inc()
}()

body, err := rekorEntryRequest(cert, priv)
if err != nil {
return fmt.Errorf("rekor entry: %w", err)
resp, t, err := makeRekorRequest(cert, priv, hostPath)

Check failure on line 172 in cmd/prober/write.go

View workflow job for this annotation

GitHub Actions / lint

response body must be closed (bodyclose)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can we move this into the for loop?

// A new body should be created when it is conflicted
for resp.StatusCode == http.StatusConflict {
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Does it make sense it be in a while? Or should I set a number of attempts ?

resp, t, err = makeRekorRequest(cert, priv, hostPath)

Check failure on line 175 in cmd/prober/write.go

View workflow job for this annotation

GitHub Actions / lint

response body must be closed (bodyclose)
}
req, err := retryablehttp.NewRequest(http.MethodPost, hostPath, bytes.NewBuffer(body))
if err != nil {
return fmt.Errorf("new request: %w", err)
}

setHeaders(req, "")

t := time.Now()
resp, err := retryableClient.Do(req)
latency := time.Since(t).Milliseconds()
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can we record latency and response data for each request to Rekor? This would measure the total time for all requests.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You mean we should also record the responses that has StatusConflict?

if err != nil {
return fmt.Errorf("error adding entry: %w", err)
Expand Down
Loading