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