智能合约中如何返回mapping

终有链响

发布于 2024-07-29 20:15:18

600

发布于 2024-07-29 20:15:18

文章被收录于专栏：终有链响

在 Solidity 中，直接返回一个 mapping 的所有数据是不可能的，因为 mapping 本身不支持直接遍历。但是，可以使用一些技巧来实现这一目标，例如通过维护一个额外的数组来跟踪 mapping 中的键，并通过这个数组来返回所有的键值对。

示例代码

假设有一个 mapping(uint256 => uint256)，并且想要创建一个函数来返回该 mapping 的所有键值对，可以按照以下步骤操作：

维护一个额外的数组来存储所有的键。
创建一个新的数组来存储键值对。
使用循环来填充新数组。
返回新数组。

示例代码

pragma solidity ^0.8.0;

contract MappingIteration {
    mapping(uint256 => uint256) public pledgeMap;
    uint256[] public pledgeKeys;

    function addEntry(uint256 _key, uint256 _value) public {
        pledgeMap[_key] = _value;
        // 如果键不存在于数组中，则添加到数组
        if (!contains(_key)) {
            pledgeKeys.push(_key);
        }
    }

    function contains(uint256 _key) private view returns (bool) {
        for (uint256 i = 0; i < pledgeKeys.length; i++) {
            if (pledgeKeys[i] == _key) {
                return true;
            }
        }
        return false;
    }

    function getAllEntries() public view returns (uint256[] memory, uint256[] memory) {
        uint256[] memory keys = new uint256[](pledgeKeys.length);
        uint256[] memory values = new uint256[](pledgeKeys.length);

        for (uint256 i = 0; i < pledgeKeys.length; i++) {
            keys[i] = pledgeKeys[i];
            values[i] = pledgeMap[keys[i]];
        }

        return (keys, values);
    }
}

解释

添加条目:
- addEntry 函数在向 mapping 添加条目时，还会检查该键是否已经存在于 pledgeKeys 数组中，如果不存在，则添加到数组中。
遍历:
- getAllEntries 函数创建两个数组 keys 和 values，并将 mapping 中的所有键值对复制到这两个数组中。
包含检查:
- contains 函数用于检查键是否已经存在于 pledgeKeys 数组中。

示例用法

假设想要从合约中获取 pledgeMap 的所有键值对，可以这样调用函数：

MappingIteration myContract = new MappingIteration();
myContract.addEntry(1, 100);
myContract.addEntry(2, 200);
myContract.addEntry(3, 300);

(uint256[] memory keys, uint256[] memory values) = myContract.getAllEntries();

在这个例子中，keys 和 values 数组将分别包含 pledgeMap 中的所有键和值。