90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
package app
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.pbiernat.dev/golang/rest-api-prototype/internal/app/handler"
|
|
)
|
|
|
|
type Server struct {
|
|
*http.Server
|
|
}
|
|
|
|
func NewServer(env *handler.Env) *Server {
|
|
return &Server{
|
|
&http.Server{
|
|
Handler: SetupRouter(env),
|
|
Addr: env.Addr,
|
|
WriteTimeout: 15 * time.Second,
|
|
ReadTimeout: 15 * time.Second,
|
|
IdleTimeout: 60 * time.Second,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (s *Server) Start() {
|
|
log.Println("Server listening on " + s.Addr)
|
|
if err := s.ListenAndServe(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
// func (s *Server) Shutdown(ctx context.Context) {
|
|
// log.Println("Shutting down...")
|
|
|
|
// if err := s.Shutdown(ctx); err != nil {
|
|
// log.Panicln(err)
|
|
// }
|
|
// }
|
|
|
|
func PrepareHeadersMiddleware(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.Header().Set("Connection", "keep-alive")
|
|
w.Header().Set("Keep-Alive", "timeout=5")
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
func ValidateJsonBodyMiddleware(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
buf, _ := ioutil.ReadAll(r.Body)
|
|
r.Body = ioutil.NopCloser(bytes.NewReader(buf)) // rollack *Request to original state
|
|
|
|
// if len(buf) > 0 {
|
|
// next.ServeHTTP(w, r)
|
|
|
|
// return
|
|
// }
|
|
|
|
// if !json.Valid(buf) {
|
|
// w.WriteHeader(http.StatusBadRequest)
|
|
// json.NewEncoder(w).Encode(handler.Error("Unable to parse JSON: " + string(buf)))
|
|
|
|
// return
|
|
// }
|
|
|
|
if len(buf) > 0 && !json.Valid(buf) {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
json.NewEncoder(w).Encode(handler.Error("Unable to parse JSON: " + string(buf)))
|
|
|
|
return
|
|
}
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
func LoggingMiddleware(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
log.Println("Request: " + r.RequestURI + " remote: " + r.RemoteAddr + " via: " + r.UserAgent())
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|