com.hedera.node.app.hapi.fees.usage.QueryUsage 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.fees.usage.SingletonEstimatorUtils.ESTIMATOR_UTILS;
import static com.hedera.node.app.hapi.utils.fee.FeeBuilder.BASIC_QUERY_HEADER;
import static com.hedera.node.app.hapi.utils.fee.FeeBuilder.BASIC_QUERY_RES_HEADER;

import com.hederahashgraph.api.proto.java.FeeComponents;
import com.hederahashgraph.api.proto.java.FeeData;
import com.hederahashgraph.api.proto.java.ResponseType;

public class QueryUsage {
    private long sb = 0;
    private long tb = BASIC_QUERY_HEADER;
    private long rb = BASIC_QUERY_RES_HEADER;

    /* Once state proofs are supported, this will be needed to compute {@code rb}. */
    // Suppressing the warning that the field is not used
    @SuppressWarnings("java:S1068")
    private final ResponseType responseType;

    public QueryUsage(final ResponseType responseType) {
        this.responseType = responseType;
    }

    public FeeData get() {
        final var usage =
                FeeComponents.newBuilder().setBpt(tb).setBpr(rb).setSbpr(sb).build();
        return ESTIMATOR_UTILS.withDefaultQueryPartitioning(usage);
    }

    public void addRb(final long amount) {
        rb += amount;
    }

    public void addTb(final long amount) {
        tb += amount;
    }

    public void addSb(final long amount) {
        sb += amount;
    }
}