51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
|
package simpleserver
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
func RunServerWithMiddleware() {
|
||
|
|
getOnly := func(next http.Handler) http.Handler {
|
||
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
if r.Method != http.MethodGet {
|
||
|
|
w.WriteHeader(http.StatusMethodNotAllowed)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
next.ServeHTTP(w, r)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
handler := func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
_, _ = w.Write([]byte("pong"))
|
||
|
|
}
|
||
|
|
|
||
|
|
err := http.ListenAndServe(":8080", getOnly(http.HandlerFunc(handler)))
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func UnifiedErrorMiddleware() {
|
||
|
|
wrapErrorReply := func(h func(w http.ResponseWriter, r *http.Request) error) http.Handler {
|
||
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
if err := h(w, r); err != nil {
|
||
|
|
w.WriteHeader(http.StatusBadRequest)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
handler := func(w http.ResponseWriter, r *http.Request) error {
|
||
|
|
if r.URL.Query().Get("secret") != "FtP8lu70XjWj8Stt" {
|
||
|
|
return errors.New("secret mismatch")
|
||
|
|
}
|
||
|
|
_, _ = w.Write([]byte("pong"))
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
err := http.ListenAndServe(":8080", wrapErrorReply(handler))
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|