2 * Sonitus - DelayFilter.java - Copyright © 2013 David Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.sonitus.data.filter;
20 import java.io.IOException;
21 import java.util.logging.Logger;
23 import net.pterodactylus.sonitus.data.Metadata;
26 * Rate limiting filter that only passes a specified amount of data per second
27 * from its {@link net.pterodactylus.sonitus.data.Source} to its {@link
28 * net.pterodactylus.sonitus.data.Sink}.
30 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
32 public class RateLimitingFilter extends DummyFilter {
35 private static final Logger logger = Logger.getLogger(RateLimitingFilter.class.getName());
37 /** The limiting rate in bytes/second. */
38 private final int rate;
40 /** The start time. */
41 private long startTime;
43 /** The number of bytes. */
47 * Creates a new rate limiting filter.
50 * The limiting rate (in bytes/second)
52 public RateLimitingFilter(int rate) {
57 * Creates a new rate limiting filter.
60 * The limiting rate (in bytes/second)
61 * @param fastStartTime
62 * The amount of time at the start of the filtering during which no delay
63 * will occur (in milliseconds)
65 public RateLimitingFilter(int rate, long fastStartTime) {
67 this.counter = (long) (-rate * (fastStartTime / 1000.0));
75 public void open(Metadata metadata) throws IOException {
77 startTime = System.currentTimeMillis();
81 public void process(byte[] buffer) throws IOException {
82 super.process(buffer);
84 counter += buffer.length;
85 long waitTime = (long) (counter / (rate / 1000.0));
86 while ((System.currentTimeMillis() - startTime) < waitTime) {
88 long limitDelay = waitTime - (System.currentTimeMillis() - startTime);
89 logger.finest(String.format("Waiting %d ms...", limitDelay));
90 Thread.sleep(limitDelay);
91 } catch (InterruptedException ie1) {
92 /* ignore, keep looping. */
95 logger.finest(String.format("Processed %d Bytes during %d ms, that’s %.1f bytes/s.", counter, System.currentTimeMillis() - startTime, counter / ((System.currentTimeMillis() - startTime) / 1000.0)));