44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
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)
|
|
}
|