86 lines
1.9 KiB
Go
86 lines
1.9 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) ExecuteHandler() {}
|
||
|
|
|
||
|
|
// 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 {
|
||
|
|
// log.Println("empty request body2") // FIXME #1
|
||
|
|
next.ServeHTTP(w, r)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if !json.Valid(buf) {
|
||
|
|
log.Println("Unable to parse JSON: ", string(buf))
|
||
|
|
w.WriteHeader(http.StatusInternalServerError)
|
||
|
|
|
||
|
|
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)
|
||
|
|
})
|
||
|
|
}
|