import net.pterodactylus.sonitus.data.Filter;
import net.pterodactylus.sonitus.data.Format;
import net.pterodactylus.sonitus.data.Source;
+import net.pterodactylus.sonitus.io.InputStreamDrainer;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableList;
public abstract class ExternalFilter implements Filter {
/** The logger. */
- private static final Logger logger = Logger.getLogger(ExternalFilter.class.getName());
-
- /** The binary to execute. */
- private final String binary;
-
- /** The parameters for the binary. */
- private final Iterable<String> parameters;
+ private final Logger logger = Logger.getLogger(getClass().getName());
/** The format of the source. */
private Format format;
/** The input stream that will hold the converted source. */
private PipedInputStream pipedInputStream;
- /**
- * Creates a new external filter.
- *
- * @param binary
- * The binary to execute
- * @param parameters
- * The parameter for the binary
- */
- public ExternalFilter(String binary, Iterable<String> parameters) {
- this.binary = binary;
- this.parameters = parameters;
- }
-
//
// FILTER METHODS
+ //
@Override
public Format format() {
public byte[] get(int bufferSize) throws EOFException, IOException {
byte[] buffer = new byte[bufferSize];
int read = pipedInputStream.read(buffer);
+ if (read == -1) {
+ throw new EOFException();
+ }
return Arrays.copyOf(buffer, read);
}
format = source.format();
try {
- Process process = Runtime.getRuntime().exec(Iterables.toArray(ImmutableList.<String>builder().add(binary).addAll(parameters).build(), String.class));
+ final Process process = Runtime.getRuntime().exec(Iterables.toArray(ImmutableList.<String>builder().add(binary(format)).addAll(parameters(format)).build(), String.class));
final InputStream processOutput = process.getInputStream();
final OutputStream processInput = process.getOutputStream();
final InputStream processError = process.getErrorStream();
final PipedOutputStream pipedOutputStream = new PipedOutputStream();
pipedInputStream = new PipedInputStream(pipedOutputStream);
- new Thread(new Runnable() {
-
- @Override
- public void run() {
- try {
- drainInputStream(processError);
- } catch (IOException ioe1) {
- /* ignore, just let the thread exit. */
- }
- logger.finest("ExternalFilter: Reading stderr finished.");
- }
- }).start();
+ new Thread(new InputStreamDrainer(processError)).start();
new Thread(new Runnable() {
@Override
} catch (IOException ioe1) {
/* okay, just exit. */
}
- logger.finest("ExternalFilter: Reading stdout finished.");
+ logger.finest("Reading stdout finished.");
}
}).start();
new Thread(new Connection(source) {
@Override
protected void feed(byte[] buffer) throws IOException {
processInput.write(buffer);
+ processInput.flush();
+ }
+
+ @Override
+ protected void finish() throws IOException {
+ processInput.close();
+ processOutput.close();
+ processError.close();
}
}).start();
} catch (IOException ioe1) {
}
//
- // STATIC METHODS
+ // SUBCLASS METHODS
//
- private static void drainInputStream(InputStream inputStream) throws IOException {
- byte[] buffer = new byte[4096];
- int read;
- while ((read = inputStream.read(buffer)) != -1) {
- logger.finest(String.format("ExternalFilter: Drained %d Bytes.", read));
- /* do nothing, just read the damn thing. */
- }
- }
+ /**
+ * Returns the location of the binary to execute.
+ *
+ * @param format
+ * The format being processed
+ * @return The location of the binary to execute
+ */
+ protected abstract String binary(Format format);
+
+ /**
+ * Returns the parameters for the binary.
+ *
+ * @param format
+ * The format being processed
+ * @return The parameters for the binary
+ */
+ protected abstract Iterable<String> parameters(Format format);
}