main.go

package main

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

func load(response http.ResponseWriter, _ *http.Request) {
	done := make(chan int)

	for i := 0; i < runtime.NumCPU(); i++ {
		go func() {
			for {
				select {
				case <-done:
					return
				default:
				}
			}
		}()
	}

	time.Sleep(time.Second * 10)
	close(done)
	_, err := response.Write([]byte("Load generation done."))
	if err != nil {
		log.Printf("failed to write response (%v)", err)
	}
}

func health(response http.ResponseWriter, _ *http.Request) {
	_, err := response.Write([]byte("OK"))
	if err != nil {
		log.Printf("failed to write response (%v)", err)
	}
}

func main() {
	http.HandleFunc("/health", health)
	http.HandleFunc("/load", load)
	err := http.ListenAndServe(":8080", nil)
	if err != nil {
		log.Fatalf("failed to launch webserver (%v)", err)
	}
}