File size: 3,350 Bytes
5ac0758 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 | package main
import (
"log/slog"
"net/http"
"os"
"github.com/go-chi/chi/v5"
"github.com/go-chi/chi/v5/middleware"
)
func main() {
port := os.Getenv("PORT")
if port == "" {
port = "7860" // Default port for Hugging Face Spaces
}
addr := "0.0.0.0:" + port
slog.Info("Starting OpenMeter Native Mock Server", "address", addr)
r := chi.NewRouter()
r.Use(middleware.RequestID)
r.Use(middleware.RealIP)
r.Use(middleware.Logger)
r.Use(middleware.Recoverer)
// Redirect root to api-docs
r.Get("/", func(w http.ResponseWriter, r *http.Request) {
http.Redirect(w, r, "/api-docs", http.StatusFound)
})
// /health
r.Get("/health", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/plain")
w.WriteHeader(http.StatusOK)
_, _ = w.Write([]byte("OK"))
})
// /api/openapi.yaml
r.Get("/api/openapi.yaml", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/yaml")
// We'll read the openapi.yaml copied to the container
http.ServeFile(w, r, "/usr/local/bin/openapi.yaml")
})
// /api-docs
r.Get("/api-docs", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/html")
w.WriteHeader(http.StatusOK)
_, _ = w.Write([]byte(`<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8" />
<title>OpenMeter API Documentation (Native Mock Server)</title>
<link rel="stylesheet" href="https://unpkg.com/swagger-ui-dist@5/swagger-ui.css" />
<link rel="icon" type="image/png" href="https://unpkg.com/swagger-ui-dist@5/favicon-32x32.png" sizes="32x32" />
</head>
<body>
<div id="swagger-ui"></div>
<script src="https://unpkg.com/swagger-ui-dist@5/swagger-ui-bundle.js"></script>
<script>
window.onload = () => {
window.ui = SwaggerUIBundle({
url: '/api/openapi.yaml',
dom_id: '#swagger-ui',
deepLinking: true,
presets: [
SwaggerUIBundle.presets.apis,
SwaggerUIBundle.swaggerInterceptor
]
});
};
</script>
</body>
</html>`))
})
// GET /api/v1/meters
r.Get("/api/v1/meters", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
_, _ = w.Write([]byte(`[
{
"id": "api_requests_total",
"slug": "api_requests_total",
"description": "API Requests",
"eventType": "request",
"aggregation": "COUNT"
},
{
"id": "tokens_total",
"slug": "tokens_total",
"description": "AI Token Usage",
"eventType": "prompt",
"aggregation": "SUM"
}
]`))
})
// POST /api/v1/events
r.Post("/api/v1/events", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusAccepted)
_, _ = w.Write([]byte(`{"status": "accepted"}`))
})
// GET /api/v1/meters/{meterIdOrSlug}/query
r.Get("/api/v1/meters/{meterIdOrSlug}/query", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
_, _ = w.Write([]byte(`{
"data": [
{
"value": 150,
"windowStart": "2026-07-07T00:00:00Z",
"windowEnd": "2026-07-07T01:00:00Z"
}
]
}`))
})
err := http.ListenAndServe(addr, r)
if err != nil {
slog.Error("Mock server stopped with error", "error", err)
os.Exit(1)
}
}
|