perf: optimize withdraw function

sablier-labs · Oct 14, 2023 · f291f6e · f291f6e
1 parent 6189bbc
commit f291f6e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/abstracts/SablierV2Lockup.sol b/src/abstracts/SablierV2Lockup.sol
@@ -251,8 +251,11 @@ abstract contract SablierV2Lockup is
             revert Errors.SablierV2Lockup_Unauthorized(streamId, msg.sender);
         }
 
+        // Retrieve the recipient from storage.
+        address recipient = _ownerOf(streamId);
+
         // Checks: if `msg.sender` is the stream's sender, the withdrawal address must be the recipient.
-        if (_isCallerStreamSender(streamId) && to != _ownerOf(streamId)) {
+        if (_isCallerStreamSender(streamId) && to != recipient) {
             revert Errors.SablierV2Lockup_InvalidSenderWithdrawal(streamId, msg.sender, to);
         }
 
@@ -275,9 +278,6 @@ abstract contract SablierV2Lockup is
         // Effects and Interactions: make the withdrawal.
         _withdraw(streamId, to, amount);
 
-        // Retrieve the recipient from storage.
-        address recipient = _ownerOf(streamId);
-
         // Interactions: if `msg.sender` is not the recipient and the recipient is a contract, try to invoke the
         // withdraw hook on it without reverting if the hook is not implemented, and also without bubbling up
         // any potential revert.