Examples with AccountCapsule org.tron.core.capsule.AccountCapsule used on opensource projects

Example 6 with AccountCapsule

use of org.tron.core.capsule.AccountCapsule in project java-tron by tronprotocol.

the class VoteWitnessActuator method countVoteAccount.

private void countVoteAccount(VoteWitnessContract voteContract) {
    AccountCapsule accountCapsule = dbManager.getAccountStore().get(voteContract.getOwnerAddress().toByteArray());
    accountCapsule.setInstance(accountCapsule.getInstance().toBuilder().clearVotes().build());
    voteContract.getVotesList().forEach(vote -> {
        // String toStringUtf8 = vote.getVoteAddress().toStringUtf8();
        logger.debug("countVoteAccount,address[{}]", ByteArray.toHexString(vote.getVoteAddress().toByteArray()));
        accountCapsule.addVotes(vote.getVoteAddress(), vote.getVoteCount());
    });
    dbManager.getAccountStore().put(accountCapsule.createDbKey(), accountCapsule);
}

Example 7 with AccountCapsule

use of org.tron.core.capsule.AccountCapsule in project java-tron by tronprotocol.

the class Wallet method getBalance.

public Account getBalance(Account account) {
    AccountStore accountStore = dbManager.getAccountStore();
    AccountCapsule accountCapsule = accountStore.get(account.getAddress().toByteArray());
    return accountCapsule == null ? null : accountCapsule.getInstance();
}

Example 8 with AccountCapsule

use of org.tron.core.capsule.AccountCapsule in project java-tron by tronprotocol.

the class AssetIssueActuator method execute.

@Override
public boolean execute(TransactionResultCapsule ret) throws ContractExeException {
    long fee = calcFee();
    try {
        if (!this.contract.is(AssetIssueContract.class)) {
            throw new ContractExeException();
        }
        if (dbManager == null) {
            throw new ContractExeException();
        }
        AssetIssueContract assetIssueContract = contract.unpack(AssetIssueContract.class);
        AssetIssueCapsule assetIssueCapsule = new AssetIssueCapsule(assetIssueContract);
        dbManager.getAssetIssueStore().put(assetIssueCapsule.getName().toByteArray(), assetIssueCapsule);
        dbManager.adjustBalance(assetIssueContract.getOwnerAddress().toByteArray(), -calcFee());
        ret.setStatus(fee, code.SUCESS);
        AccountCapsule accountCapsule = dbManager.getAccountStore().get(assetIssueContract.getOwnerAddress().toByteArray());
        accountCapsule.addAsset(ByteArray.toStr(assetIssueContract.getName().toByteArray()), assetIssueContract.getTotalSupply());
        dbManager.getAccountStore().put(assetIssueContract.getOwnerAddress().toByteArray(), accountCapsule);
    } catch (InvalidProtocolBufferException e) {
        ret.setStatus(fee, code.FAILED);
        throw new ContractExeException();
    } catch (BalanceInsufficientException e) {
        ret.setStatus(fee, code.FAILED);
        throw new ContractExeException();
    }
    return true;
}

Example 9 with AccountCapsule

use of org.tron.core.capsule.AccountCapsule in project java-tron by tronprotocol.

the class ParticipateAssetIssueActuator method execute.

@Override
public boolean execute(TransactionResultCapsule ret) throws ContractExeException {
    long fee = calcFee();
    try {
        Contract.ParticipateAssetIssueContract participateAssetIssueContract = contract.unpack(Contract.ParticipateAssetIssueContract.class);
        long cost = participateAssetIssueContract.getAmount();
        // subtract from owner address
        byte[] ownerAddressBytes = participateAssetIssueContract.getOwnerAddress().toByteArray();
        AccountCapsule ownerAccount = this.dbManager.getAccountStore().get(ownerAddressBytes);
        ownerAccount.setBalance(ownerAccount.getBalance() - cost - fee);
        // calculate the exchange amount
        AssetIssueCapsule assetIssueCapsule = this.dbManager.getAssetIssueStore().get(participateAssetIssueContract.getAssetName().toByteArray());
        long exchangeAmount = cost * assetIssueCapsule.getNum() / assetIssueCapsule.getTrxNum();
        ownerAccount.addAssetAmount(assetIssueCapsule.getName(), exchangeAmount);
        // add to to_address
        byte[] toAddressBytes = participateAssetIssueContract.getToAddress().toByteArray();
        AccountCapsule toAccount = this.dbManager.getAccountStore().get(toAddressBytes);
        toAccount.setBalance(toAccount.getBalance() + cost);
        if (!toAccount.reduceAssetAmount(assetIssueCapsule.getName(), exchangeAmount)) {
            throw new ContractExeException("reduceAssetAmount failed !");
        }
        // write to db
        dbManager.getAccountStore().put(ownerAddressBytes, ownerAccount);
        dbManager.getAccountStore().put(toAddressBytes, toAccount);
        ret.setStatus(fee, Protocol.Transaction.Result.code.SUCESS);
        return true;
    } catch (InvalidProtocolBufferException e) {
        ret.setStatus(fee, Protocol.Transaction.Result.code.FAILED);
        logger.debug(e.getMessage(), e);
        throw new ContractExeException(e.getMessage());
    }
}

