closenotifier.go

closenotifier.go

package main

import (
	"net/http"
	"log"
	"time"
)

func main() {
	http.HandleFunc("/", func (w http.ResponseWriter, r *http.Request) {
		finished := make(chan struct{})
		defer close(finished)
		disconnected := w.(http.CloseNotifier).CloseNotify()
		go func () {
			select {
			case <-finished:
				log.Println("finished 1")
				return
			case <-disconnected:
				log.Println("disconnected 1")
				return
			}
		}()
		go func () {
			select {
			case <-finished:
				log.Println("finished 2")
				return
			case <-disconnected:
				log.Println("disconnected 2")
				return
			}
		}()
		time.Sleep(3*time.Second)
		w.Write([]byte("OK"))
	})
	http.ListenAndServe(":9876", nil)
}