experiment

context/server.go

@@ -1,20 +1,32 @@
 package main
 
 import (
+	"context"
 	"fmt"
+	"log"
 	"net/http"
+	"time"
 )
 
 type ServerHandler func(http.Handler) http.Handler
 
 type Server struct {
+	timeout time.Duration
 }
 
-func (s Server) Handle(addr string, handlers ...ServerHandler) {
-	http.Handle(addr, handleMiddlewares(handlers))
+func NewServer(timeout time.Duration) *Server {
+	return &Server{
+		timeout,
+	}
 }
 
-func (s Server) Listen(port int) error {
+func (s *Server) Handle(addr string, handlers ...ServerHandler) {
+	middlewares := append([]ServerHandler{s.ctxMiddleware}, handlers...)
+
+	http.Handle(addr, handleMiddlewares(middlewares))
+}
+
+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())
 	}
@@ -22,6 +34,28 @@ func (s Server) Listen(port int) error {
 	return nil
 }
 
+func (s *Server) ctxMiddleware(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		ctx, cancel := context.WithTimeout(r.Context(), s.timeout)
+		defer cancel()
+
+		processDone := make(chan bool)
+
+		go func() {
+			next.ServeHTTP(w, r.WithContext(ctx))
+			processDone <- true
+		}()
+
+		select {
+		case <-ctx.Done():
+			w.Write([]byte(`{"error": "context expired"}`))
+			log.Panicln("context expired")
+		case <-processDone:
+			fmt.Println("process done")
+		}
+	})
+}
+
 func handleMiddlewares(handlers []ServerHandler) http.Handler {
 	var handler http.Handler