Test calls on datapool

Author: james-toussaint

contracts/registries/datasets/Datapool.sol

@@ -0,0 +1,40 @@
+// SPDX-FileCopyrightText: 2023-2024 IEXEC BLOCKCHAIN TECH <[email protected]>
+// SPDX-License-Identifier: Apache-2.0
+
+pragma solidity ^0.8.0;
+
+contract Datapool {
+    event DatasetAdded(address dataset, uint256 versionId, uint256 datasetCount);
+    event DatasetRemoved(address dataset, uint256 versionId, uint256 datasetCount);
+
+    uint256 public versionId;
+    mapping(address dataset => uint256 addedVersionId) public addedVersionId;
+    address[] public datasets;
+    mapping(address dataset => uint256 index) public datasetIndexes;
+
+    constructor() {}
+
+    function datasetCount() external view returns (uint256) {
+        return datasets.length;
+    }
+
+    function addDataset(address dataset) external {
+        versionId++;
+        datasets.push(dataset);
+        datasetIndexes[dataset] = datasets.length;
+        addedVersionId[dataset] = versionId;
+        emit DatasetAdded(dataset, versionId, datasets.length);
+    }
+
+    function removeDataset(address removedDataset) external {
+        versionId++;
+        uint256 removedDatasetIndex = datasetIndexes[removedDataset];
+        address movingDataset = datasets[datasets.length - 1];
+        datasets[removedDatasetIndex] = movingDataset; // move last dataset to removed index slot
+        datasetIndexes[movingDataset] = removedDatasetIndex;
+        datasets.pop(); // removed already moved dataset
+        datasetIndexes[removedDataset] = 0;
+        addedVersionId[removedDataset] = 0;
+        emit DatasetRemoved(removedDataset, versionId, datasets.length);
+    }
+}

package-lock.json

@@ -59,6 +59,7 @@
     "@nomiclabs/hardhat-web3": "^2.0.0",
     "@openzeppelin/test-helpers": "^0.5.16",
     "@openzeppelin/upgrades-core": "^1.37.1",
+    "ethers-multicall-provider": "^3.1.2",
     "hardhat": "^2.14.1",
     "hardhat-dependency-compiler": "^1.2.1",
     "hardhat-deploy": "^0.11.43",

package.json

@@ -0,0 +1,62 @@
+import { MulticallWrapper } from 'ethers-multicall-provider';
+import { ethers } from 'hardhat';
+import { Datapool, Datapool__factory } from '../typechain';
+
+(async () => {
+    await deploy();
+})();
+
+export async function deploy() {
+    console.log('Deploying datapool..');
+    const [deployer] = await ethers.getSigners();
+    console.log(`Deployer: ${deployer.address}`);
+    const datapoolAddress = await new Datapool__factory(deployer)
+        .deploy()
+        .then((tx) => tx.deployed())
+        .then((contract) => contract.address);
+    console.log(`Datapool: ${datapoolAddress}`);
+    // const provider = ethers.getDefaultProvider();
+    const provider = MulticallWrapper.wrap(ethers.getDefaultProvider());
+    // ethers might already use batch-calls provider internally, lets force one
+    console.log(`Using batch-calls provider ${MulticallWrapper.isMulticallProvider(provider)}`);
+    let datapool = new ethers.Contract(
+        datapoolAddress,
+        new Datapool__factory().interface,
+        provider,
+    ).connect(deployer) as Datapool;
+
+    const datasets = 5;
+    await addDatasets(datasets);
+    await getDatasets();
+
+    async function addDatasets(datasets: number) {
+        console.log('From dataset providers:');
+        console.log(`Requested ${datasets} datasets`);
+        const addDatasetCalls = [];
+        for (let i = 0; i < datasets; i++) {
+            const datasetAddress = await ethers.Wallet.createRandom().getAddress();
+            addDatasetCalls.push(datapool.addDataset(datasetAddress));
+            process.stdout.write(`Preparing datasets: ${i + 1}.. \r`);
+        }
+        await Promise.all(addDatasetCalls);
+        const datasetsCount = Number(await datapool.datasetCount());
+        console.log();
+        console.log(`Datasets added`);
+        console.log(`Dataset count is now: ${datasetsCount}`);
+    }
+
+    async function getDatasets() {
+        console.log('From SMS:');
+        const start = new Date().getTime();
+        const datasetsCount = Number(await datapool.datasetCount());
+        const getDatasetCalls = [];
+        for (let i = 0; i < datasetsCount; i++) {
+            getDatasetCalls.push(datapool.datasets(i));
+        }
+        const allDatasets = await Promise.all(getDatasetCalls);
+        console.log(allDatasets);
+        console.log(`Dataset found: ${allDatasets.length}`);
+        const end = new Date().getTime();
+        console.log(`Took ${end - start}ms`);
+    }
+}