Merge pull request #10 from practicalgo/publish_code

Publish chap7/abort-processing-timeout

Author: amitsaha

chap7/abort-processing-timeout/go.mod

@@ -0,0 +1,3 @@
+module github.com/practicalgo/code/chap7/abort-processing-timeout
+
+go 1.16

chap7/abort-processing-timeout/server.go

@@ -0,0 +1,34 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+	"time"
+)
+
+func handleUserAPI(w http.ResponseWriter, r *http.Request) {
+	log.Println("I started processing the request")
+	time.Sleep(15 * time.Second)
+
+	log.Println("Before continuing, i will check if the timeout has already expired")
+	if r.Context().Err() != nil {
+		log.Printf("Aborting further processing: %v\n", r.Context().Err())
+		return
+	}
+	fmt.Fprintf(w, "Hello world!")
+	log.Println("I finished processing the request")
+}
+
+func main() {
+
+	timeoutDuration := 14 * time.Second
+
+	userHandler := http.HandlerFunc(handleUserAPI)
+	hTimeout := http.TimeoutHandler(userHandler, timeoutDuration, "I ran out of time")
+
+	mux := http.NewServeMux()
+	mux.Handle("/api/users/", hTimeout)
+
+	log.Fatal(http.ListenAndServe(":8080", mux))
+}