feat: set height header for GRPC query (#13)

terra-money · Dec 20, 2021 · 86f44af · 86f44af
1 parent 98f46da
commit 86f44af
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bin/v0.34.x/rpc/register.go b/bin/v0.34.x/rpc/register.go
@@ -90,10 +90,13 @@ func StartRPC(
 	// caching middleware
 	apiSrv.Router.Use(func(next http.Handler) http.Handler {
 		return http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
-			height, err := strconv.ParseInt(request.URL.Query().Get("height"), 10, 64)
+			heightQuery := request.URL.Query().Get("height")
+			height, err := strconv.ParseInt(heightQuery, 10, 64)
 
 			// don't use archival cache if height is 0 or error
 			if err == nil && height > 0 {
+				// GRPC query parses height from header
+				request.Header.Add("x-cosmos-block-height", heightQuery)
 				archivalCache.HandleCachedHTTP(writer, request, next)
 			} else {
 				cache.HandleCachedHTTP(writer, request, next)