86 lines
1.9 KiB
Java
86 lines
1.9 KiB
Java
/*
|
|
* This file is part of ComputerCraft - http://www.computercraft.info
|
|
* Copyright Daniel Ratcliffe, 2011-2021. 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.
|
|
*
|
|
* @param <T> The type of the resource this group manages.
|
|
*/
|
|
public class ResourceGroup<T extends Resource<T>>
|
|
{
|
|
public static final int DEFAULT_LIMIT = 512;
|
|
public static final IntSupplier DEFAULT = () -> DEFAULT_LIMIT;
|
|
|
|
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()
|
|
{
|
|
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 );
|
|
}
|
|
}
|