Package org.apache.nifi.stream.io

Class LeakyBucketStreamThrottler

java.lang.Object

org.apache.nifi.stream.io.LeakyBucketStreamThrottler

All Implemented Interfaces:

Closeable, AutoCloseable, StreamThrottler

public class LeakyBucketStreamThrottler
extends Object
implements StreamThrottler

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
private class	LeakyBucketStreamThrottler.Drain	This class is responsible for draining water from the leaky bucket.
private static class	LeakyBucketStreamThrottler.Request
private static class	LeakyBucketStreamThrottler.Response

Field Summary

Fields

Modifier and Type	Field	Description
private final ScheduledExecutorService	executorService
private final int	maxBytesPerSecond
private final BlockingQueue<LeakyBucketStreamThrottler.Request>	requestQueue
private final AtomicBoolean	shutdown

Constructor Summary

Constructors

Constructor	Description
LeakyBucketStreamThrottler(int maxBytesPerSecond)

Method Summary

Modifier and Type	Method	Description
void	close()
long	copy(InputStream in, OutputStream out)
long	copy(InputStream in, OutputStream out, long maxBytes)
InputStream	newThrottledInputStream(InputStream toWrap)
OutputStream	newThrottledOutputStream(OutputStream toWrap)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

maxBytesPerSecond

private final int maxBytesPerSecond

requestQueue

private final BlockingQueue<LeakyBucketStreamThrottler.Request> requestQueue

executorService

private final ScheduledExecutorService executorService

shutdown

private final AtomicBoolean shutdown

Constructor Details

LeakyBucketStreamThrottler

public LeakyBucketStreamThrottler(int maxBytesPerSecond)

Method Details

close

public void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Closeable

newThrottledOutputStream

public OutputStream newThrottledOutputStream(OutputStream toWrap)

Specified by:

newThrottledOutputStream in interface StreamThrottler

newThrottledInputStream

public InputStream newThrottledInputStream(InputStream toWrap)

Specified by:

newThrottledInputStream in interface StreamThrottler

copy

public long copy(InputStream in,
 OutputStream out)
          throws IOException

Specified by:

copy in interface StreamThrottler

Throws:

IOException

copy

public long copy(InputStream in,
 OutputStream out,
 long maxBytes)
          throws IOException

Specified by:

copy in interface StreamThrottler

Throws:

IOException