@@ -12,7 +12,7 @@ import { Allocation } from "../libraries/Allocation.sol";
1212import { LegacyAllocation } from "../libraries/LegacyAllocation.sol " ;
1313import { PPMMath } from "@graphprotocol/horizon/contracts/libraries/PPMMath.sol " ;
1414import { 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