com.hedera.node.app.service.token.impl.schemas.V0530TokenSchema Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of app-service-token-impl Show documentation

Default Hedera Token Service Implementation

There is a newer version: 0.57.3

Show newest version

/*
 * Copyright (C) 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.service.token.impl.schemas;

import static com.hedera.node.app.service.token.impl.schemas.V0490TokenSchema.STAKING_INFO_KEY;

import com.hedera.hapi.node.base.PendingAirdropId;
import com.hedera.hapi.node.base.SemanticVersion;
import com.hedera.hapi.node.state.common.EntityNumber;
import com.hedera.hapi.node.state.token.AccountPendingAirdrop;
import com.hedera.hapi.node.state.token.StakingNodeInfo;
import com.swirlds.state.spi.MigrationContext;
import com.swirlds.state.spi.StateDefinition;
import edu.umd.cs.findbugs.annotations.NonNull;
import java.util.Set;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

public class V0530TokenSchema extends StakingInfoManagementSchema {
    private static final Logger logger = LogManager.getLogger(V0530TokenSchema.class);
    private static final long MAX_PENDING_AIRDROPS = 1_000_000_000L;
    public static final String AIRDROPS_KEY = "PENDING_AIRDROPS";

    private static final SemanticVersion VERSION =
            SemanticVersion.newBuilder().major(0).minor(53).patch(0).build();

    public V0530TokenSchema() {
        super(VERSION);
    }

    @SuppressWarnings("rawtypes")
    @NonNull
    @Override
    public Set statesToCreate() {
        return Set.of(StateDefinition.onDisk(
                AIRDROPS_KEY, PendingAirdropId.PROTOBUF, AccountPendingAirdrop.PROTOBUF, MAX_PENDING_AIRDROPS));
    }

    @Override
    public void migrate(@NonNull final MigrationContext ctx) {
        setMinStakeToZero(ctx);
    }

    private void setMinStakeToZero(final MigrationContext ctx) {
        final var stakingInfoState = ctx.newStates().get(STAKING_INFO_KEY);
        final var addressBook = ctx.networkInfo().addressBook();
        logger.info("Setting minStake to 0 for all nodes in the address book");
        for (final var node : addressBook) {
            final var nodeNumber =
                    EntityNumber.newBuilder().number(node.nodeId()).build();
            final var stakingInfo = (StakingNodeInfo) stakingInfoState.get(nodeNumber);
            if (stakingInfo != null) {
                stakingInfoState.put(
                        nodeNumber, stakingInfo.copyBuilder().minStake(0).build());
            }
        }
    }
}