087d8085b7<!-- Provide a brief summary of your changes --> ## Motivation and Context <!-- Why is this change needed? What problem does it solve? --> Follow up from #658 addressing the feedback from @tadasant ## How Has This Been Tested? <!-- Have you tested this in a real application? Which scenarios were tested? --> ## Breaking Changes <!-- Will users need to update their code or configurations? --> ## Types of changes <!-- What types of changes does your code introduce? Put an `x` in all the boxes that apply: --> - [ ] Bug fix (non-breaking change which fixes an issue) - [ ] New feature (non-breaking change which adds functionality) - [ ] Breaking change (fix or feature that would cause existing functionality to change) - [ ] Documentation update ## Checklist <!-- Go over all the following points, and put an `x` in all the boxes that apply. --> - [ ] I have read the [MCP Documentation](https://modelcontextprotocol.io) - [ ] My code follows the repository's style guidelines - [ ] New and existing tests pass locally - [ ] I have added appropriate error handling - [ ] I have added or updated documentation as needed ## Additional context <!-- Add any other context, implementation notes, or design decisions --> Signed-off-by: Radoslav Dimitrov <radoslav@stacklok.com>
54 lines
1.8 KiB
Go
54 lines
1.8 KiB
Go
package v0
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/danielgtaylor/huma/v2"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/metric"
|
|
|
|
"github.com/modelcontextprotocol/registry/internal/config"
|
|
"github.com/modelcontextprotocol/registry/internal/telemetry"
|
|
)
|
|
|
|
// HealthBody represents the health check response body
|
|
type HealthBody struct {
|
|
Status string `json:"status" example:"ok" doc:"Health status"`
|
|
GitHubClientID string `json:"github_client_id,omitempty" doc:"GitHub OAuth App Client ID"`
|
|
}
|
|
|
|
// RegisterHealthEndpoint registers the health check endpoint with a custom path prefix
|
|
func RegisterHealthEndpoint(api huma.API, pathPrefix string, cfg *config.Config, metrics *telemetry.Metrics) {
|
|
huma.Register(api, huma.Operation{
|
|
OperationID: "get-health" + strings.ReplaceAll(pathPrefix, "/", "-"),
|
|
Method: http.MethodGet,
|
|
Path: pathPrefix + "/health",
|
|
Summary: "Health check",
|
|
Description: "Check the health status of the API",
|
|
Tags: []string{"health"},
|
|
}, func(ctx context.Context, _ *struct{}) (*Response[HealthBody], error) {
|
|
// Record the health check metrics
|
|
recordHealthMetrics(ctx, metrics, pathPrefix+"/health", cfg.Version)
|
|
|
|
return &Response[HealthBody]{
|
|
Body: HealthBody{
|
|
Status: "ok",
|
|
GitHubClientID: cfg.GithubClientID,
|
|
},
|
|
}, nil
|
|
})
|
|
}
|
|
|
|
// recordHealthMetrics records the health check metrics
|
|
func recordHealthMetrics(ctx context.Context, metrics *telemetry.Metrics, path string, version string) {
|
|
attrs := []attribute.KeyValue{
|
|
attribute.String("path", path),
|
|
attribute.String("version", version),
|
|
attribute.String("service", telemetry.Namespace),
|
|
}
|
|
|
|
// metric : Up status (1 = healthy, 0 = unhealthy)
|
|
metrics.Up.Record(ctx, 1, metric.WithAttributes(attrs...))
|
|
}
|