智能合約開發實戰(17) - Library

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.13;
library SafeMath {
　　//數字新增檢查
　　function add(uint x, uint y) internal pure returns(uint) {
　　　　uint result = x + y;
　　　　require(result >= x, "Overflow!");
　　　　return result;
　　}
}
library Array {
　　//陣列移除某內容
　　function remove(uint[] storage arr, uint index) public {
　　　　arr[index] = arr[arr.length -1];
　　　　arr.pop();
　　}
}

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.13;
import "./Library.sol";
contract TestSafeMath {
　　using SafeMath for uint;//引用library到uint屬性內
　　function testAdd(uint x, uint y) public pure returns (uint) {
　　　　uint result = x.add(y);
　　　　//SafeMath.add(x,y);
　　　　return result;
　　}
}
contract TestArray {
　　using Array for uint[];//引用Library
　　uint[] public testArr;
　　function testArrayRemove() public {
　　　　testArr.push(1);
　　　　testArr.push(2);
　　　　testArr.push(3);
　　　　//testArr = [1,2,3]
　　　　testArr.remove(1);
　　　　//testArr = [1,3]
　　　　//驗證
　　　　assert(testArr.length == 2);
　　　　assert(testArr[0] == 1);
　　　　assert(testArr[1] == 3);
　　}
}