add deprecation headers to custom rest handlers

terra-money · May 26, 2021 · c8c8f9e · c8c8f9e
1 parent 5ac5439
commit c8c8f9e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/custom/auth/client/rest/rest.go b/custom/auth/client/rest/rest.go
@@ -4,9 +4,11 @@ import (
 	"github.com/gorilla/mux"
 
 	"github.com/cosmos/cosmos-sdk/client"
+	clientrest "github.com/cosmos/cosmos-sdk/client/rest"
 )
 
 // RegisterTxRoutes registers registers terra custom transaction routes on the provided router.
 func RegisterTxRoutes(clientCtx client.Context, rtr *mux.Router) {
-	rtr.HandleFunc("/txs/estimate_fee", EstimateTxFeeRequestHandlerFn(clientCtx)).Methods("POST")
+	r := clientrest.WithHTTPDeprecationHeaders(rtr)
+	r.HandleFunc("/txs/estimate_fee", EstimateTxFeeRequestHandlerFn(clientCtx)).Methods("POST")
 }
diff --git a/custom/bank/client/rest/tx.go b/custom/bank/client/rest/tx.go
@@ -21,8 +21,8 @@ import (
 func RegisterRoutes(clientCtx client.Context, rtr *mux.Router) {
 	r := clientrest.WithHTTPDeprecationHeaders(rtr)
 
-	rtr.HandleFunc("/bank/accounts/{address}/transfers", sendRequestHandlerFn(clientCtx)).Methods("POST")
-	bankrest.RegisterHandlers(clientCtx, r)
+	r.HandleFunc("/bank/accounts/{address}/transfers", sendRequestHandlerFn(clientCtx)).Methods("POST")
+	bankrest.RegisterHandlers(clientCtx, rtr)
 }
 
 // SendReq defines the properties of a send request's body.