如何限制OpenZeppelin PaymentSplitter合约的release()函数调用权限？

不需要复制完整的PaymentSplitter合约，通过继承+覆写结合权限控制合约就能优雅解决问题，核心思路是在覆写的release()函数中先做权限校验，再调用父类的原有逻辑，避免访问私有变量导致编译失败。

方案一：仅所有者可调用（基于Ownable） #

继承PaymentSplitter和Ownable，覆写release()函数并添加onlyOwner修饰符，校验通过后调用父类的release()完成资金分发：

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;

import "@openzeppelin/contracts/finance/PaymentSplitter.sol";
import "@openzeppelin/contracts/access/Ownable.sol";

contract RestrictedPaymentSplitter is PaymentSplitter, Ownable {
    constructor(address[] memory payees, uint256[] memory shares) 
        PaymentSplitter(payees, shares) 
    {}

    function release(address payable account) public virtual override onlyOwner {
        super.release(account);
    }
}

方案二：多角色权限控制（基于AccessControl） #

如果需要多个地址拥有调用权限，可结合AccessControl自定义角色，只有持有指定角色的地址才能触发release()：

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;

import "@openzeppelin/contracts/finance/PaymentSplitter.sol";
import "@openzeppelin/contracts/access/AccessControl.sol";

contract RoleRestrictedPaymentSplitter is PaymentSplitter, AccessControl {
    bytes32 public constant RELEASE_ROLE = keccak256("RELEASE_ROLE");

    constructor(
        address[] memory payees, 
        uint256[] memory shares, 
        address admin
    ) PaymentSplitter(payees, shares) {
        _grantRole(DEFAULT_ADMIN_ROLE, admin);
        _grantRole(RELEASE_ROLE, admin);
    }

    function release(address payable account) public virtual override onlyRole(RELEASE_ROLE) {
        super.release(account);
    }

    // 覆写supportsInterface解决多继承冲突
    function supportsInterface(bytes4 interfaceId) 
        public view virtual override(PaymentSplitter, AccessControl) 
        returns (bool) 
    {
        return super.supportsInterface(interfaceId);
    }
}

方案优势 #

• 保留OpenZeppelin官方合约的审计安全性，无需自己维护完整的资金分发逻辑
• 代码简洁易维护，后续可直接升级OpenZeppelin依赖获取安全更新
• 权限控制逻辑与业务逻辑解耦，可根据需求灵活切换Ownable或AccessControl模式

内容的提问来源于stack exchange，提问作者Jim Dee