snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/pkg/apigw/helpers.go

@@ -0,0 +1,43 @@
+package apigw
+
+import (
+	"net/http"
+
+	"github.com/cortezaproject/corteza/server/pkg/apigw/profiler"
+	"github.com/cortezaproject/corteza/server/pkg/apigw/types"
+	h "github.com/cortezaproject/corteza/server/pkg/http"
+	"go.uber.org/zap"
+)
+
+func helperDefaultResponse(cfg types.Config, pr *profiler.Profiler, log *zap.Logger) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		addToProfiler(cfg, pr, log, r, http.StatusNotFound)
+		http.Error(w, "", http.StatusNotFound)
+	}
+}
+
+func helperMethodNotAllowed(cfg types.Config, pr *profiler.Profiler, log *zap.Logger) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		addToProfiler(cfg, pr, log, r, http.StatusMethodNotAllowed)
+		http.Error(w, "", http.StatusMethodNotAllowed)
+	}
+}
+
+func addToProfiler(cfg types.Config, pr *profiler.Profiler, log *zap.Logger, r *http.Request, status int) {
+	if !(cfg.Profiler.Enabled && cfg.Profiler.Global) {
+		return
+	}
+
+	// add to profiler
+	ar, err := h.NewRequest(r)
+
+	if err != nil {
+		log.Warn("could not create request wrapper, not adding to profiler")
+		return
+	}
+
+	h := pr.Hit(ar)
+	h.Status = status
+
+	pr.Push(h)
+}