mirror of
https://github.com/SquidDev-CC/CC-Tweaked
synced 2024-06-26 07:03:22 +00:00
82 lines
1.7 KiB
Java
82 lines
1.7 KiB
Java
![]() |
/*
|
||
|
* This file is part of ComputerCraft - http://www.computercraft.info
|
||
|
* Copyright Daniel Ratcliffe, 2011-2019. Do not distribute without permission.
|
||
|
* Send enquiries to dratcliffe@gmail.com
|
||
|
*/
|
||
|
|
||
|
package dan200.computercraft.core.apis.http;
|
||
|
|
||
|
import java.util.Collections;
|
||
|
import java.util.Set;
|
||
|
import java.util.concurrent.ConcurrentHashMap;
|
||
|
import java.util.function.IntSupplier;
|
||
|
import java.util.function.Supplier;
|
||
|
|
||
|
/**
|
||
|
* A collection of {@link Resource}s, with an upper bound on capacity.
|
||
|
*/
|
||
|
public class ResourceGroup<T extends Resource<T>>
|
||
|
{
|
||
|
private static final IntSupplier ZERO = () -> 0;
|
||
|
|
||
|
final IntSupplier limit;
|
||
|
|
||
|
boolean active = false;
|
||
|
|
||
|
final Set<T> resources = Collections.newSetFromMap( new ConcurrentHashMap<>() );
|
||
|
|
||
|
public ResourceGroup( IntSupplier limit )
|
||
|
{
|
||
|
this.limit = limit;
|
||
|
}
|
||
|
|
||
|
public ResourceGroup()
|
||
|
{
|
||
|
this.limit = ZERO;
|
||
|
}
|
||
|
|
||
|
public void startup()
|
||
|
{
|
||
|
active = true;
|
||
|
}
|
||
|
|
||
|
public synchronized void shutdown()
|
||
|
{
|
||
|
active = false;
|
||
|
|
||
|
for( T resource : resources ) resource.close();
|
||
|
resources.clear();
|
||
|
|
||
|
Resource.cleanup();
|
||
|
}
|
||
|
|
||
|
|
||
|
public final boolean queue( T resource, Runnable setup )
|
||
|
{
|
||
|
return queue( () -> {
|
||
|
setup.run();
|
||
|
return resource;
|
||
|
} );
|
||
|
}
|
||
|
|
||
|
public synchronized boolean queue( Supplier<T> resource )
|
||
|
{
|
||
|
Resource.cleanup();
|
||
|
if( !active ) return false;
|
||
|
|
||
|
int limit = this.limit.getAsInt();
|
||
|
if( limit <= 0 || resources.size() < limit )
|
||
|
{
|
||
|
resources.add( resource.get() );
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
public synchronized void release( T resource )
|
||
|
{
|
||
|
resources.remove( resource );
|
||
|
}
|
||
|
}
|