Solidity学习之路 - 一个记录邀请关系的通用合约

原创

已于 2022-04-24 02:15:20 修改 · 1.2k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#智能合约

于 2022-04-24 02:07:47 首次发布

本文介绍了使用Solidity编写的智能合约，展示了如何通过IInvite接口设计一个邀请与层级管理系统，包括添加记录、查询上级和下级、统计邀请数量等功能。

// SPDX-License-Identifier: MIT
pragma solidity >=0.8.0;

interface IInvite {
    function addRecord(address) external returns(bool);
    function getParents(address) external view returns(address[2] memory);
    function getChilds(address) external view returns(address[] memory);
    function getInviteNum(address) external view returns(uint256[2] memory);
}

// SPDX-License-Identifier: MIT
pragma solidity >=0.8.0;

import './interfaces/IInvite.sol'; // 注意路径，我把所有的接口合约都放在了interfaces目录下

contract Invite is IInvite {
    address public factory; // 记录合约发布者地址
    mapping(address => address[]) public inviteRecords; // 邀请记录  邀请人地址 => 被邀请人地址数组
    mapping(address => address) public parents; // 记录上级  我的地址 => 我的上级地址
    mapping(address => uint256[2]) public inviteNumRecords; // 记录邀请数量  我的地址 => [邀请的一级用户数量,邀请的二级用户数量]
    address public firstAddress; // 合约发布时需要初始化第一个用户地址，否则无法往下绑定用户
    uint256 public totalPeople;

    constructor() {