在复杂的智能合约系统中,代码复用与模块化至关重要。Solidity 提供了 库(Library) 机制,用来组织可重用逻辑,避免重复开发与部署,提升合约的可维护性与安全性。
payable fallback/receive)。using for 调用。internal 或 public:internal:编译时内联进调用合约,节省 gas。public:作为独立字节码部署,类似合约调用。// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;
library MathLib {
function add(uint256 a, uint256 b) internal pure returns (uint256) {
return a + b;
}
}
contract Calculator {
function sum(uint256 x, uint256 y) external pure returns (uint256) {
return MathLib.add(x, y);
}
}调用 Calculator.sum(1, 2) → 返回 3。
delegatecall 被调用。// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;
library ArrayLib {
function find(uint256[] storage arr, uint256 value) internal view returns (bool) {
for (uint i = 0; i < arr.length; i++) {
if (arr[i] == value) return true;
}
return false;
}
}
contract DataStore {
using ArrayLib for uint256[];
uint256[] private data;
function add(uint256 v) external {
data.push(v);
}
function exists(uint256 v) external view returns (bool) {
return data.find(v);
}
}这里 using ArrayLib for uint256[]; 让数组 data 直接拥有 find 方法。
using for 的魔力using A for B; 让类型 B 自动绑定库 A 的函数,相当于“扩展方法”。
示例:
using MathLib for uint256;
function example() external pure returns (uint256) {
uint256 x = 5;
return x.add(10); // 等价于 MathLib.add(x, 10)
}场景 | 示例 | 原因 |
|---|---|---|
数学运算 | SafeMath(0.8 前防溢出库) | 避免重复实现、提升安全性 |
数据结构 | EnumerableSet、LinkedList | Solidity 原生不支持复杂结构 |
字符串处理 | Strings 库(OpenZeppelin) | 转换 |
地址工具 | Address 库(OpenZeppelin) | 判断合约地址、低级调用工具 |
下面我们用 Foundry 写一个完整示例,展示如何测试 ArrayLib 库。
src/ArrayLib.sol
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;
library ArrayLib {
function contains(uint256[] storage arr, uint256 v) internal view returns (bool) {
for (uint i = 0; i < arr.length; i++) {
if (arr[i] == v) return true;
}
return false;
}
}src/DataStore.sol
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;
import {ArrayLib} from "./ArrayLib.sol";
contract DataStore {
using ArrayLib for uint256[];
uint256[] private data;
function add(uint256 v) external {
data.push(v);
}
function exists(uint256 v) external view returns (bool) {
return data.contains(v);
}
}test/DataStore.t.sol
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;
import "forge-std/Test.sol";
import "../src/DataStore.sol";
contract DataStoreTest is Test {
DataStore ds;
function setUp() public {
ds = new DataStore();
}
function testAddAndFind() public {
ds.add(42);
assertTrue(ds.exists(42));
assertFalse(ds.exists(7));
}
}执行测试命令:
➜ counter git:(main) ✗ forge test --match-path test/DataStore.t.sol -vvv
[⠊] Compiling...
[⠒] Compiling 3 files with Solc 0.8.29
[⠢] Solc 0.8.29 finished in 1.06s
Compiler run successful!
Ran 1 test for test/DataStore.t.sol:DataStoreTest
[PASS] testAddAndFind() (gas: 58002)
Suite result: ok. 1 passed; 0 failed; 0 skipped; finished in 8.17ms (2.16ms CPU time)
Ran 1 test suite in 378.46ms (8.17ms CPU time): 1 tests passed, 0 failed, 0 skipped (1 total tests)using for 机制让库函数更贴近对象方法风格。