simple server with middlewares

This commit is contained in:
unintendedfraud
2024-03-07 02:25:55 +01:00
commit ef3410470a
5 changed files with 80 additions and 0 deletions
+4
View File
@@ -0,0 +1,4 @@
package main
func main() {
}
+2
View File
@@ -0,0 +1,2 @@
package main
+38
View File
@@ -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)
})
}
+33
View File
@@ -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
}
+3
View File
@@ -0,0 +1,3 @@
module goplayground
go 1.22.0