Search in sources :

Example 1 with AssetIssueCapsule

use of org.tron.core.capsule.AssetIssueCapsule 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;
}
Also used : AccountCapsule(org.tron.core.capsule.AccountCapsule) BalanceInsufficientException(org.tron.core.exception.BalanceInsufficientException) InvalidProtocolBufferException(com.google.protobuf.InvalidProtocolBufferException) AssetIssueCapsule(org.tron.core.capsule.AssetIssueCapsule) AssetIssueContract(org.tron.protos.Contract.AssetIssueContract) ContractExeException(org.tron.core.exception.ContractExeException)

Example 2 with AssetIssueCapsule

use of org.tron.core.capsule.AssetIssueCapsule 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());
    }
}
Also used : AccountCapsule(org.tron.core.capsule.AccountCapsule) InvalidProtocolBufferException(com.google.protobuf.InvalidProtocolBufferException) AssetIssueCapsule(org.tron.core.capsule.AssetIssueCapsule) Contract(org.tron.protos.Contract) ContractExeException(org.tron.core.exception.ContractExeException)

Example 3 with AssetIssueCapsule

use of org.tron.core.capsule.AssetIssueCapsule 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;
}
Also used : AccountCapsule(org.tron.core.capsule.AccountCapsule) ContractValidateException(org.tron.core.exception.ContractValidateException) InvalidProtocolBufferException(com.google.protobuf.InvalidProtocolBufferException) AssetIssueCapsule(org.tron.core.capsule.AssetIssueCapsule) Contract(org.tron.protos.Contract) DateTime(org.joda.time.DateTime)

Example 4 with AssetIssueCapsule

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

the class AssetIssueActuatorTest method rightAssetIssue.

@Test
public void rightAssetIssue() {
    AssetIssueActuator actuator = new AssetIssueActuator(getContract(), dbManager);
    TransactionResultCapsule ret = new TransactionResultCapsule();
    try {
        actuator.validate();
        actuator.execute(ret);
        Assert.assertEquals(ret.getInstance().getRet(), code.SUCESS);
        AccountCapsule owner = dbManager.getAccountStore().get(ByteArray.fromHexString(OWNER_ADDRESS));
        AssetIssueCapsule assetIssueCapsule = dbManager.getAssetIssueStore().get(ByteString.copyFromUtf8(NAME).toByteArray());
        Assert.assertNotNull(assetIssueCapsule);
        Assert.assertEquals(owner.getBalance(), 0L);
        Assert.assertEquals(owner.getAssetMap().get(NAME).longValue(), 10000L);
    } catch (ContractValidateException e) {
        Assert.assertFalse(e instanceof ContractValidateException);
    } catch (ContractExeException e) {
        Assert.assertFalse(e instanceof ContractExeException);
    } finally {
        dbManager.getAssetIssueStore().delete(ByteArray.fromString(NAME));
    }
}
Also used : AccountCapsule(org.tron.core.capsule.AccountCapsule) TransactionResultCapsule(org.tron.core.capsule.TransactionResultCapsule) ContractValidateException(org.tron.core.exception.ContractValidateException) AssetIssueCapsule(org.tron.core.capsule.AssetIssueCapsule) ContractExeException(org.tron.core.exception.ContractExeException) Test(org.junit.Test)

Example 5 with AssetIssueCapsule

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

the class AssetIssueActuatorTest method repeatAssetIssue.

@Test
public void repeatAssetIssue() {
    AssetIssueActuator actuator = new AssetIssueActuator(getContract(), dbManager);
    TransactionResultCapsule ret = new TransactionResultCapsule();
    try {
        dbManager.getAssetIssueStore().put(ByteArray.fromString(NAME), new AssetIssueCapsule(getContract().unpack(Contract.AssetIssueContract.class)));
        actuator.validate();
        actuator.execute(ret);
        Assert.assertTrue(false);
    } catch (ContractValidateException e) {
        Assert.assertTrue(e instanceof ContractValidateException);
        AccountCapsule owner = dbManager.getAccountStore().get(ByteArray.fromHexString(OWNER_ADDRESS));
        AssetIssueCapsule assetIssueCapsule = dbManager.getAssetIssueStore().get(ByteArray.fromString(NAME));
        Assert.assertNotNull(assetIssueCapsule);
        Assert.assertNull(owner.getInstance().getAssetMap().get(NAME));
    } catch (ContractExeException e) {
        Assert.assertFalse(e instanceof ContractExeException);
    } catch (InvalidProtocolBufferException e) {
        Assert.assertFalse(e instanceof InvalidProtocolBufferException);
    } finally {
        dbManager.getAssetIssueStore().delete(ByteArray.fromString(NAME));
    }
}
Also used : AccountCapsule(org.tron.core.capsule.AccountCapsule) TransactionResultCapsule(org.tron.core.capsule.TransactionResultCapsule) ContractValidateException(org.tron.core.exception.ContractValidateException) InvalidProtocolBufferException(com.google.protobuf.InvalidProtocolBufferException) AssetIssueCapsule(org.tron.core.capsule.AssetIssueCapsule) Contract(org.tron.protos.Contract) ContractExeException(org.tron.core.exception.ContractExeException) Test(org.junit.Test)

Aggregations

AssetIssueCapsule (org.tron.core.capsule.AssetIssueCapsule)7 AccountCapsule (org.tron.core.capsule.AccountCapsule)6 InvalidProtocolBufferException (com.google.protobuf.InvalidProtocolBufferException)4 ContractExeException (org.tron.core.exception.ContractExeException)4 ContractValidateException (org.tron.core.exception.ContractValidateException)3 Contract (org.tron.protos.Contract)3 AssetIssueContract (org.tron.protos.Contract.AssetIssueContract)3 Test (org.junit.Test)2 TransactionResultCapsule (org.tron.core.capsule.TransactionResultCapsule)2 DateTime (org.joda.time.DateTime)1 Before (org.junit.Before)1 BalanceInsufficientException (org.tron.core.exception.BalanceInsufficientException)1