Skip to content

Commit

Permalink
Add MetadataObfuscator
Browse files Browse the repository at this point in the history
  • Loading branch information
robertjndw committed Jan 26, 2024
1 parent 38af03d commit 3de6652
Show file tree
Hide file tree
Showing 3 changed files with 25 additions and 1 deletion.
2 changes: 1 addition & 1 deletion HadesAPI/go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ require (
github.com/Mtze/HadesCI/shared v0.0.0
github.com/gin-gonic/gin v1.9.1
github.com/hibiken/asynq v0.24.1
github.com/hibiken/asynqmon v0.7.2
github.com/sirupsen/logrus v1.9.3
)

Expand All @@ -27,7 +28,6 @@ require (
github.com/google/go-cmp v0.6.0 // indirect
github.com/google/uuid v1.6.0 // indirect
github.com/gorilla/mux v1.8.1 // indirect
github.com/hibiken/asynqmon v0.7.2 // indirect
github.com/joho/godotenv v1.5.1 // indirect
github.com/json-iterator/go v1.1.12 // indirect
github.com/klauspost/cpuid/v2 v2.2.4 // indirect
Expand Down
1 change: 1 addition & 0 deletions HadesAPI/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,7 @@ func main() {
RootPath: "/monitoring", // RootPath specifies the root for asynqmon app
RedisConnOpt: asynq_client_opts,
PrometheusAddress: cfg.PrometheusAddress,
PayloadFormatter: MetadataObfuscator,
})
r.Any("/monitoring/*a", gin.WrapH(h))

Expand Down
23 changes: 23 additions & 0 deletions HadesAPI/monitoring.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
package main

import (
"encoding/json"

"github.com/Mtze/HadesCI/shared/payload"
"github.com/hibiken/asynqmon"
log "github.com/sirupsen/logrus"
)

var MetadataObfuscator = asynqmon.PayloadFormatterFunc(func(_ string, bytes []byte) string {
var job payload.QueuePayload
if err := json.Unmarshal(bytes, &job); err != nil {
log.WithError(err).Error("Failed to unmarshal task payload")
return ""
}
for i := range job.Steps {
job.Steps[i].Metadata = make(map[string]string)
}

json_tmp, _ := json.Marshal(job)
return string(json_tmp)
})

0 comments on commit 3de6652

Please sign in to comment.