simple server with middlewares
This commit is contained in:
@@ -0,0 +1,4 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
}
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
@@ -0,0 +1,38 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"math/rand"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
const PORT = 8080
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
server := Server{}
|
||||||
|
|
||||||
|
server.Handle(
|
||||||
|
"/get-random-number",
|
||||||
|
printHello,
|
||||||
|
handleGetRandomNumber,
|
||||||
|
)
|
||||||
|
|
||||||
|
if err := server.Listen(PORT); err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
fmt.Println("listening on :", PORT)
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleGetRandomNumber(next http.Handler) http.Handler {
|
||||||
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
fmt.Println("Random number: ", rand.Int())
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func printHello(next http.Handler) http.Handler {
|
||||||
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
fmt.Println("hello")
|
||||||
|
|
||||||
|
next.ServeHTTP(w, r)
|
||||||
|
})
|
||||||
|
}
|
||||||
@@ -0,0 +1,33 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ServerHandler func(http.Handler) http.Handler
|
||||||
|
|
||||||
|
type Server struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s Server) Handle(addr string, handlers ...ServerHandler) {
|
||||||
|
http.Handle(addr, handleMiddlewares(handlers))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s Server) Listen(port int) error {
|
||||||
|
if err := http.ListenAndServe(fmt.Sprintf(":%d", port), nil); err != nil {
|
||||||
|
return fmt.Errorf("server broke: %s", err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func handleMiddlewares(handlers []ServerHandler) http.Handler {
|
||||||
|
var handler http.Handler
|
||||||
|
|
||||||
|
for i := range handlers {
|
||||||
|
handler = handlers[len(handlers)-1-i](handler)
|
||||||
|
}
|
||||||
|
|
||||||
|
return handler
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user