Fix: get last_action for contract from database

baking-bad · Jul 11, 2023 · 9c33c53 · 9c33c53
1 parent 8306993
commit 9c33c53
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/internal/postgres/operation/storage.go b/internal/postgres/operation/storage.go
@@ -440,11 +440,20 @@ func (storage *Storage) ContractStats(address string) (stats operation.ContractS
 		return
 	}
 
-	var result lastTimestampResult
-	if _, err := storage.DB.QueryOne(&result, cteContractStatsTemplate, accountID, accountID); err != nil {
-		return stats, err
+	if bcd.IsContractLazy(address) {
+		if err := storage.DB.Model((*contract.Contract)(nil)).
+			Column("last_action").
+			Where("account_id = ?", accountID).
+			Select(&stats.LastAction); err != nil {
+			return stats, err
+		}
+	} else {
+		var result lastTimestampResult
+		if _, err := storage.DB.QueryOne(&result, cteContractStatsTemplate, accountID, accountID); err != nil {
+			return stats, err
+		}
+		stats.LastAction = result.Timestamp
 	}
-	stats.LastAction = result.Timestamp
 
 	count, err := storage.DB.Model((*operation.Operation)(nil)).WhereGroup(
 		func(q *orm.Query) (*orm.Query, error) {