Example 10 with AccountCapsule

use of org.tron.core.capsule.AccountCapsule in project java-tron by tronprotocol.

the class ParticipateAssetIssueActuator method validate.

@Override
public boolean validate() throws ContractValidateException {
    if (!this.contract.is(Contract.ParticipateAssetIssueContract.class)) {
        throw new ContractValidateException();
    }
    try {
        final Contract.ParticipateAssetIssueContract participateAssetIssueContract = this.contract.unpack(Contract.ParticipateAssetIssueContract.class);
        Preconditions.checkNotNull(participateAssetIssueContract.getOwnerAddress(), "OwnerAddress is null");
        Preconditions.checkNotNull(participateAssetIssueContract.getToAddress(), "ToAddress is null");
        Preconditions.checkNotNull(participateAssetIssueContract.getAssetName(), "trx name is null");
        if (participateAssetIssueContract.getAmount() <= 0) {
            throw new ContractValidateException("Trx Num must be positive!");
        }
        if (participateAssetIssueContract.getOwnerAddress().equals(participateAssetIssueContract.getToAddress())) {
            throw new ContractValidateException("Cannot participate asset Issue yourself !");
        }
        byte[] addressBytes = participateAssetIssueContract.getOwnerAddress().toByteArray();
        // Whether the account exist
        if (!this.dbManager.getAccountStore().has(addressBytes)) {
            throw new ContractValidateException("Account does not exist!");
        }
        AccountCapsule ac = this.dbManager.getAccountStore().get(addressBytes);
        long fee = calcFee();
        // Whether the balance is enough
        if (ac.getBalance() < participateAssetIssueContract.getAmount() + fee) {
            throw new ContractValidateException("No enough balance !");
        }
        // Whether have the mapping
        if (!this.dbManager.getAssetIssueStore().has(participateAssetIssueContract.getAssetName().toByteArray())) {
            throw new ContractValidateException("No asset named " + ByteArray.toStr(participateAssetIssueContract.getAssetName().toByteArray()));
        }
        AssetIssueCapsule assetIssueCapsule = this.dbManager.getAssetIssueStore().get(participateAssetIssueContract.getAssetName().toByteArray());
        if (!participateAssetIssueContract.getToAddress().equals(assetIssueCapsule.getOwnerAddress())) {
            throw new ContractValidateException("The asset is not issued by " + ByteArray.toHexString(participateAssetIssueContract.getToAddress().toByteArray()));
        }
        // Whether the exchange can be processed: to see if the exchange can be the exact int
        long cost = participateAssetIssueContract.getAmount();
        DateTime now = DateTime.now();
        if (now.getMillis() >= assetIssueCapsule.getEndTime() || now.getMillis() < assetIssueCapsule.getStartTime()) {
            throw new ContractValidateException("No longer valid period!");
        }
        int trxNum = assetIssueCapsule.getTrxNum();
        int num = assetIssueCapsule.getNum();
        long exchangeAmount = cost * num / trxNum;
        if (exchangeAmount == 0) {
            throw new ContractValidateException("Can not process the exchange!");
        }
        AccountCapsule toAccount = this.dbManager.getAccountStore().get(participateAssetIssueContract.getToAddress().toByteArray());
        if (!toAccount.assetBalanceEnough(assetIssueCapsule.getName(), exchangeAmount)) {
            throw new ContractValidateException("Asset balance is not enough !");
        }
    } catch (InvalidProtocolBufferException e) {
        throw new ContractValidateException();
    }
    return true;
}