handler.go

package connecthttp

import (
	"context"
	"net/http"
)

type Handler struct {
	impl func(w http.ResponseWriter, r *http.Request)
}

func NewHandler[Req, Res any](
	_ string,
	unary func(context.Context, *Request[Req]) (*Response[Res], error),
	options ...HandlerOption,
) *Handler {
	config := newHandlerConfig(options...)
	impl := func(w http.ResponseWriter, r *http.Request) {
		var req Req
		if err := config.drf(r, &req); err != nil {
			config.eef(w, r, err)
			return
		}

		response, err := unary(r.Context(), NewRequest(&req))
		if err != nil {
			config.eef(w, r, err)
			return
		}

		if err := config.erf(w, r, response.Any()); err != nil {
			config.eef(w, r, err)
			return
		}
	}

	return &Handler{
		impl: impl,
	}
}

func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	tr := &Transport{
		request:  r,
		response: w,
	}
	tr.request = r.WithContext(NewTransportContext(ctx, tr))
	h.impl(w, tr.request)
}