mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
f335d34636
we're going to want to do this in our service version of this thing, but adding this here so that it's usable by everyone. just an option, can add it to server configuration, but response is nicely formatted, etc. closes #277
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type ServerOption func(*Server)
|
|
|
|
func EnableShutdownEndpoint(halt context.CancelFunc) ServerOption {
|
|
return func(s *Server) {
|
|
s.Router.GET("/shutdown", s.handleShutdown(halt))
|
|
}
|
|
}
|
|
|
|
func LimitRequestBody(max int64) ServerOption {
|
|
return func(s *Server) {
|
|
s.Router.Use(limitRequestBody(max))
|
|
}
|
|
}
|
|
|
|
func limitRequestBody(max int64) func(c *gin.Context) {
|
|
return func(c *gin.Context) {
|
|
cl := int64(c.Request.ContentLength)
|
|
if cl > max {
|
|
// try to deny this quickly, instead of just letting it get lopped off
|
|
|
|
handleErrorResponse(c, errTooBig{cl, max})
|
|
c.Abort()
|
|
return
|
|
}
|
|
|
|
// if no Content-Length specified, limit how many bytes we read and error
|
|
// if we hit the max (intercontinental anti-air missile defense system).
|
|
// read http.MaxBytesReader for gritty details..
|
|
c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, max)
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
// models.APIError
|
|
type errTooBig struct {
|
|
n, max int64
|
|
}
|
|
|
|
func (e errTooBig) Code() int { return http.StatusRequestEntityTooLarge }
|
|
func (e errTooBig) Error() string {
|
|
return fmt.Sprintf("Content-Length too large for this server, %d > max %d", e.n, e.max)
|
|
}
|