mirror of
https://github.com/foomo/keel.git
synced 2025-10-16 12:35:34 +00:00
41 lines
822 B
Go
41 lines
822 B
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/foomo/keel"
|
|
keelhttp "github.com/foomo/keel/net/http"
|
|
"github.com/foomo/keel/net/http/middleware"
|
|
"github.com/foomo/keel/service"
|
|
)
|
|
|
|
func main() {
|
|
svr := keel.NewServer()
|
|
|
|
// get logger
|
|
l := svr.Logger()
|
|
|
|
// custom request id generator
|
|
requestIDGenerator := func() string {
|
|
return "my-custom-request-id"
|
|
}
|
|
|
|
// create demo service
|
|
svs := http.NewServeMux()
|
|
svs.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
_, _ = w.Write([]byte(r.Header.Get(keelhttp.HeaderXRequestID)))
|
|
})
|
|
|
|
svr.AddService(
|
|
service.NewHTTP(l, "demo", "localhost:8080", svs,
|
|
middleware.RequestID(
|
|
middleware.RequestIDWithSetResponseHeader(true),
|
|
middleware.RequestIDWithGenerator(requestIDGenerator),
|
|
),
|
|
),
|
|
)
|
|
|
|
svr.Run()
|
|
}
|