mod: optimize aggregator

Author: toannhu96

internal/config/config.go

@@ -22,7 +22,7 @@ var (
 	defaultPhishingDb = "https://cfg.rpchub.io/agg/scam-addresses.json"
 
 	_Config = &Config{
-		RequestTimeout:           60,
+		RequestTimeout:           90,
 		MaxRetries:               3,
 		PhishingDb:               []string{defaultPhishingDb},
 		PhishingDbUpdateInterval: 3600,

internal/middleware/plugins/http_proxy.go

@@ -76,14 +76,15 @@ func (m *HttpProxyMiddleware) OnProcess(session *rpc.Session) error {
 	if ctx, ok := session.RequestCtx.(*fasthttp.RequestCtx); ok {
 		log.Debug("relay rpc -> "+session.RpcMethod(), "sid", session.SId(), "node", session.NodeName, "isTx", session.IsWriteRpcMethod, "tries", session.Tries)
 
+		var node *aggregator.Node
 		if _, ok := m.disableEndpoints.Get(session.NodeName); ok {
 			log.Debug("disabled endpoint", "node", session.NodeName)
 			retries := 3
 			for {
 				if retries == 0 {
 					return aggregator.ErrServerError
 				}
-				node := loadbalance.NextNode(session.Chain)
+				node = loadbalance.NextNode(session.Chain)
 				if node != nil {
 					session.NodeName = node.Name
 					ctx.Request.SetRequestURI(node.Endpoint)
@@ -114,13 +115,16 @@ func (m *HttpProxyMiddleware) OnProcess(session *rpc.Session) error {
 		}, policies...)
 
 		// TODO: add response headers
-		//if ctx, ok := session.RequestCtx.(*fasthttp.RequestCtx); ok {
-		//	ctx.Response.Header.Set("Access-Control-Max-Age", "86400")
-		//	ctx.Response.Header.Set("Access-Control-Allow-Origin", "*")
-		//	ctx.Response.Header.Set("Access-Control-Allow-Methods", "POST, GET, PUT, DELETE, OPTIONS")
-		//	ctx.Response.Header.Set("Access-Control-Allow-Credentials", "true")
-		//	ctx.Response.Header.Set("X-Do-Node", session.NodeName)
-		//}
+		if node != nil && strings.Contains(node.Endpoint, "coin98.com") {
+			if ctx, ok := session.RequestCtx.(*fasthttp.RequestCtx); ok {
+				ctx.Response.Header.Set("origin", "https://dex.saros.xyz")
+				//	ctx.Response.Header.Set("Access-Control-Max-Age", "86400")
+				//	ctx.Response.Header.Set("Access-Control-Allow-Origin", "*")
+				//	ctx.Response.Header.Set("Access-Control-Allow-Methods", "POST, GET, PUT, DELETE, OPTIONS")
+				//	ctx.Response.Header.Set("Access-Control-Allow-Credentials", "true")
+				//	ctx.Response.Header.Set("X-Do-Node", session.NodeName)
+			}
+		}
 
 		var shouldDisableEndpoint = false
 		// alert if error
@@ -177,7 +181,13 @@ func (m *HttpProxyMiddleware) OnProcess(session *rpc.Session) error {
 
 		// check response body
 		var response map[string]interface{}
-		if err1 := json.Unmarshal(ctx.Response.Body(), &response); err1 == nil {
+		if strings.Contains(string(ctx.Response.Body()), "error status code 429") {
+			log.Error("error response", "node", session.NodeName, "response", string(ctx.Response.Body()))
+			err = fmt.Errorf("error response %s", string(ctx.Response.Body()))
+			shouldDisableEndpoint = true
+			ctx.SetStatusCode(429)
+			return err
+		} else if err1 := json.Unmarshal(ctx.Response.Body(), &response); err1 == nil {
 			if _, ok := response["error"]; ok {
 				log.Error("error response", "node", session.NodeName, "response", string(ctx.Response.Body()))
 				err = fmt.Errorf("error response %s", string(ctx.Response.Body()))