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;
}
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());
}
}
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;
}
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));
}
}
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));
}
}
Aggregations