-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #8 from volmedo/panic-recovery
Add a basic panic recovery middleware to send 500 Internal Server Error responses to clients when something goes wrong.
- Loading branch information
Showing
6 changed files
with
68 additions
and
60 deletions.
There are no files selected for viewing
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,59 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "net/http" | ||
| "time" | ||
|
|
||
| "github.com/prometheus/client_golang/prometheus/promhttp" | ||
| metrics "github.com/slok/go-http-metrics/metrics/prometheus" | ||
| "github.com/slok/go-http-metrics/middleware" | ||
| "github.com/ulule/limiter/v3" | ||
| "github.com/ulule/limiter/v3/drivers/middleware/stdlib" | ||
| "github.com/ulule/limiter/v3/drivers/store/memory" | ||
| "github.com/unrolled/recovery" | ||
| ) | ||
|
|
||
| // newMeasuredHandler creates a middleware that take essential metrics about | ||
| // the handler being measured, such as number of requests, duration of each request, | ||
| // concurrent or in-flight requests and response size. | ||
| // This function returns two handlers, the handler being measured and a Prometheus | ||
| // handler that exposes the metrics being collected | ||
| func newMeasuredHandler(handler http.Handler) (measuredH http.Handler, metricsH http.Handler) { | ||
| recorder := metrics.NewRecorder(metrics.Config{ | ||
| Prefix: "pAPI", | ||
| }) | ||
| mdlw := middleware.New(middleware.Config{ | ||
| Recorder: recorder, | ||
| }) | ||
|
|
||
| return mdlw.Handler("", handler), promhttp.Handler() | ||
| } | ||
|
|
||
| // newRateLimitedHandler creates a new middleware based on ulule/limiter package that | ||
| // limits the request rate that is sent to the specified handler. | ||
| // The returned rate-limited handler will allow up to rps requests per second to | ||
| // handler. When the rate exceeds the limit, a "429 Too Many Requests" response will be | ||
| // sent back without invoking the wrapped handler. | ||
| func newRateLimitedHandler(rps int64, handler http.Handler) (http.Handler, error) { | ||
| if rps <= 0 { | ||
| return nil, fmt.Errorf("rps cannot be negative (rps = %d)", rps) | ||
| } | ||
|
|
||
| store := memory.NewStore() | ||
| rate := limiter.Rate{ | ||
| Period: time.Second, | ||
| Limit: rps, | ||
| } | ||
| instance := limiter.New(store, rate) | ||
| middleware := stdlib.NewMiddleware(instance) | ||
|
|
||
| return middleware.Handler(handler), nil | ||
| } | ||
|
|
||
| // newRecoveredHandler adds a basic panic recovery middleware so that clients | ||
| // get a 500 Internal Server Error when something goes wrong | ||
| func newRecoverableHandler(handler http.Handler) http.Handler { | ||
| rec := recovery.New() | ||
| return rec.Handler(handler) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters