@@ -12,7 +12,7 @@ import { Allocation } from "../libraries/Allocation.sol";
12
12
import { LegacyAllocation } from "../libraries/LegacyAllocation.sol " ;
13
13
import { PPMMath } from "@graphprotocol/horizon/contracts/libraries/PPMMath.sol " ;
14
14
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 " ;
16
16
17
17
/**
18
18
* @title AllocationManager contract
@@ -202,12 +202,12 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
202
202
bytes memory _allocationProof ,
203
203
uint32 _delegationRatio
204
204
) internal {
205
- AllocationManagerLib .allocate (
205
+ AllocationHandler .allocate (
206
206
_allocations,
207
207
_legacyAllocations,
208
208
allocationProvisionTracker,
209
209
_subgraphAllocatedTokens,
210
- AllocationManagerLib .AllocateParams ({
210
+ AllocationHandler .AllocateParams ({
211
211
_allocationId: _allocationId,
212
212
_allocationProof: _allocationProof,
213
213
_encodeAllocationProof: _encodeAllocationProof (_indexer, _allocationId),
@@ -257,11 +257,11 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
257
257
address _paymentsDestination
258
258
) internal returns (uint256 ) {
259
259
return
260
- AllocationManagerLib .presentPOI (
260
+ AllocationHandler .presentPOI (
261
261
_allocations,
262
262
allocationProvisionTracker,
263
263
_subgraphAllocatedTokens,
264
- AllocationManagerLib .PresentParams ({
264
+ AllocationHandler .PresentParams ({
265
265
maxPOIStaleness: maxPOIStaleness,
266
266
graphEpochManager: _graphEpochManager (),
267
267
graphStaking: _graphStaking (),
@@ -294,7 +294,7 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
294
294
* @param _delegationRatio The delegation ratio to consider when locking tokens
295
295
*/
296
296
function _resizeAllocation (address _allocationId , uint256 _tokens , uint32 _delegationRatio ) internal {
297
- AllocationManagerLib .resizeAllocation (
297
+ AllocationHandler .resizeAllocation (
298
298
_allocations,
299
299
allocationProvisionTracker,
300
300
_subgraphAllocatedTokens,
@@ -319,7 +319,7 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
319
319
* @param _forceClosed Whether the allocation was force closed
320
320
*/
321
321
function _closeAllocation (address _allocationId , bool _forceClosed ) internal {
322
- AllocationManagerLib .closeAllocation (
322
+ AllocationHandler .closeAllocation (
323
323
_allocations,
324
324
allocationProvisionTracker,
325
325
_subgraphAllocatedTokens,
@@ -357,11 +357,6 @@ abstract contract AllocationManager is EIP712Upgradeable, GraphDirectory, Alloca
357
357
*/
358
358
function _isOverAllocated (address _indexer , uint32 _delegationRatio ) internal view returns (bool ) {
359
359
return
360
- AllocationManagerLib.isOverAllocated (
361
- allocationProvisionTracker,
362
- _graphStaking (),
363
- _indexer,
364
- _delegationRatio
365
- );
360
+ AllocationHandler.isOverAllocated (allocationProvisionTracker, _graphStaking (), _indexer, _delegationRatio);
366
361
}
367
362
}
0 commit comments