com.hedera.node.app.hapi.fees.usage.SingletonEstimatorUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of app-hapi-fees Show documentation
Hedera Services API Fees
There is a newer version: 0.57.2
/*
 * Copyright (C) 2020-2024 Hedera Hashgraph, LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.hedera.node.app.hapi.fees.usage;

import static com.hedera.node.app.hapi.utils.fee.FeeBuilder.BASIC_ACCOUNT_AMT_SIZE;
import static com.hedera.node.app.hapi.utils.fee.FeeBuilder.BASIC_RECEIPT_SIZE;
import static com.hedera.node.app.hapi.utils.fee.FeeBuilder.BASIC_TX_RECORD_SIZE;
import static com.hedera.node.app.hapi.utils.fee.FeeBuilder.FEE_MATRICES_CONST;
import static com.hedera.node.app.hapi.utils.fee.FeeBuilder.INT_SIZE;
import static com.hedera.node.app.hapi.utils.fee.FeeBuilder.RECEIPT_STORAGE_TIME_SEC;

import com.hederahashgraph.api.proto.java.FeeComponents;
import com.hederahashgraph.api.proto.java.FeeData;
import com.hederahashgraph.api.proto.java.SubType;
import com.hederahashgraph.api.proto.java.TransactionBody;
import com.hederahashgraph.api.proto.java.TransferList;

public enum SingletonEstimatorUtils implements EstimatorUtils {
    ESTIMATOR_UTILS;

    @Override
    public long baseNetworkRbs() {
        return BASIC_RECEIPT_SIZE * RECEIPT_STORAGE_TIME_SEC;
    }

    @Override
    public UsageEstimate baseEstimate(TransactionBody txn, SigUsage sigUsage) {
        var base = FeeComponents.newBuilder()
                .setBpr(INT_SIZE)
                .setVpt(sigUsage.numSigs())
                .setBpt(baseBodyBytes(txn) + sigUsage.sigsSize());
        var estimate = new UsageEstimate(base);
        estimate.addRbs(baseRecordBytes(txn) * RECEIPT_STORAGE_TIME_SEC);
        return estimate;
    }

    public FeeData withDefaultTxnPartitioning(FeeComponents usage, SubType subType, long networkRbh, int numPayerKeys) {
        var usages = FeeData.newBuilder();

        var network = FeeComponents.newBuilder()
                .setConstant(FEE_MATRICES_CONST)
                .setBpt(usage.getBpt())
                .setVpt(usage.getVpt())
                .setRbh(networkRbh);
        var node = FeeComponents.newBuilder()
                .setConstant(FEE_MATRICES_CONST)
                .setBpt(usage.getBpt())
                .setVpt(numPayerKeys)
                .setBpr(usage.getBpr())
                .setSbpr(usage.getSbpr());
        var service = FeeComponents.newBuilder()
                .setConstant(FEE_MATRICES_CONST)
                .setRbh(usage.getRbh())
                .setSbh(usage.getSbh())
                .setTv(usage.getTv());
        return usages.setNetworkdata(network)
                .setNodedata(node)
                .setServicedata(service)
                .setSubType(subType)
                .build();
    }

    @Override
    public FeeData withDefaultQueryPartitioning(FeeComponents usage) {
        var usages = FeeData.newBuilder();
        var node = FeeComponents.newBuilder()
                .setConstant(FEE_MATRICES_CONST)
                .setBpt(usage.getBpt())
                .setBpr(usage.getBpr())
                .setSbpr(usage.getSbpr());
        return usages.setNodedata(node).build();
    }

    int baseRecordBytes(TransactionBody txn) {
        return BASIC_TX_RECORD_SIZE
                + txn.getMemoBytes().size()
                + transferListBytes(txn.getCryptoTransfer().getTransfers());
    }

    private int transferListBytes(TransferList transfers) {
        return BASIC_ACCOUNT_AMT_SIZE * transfers.getAccountAmountsCount();
    }
}