Skip to content

Commit 54ef3eb

Browse files
f: rename to AllocationHandler
1 parent 91b341a commit 54ef3eb

File tree

2 files changed

+9
-14
lines changed

2 files changed

+9
-14
lines changed

packages/subgraph-service/contracts/libraries/AllocationManagerLib.sol renamed to packages/subgraph-service/contracts/libraries/AllocationHandler.sol

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ import { Allocation } from "../libraries/Allocation.sol";
1616
import { LegacyAllocation } from "../libraries/LegacyAllocation.sol";
1717
import { AllocationManager } from "../utilities/AllocationManager.sol";
1818

19-
library AllocationManagerLib {
19+
library AllocationHandler {
2020
using ProvisionTracker for mapping(address => uint256);
2121
using Allocation for mapping(address => Allocation.State);
2222
using Allocation for Allocation.State;

packages/subgraph-service/contracts/utilities/AllocationManager.sol

Lines changed: 8 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ import { Allocation } from "../libraries/Allocation.sol";
1212
import { LegacyAllocation } from "../libraries/LegacyAllocation.sol";
1313
import { PPMMath } from "@graphprotocol/horizon/contracts/libraries/PPMMath.sol";
1414
import { ProvisionTracker } from "@graphprotocol/horizon/contracts/data-service/libraries/ProvisionTracker.sol";
15-
import { AllocationManagerLib } from "../libraries/AllocationManagerLib.sol";
15+
import { AllocationHandler } from "../libraries/AllocationHandler.sol";
1616

1717
/**
1818
* @title AllocationManager contract
@@ -202,12 +202,12 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
202202
bytes memory _allocationProof,
203203
uint32 _delegationRatio
204204
) internal {
205-
AllocationManagerLib.allocate(
205+
AllocationHandler.allocate(
206206
_allocations,
207207
_legacyAllocations,
208208
allocationProvisionTracker,
209209
_subgraphAllocatedTokens,
210-
AllocationManagerLib.AllocateParams({
210+
AllocationHandler.AllocateParams({
211211
_allocationId: _allocationId,
212212
_allocationProof: _allocationProof,
213213
_encodeAllocationProof: _encodeAllocationProof(_indexer, _allocationId),
@@ -257,11 +257,11 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
257257
address _paymentsDestination
258258
) internal returns (uint256) {
259259
return
260-
AllocationManagerLib.presentPOI(
260+
AllocationHandler.presentPOI(
261261
_allocations,
262262
allocationProvisionTracker,
263263
_subgraphAllocatedTokens,
264-
AllocationManagerLib.PresentParams({
264+
AllocationHandler.PresentParams({
265265
maxPOIStaleness: maxPOIStaleness,
266266
graphEpochManager: _graphEpochManager(),
267267
graphStaking: _graphStaking(),
@@ -294,7 +294,7 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
294294
* @param _delegationRatio The delegation ratio to consider when locking tokens
295295
*/
296296
function _resizeAllocation(address _allocationId, uint256 _tokens, uint32 _delegationRatio) internal {
297-
AllocationManagerLib.resizeAllocation(
297+
AllocationHandler.resizeAllocation(
298298
_allocations,
299299
allocationProvisionTracker,
300300
_subgraphAllocatedTokens,
@@ -319,7 +319,7 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
319319
* @param _forceClosed Whether the allocation was force closed
320320
*/
321321
function _closeAllocation(address _allocationId, bool _forceClosed) internal {
322-
AllocationManagerLib.closeAllocation(
322+
AllocationHandler.closeAllocation(
323323
_allocations,
324324
allocationProvisionTracker,
325325
_subgraphAllocatedTokens,
@@ -357,11 +357,6 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
357357
*/
358358
function _isOverAllocated(address _indexer, uint32 _delegationRatio) internal view returns (bool) {
359359
return
360-
AllocationManagerLib.isOverAllocated(
361-
allocationProvisionTracker,
362-
_graphStaking(),
363-
_indexer,
364-
_delegationRatio
365-
);
360+
AllocationHandler.isOverAllocated(allocationProvisionTracker, _graphStaking(), _indexer, _delegationRatio);
366361
}
367362
}

0 commit comments

Comments
 (0)