feat: add proxy for unhandled requests (#67)

config/config.go

@@ -19,6 +19,7 @@ type Config struct {
 	ChainID              string
 	RPCEndpoints         []string
 	WSEndpoints          []string
+	LCDEndpoints         []string
 	MantlemintDB         string
 	IndexerDB            string
 	DisableSync          bool
@@ -67,6 +68,12 @@ func newConfig() Config {
 			return strings.Split(endpoints, ",")
 		}(),
 
+		// LCDEndpoints is where to forward unhandled queries to a node
+		LCDEndpoints: func() []string {
+			endpoints := getValidEnv("LCD_ENDPOINTS")
+			return strings.Split(endpoints, ",")
+		}(),
+
 		// MantlemintDB is the db name for mantlemint. Defaults to mantlemint
 		MantlemintDB: func() string {
 			mantlemintDB := getValidEnv("MANTLEMINT_DB")

rpc/proxy.go

@@ -0,0 +1,43 @@
+package rpc
+
+import (
+	"github.com/tendermint/tendermint/libs/rand"
+	"net/http"
+	"net/http/httptest"
+	"net/http/httputil"
+	"net/url"
+)
+
+type ProxyMiddleware struct {
+	lcdUrls []string
+	proxies []*httputil.ReverseProxy
+}
+
+func NewProxyMiddleware(lcdUrls []string) ProxyMiddleware {
+	var proxies []*httputil.ReverseProxy
+	for _, u := range lcdUrls {
+		lcdUrl, err := url.Parse(u)
+		if err != nil {
+			panic(err)
+		}
+		proxies = append(proxies, httputil.NewSingleHostReverseProxy(lcdUrl))
+	}
+	return ProxyMiddleware{
+		lcdUrls: lcdUrls,
+		proxies: proxies,
+	}
+}
+
+func (pm ProxyMiddleware) HandleRequest(writer http.ResponseWriter, request *http.Request, handler http.Handler) {
+	recorder := httptest.NewRecorder()
+	handler.ServeHTTP(recorder, request)
+	if recorder.Code != http.StatusOK {
+		// randomly pick a proxy from the list
+		proxyToUse := rand.NewRand().Intn(len(pm.proxies))
+		pm.proxies[proxyToUse].ServeHTTP(writer, request)
+		return
+	}
+	writer.WriteHeader(recorder.Code)
+	writer.Write(recorder.Body.Bytes())
+	return
+}

rpc/register.go

@@ -111,6 +111,14 @@ func StartRPC(
 		})
 	})
 
+	pm := NewProxyMiddleware(mantlemintConfig.LCDEndpoints)
+	// proxy middleware to handle unimplemented queries
+	apiSrv.Router.Use(func(next http.Handler) http.Handler {
+		return http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
+			pm.HandleRequest(writer, request, next)
+		})
+	})
+
 	// start api server in goroutine
 	go func() {
 		if err := apiSrv.Start(cfg); err != nil {