mirror of
https://github.com/hanzoai/status.git
synced 2026-10-30 18:21:48 +00:00
zeekayandHanzo Dev
78ab4f2f83
zip's registration is a verb method on the group, so a route that is not a typed operation says which method it answers with. The security package's own two-method router goes with it: it existed so both the app and a sub-router could satisfy one name, and restating a routing surface is what let the two disagree — the methods it named no longer exist. The app spells itself as a group, which registers at exactly the paths it did. The toolchain and the pin move together, because below zip v1.36.50 the JSON encoder was selected on a build tag: a repo whose go directive rises first writes a wire that is neither the old release's nor the new one's until the pin catches up. metrics_test.go still does not compile, before and after: it compares a metric registry against a gatherer from a different metrics library, so the two Gather signatures disagree. That is its own change. Co-authored-by: Hanzo Dev <[email protected]>
93 lines
3.6 KiB
Go
93 lines
3.6 KiB
Go
package api
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/TwiN/logr"
|
|
"github.com/zap-proto/zip"
|
|
"hanzo.ai/status/config"
|
|
"hanzo.ai/status/config/endpoint"
|
|
"hanzo.ai/status/metrics"
|
|
"hanzo.ai/status/storage/store"
|
|
"hanzo.ai/status/storage/store/common"
|
|
"hanzo.ai/status/watchdog"
|
|
)
|
|
|
|
func CreateExternalEndpointResult(cfg *config.Config) zip.Handler {
|
|
extraLabels := cfg.GetUniqueExtraMetricLabels()
|
|
return func(c *zip.Ctx) error {
|
|
// Check if the success query parameter is present
|
|
success, exists := c.Fiber().Queries()["success"]
|
|
if !exists || (success != "true" && success != "false") {
|
|
return c.String(400, "missing or invalid success query parameter")
|
|
}
|
|
// Check if the authorization bearer token header is correct
|
|
authorizationHeader := c.Header("Authorization")
|
|
if !strings.HasPrefix(authorizationHeader, "Bearer ") {
|
|
return c.String(401, "invalid Authorization header")
|
|
}
|
|
token := strings.TrimSpace(strings.TrimPrefix(authorizationHeader, "Bearer "))
|
|
if len(token) == 0 {
|
|
return c.String(401, "bearer token must not be empty")
|
|
}
|
|
key := c.Param("key")
|
|
externalEndpoint := cfg.GetExternalEndpointByKey(key)
|
|
if externalEndpoint == nil {
|
|
logr.Errorf("[api.CreateExternalEndpointResult] External endpoint with key=%s not found", key)
|
|
return c.String(404, "not found")
|
|
}
|
|
if externalEndpoint.Token != token {
|
|
logr.Errorf("[api.CreateExternalEndpointResult] Invalid token for external endpoint with key=%s", key)
|
|
return c.String(401, "invalid token")
|
|
}
|
|
// Persist the result in the storage
|
|
result := &endpoint.Result{
|
|
Timestamp: time.Now(),
|
|
Success: success == "true",
|
|
Errors: []string{},
|
|
}
|
|
if len(c.Query("duration")) > 0 {
|
|
parsedDuration, err := time.ParseDuration(c.Query("duration"))
|
|
if err != nil {
|
|
logr.Errorf("[api.CreateExternalEndpointResult] Invalid duration from string=%s with error: %s", c.Query("duration"), err.Error())
|
|
return c.String(400, "invalid duration: "+err.Error())
|
|
}
|
|
result.Duration = parsedDuration
|
|
}
|
|
if errorFromQuery := c.Query("error"); !result.Success && len(errorFromQuery) > 0 {
|
|
result.AddError(strings.Clone(errorFromQuery))
|
|
}
|
|
convertedEndpoint := externalEndpoint.ToEndpoint()
|
|
if err := store.Get().InsertEndpointResult(convertedEndpoint, result); err != nil {
|
|
if errors.Is(err, common.ErrEndpointNotFound) {
|
|
return c.String(404, err.Error())
|
|
}
|
|
logr.Errorf("[api.CreateExternalEndpointResult] Failed to insert result in storage: %s", err.Error())
|
|
return c.String(500, err.Error())
|
|
}
|
|
logr.Infof("[api.CreateExternalEndpointResult] Successfully inserted result for external endpoint with key=%s and success=%s", c.Param("key"), success)
|
|
inEndpointMaintenanceWindow := false
|
|
for _, maintenanceWindow := range externalEndpoint.MaintenanceWindows {
|
|
if maintenanceWindow.IsUnderMaintenance() {
|
|
logr.Debug("[api.CreateExternalEndpointResult] Under endpoint maintenance window")
|
|
inEndpointMaintenanceWindow = true
|
|
}
|
|
}
|
|
// Check if an alert should be triggered or resolved
|
|
if !cfg.Maintenance.IsUnderMaintenance() && !inEndpointMaintenanceWindow {
|
|
watchdog.HandleAlerting(convertedEndpoint, result, cfg.Alerting)
|
|
externalEndpoint.NumberOfSuccessesInARow = convertedEndpoint.NumberOfSuccessesInARow
|
|
externalEndpoint.NumberOfFailuresInARow = convertedEndpoint.NumberOfFailuresInARow
|
|
} else {
|
|
logr.Debug("[api.CreateExternalEndpointResult] Not handling alerting because currently in the maintenance window")
|
|
}
|
|
if cfg.Metrics {
|
|
metrics.PublishMetricsForEndpoint(convertedEndpoint, result, extraLabels)
|
|
}
|
|
// Return the result
|
|
return c.String(200, "")
|
|
}
|
|
}
|