Skip to content

Ignore any calls to increment_metric as irrelevant #93

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.

Sign up for GitHub

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

Merged
brrygrdn merged 2 commits into from
Apr 6, 2023
Merged

Ignore any calls to increment_metric as irrelevant #93

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
651c176
Ignore any calls to increment_metric as irrelevant
brrygrdn Apr 6, 2023
a7a009a
Let's output the metrics data instead of swallowing it
brrygrdn Apr 6, 2023
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
5 changes: 5 additions & 0 deletions internal/model/update.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -58,3 +58,8 @@ type RecordUpdateJobError struct {
ErrorType string `json:"error-type" yaml:"error-type"`
ErrorDetails map[string]any `json:"error-details" yaml:"error-details"`
}

type IncrementMetric struct {
Metric string `json:"metric" yaml:"metric"`
Tags map[string]any `json:"tags" yaml:"tags"`
}
32 changes: 22 additions & 10 deletions internal/server/api.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -118,22 +118,19 @@ func (a *API) ServeHTTP(_ http.ResponseWriter, r *http.Request) {
a.pushError(err)
}

if kind == "increment_metric" {
// Let's just output the metrics data and stop
a.outputRequestData(kind, actual)
return
}

if err := a.pushResult(kind, actual); err != nil {
a.pushError(err)
return
}

if !a.hasExpectations {
if a.writer != nil {
// output the data received to stdout
if err = json.NewEncoder(a.writer).Encode(map[string]any{
"type": kind,
"data": actual.Data,
}); err != nil {
// Fail so the user knows stdout is not working
log.Panicln("Failed to write to stdout: ", err)
}
}
a.outputRequestData(kind, actual)
return
}

Expand Down Expand Up @@ -167,6 +164,19 @@ func (a *API) assertExpectation(kind string, actual *model.UpdateWrapper) {
}
}

func (a *API) outputRequestData(kind string, actual *model.UpdateWrapper) {
if a.writer != nil {
// output the data received to stdout
if err := json.NewEncoder(a.writer).Encode(map[string]any{
"type": kind,
"data": actual.Data,
}); err != nil {
// Fail so the user knows stdout is not working
log.Panicln("Failed to write to stdout: ", err)
}
}
}

func (a *API) pushError(err error) {
escapedError := strings.ReplaceAll(err.Error(), "\n", "")
escapedError = strings.ReplaceAll(escapedError, "\r", "")
Expand Down Expand Up @@ -213,6 +223,8 @@ func decodeWrapper(kind string, data []byte) (actual *model.UpdateWrapper, err e
actual.Data, err = decode[model.RecordPackageManagerVersion](data)
case "record_update_job_error":
actual.Data, err = decode[model.RecordUpdateJobError](data)
case "increment_metric":
actual.Data, err = decode[model.IncrementMetric](data)
default:
return nil, fmt.Errorf("unexpected output type: %s", kind)
}
Expand Down