Add response compression to horizon and federation

stellar · Mar 28, 2018 · 58bc4ff · 58bc4ff
1 parent 74a7175
commit 58bc4ff
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/services/federation/main.go b/services/federation/main.go
@@ -7,6 +7,7 @@ import (
 	"goji.io"
 	"goji.io/pat"
 
+	"github.com/go-chi/chi/middleware"
 	"github.com/rs/cors"
 	"github.com/spf13/cobra"
 	"github.com/stellar/go/handlers/federation"
@@ -138,8 +139,9 @@ func initMux(driver federation.Driver) *goji.Mux {
 	})
 	mux.Use(c.Handler)
 	mux.Use(log.HTTPMiddleware)
+	mux.Use(middleware.DefaultCompress)
 
-	fed := &federation.Handler{driver}
+	fed := &federation.Handler{Driver: driver}
 
 	mux.Handle(pat.Get("/federation"), fed)
 	mux.Handle(pat.Get("/federation/"), fed)

diff --git a/services/horizon/internal/init_web.go b/services/horizon/internal/init_web.go
@@ -1,12 +1,14 @@
 package horizon
 
 import (
+	"compress/flate"
 	"database/sql"
 	"net/http"
 	"strings"
 
 	"github.com/PuerkitoBio/throttled"
 	"github.com/PuerkitoBio/throttled/store"
+	chimiddleware "github.com/go-chi/chi/middleware"
 	metrics "github.com/rcrowley/go-metrics"
 	"github.com/rs/cors"
 	"github.com/sebest/xff"
@@ -61,6 +63,7 @@ func initWebMiddleware(app *App) {
 	r.Use(requestMetricsMiddleware)
 	r.Use(RecoverMiddleware)
 	r.Use(middleware.AutomaticOptions)
+	r.Use(chimiddleware.Compress(flate.DefaultCompression, "application/hal+json"))
 
 	c := cors.New(cors.Options{
 		AllowedOrigins: []string{"*"},