library Checkpoints:Transform openzeppelin

使用时间戳替换区块号，check数字支持uint256。

1. // SPDX-License-Identifier: MIT
3. pragma solidity ^0.8.0;
5. library Checkpoints {
6.         struct data{
7.         uint32 fromTime;
8.         uint256 quantity;
9.         }
11.     struct Checkpoint {
12.                 mapping (address => mapping (uint32 => data)) checkpoints;
13.                 mapping (address => uint32) numCheckpoints;
14.     }
16.     // --------------checkpoint-----------
17.         function _getCurrentQuantity(Checkpoint storage checkpoint,address account) internal view returns (uint256) {
18.         uint32 nCheckpoints = checkpoint.numCheckpoints[account];
19.         return nCheckpoints > 0 ? checkpoint.checkpoints[account][nCheckpoints - 1].quantity : 0;
20.     }
22.         function getCurrentQuantity(Checkpoint storage checkpoint,address account) external view returns (uint256) {
23.         return _getCurrentQuantity(checkpoint,account);
24.     }
26.     function getPriorQuantity(Checkpoint storage checkpoint,address account, uint blockTimestamp) external view returns (uint256) {
27.         return _getPriorQuantity(checkpoint,account,blockTimestamp);
28.     }
30.     function _getPriorQuantity(Checkpoint storage checkpoint,address account, uint blockTimestamp) internal view returns (uint256) {
31.         require(blockTimestamp < block.timestamp, "ShipSell::getPriorQuantity: not yet determined");
33.         uint32 nCheckpoints = checkpoint.numCheckpoints[account];
34.         if (nCheckpoints == 0) {
35.             return 0;
36.         }
38.         // First check most recent balance
39.         if (checkpoint.checkpoints[account][nCheckpoints - 1].fromTime <= blockTimestamp) {
40.             return checkpoint.checkpoints[account][nCheckpoints - 1].quantity;
41.         }
43.         // Next check implicit zero balance
44.         if (checkpoint.checkpoints[account][0].fromTime > blockTimestamp) {
45.             return 0;
46.         }
48.         uint32 lower = 0;
49.         uint32 upper = nCheckpoints - 1;
50.         while (upper > lower) {
51.             uint32 center = upper - (upper - lower) / 2; // ceil, avoiding overflow
52.             data memory cp = checkpoint.checkpoints[account][center];
53.             if (cp.fromTime == blockTimestamp) {
54.                 return cp.quantity;
55.             } else if (cp.fromTime < blockTimestamp) {
56.                 lower = center;
57.             } else {
58.                 upper = center - 1;
59.             }
60.         }
61.         return checkpoint.checkpoints[account][lower].quantity;
62.     }
64.     function _writeCheckpoint(Checkpoint storage checkpoint,address account, uint256 newQuantity) internal {
65.       uint32 blockTimestamp = uint32(block.timestamp);
66.       uint32 nCheckpoints = checkpoint.numCheckpoints[msg.sender];
68.       if (nCheckpoints > 0 && checkpoint.checkpoints[account][nCheckpoints - 1].fromTime == blockTimestamp) {
69.           checkpoint.checkpoints[account][nCheckpoints - 1].quantity = newQuantity;
70.       } else {
71.           checkpoint.checkpoints[account][nCheckpoints] = data(blockTimestamp, newQuantity);
72.           checkpoint.numCheckpoints[account] = nCheckpoints + 1;
73.       }
74.     }
76.     function WriteCheckpoint(Checkpoint storage checkpoint,address account, uint256 newQuantity) external {
77.         _writeCheckpoint(checkpoint,account,newQuantity);
78.     }
79. }

复制代码

mock:

1. // SPDX-License-Identifier: MIT
3. pragma solidity 0.8.7;
5. import "./Checkpoints.sol";
7. contract CheckpointsMock{
8.     using Checkpoints for Checkpoints.Checkpoint;
9.     Checkpoints.Checkpoint example;
11.     function write(uint256 amount)external{
12.         uint256 current = Checkpoints.getCurrentQuantity(example,msg.sender);
13.         Checkpoints.WriteCheckpoint(example,msg.sender,current+amount);
14.     }
16.     function readCurrent()public view returns (uint256){
17.         return Checkpoints.getCurrentQuantity(example,msg.sender);
18.     }
20.     function readPrior(uint256 fromTime)public view returns (uint256){
21.         return Checkpoints.getPriorQuantity(example,msg.sender,fromTime);
22.     }
23. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！