46 lines
917 B
Go
46 lines
917 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/rand"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
const PORT = 8080
|
|
|
|
func main() {
|
|
server := NewServer(3 * time.Second)
|
|
|
|
server.Handle(
|
|
"/get-random-number",
|
|
addValueToContext,
|
|
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: ", r.Context().Value("number"))
|
|
w.Write([]byte(`{"error": "process timeout"}`))
|
|
})
|
|
}
|
|
|
|
func addValueToContext(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
n := rand.Intn(50)
|
|
ctx := context.WithValue(r.Context(), "number", n)
|
|
fmt.Println("add value to context: ", n)
|
|
|
|
time.Sleep(10 * time.Second)
|
|
|
|
next.ServeHTTP(w, r.WithContext(ctx))
|
|
})
|
|
}
|