Add cache for prover

pkg/api/handler.go

@@ -8,6 +8,7 @@ import (
 	"sync"
 	"time"
 
+	"github.com/tonkeeper/claim-api-go/pkg/utils"
 	"github.com/tonkeeper/tongo/tvm"
 
 	"github.com/avast/retry-go"
@@ -34,6 +35,9 @@ type Handler struct {
 	cli          *liteapi.Client
 	config       string
 
+	proofsCache      utils.Cache[ton.AccountID, prover.WalletAirdrop]
+	keyNotFoundCache utils.Cache[ton.AccountID, struct{}]
+
 	mu                     sync.RWMutex
 	jettonMasterStateCache map[ton.AccountID][2]string
 }
@@ -81,6 +85,8 @@ func NewHandler(logger *zap.Logger, config Config) (*Handler, error) {
 		jettonMaster:           config.JettonMaster,
 		jettonMasterStateCache: map[ton.AccountID][2]string{},
 		config:                 blockchainConfig,
+		proofsCache:            utils.NewLRUCache[ton.AccountID, prover.WalletAirdrop](700_000, "proofs"),
+		keyNotFoundCache:       utils.NewLRUCache[ton.AccountID, struct{}](700_000, "keyNotFound"),
 	}, nil
 }
 
@@ -135,6 +141,18 @@ func (h *Handler) GetWalletInfo(ctx context.Context, params oas.GetWalletInfoPar
 	if err != nil {
 		return nil, BadRequest("failed to parse account id")
 	}
+
+	if proof, ok := h.proofsCache.Get(accountID); ok {
+		info, err := h.convertToWalletInfo(ctx, proof)
+		if err != nil {
+			return nil, InternalError(err)
+		}
+		return info, nil
+	}
+	if _, ok := h.keyNotFoundCache.Get(accountID); ok {
+		return nil, NotFound("account not found")
+	}
+
 	responseCh := make(chan prover.ProofResponse, 1)
 	h.prover.Queue() <- prover.ProofRequest{
 		AccountID:  accountID,
@@ -145,11 +163,13 @@ func (h *Handler) GetWalletInfo(ctx context.Context, params oas.GetWalletInfoPar
 		return nil, BadRequest("timeout")
 	case resp := <-responseCh:
 		if resp.Err != nil && strings.Contains(resp.Err.Error(), "key is not found") {
-			return nil, NotFound("account not not found")
+			h.keyNotFoundCache.Set(accountID, struct{}{})
+			return nil, NotFound("account not found")
 		}
 		if resp.Err != nil {
 			return nil, InternalError(resp.Err)
 		}
+		h.proofsCache.Set(accountID, resp.WalletAirdrop, utils.WithExpiration(7*time.Minute))
 		info, err := h.convertToWalletInfo(ctx, resp.WalletAirdrop)
 		if err != nil {
 			return nil, InternalError(err)