-
Notifications
You must be signed in to change notification settings - Fork 1.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'develop' into feature/re-1696-add-ci-lint-chaincli
- Loading branch information
Showing
13 changed files
with
2,016 additions
and
109 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
175 changes: 175 additions & 0 deletions
175
contracts/src/v0.8/dev/vrf/testhelpers/VRFV2PlusWrapperLoadTestConsumer.sol
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,175 @@ | ||
| // SPDX-License-Identifier: MIT | ||
| pragma solidity ^0.8.6; | ||
|
|
||
| import "../VRFV2PlusWrapperConsumerBase.sol"; | ||
| import "../../../shared/access/ConfirmedOwner.sol"; | ||
| import "../../../ChainSpecificUtil.sol"; | ||
|
|
||
| contract VRFV2PlusWrapperLoadTestConsumer is VRFV2PlusWrapperConsumerBase, ConfirmedOwner { | ||
| uint256 public s_responseCount; | ||
| uint256 public s_requestCount; | ||
| uint256 public s_averageFulfillmentInMillions = 0; // in millions for better precision | ||
| uint256 public s_slowestFulfillment = 0; | ||
| uint256 public s_fastestFulfillment = 999; | ||
| uint256 public s_lastRequestId; | ||
| mapping(uint256 => uint256) requestHeights; // requestIds to block number when rand request was made | ||
|
|
||
| event WrappedRequestFulfilled(uint256 requestId, uint256[] randomWords, uint256 payment); | ||
| event WrapperRequestMade(uint256 indexed requestId, uint256 paid); | ||
|
|
||
| struct RequestStatus { | ||
| uint256 paid; | ||
| bool fulfilled; | ||
| uint256[] randomWords; | ||
| uint requestTimestamp; | ||
| uint fulfilmentTimestamp; | ||
| uint256 requestBlockNumber; | ||
| uint256 fulfilmentBlockNumber; | ||
| bool native; | ||
| } | ||
|
|
||
| mapping(uint256 => RequestStatus) /* requestId */ /* requestStatus */ public s_requests; | ||
|
|
||
| constructor( | ||
| address _link, | ||
| address _vrfV2PlusWrapper | ||
| ) ConfirmedOwner(msg.sender) VRFV2PlusWrapperConsumerBase(_link, _vrfV2PlusWrapper) {} | ||
|
|
||
| function makeRequests( | ||
| uint32 _callbackGasLimit, | ||
| uint16 _requestConfirmations, | ||
| uint32 _numWords, | ||
| uint16 _requestCount | ||
| ) external onlyOwner { | ||
| for (uint16 i = 0; i < _requestCount; i++) { | ||
| uint256 requestId = requestRandomness(_callbackGasLimit, _requestConfirmations, _numWords); | ||
| s_lastRequestId = requestId; | ||
|
|
||
| uint256 requestBlockNumber = ChainSpecificUtil.getBlockNumber(); | ||
| uint256 paid = VRF_V2_PLUS_WRAPPER.calculateRequestPrice(_callbackGasLimit); | ||
| s_requests[requestId] = RequestStatus({ | ||
| paid: paid, | ||
| fulfilled: false, | ||
| randomWords: new uint256[](0), | ||
| requestTimestamp: block.timestamp, | ||
| requestBlockNumber: requestBlockNumber, | ||
| fulfilmentTimestamp: 0, | ||
| fulfilmentBlockNumber: 0, | ||
| native: false | ||
| }); | ||
| s_requestCount++; | ||
| requestHeights[requestId] = requestBlockNumber; | ||
| emit WrapperRequestMade(requestId, paid); | ||
| } | ||
| } | ||
|
|
||
| function makeRequestsNative( | ||
| uint32 _callbackGasLimit, | ||
| uint16 _requestConfirmations, | ||
| uint32 _numWords, | ||
| uint16 _requestCount | ||
| ) external onlyOwner { | ||
| for (uint16 i = 0; i < _requestCount; i++) { | ||
| uint256 requestId = requestRandomnessPayInNative(_callbackGasLimit, _requestConfirmations, _numWords); | ||
| s_lastRequestId = requestId; | ||
|
|
||
| uint256 requestBlockNumber = ChainSpecificUtil.getBlockNumber(); | ||
| uint256 paid = VRF_V2_PLUS_WRAPPER.calculateRequestPriceNative(_callbackGasLimit); | ||
| s_requests[requestId] = RequestStatus({ | ||
| paid: paid, | ||
| fulfilled: false, | ||
| randomWords: new uint256[](0), | ||
| requestTimestamp: block.timestamp, | ||
| requestBlockNumber: requestBlockNumber, | ||
| fulfilmentTimestamp: 0, | ||
| fulfilmentBlockNumber: 0, | ||
| native: true | ||
| }); | ||
| s_requestCount++; | ||
| requestHeights[requestId] = requestBlockNumber; | ||
| emit WrapperRequestMade(requestId, paid); | ||
| } | ||
| } | ||
|
|
||
| function fulfillRandomWords(uint256 _requestId, uint256[] memory _randomWords) internal override { | ||
| require(s_requests[_requestId].paid > 0, "request not found"); | ||
| uint256 fulfilmentBlockNumber = ChainSpecificUtil.getBlockNumber(); | ||
| uint256 requestDelay = fulfilmentBlockNumber - requestHeights[_requestId]; | ||
| uint256 requestDelayInMillions = requestDelay * 1_000_000; | ||
|
|
||
| if (requestDelay > s_slowestFulfillment) { | ||
| s_slowestFulfillment = requestDelay; | ||
| } | ||
| s_fastestFulfillment = requestDelay < s_fastestFulfillment ? requestDelay : s_fastestFulfillment; | ||
| s_averageFulfillmentInMillions = s_responseCount > 0 | ||
| ? (s_averageFulfillmentInMillions * s_responseCount + requestDelayInMillions) / (s_responseCount + 1) | ||
| : requestDelayInMillions; | ||
|
|
||
| s_responseCount++; | ||
| s_requests[_requestId].fulfilled = true; | ||
| s_requests[_requestId].randomWords = _randomWords; | ||
| s_requests[_requestId].fulfilmentTimestamp = block.timestamp; | ||
| s_requests[_requestId].fulfilmentBlockNumber = fulfilmentBlockNumber; | ||
|
|
||
| emit WrappedRequestFulfilled(_requestId, _randomWords, s_requests[_requestId].paid); | ||
| } | ||
|
|
||
| function getRequestStatus( | ||
| uint256 _requestId | ||
| ) | ||
| external | ||
| view | ||
| returns ( | ||
| uint256 paid, | ||
| bool fulfilled, | ||
| uint256[] memory randomWords, | ||
| uint requestTimestamp, | ||
| uint fulfilmentTimestamp, | ||
| uint256 requestBlockNumber, | ||
| uint256 fulfilmentBlockNumber | ||
| ) | ||
| { | ||
| require(s_requests[_requestId].paid > 0, "request not found"); | ||
| RequestStatus memory request = s_requests[_requestId]; | ||
| return ( | ||
| request.paid, | ||
| request.fulfilled, | ||
| request.randomWords, | ||
| request.requestTimestamp, | ||
| request.fulfilmentTimestamp, | ||
| request.requestBlockNumber, | ||
| request.fulfilmentBlockNumber | ||
| ); | ||
| } | ||
|
|
||
| function reset() external { | ||
| s_averageFulfillmentInMillions = 0; // in millions for better precision | ||
| s_slowestFulfillment = 0; | ||
| s_fastestFulfillment = 999; | ||
| s_requestCount = 0; | ||
| s_responseCount = 0; | ||
| } | ||
|
|
||
| /// @notice withdrawLink withdraws the amount specified in amount to the owner | ||
| /// @param amount the amount to withdraw, in juels | ||
| function withdrawLink(uint256 amount) external onlyOwner { | ||
| LINK.transfer(owner(), amount); | ||
| } | ||
|
|
||
| /// @notice withdrawNative withdraws the amount specified in amount to the owner | ||
| /// @param amount the amount to withdraw, in wei | ||
| function withdrawNative(uint256 amount) external onlyOwner { | ||
| (bool success, ) = payable(owner()).call{value: amount}(""); | ||
| require(success, "withdrawNative failed"); | ||
| } | ||
|
|
||
| function getWrapper() external view returns (VRFV2PlusWrapperInterface) { | ||
| return VRF_V2_PLUS_WRAPPER; | ||
| } | ||
|
|
||
| receive() external payable {} | ||
|
|
||
| function getBalance() public view returns (uint) { | ||
| return address(this).balance; | ||
| } | ||
| } |
1,210 changes: 1,210 additions & 0 deletions
1,210
...rs/generated/vrfv2plus_wrapper_load_test_consumer/vrfv2plus_wrapper_load_test_consumer.go
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.