chore: Use dispatch's FeeCharging strategy in HandleContext

hedera-node/hedera-app/src/main/java/com/hedera/node/app/fees/AppFeeCharging.java

@@ -49,7 +49,7 @@ public ValidationResult validate(
             @NonNull final AccountID creatorId,
             @NonNull final Fees fees,
             @NonNull final TransactionBody body,
-            boolean isDuplicate,
+            final boolean isDuplicate,
             @NonNull final HederaFunctionality function,
             @NonNull final HandleContext.TransactionCategory category) {
         requireNonNull(payer);

hedera-node/hedera-app/src/main/java/com/hedera/node/app/workflows/handle/DispatchHandleContext.java

@@ -5,6 +5,7 @@
 import static com.hedera.hapi.util.HapiUtils.functionOf;
 import static com.hedera.node.app.spi.workflows.HandleContext.DispatchMetadata.Type.INNER_TRANSACTION_BYTES;
 import static com.hedera.node.app.spi.workflows.HandleContext.TransactionCategory.BATCH_INNER;
+import static com.hedera.node.app.spi.workflows.HandleContext.TransactionCategory.CHILD;
 import static com.hedera.node.app.workflows.handle.stack.SavepointStackImpl.castBuilder;
 import static java.util.Collections.emptyMap;
 import static java.util.Objects.requireNonNull;
@@ -31,6 +32,7 @@
 import com.hedera.node.app.spi.fees.ExchangeRateInfo;
 import com.hedera.node.app.spi.fees.FeeCalculator;
 import com.hedera.node.app.spi.fees.FeeCalculatorFactory;
+import com.hedera.node.app.spi.fees.FeeCharging;
 import com.hedera.node.app.spi.fees.FeeContext;
 import com.hedera.node.app.spi.fees.Fees;
 import com.hedera.node.app.spi.fees.ResourcePriceCalculator;
@@ -55,6 +57,7 @@
 import com.hedera.node.app.workflows.TransactionInfo;
 import com.hedera.node.app.workflows.dispatcher.TransactionDispatcher;
 import com.hedera.node.app.workflows.handle.dispatch.ChildDispatchFactory;
+import com.hedera.node.app.workflows.handle.dispatch.ValidationResult;
 import com.hedera.node.app.workflows.handle.stack.SavepointStackImpl;
 import com.hedera.node.app.workflows.handle.validation.AttributeValidatorImpl;
 import com.hedera.node.app.workflows.handle.validation.ExpiryValidatorImpl;
@@ -70,11 +73,12 @@
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.function.ObjLongConsumer;
 
 /**
  * The {@link HandleContext} implementation.
  */
-public class DispatchHandleContext implements HandleContext, FeeContext {
+public class DispatchHandleContext implements HandleContext, FeeContext, FeeCharging.Context {
     private final Instant consensusNow;
     private final NodeInfo creatorInfo;
     private final TransactionInfo txnInfo;
@@ -83,6 +87,7 @@
     private final BlockRecordInfo blockRecordInfo;
     private final ResourcePriceCalculator resourcePriceCalculator;
     private final FeeManager feeManager;
+    private final FeeCharging feeCharging;
     private final StoreFactoryImpl storeFactory;
     private final AccountID payerId;
     private final AppKeyVerifier verifier;
@@ -99,11 +104,13 @@
     private final DispatchProcessor dispatchProcessor;
     private final ThrottleAdviser throttleAdviser;
     private final FeeAccumulator feeAccumulator;
-    private Map<AccountID, Long> dispatchPaidRewards;
     private final DispatchMetadata dispatchMetaData;
     private final TransactionChecker transactionChecker;
     private final TransactionCategory transactionCategory;
 
+    @Nullable
+    private Map<AccountID, Long> dispatchPaidRewards;
+
     // This is used to store the pre-handle results for the inner transactions
     // in an atomic batch, null otherwise
     @Nullable
@@ -121,6 +128,7 @@
             @NonNull final BlockRecordInfo blockRecordInfo,
             @NonNull final ResourcePriceCalculator resourcePriceCalculator,
             @NonNull final FeeManager feeManager,
+            @NonNull final FeeCharging feeCharging,
             @NonNull final StoreFactoryImpl storeFactory,
             @NonNull final AccountID payerId,
             @NonNull final AppKeyVerifier verifier,
@@ -148,6 +156,7 @@
         this.blockRecordInfo = requireNonNull(blockRecordInfo);
         this.resourcePriceCalculator = requireNonNull(resourcePriceCalculator);
         this.feeManager = requireNonNull(feeManager);
+        this.feeCharging = requireNonNull(feeCharging);
         this.storeFactory = requireNonNull(storeFactory);
         this.payerId = requireNonNull(payerId);
         this.verifier = requireNonNull(verifier);
@@ -195,16 +204,19 @@
         if (amount < 0) {
             throw new IllegalArgumentException("Cannot charge negative amount " + amount);
         }
-        return feeAccumulator.chargeFee(accountId, amount, null).networkFee() == amount;
+        return feeCharging
+                        .charge(this, ValidationResult.newSuccess(creatorInfo.accountId()), new Fees(0, amount, 0))
+                        .totalFee()
+                == amount;
     }
 
     @Override
     public void refundBestEffort(@NonNull final AccountID accountId, final long amount) {
         requireNonNull(accountId);
         if (amount < 0) {
-            throw new IllegalArgumentException("Cannot charge negative amount " + amount);
+            throw new IllegalArgumentException("Cannot refund negative amount " + amount);
         }
-        feeAccumulator.refundFee(accountId, amount);
+        feeCharging.refund(this, new Fees(0, amount, 0));
     }
 
     @NonNull
@@ -471,4 +483,48 @@
     public DispatchMetadata dispatchMetadata() {
         return dispatchMetaData;
     }
+
+    @Override
+    public AccountID payerId() {
+        return payerId;
+    }
+
+    @Override
+    public AccountID nodeAccountId() {
+        return creatorInfo.accountId();
+    }
+
+    @Override
+    public Fees charge(
+            @NonNull final AccountID payerId, @NonNull final Fees fees, @Nullable final ObjLongConsumer<AccountID> cb) {
+        return feeAccumulator.chargeFee(payerId, fees.totalFee(), cb);
+    }
+
+    @Override
+    public void refund(@NonNull final AccountID receiverId, @NonNull final Fees fees) {
+        feeAccumulator.refundFee(receiverId, fees.totalFee());
+    }
+
+    @Override
+    public Fees charge(
+            @NonNull final AccountID payerId,
+            @NonNull final Fees fees,
+            @NonNull final AccountID nodeAccountId,
+            @Nullable final ObjLongConsumer<AccountID> cb) {
+        return feeAccumulator.chargeFees(payerId, nodeAccountId, fees, cb);
+    }
+
+    @Override
+    public void refund(
+            @NonNull final AccountID payerId, @NonNull final Fees fees, @NonNull final AccountID nodeAccountId) {
+        feeAccumulator.refundFees(payerId, fees, nodeAccountId);
+    }
+
+    @Override
+    public TransactionCategory category() {
+        // When the DispatchHandleContext is used as a fee charging context, always report
+        // CHILD category to stay backward compatible with the calls made to FeeAccumulator
+        // when it was invoked directly
+        return CHILD;
+    }
 }

hedera-node/hedera-app/src/main/java/com/hedera/node/app/workflows/handle/dispatch/ChildDispatchFactory.java

@@ -26,6 +26,7 @@
 import com.hedera.hapi.node.transaction.SignedTransaction;
 import com.hedera.hapi.node.transaction.TransactionBody;
 import com.hedera.hapi.util.UnknownHederaFunctionality;
+import com.hedera.node.app.fees.AppFeeCharging;
 import com.hedera.node.app.fees.ExchangeRateManager;
 import com.hedera.node.app.fees.FeeAccumulator;
 import com.hedera.node.app.fees.FeeManager;
@@ -104,6 +105,7 @@ public class ChildDispatchFactory {
     private final Authorizer authorizer;
     private final NetworkInfo networkInfo;
     private final FeeManager feeManager;
+    private final AppFeeCharging appFeeCharging;
     private final DispatchProcessor dispatchProcessor;
     private final ServiceScopeLookup serviceScopeLookup;
     private final ExchangeRateManager exchangeRateManager;
@@ -116,6 +118,7 @@ public ChildDispatchFactory(
             @NonNull final Authorizer authorizer,
             @NonNull final NetworkInfo networkInfo,
             @NonNull final FeeManager feeManager,
+            @NonNull final AppFeeCharging appFeeCharging,
             @NonNull final DispatchProcessor dispatchProcessor,
             @NonNull final ServiceScopeLookup serviceScopeLookup,
             @NonNull final ExchangeRateManager exchangeRateManager,
@@ -125,6 +128,7 @@ public ChildDispatchFactory(
         this.authorizer = requireNonNull(authorizer);
         this.networkInfo = requireNonNull(networkInfo);
         this.feeManager = requireNonNull(feeManager);
+        this.appFeeCharging = requireNonNull(appFeeCharging);
         this.dispatchProcessor = requireNonNull(dispatchProcessor);
         this.serviceScopeLookup = requireNonNull(serviceScopeLookup);
         this.exchangeRateManager = requireNonNull(exchangeRateManager);
@@ -239,7 +243,7 @@ private RecordDispatch newChildDispatch(
             @NonNull final Instant consensusNow,
             @NonNull final DispatchMetadata dispatchMetadata,
             @NonNull final ConsensusThrottling consensusThrottling,
-            @Nullable FeeCharging customFeeCharging,
+            @Nullable final FeeCharging customFeeCharging,
             // @UserTxnScope
             @NonNull final NodeInfo creatorInfo,
             @NonNull final Configuration config,
@@ -266,6 +270,7 @@ private RecordDispatch newChildDispatch(
         final var storeFactory = new StoreFactoryImpl(readableStoreFactory, writableStoreFactory, serviceApiFactory);
         final var childFeeAccumulator = new FeeAccumulator(
                 serviceApiFactory.getApi(TokenServiceApi.class), (FeeStreamBuilder) builder, childStack);
+        final var feeCharging = customFeeCharging != null ? customFeeCharging : appFeeCharging;
         final var dispatchHandleContext = new DispatchHandleContext(
                 consensusNow,
                 creatorInfo,
@@ -275,6 +280,7 @@ private RecordDispatch newChildDispatch(
                 blockRecordInfo,
                 priceCalculator,
                 feeManager,
+                feeCharging,
                 storeFactory,
                 payerId,
                 keyVerifier,

hedera-node/hedera-app/src/main/java/com/hedera/node/app/workflows/handle/dispatch/ValidationResult.java

@@ -135,6 +135,17 @@ public static ValidationResult newSuccess(@NonNull final AccountID creatorId, @N
         return new ValidationResult(creatorId, null, payer, null, CAN_PAY_SERVICE_FEE, NO_DUPLICATE);
     }
 
+    /**
+     * Creates an error report with no error.
+     * @param creatorId the creator account ID
+     * @return the error report
+     */
+    @NonNull
+    public static ValidationResult newSuccess(@NonNull final AccountID creatorId) {
+        requireNonNull(creatorId);
+        return new ValidationResult(creatorId, null, null, null, CAN_PAY_SERVICE_FEE, NO_DUPLICATE);
+    }
+
     /**
      * Creates an error report with no error.
      * @param creatorId the creator account ID
@@ -187,15 +198,6 @@ public ResponseCodeEnum creatorErrorOrThrow() {
         return requireNonNull(creatorError);
     }
 
-    /**
-     * Checks if there is a payer.
-     * @return payer account if there is a payer. Otherwise, throws an exception.
-     */
-    @NonNull
-    public Account payerOrThrow() {
-        return requireNonNull(payer);
-    }
-
     /**
      * Returns the error report with all fees except service fee.
      * @return the error report

hedera-node/hedera-app/src/main/java/com/hedera/node/app/workflows/handle/steps/ParentTxnFactory.java

@@ -24,6 +24,7 @@
 import com.hedera.node.app.blocks.BlockStreamManager;
 import com.hedera.node.app.blocks.impl.BoundaryStateChangeListener;
 import com.hedera.node.app.blocks.impl.ImmediateStateChangeListener;
+import com.hedera.node.app.fees.AppFeeCharging;
 import com.hedera.node.app.fees.ExchangeRateManager;
 import com.hedera.node.app.fees.FeeAccumulator;
 import com.hedera.node.app.fees.FeeManager;
@@ -96,6 +97,7 @@ public class ParentTxnFactory {
     private final Authorizer authorizer;
     private final NetworkInfo networkInfo;
     private final FeeManager feeManager;
+    private final AppFeeCharging appFeeCharging;
     private final DispatchProcessor dispatchProcessor;
     private final ServiceScopeLookup serviceScopeLookup;
     private final ExchangeRateManager exchangeRateManager;
@@ -117,6 +119,7 @@ public ParentTxnFactory(
             @NonNull final Authorizer authorizer,
             @NonNull final NetworkInfo networkInfo,
             @NonNull final FeeManager feeManager,
+            @NonNull final AppFeeCharging appFeeCharging,
             @NonNull final DispatchProcessor dispatchProcessor,
             @NonNull final ServiceScopeLookup serviceScopeLookup,
             @NonNull final ExchangeRateManager exchangeRateManager,
@@ -134,6 +137,7 @@ public ParentTxnFactory(
         this.authorizer = requireNonNull(authorizer);
         this.networkInfo = requireNonNull(networkInfo);
         this.feeManager = requireNonNull(feeManager);
+        this.appFeeCharging = requireNonNull(appFeeCharging);
         this.dispatcher = requireNonNull(dispatcher);
         this.networkUtilizationManager = requireNonNull(networkUtilizationManager);
         this.dispatchProcessor = requireNonNull(dispatchProcessor);
@@ -326,7 +330,6 @@ private Dispatch createDispatch(
         final var throttleAdvisor = new AppThrottleAdviser(networkUtilizationManager, consensusNow);
         final var feeAccumulator = new FeeAccumulator(
                 serviceApiFactory.getApi(TokenServiceApi.class), (FeeStreamBuilder) baseBuilder, stack);
-
         final var dispatchHandleContext = new DispatchHandleContext(
                 consensusNow,
                 creatorInfo,
@@ -336,6 +339,7 @@ private Dispatch createDispatch(
                 streamMode != RECORDS ? blockStreamManager : blockRecordManager,
                 priceCalculator,
                 feeManager,
+                appFeeCharging,
                 storeFactory,
                 requireNonNull(txnInfo.payerID()),
                 keyVerifier,

hedera-node/hedera-app/src/main/java/com/hedera/node/app/workflows/standalone/impl/StandaloneDispatchFactory.java

@@ -15,6 +15,7 @@
 import com.hedera.hapi.node.transaction.TransactionBody;
 import com.hedera.node.app.blocks.impl.BoundaryStateChangeListener;
 import com.hedera.node.app.blocks.impl.ImmediateStateChangeListener;
+import com.hedera.node.app.fees.AppFeeCharging;
 import com.hedera.node.app.fees.ExchangeRateManager;
 import com.hedera.node.app.fees.FeeAccumulator;
 import com.hedera.node.app.fees.FeeManager;
@@ -75,6 +76,7 @@
 @Singleton
 public class StandaloneDispatchFactory {
     private final FeeManager feeManager;
+    private final AppFeeCharging appFeeCharging;
     private final Authorizer authorizer;
     private final NetworkInfo networkInfo;
     private final ConfigProvider configProvider;
@@ -92,6 +94,7 @@ public class StandaloneDispatchFactory {
     @Inject
     public StandaloneDispatchFactory(
             @NonNull final FeeManager feeManager,
+            @NonNull final AppFeeCharging appFeeCharging,
             @NonNull final Authorizer authorizer,
             @NonNull final NetworkInfo networkInfo,
             @NonNull final ConfigProvider configProvider,
@@ -106,6 +109,7 @@ public StandaloneDispatchFactory(
             @NonNull final TransactionChecker transactionChecker,
             @NonNull final ScheduleServiceImpl scheduleService) {
         this.feeManager = requireNonNull(feeManager);
+        this.appFeeCharging = requireNonNull(appFeeCharging);
         this.authorizer = requireNonNull(authorizer);
         this.networkInfo = requireNonNull(networkInfo);
         this.configProvider = requireNonNull(configProvider);
@@ -181,6 +185,7 @@ public Dispatch newDispatch(
                 blockRecordInfo,
                 priceCalculator,
                 feeManager,
+                appFeeCharging,
                 storeFactory,
                 requireNonNull(txnInfo.payerID()),
                 NO_OP_KEY_VERIFIER,