chore: add ActivityTrace for CurrencyAcount

limebell · Aug 8, 2024 · 40eaf98 · 40eaf98
1 parent 6db26e1
commit 40eaf98
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/Libplanet.Action/State/CurrencyAccount.cs b/src/Libplanet.Action/State/CurrencyAccount.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Diagnostics;
 using System.Numerics;
 using Bencodex.Types;
 using Libplanet.Crypto;
@@ -20,12 +21,14 @@ public sealed class CurrencyAccount
         /// </summary>
         public static readonly Address TotalSupplyAddress =
             new Address("1000000000000000000000000000000000000000");
+        private readonly ActivitySource _activitySource;
 
         public CurrencyAccount(ITrie trie, int worldVersion, Currency currency)
         {
             Trie = trie;
             WorldVersion = worldVersion;
             Currency = currency;
+            _activitySource = new ActivitySource("Libplanet.Action.State");
         }
 
         public ITrie Trie { get; }
@@ -36,6 +39,9 @@ public CurrencyAccount(ITrie trie, int worldVersion, Currency currency)
 
         public FungibleAssetValue GetBalance(Address address, Currency currency)
         {
+            using Activity? a = _activitySource
+                .StartActivity(ActivityKind.Internal)?
+                .AddTag("Address", address.ToString());
             CheckCurrency(currency);
 #pragma warning disable SA1118  // The parameter spans multiple lines
             return FungibleAssetValue.FromRawValue(
@@ -48,6 +54,9 @@ public FungibleAssetValue GetBalance(Address address, Currency currency)
 
         public FungibleAssetValue GetTotalSupply(Currency currency)
         {
+            using Activity? a = _activitySource
+                .StartActivity(ActivityKind.Internal)?
+                .AddTag("Currency", currency.ToString());
             CheckCurrency(currency);
 #pragma warning disable SA1118  // The parameter spans multiple lines
             return FungibleAssetValue.FromRawValue(
@@ -97,6 +106,11 @@ public CurrencyAccount TransferAsset(
             Address recipient,
             FungibleAssetValue value)
         {
+            using Activity? a = _activitySource
+                .StartActivity(ActivityKind.Internal)?
+                .AddTag("Sender", sender.ToString())
+                .AddTag("Recipient", recipient.ToString())
+                .AddTag("Value", value.ToString());
             CheckCurrency(value.Currency);
             if (value.Sign <= 0)
             {
@@ -118,6 +132,11 @@ public CurrencyAccount TransferAssetV0(
             Address recipient,
             FungibleAssetValue value)
         {
+            using Activity? a = _activitySource
+                .StartActivity(ActivityKind.Internal)?
+                .AddTag("Sender", sender.ToString())
+                .AddTag("Recipient", recipient.ToString())
+                .AddTag("Value", value.ToString());
             CheckCurrency(value.Currency);
             if (value.Sign <= 0)
             {