use of com.hedera.services.bdd.spec.HapiSpecOperation in project hedera-services by hashgraph.
the class TokenTransferBasicLoadTest method tokenCreatesFactory.
private Function<HapiApiSpec, OpProvider> tokenCreatesFactory(PerfTestLoadSettings settings) {
int numTotalTokens = settings.getTotalTokens();
int totalClients = settings.getTotalClients();
int numActiveTokens = (totalClients >= 1) ? numTotalTokens / totalClients : numTotalTokens;
AtomicInteger remaining = new AtomicInteger(numActiveTokens - 1);
return spec -> new OpProvider() {
@Override
public List<HapiSpecOperation> suggestedInitializers() {
return Collections.emptyList();
}
@Override
public Optional<HapiSpecOperation> get() {
int next;
if ((next = remaining.getAndDecrement()) < 0) {
return Optional.empty();
}
var payingTreasury = String.format("0.0.%d", settings.getTestTreasureStartAccount() + next);
var op = tokenCreate(tokenRegistryName(next)).payingWith(DEFAULT_PAYER).signedBy(DEFAULT_PAYER).fee(ONE_HUNDRED_HBARS).initialSupply(100_000_000_000L).treasury(payingTreasury).hasRetryPrecheckFrom(BUSY, PLATFORM_TRANSACTION_NOT_CREATED, DUPLICATE_TRANSACTION, INSUFFICIENT_PAYER_BALANCE).hasPrecheckFrom(DUPLICATE_TRANSACTION, OK).hasKnownStatusFrom(SUCCESS, TOKEN_ALREADY_ASSOCIATED_TO_ACCOUNT, TOKENS_PER_ACCOUNT_LIMIT_EXCEEDED, FAIL_INVALID).suppressStats(true).noLogging();
return Optional.of(op);
}
};
}
use of com.hedera.services.bdd.spec.HapiSpecOperation in project hedera-services by hashgraph.
the class TokenTransfersLoadProvider method tokenTransfersFactory.
private Function<HapiApiSpec, OpProvider> tokenTransfersFactory() {
var firstDir = new AtomicBoolean(Boolean.TRUE);
var balanceInit = new AtomicLong();
var tokensPerTxn = new AtomicInteger();
var sendingAccountsPerToken = new AtomicInteger();
var receivingAccountsPerToken = new AtomicInteger();
List<String> treasuries = new ArrayList<>();
Map<String, List<String>> senders = new HashMap<>();
Map<String, List<String>> receivers = new HashMap<>();
return spec -> new OpProvider() {
@Override
public List<HapiSpecOperation> suggestedInitializers() {
var ciProps = spec.setup().ciPropertiesMap();
balanceInit.set(ciProps.getLong("balanceInit"));
tokensPerTxn.set(ciProps.getInteger("tokensPerTxn"));
sendingAccountsPerToken.set(ciProps.getInteger("sendingAccountsPerToken"));
receivingAccountsPerToken.set(ciProps.getInteger("receivingAccountsPerToken"));
var initialSupply = (sendingAccountsPerToken.get() + receivingAccountsPerToken.get()) * balanceInit.get();
List<HapiSpecOperation> initializers = new ArrayList<>();
initializers.add(tokenOpsEnablement());
/* Temporary, can be removed after the public testnet state used in
restart tests includes a fee schedule with HTS resource prices. */
if (spec.setup().defaultNode().equals(asAccount("0.0.3"))) {
initializers.add(uploadDefaultFeeSchedules(GENESIS));
} else {
initializers.add(withOpContext((spec, opLog) -> {
log.info("\n\n" + bannerWith("Waiting for a fee schedule with token ops!"));
boolean hasKnownHtsFeeSchedules = false;
SysFileSerde<String> serde = new FeesJsonToGrpcBytes();
while (!hasKnownHtsFeeSchedules) {
var query = QueryVerbs.getFileContents(FEE_SCHEDULE).fee(10_000_000_000L);
try {
allRunFor(spec, query);
var contents = query.getResponse().getFileGetContents().getFileContents().getContents();
var schedules = serde.fromRawFile(contents.toByteArray());
hasKnownHtsFeeSchedules = schedules.contains("TokenCreate");
} catch (Exception e) {
var msg = e.toString();
msg = msg.substring(msg.indexOf(":") + 2);
log.info("Couldn't check for HTS fee schedules---'{}'", msg);
}
TimeUnit.SECONDS.sleep(3);
}
log.info("\n\n" + bannerWith("A fee schedule with token ops now available!"));
spec.tryReinitializingFees();
}));
}
for (int i = 0; i < tokensPerTxn.get(); i++) {
var token = "token" + i;
var treasury = "treasury" + i;
initializers.add(cryptoCreate(treasury));
initializers.add(tokenCreate(token).treasury(treasury).initialSupply(initialSupply));
treasuries.add(treasury);
for (int j = 0; j < sendingAccountsPerToken.get(); j++) {
var sender = token + "sender" + j;
senders.computeIfAbsent(token, ignore -> new ArrayList<>()).add(sender);
initializers.add(cryptoCreate(sender));
initializers.add(tokenAssociate(sender, token));
initializers.add(cryptoTransfer(moving(balanceInit.get(), token).between(treasury, sender)));
}
for (int j = 0; j < receivingAccountsPerToken.get(); j++) {
var receiver = token + "receiver" + j;
receivers.computeIfAbsent(token, ignore -> new ArrayList<>()).add(receiver);
initializers.add(cryptoCreate(receiver));
initializers.add(tokenAssociate(receiver, token));
initializers.add(cryptoTransfer(moving(balanceInit.get(), token).between(treasury, receiver)));
}
}
for (HapiSpecOperation op : initializers) {
if (op instanceof HapiTxnOp) {
((HapiTxnOp) op).hasRetryPrecheckFrom(NOISY_RETRY_PRECHECKS);
}
}
return initializers;
}
@Override
public Optional<HapiSpecOperation> get() {
HapiSpecOperation op;
var numTokens = tokensPerTxn.get();
var numSenders = sendingAccountsPerToken.get();
var numReceivers = receivingAccountsPerToken.get();
if (firstDir.get()) {
var xfers = new TokenMovement[numTokens * numSenders];
for (int i = 0; i < numTokens; i++) {
var token = "token" + i;
for (int j = 0; j < numSenders; j++) {
var receivers = new String[numReceivers];
for (int k = 0; k < numReceivers; k++) {
receivers[k] = token + "receiver" + k;
}
xfers[i * numSenders + j] = moving(numReceivers, token).distributing(token + "sender" + j, receivers);
}
}
op = cryptoTransfer(xfers).hasKnownStatusFrom(OK, DUPLICATE_TRANSACTION, SUCCESS, UNKNOWN, INSUFFICIENT_PAYER_BALANCE).hasRetryPrecheckFrom(NOISY_RETRY_PRECHECKS).hasPrecheckFrom(OK, PLATFORM_NOT_ACTIVE).noLogging().deferStatusResolution();
firstDir.set(Boolean.FALSE);
} else {
var xfers = new TokenMovement[numTokens * numReceivers];
for (int i = 0; i < numTokens; i++) {
var token = "token" + i;
for (int j = 0; j < numReceivers; j++) {
var senders = new String[numSenders];
for (int k = 0; k < numSenders; k++) {
senders[k] = token + "sender" + k;
}
xfers[i * numReceivers + j] = moving(numSenders, token).distributing(token + "receiver" + j, senders);
}
}
op = cryptoTransfer(xfers).hasKnownStatusFrom(OK, DUPLICATE_TRANSACTION, SUCCESS, UNKNOWN, INSUFFICIENT_PAYER_BALANCE).hasRetryPrecheckFrom(NOISY_RETRY_PRECHECKS).hasPrecheckFrom(OK, PLATFORM_NOT_ACTIVE).noLogging().deferStatusResolution();
firstDir.set(Boolean.TRUE);
}
return Optional.of(op);
}
};
}
use of com.hedera.services.bdd.spec.HapiSpecOperation in project hedera-services by hashgraph.
the class UniqueTokenStateSetup method nftFactory.
private Function<HapiApiSpec, OpProvider> nftFactory() {
final AtomicInteger uniqueTokensCreated = new AtomicInteger(0);
final AtomicInteger nftsMintedForCurrentUniqueToken = new AtomicInteger(0);
final AtomicBoolean done = new AtomicBoolean(false);
final AtomicReference<String> currentUniqueToken = new AtomicReference<>(uniqueTokenNameFn.apply(0));
return spec -> new OpProvider() {
@Override
public List<HapiSpecOperation> suggestedInitializers() {
final var numTreasuries = numTokens.get() / UNIQ_TOKENS_PER_TREASURY + Math.min(1, numTokens.get() % UNIQ_TOKENS_PER_TREASURY);
final List<HapiSpecOperation> inits = new ArrayList<>();
inits.add(inParallel(IntStream.range(0, numTreasuries).mapToObj(i -> cryptoCreate(treasuryNameFn.apply(i)).payingWith(GENESIS).balance(0L).noLogging().key(GENESIS).hasRetryPrecheckFrom(DUPLICATE_TRANSACTION).hasKnownStatusFrom(SUCCESS, UNKNOWN, TRANSACTION_EXPIRED).deferStatusResolution()).toArray(HapiSpecOperation[]::new)));
inits.add(sleepFor(5_000L));
inits.addAll(burstedUniqCreations(UNIQ_TOKENS_BURST_SIZE, numTreasuries, UNIQ_TOKENS_POST_BURST_PAUSE_MS));
return inits;
}
@Override
public Optional<HapiSpecOperation> get() {
if (done.get()) {
return Optional.empty();
}
final var currentToken = currentUniqueToken.get();
if (nftsMintedForCurrentUniqueToken.get() < numNftsPerToken.get()) {
final List<ByteString> allMeta = new ArrayList<>();
final int noMoreThan = numNftsPerToken.get() - nftsMintedForCurrentUniqueToken.get();
for (int i = 0, n = Math.min(noMoreThan, numNftsPerMintOp.get()); i < n; i++) {
final var nextSerialNo = nftsMintedForCurrentUniqueToken.incrementAndGet();
allMeta.add(metadataFor(currentToken, nextSerialNo));
}
final var op = mintToken(currentToken, allMeta).payingWith(GENESIS).rememberingNothing().deferStatusResolution().fee(ONE_HBAR).hasPrecheckFrom(OK, DUPLICATE_TRANSACTION).hasKnownStatusFrom(SUCCESS, UNKNOWN, OK, TRANSACTION_EXPIRED, INVALID_TOKEN_ID).noLogging();
return Optional.of(op);
} else {
nftsMintedForCurrentUniqueToken.set(0);
final var nextUniqTokenNo = uniqueTokensCreated.incrementAndGet();
currentUniqueToken.set(uniqueTokenNameFn.apply(nextUniqTokenNo));
if (nextUniqTokenNo >= numTokens.get()) {
System.out.println("Done creating " + nextUniqTokenNo + " unique tokens w/ approximately " + (numNftsPerToken.get() * nextUniqTokenNo) + " NFTs");
done.set(true);
}
return Optional.empty();
}
}
};
}
use of com.hedera.services.bdd.spec.HapiSpecOperation in project hedera-services by hashgraph.
the class AccountBalancesClientSaveLoadTest method accountsCreate.
private Function<HapiApiSpec, OpProvider> accountsCreate(PerfTestLoadSettings settings) {
totalTestTokens = settings.getTotalTokens() > 10 ? settings.getTotalTokens() : TOTAL_TEST_TOKENS;
totalTreasureAccounts = totalTestTokens;
totalAccounts = settings.getTotalAccounts() > 100 ? settings.getTotalAccounts() : TOTAL_ACCOUNT;
log.info("Total accounts: {}", totalAccounts);
log.info("Total tokens: {}", totalTestTokens);
AtomicInteger moreToCreate = new AtomicInteger(totalAccounts);
return spec -> new OpProvider() {
@Override
public List<HapiSpecOperation> suggestedInitializers() {
log.info("Now running accountsCreate initializer");
return Collections.emptyList();
}
@Override
public Optional<HapiSpecOperation> get() {
int next;
next = moreToCreate.getAndDecrement();
if (next <= 0) {
return Optional.empty();
}
var op = cryptoCreate(String.format("%s%d", ACCT_NAME_PREFIX, next)).balance((long) (r.nextInt((int) ONE_HBAR) + MIN_ACCOUNT_BALANCE)).key(GENESIS).fee(ONE_HUNDRED_HBARS).withRecharging().rechargeWindow(30).hasRetryPrecheckFrom(NOISY_RETRY_PRECHECKS).hasPrecheckFrom(DUPLICATE_TRANSACTION, OK, INSUFFICIENT_TX_FEE).hasKnownStatusFrom(SUCCESS, INVALID_SIGNATURE).noLogging().deferStatusResolution();
return Optional.of(op);
}
};
}
use of com.hedera.services.bdd.spec.HapiSpecOperation in project hedera-services by hashgraph.
the class AccountBalancesClientSaveLoadTest method tokensCreate.
private Function<HapiApiSpec, OpProvider> tokensCreate(PerfTestLoadSettings settings) {
AtomicInteger createdSofar = new AtomicInteger(0);
return spec -> new OpProvider() {
@Override
public List<HapiSpecOperation> suggestedInitializers() {
log.info("Now running tokensCreate initializer");
return Collections.emptyList();
}
@Override
public Optional<HapiSpecOperation> get() {
int next;
next = createdSofar.getAndIncrement();
if (next >= totalTestTokens) {
return Optional.empty();
}
var payingTreasury = String.format(ACCT_NAME_PREFIX + next);
var op = tokenCreate(TOKEN_NAME_PREFIX + next).signedBy(GENESIS).fee(ONE_HUNDRED_HBARS).initialSupply(MIN_TOKEN_SUPPLY + r.nextInt(MAX_TOKEN_SUPPLY)).treasury(payingTreasury).hasRetryPrecheckFrom(NOISY_RETRY_PRECHECKS).hasPrecheckFrom(DUPLICATE_TRANSACTION, OK).hasKnownStatusFrom(SUCCESS, TOKEN_ALREADY_ASSOCIATED_TO_ACCOUNT, TOKENS_PER_ACCOUNT_LIMIT_EXCEEDED).suppressStats(true).noLogging();
if (next > 0) {
op.deferStatusResolution();
}
return Optional.of(op);
}
};
}
Aggregations