Skip to content

Commit

Permalink
Implement StakePoolParamsQuery deserialization
Browse files Browse the repository at this point in the history
Added functionality to deserialize StakePoolParamsQuery results, converting CBOR data items into PoolParams objects. Updated integration tests to validate the deserialization process.
  • Loading branch information
satran004 committed Oct 15, 2024
1 parent 2b78e82 commit 68822de
Show file tree
Hide file tree
Showing 4 changed files with 91 additions and 9 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ public PoolRegistration deserializeDI(DataItem di) {
return new PoolRegistration(poolParams);
}

private Relay deserializeRelay(DataItem relayDI) {
public static Relay deserializeRelay(DataItem relayDI) {
List<DataItem> relayItems = ((Array) relayDI).getDataItems();
int type = toInt(relayItems.get(0));

Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,14 @@
package com.bloxbean.cardano.yaci.core.protocol.localstate.queries;

import com.bloxbean.cardano.yaci.core.model.PoolParams;
import com.bloxbean.cardano.yaci.core.protocol.localstate.api.QueryResult;
import lombok.AllArgsConstructor;
import lombok.Getter;
import lombok.ToString;

@Getter
@ToString
@AllArgsConstructor
public class StakePoolParamQueryResult implements QueryResult {
private java.util.Map<String, PoolParams> poolParams;
}
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
package com.bloxbean.cardano.yaci.core.protocol.localstate.queries;

import co.nstant.in.cbor.model.Array;
import co.nstant.in.cbor.model.ByteString;
import co.nstant.in.cbor.model.DataItem;
import co.nstant.in.cbor.model.UnsignedInteger;
import co.nstant.in.cbor.model.*;
import co.nstant.in.cbor.model.Map;
import com.bloxbean.cardano.yaci.core.model.PoolParams;
import com.bloxbean.cardano.yaci.core.model.Relay;
import com.bloxbean.cardano.yaci.core.protocol.handshake.messages.AcceptVersion;
import com.bloxbean.cardano.yaci.core.protocol.localstate.api.Era;
import com.bloxbean.cardano.yaci.core.protocol.localstate.api.EraQuery;
Expand All @@ -12,7 +12,12 @@
import lombok.Getter;
import lombok.NonNull;

import java.util.List;
import java.math.BigInteger;
import java.util.*;

import static com.bloxbean.cardano.yaci.core.model.serializers.PoolRegistrationSerializer.deserializeRelay;
import static com.bloxbean.cardano.yaci.core.util.CborSerializationUtil.*;
import static com.bloxbean.cardano.yaci.core.util.CborSerializationUtil.toHex;

@Getter
@AllArgsConstructor
Expand Down Expand Up @@ -41,6 +46,65 @@ public DataItem serialize(AcceptVersion protocolVersion) {

@Override
public StakePoolParamQueryResult deserializeResult(AcceptVersion protocolVersion, DataItem[] di) {
throw new UnsupportedOperationException("Not implemented yet");
List<DataItem> dataItemList = extractResultArray(di[0]);
Map poolMap = (Map) dataItemList.get(0);

java.util.Map<String, PoolParams> poolParamsMap = new LinkedHashMap<>();
for (var key : poolMap.getKeys()) {
String poolId = HexUtil.encodeHexString(((ByteString) key).getBytes());
var poolParamDIList = ((Array)poolMap.get(key)).getDataItems();

String operator = toHex(poolParamDIList.get(0));
String vrfKeyHash = toHex(poolParamDIList.get(1));
BigInteger pledge = toBigInteger(poolParamDIList.get(2));
BigInteger cost = toBigInteger(poolParamDIList.get(3));
String margin = ((RationalNumber) poolParamDIList.get(4)).getNumerator() + "/" + ((RationalNumber) poolParamDIList.get(4)).getDenominator();
String rewardAccount = toHex(poolParamDIList.get(5));

//Pool Owners0
Set<String> poolOwners = new HashSet<>();
List<DataItem> poolOwnersDataItems = ((Array) poolParamDIList.get(6)).getDataItems();
for (DataItem poolOwnerDI : poolOwnersDataItems) {
if (poolOwnerDI == SimpleValue.BREAK)
continue;
poolOwners.add(toHex(poolOwnerDI));
}

//Relays
List<DataItem> relaysDataItems = ((Array) poolParamDIList.get(7)).getDataItems();
List<Relay> relays = new ArrayList<>();
for (DataItem relayDI : relaysDataItems) {
if (relayDI == SimpleValue.BREAK)
continue;
relays.add(deserializeRelay(relayDI));
}

//pool metadata
DataItem poolMetaDataDI = poolParamDIList.get(8);
String metadataUrl = null;
String metadataHash = null;
if (poolMetaDataDI != SimpleValue.NULL) {
List<DataItem> poolMetadataDataItems = ((Array) poolMetaDataDI).getDataItems();
metadataUrl = toUnicodeString(poolMetadataDataItems.get(0));
metadataHash = toHex(poolMetadataDataItems.get(1));
}

PoolParams poolParams = PoolParams.builder()
.operator(operator)
.vrfKeyHash(vrfKeyHash)
.pledge(pledge)
.cost(cost)
.margin(margin)
.rewardAccount(rewardAccount)
.poolOwners(poolOwners)
.relays(relays)
.poolMetadataUrl(metadataUrl)
.poolMetadataHash(metadataHash)
.build();

poolParamsMap.put(poolId, poolParams);
}

return new StakePoolParamQueryResult(poolParamsMap);
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -217,11 +217,21 @@ void stakeSnapshotsQuery() {

@Test
void stakePoolParamQuery() {
Mono<StakePoolParamQueryResult> mono = localStateQueryClient.executeQuery(new StakePoolParamsQuery(List.of("032a04334a846fdf542fd5633c9b3928998691b8276e004facbc8af1",
"0a4ed3c5cc11a044cff16f7045588c9b6f6c98f7154026a3a3f55f24")));
String poolId1 = "27aa9ad499cb47f856727913abd02dfd08bbb69dbefd1b4c098c6e96";
//String poolId2 = "1c07dbd3648fe70545170b6f6d390136c1cc66ec83326f5d94a0fd3e";
Mono<StakePoolParamQueryResult> mono = localStateQueryClient.executeQuery(new StakePoolParamsQuery(List.of(poolId1)));

StakePoolParamQueryResult result = mono.block(Duration.ofSeconds(5));
System.out.println(result);
assertThat(result.getPoolParams().get(poolId1)).isNotNull();
assertThat(result.getPoolParams().get(poolId1).getOperator()).isEqualTo(poolId1);
assertThat(result.getPoolParams().get(poolId1).getVrfKeyHash()).isEqualTo("5f97dc97196c55b7eca60909b68de9db9bedee4f04a000b27c2df4b95911c5af");
assertThat(result.getPoolParams().get(poolId1).getCost()).isNotNull();

// assertThat(result.getPoolParams().get(poolId2)).isNotNull();
// assertThat(result.getPoolParams().get(poolId2).getOperator()).isEqualTo(poolId2);
// assertThat(result.getPoolParams().get(poolId2).getVrfKeyHash()).isEqualTo("322828d4f7cfe86db0f8c3fb385e317b5415a8287963aa58c967eb3643c5ee25");
// assertThat(result.getPoolParams().get(poolId2).getCost()).isNotNull();
}

@Test
Expand Down

0 comments on commit 68822de

Please sign in to comment.