fix comments

Author: erokhinav

pkg/api/event_handlers.go

@@ -8,6 +8,7 @@ import (
 	"errors"
 	"fmt"
 	"github.com/tonkeeper/tongo/abi"
+	"go.uber.org/zap"
 	"net/http"
 	"strings"
 	"time"
@@ -353,7 +354,13 @@ func (h *Handler) EmulateMessageToAccountEvent(ctx context.Context, request *oas
 		return nil, toError(http.StatusBadRequest, err)
 	}
 	hash, err := c.HashString()
+	if err != nil {
+		return nil, toError(http.StatusBadRequest, err)
+	}
 	trace, _, err := h.storage.GetTraceWithState(ctx, hash)
+	if err != nil {
+		h.logger.Warn("get trace from storage: ", zap.Error(err))
+	}
 	if trace == nil {
 		configBase64, err := h.storage.TrimmedConfigBase64()
 		if err != nil {
@@ -379,7 +386,7 @@ func (h *Handler) EmulateMessageToAccountEvent(ctx context.Context, request *oas
 		if err != nil {
 			return nil, toError(http.StatusInternalServerError, err)
 		}
-		err = h.storage.SaveTraceWithState(hash, trace, []abi.MethodInvocation{}, 24*time.Hour)
+		err = h.storage.SaveTraceWithState(ctx, hash, trace, []abi.MethodInvocation{}, 24*time.Hour)
 		if err != nil {
 			fmt.Println("trace not saved: ", err)
 		}
@@ -407,8 +414,14 @@ func (h *Handler) EmulateMessageToEvent(ctx context.Context, request *oas.Emulat
 	}
 	trace, prs := h.mempoolEmulate.traces.Get(hash)
 	if !prs {
-		hs, _ := c.HashString()
+		hs, err := c.HashString()
+		if err != nil {
+			return nil, toError(http.StatusBadRequest, err)
+		}
 		trace, _, err = h.storage.GetTraceWithState(ctx, hs)
+		if err != nil {
+			h.logger.Warn("get trace from storage: ", zap.Error(err))
+		}
 		if trace == nil {
 			var m tlb.Message
 			if err := tlb.Unmarshal(c, &m); err != nil {
@@ -438,7 +451,7 @@ func (h *Handler) EmulateMessageToEvent(ctx context.Context, request *oas.Emulat
 			if err != nil {
 				return nil, toError(http.StatusInternalServerError, err)
 			}
-			err = h.storage.SaveTraceWithState(hs, trace, []abi.MethodInvocation{}, 24*time.Hour)
+			err = h.storage.SaveTraceWithState(ctx, hs, trace, []abi.MethodInvocation{}, 24*time.Hour)
 			if err != nil {
 				fmt.Println("trace not saved: ", err)
 			}
@@ -468,7 +481,13 @@ func (h *Handler) EmulateMessageToTrace(ctx context.Context, request *oas.Emulat
 	trace, prs := h.mempoolEmulate.traces.Get(hash)
 	if !prs {
 		hs, err := c.HashString()
+		if err != nil {
+			return nil, toError(http.StatusBadRequest, err)
+		}
 		trace, _, err = h.storage.GetTraceWithState(ctx, hs)
+		if err != nil {
+			h.logger.Warn("get trace from storage: ", zap.Error(err))
+		}
 		if trace == nil {
 			var m tlb.Message
 			err = tlb.Unmarshal(c, &m)
@@ -499,7 +518,7 @@ func (h *Handler) EmulateMessageToTrace(ctx context.Context, request *oas.Emulat
 			if err != nil {
 				return nil, toError(http.StatusInternalServerError, err)
 			}
-			err = h.storage.SaveTraceWithState(hs, trace, []abi.MethodInvocation{}, 24*time.Hour)
+			err = h.storage.SaveTraceWithState(ctx, hs, trace, []abi.MethodInvocation{}, 24*time.Hour)
 			if err != nil {
 				fmt.Println("trace not saved: ", err)
 			}
@@ -591,7 +610,13 @@ func (h *Handler) EmulateMessageToWallet(ctx context.Context, request *oas.Emula
 	}
 
 	hash, err := msgCell.HashString()
+	if err != nil {
+		return nil, toError(http.StatusBadRequest, err)
+	}
 	trace, _, err := h.storage.GetTraceWithState(ctx, hash)
+	if err != nil {
+		h.logger.Warn("get trace from storage: ", zap.Error(err))
+	}
 	if trace == nil {
 		configBase64, err := h.storage.TrimmedConfigBase64()
 		if err != nil {
@@ -633,7 +658,7 @@ func (h *Handler) EmulateMessageToWallet(ctx context.Context, request *oas.Emula
 		if err != nil {
 			return nil, toError(http.StatusInternalServerError, err)
 		}
-		err = h.storage.SaveTraceWithState(hash, trace, []abi.MethodInvocation{}, 24*time.Hour)
+		err = h.storage.SaveTraceWithState(ctx, hash, trace, []abi.MethodInvocation{}, 24*time.Hour)
 		if err != nil {
 			fmt.Println("trace not saved: ", err)
 		}

pkg/api/interfaces.go

@@ -111,7 +111,7 @@ type storage interface {
 	GetAccountMultisigs(ctx context.Context, accountID ton.AccountID) ([]core.Multisig, error)
 	GetMultisigByID(ctx context.Context, accountID ton.AccountID) (*core.Multisig, error)
 
-	SaveTraceWithState(msgHash string, trace *core.Trace, getMethods []abi.MethodInvocation, ttl time.Duration) error
+	SaveTraceWithState(ctx context.Context, msgHash string, trace *core.Trace, getMethods []abi.MethodInvocation, ttl time.Duration) error
 	GetTraceWithState(ctx context.Context, msgHash string) (*core.Trace, []abi.MethodInvocation, error)
 
 	liteStorageRaw

pkg/litestorage/litestorage.go

@@ -566,3 +566,11 @@ func (s *LiteStorage) GetAccountMultisigs(ctx context.Context, accountID ton.Acc
 func (s *LiteStorage) GetMultisigByID(ctx context.Context, accountID ton.AccountID) (*core.Multisig, error) {
 	return nil, fmt.Errorf("not implemented")
 }
+
+func (s *LiteStorage) SaveTraceWithState(ctx context.Context, msgHash string, trace *core.Trace, getMethods []abi.MethodInvocation, ttl time.Duration) error {
+	return fmt.Errorf("not implemented")
+}
+
+func (s *LiteStorage) GetTraceWithState(ctx context.Context, msgHash string) (*core.Trace, []abi.MethodInvocation, error) {
+	return nil, nil, fmt.Errorf("not implemented")
+}