fix comments

Author: erokhinav

pkg/api/event_handlers.go

@@ -392,7 +392,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 {
@@ -418,7 +424,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)
 		}
@@ -446,8 +452,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 {
@@ -477,7 +489,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)
 			}
@@ -507,7 +519,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)
@@ -538,7 +556,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)
 			}
@@ -630,7 +648,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 {
@@ -672,7 +696,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

@@ -112,7 +112,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")
+}