allow to define a custom default response encoder

config.go

@@ -1,5 +1,7 @@
 package kitty
 
+import kithttp "github.com/go-kit/kit/transport/http"
+
 // Config holds configuration info for kitty.HTTPTransport.
 type Config struct {
 	// LivenessCheckPath is the path of the health handler (default: "/alivez").
@@ -10,6 +12,8 @@ type Config struct {
 	HTTPPort int
 	// EnablePProf enables pprof urls (default: false).
 	EnablePProf bool
+	// EncodeResponse overrides go-kit reponse encoder without to have to pass the encoder to each endpoint, but is still overrided by the endpoint's encoder
+	EncodeResponse kithttp.EncodeResponseFunc
 }
 
 // DefaultConfig defines the default config of kitty.HTTPTransport.
@@ -18,4 +22,5 @@ var DefaultConfig = Config{
 	LivenessCheckPath:  "/alivez",
 	ReadinessCheckPath: "/readyz",
 	EnablePProf:        false,
+	EncodeResponse:     kithttp.EncodeJSONResponse,
 }

endpoint.go

@@ -30,7 +30,6 @@ func (t *HTTPTransport) Endpoint(method, path string, ep endpoint.Endpoint, opts
 		path:     path,
 		endpoint: ep,
 		decoder:  kithttp.NopRequestDecoder,
-		encoder:  kithttp.EncodeJSONResponse,
 	}
 	for _, opt := range opts {
 		e = opt(e)

http.go

@@ -63,6 +63,7 @@ func NewHTTPTransport(cfg Config) *HTTPTransport {
 		t.cfg.ReadinessCheckPath = cfg.ReadinessCheckPath
 	}
 	t.cfg.EnablePProf = cfg.EnablePProf
+	t.cfg.EncodeResponse = cfg.EncodeResponse
 	return t
 }
 
@@ -80,11 +81,15 @@ func (t *HTTPTransport) RegisterEndpoints(m endpoint.Middleware) error {
 
 	// register endpoints
 	for _, ep := range t.endpoints {
+		encoder := t.cfg.EncodeResponse
+		if ep.encoder != nil {
+			encoder = ep.encoder
+		}
 		t.mux.Handle(ep.method, ep.path,
 			kithttp.NewServer(
 				m(ep.endpoint),
 				ep.decoder,
-				ep.encoder,
+				encoder,
 				append(opts, ep.options...)...))
 	}