智能合約開發實戰(11) - Mapping

// SPDX-License-Identifier: MIT

pragma solidity ^0.8.11;

contract Mapping {

　　event LOG(address, uint);

　　mapping(address => uint) public myMap; //宣告一個mapping，名叫：myMap

　　address[] public myMappingAddr; //因為mapping沒辦法使用迴圈，所以用一個陣列來記錄

　　//設定mapping

　　function setMapping(address _myAddr, uint _number) public {

　　　　myMap[_myAddr] = _number;

　　　　myMappingAddr.push(_myAddr);

　　}

　　//取得mapping(傳入地址，回傳值)

　　function getMapping(address _myAddr) public view returns (uint) {

　　　　return myMap[_myAddr];

　　}

　　//刪除mapping(和陣列一樣，只刪除內容(歸零)，不會刪除key)

　　function deleteMapping(address _myAddr) public {

　　　　delete myMap[_myAddr];

　　}

　　//總計

　　function getTotal() public view returns (uint) {

　　　　require(myMappingAddr.length > 0, "The mapping has no values!"); //驗證mapping的長度是否為零

　　　　//利用陣列取出mapping內所有的值做加總

　　　　uint sum = 0;

　　　　for (uint i = 0; i < myMappingAddr.length; i++){

　　　　　　address key = myMappingAddr[i];

　　　　　　uint value = myMap[key];

　　　　　　sum += value;

　　　　}

　　　　return sum; //回傳

　　}

　　//寫一個test，測試自己的function是否正確

　　function test() public {

　　　　//模擬設定了兩個地址和參數

　　　　setMapping(0x5B38Da6a701c568545dCfcB03FcB875f56beddC4, 10);

　　　　setMapping(0x785aBdDd01bBeCA2D0a68b3461F0779eDc3A7A15, 20);

　　　　uint sum = getTotal();//取得總計

　　　　assert(sum == 30); //最後的驗證

　　}

}