package net.pterodactylus.sonitus.data.sink;
-import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
+import java.util.Collections;
+import java.util.List;
import java.util.logging.Logger;
-import net.pterodactylus.sonitus.data.ConnectException;
-import net.pterodactylus.sonitus.data.Connection;
+import net.pterodactylus.sonitus.data.AbstractControlledComponent;
+import net.pterodactylus.sonitus.data.Controller;
+import net.pterodactylus.sonitus.data.Metadata;
import net.pterodactylus.sonitus.data.Sink;
-import net.pterodactylus.sonitus.data.Source;
+import net.pterodactylus.sonitus.data.event.MetadataUpdated;
-import com.google.common.base.Preconditions;
+import com.google.common.eventbus.EventBus;
/**
- * {@link Sink} that writes all received data into a file.
+ * {@link net.pterodactylus.sonitus.data.Sink} that writes all received data
+ * into a file.
*
* @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
*/
-public class FileSink implements Sink {
+public class FileSink extends AbstractControlledComponent implements Sink {
/** The logger. */
private static final Logger logger = Logger.getLogger(FileSink.class.getName());
+ /** The event bus. */
+ private final EventBus eventBus;
+
/** The path of the file to write to. */
private final String path;
+ /** The output stream writing to the file. */
+ private FileOutputStream fileOutputStream;
+
+ /** The current metadata. */
+ private Metadata metadata;
+
/**
* Creates a new file sink that will write to the given path.
*
+ * @param eventBus
+ * The event bus
* @param path
* The path of the file to write to
*/
- public FileSink(String path) {
+ public FileSink(EventBus eventBus, String path) {
+ this.eventBus = eventBus;
this.path = path;
}
+ //
+ // CONTROLLED METHODS
+ //
+
+ @Override
+ public String name() {
+ return path;
+ }
+
+ @Override
+ public Metadata metadata() {
+ return metadata;
+ }
+
+ @Override
+ public List<Controller<?>> controllers() {
+ return Collections.emptyList();
+ }
+
+ //
+ // SINK METHODS
+ //
+
@Override
- public void connect(Source source) throws ConnectException {
- Preconditions.checkNotNull(source, "source must not be null");
+ public void open(Metadata metadata) throws IOException {
+ fileOutputStream = new FileOutputStream(path);
+ metadataUpdated(metadata);
+ }
+ @Override
+ public void close() {
try {
- final FileOutputStream fileOutputStream = new FileOutputStream(path);
- new Thread(new Connection(source) {
-
- @Override
- protected int bufferSize() {
- return 65536;
- }
-
- @Override
- protected void feed(byte[] buffer) throws IOException {
- fileOutputStream.write(buffer);
- logger.finest(String.format("FileSink: Wrote %d Bytes.", buffer.length));
- }
-
- @Override
- protected void finish() throws IOException {
- fileOutputStream.close();
- }
- }).start();
- } catch (FileNotFoundException fnfe1) {
- throw new ConnectException(fnfe1);
+ fileOutputStream.close();
+ } catch (IOException e) {
+ /* ignore. */
}
}
+ @Override
+ public void metadataUpdated(Metadata metadata) {
+ this.metadata = metadata;
+ fireMetadataUpdated(metadata);
+ eventBus.post(new MetadataUpdated(this, metadata));
+ }
+
+ @Override
+ public void process(byte[] buffer) throws IOException {
+ fileOutputStream.write(buffer);
+ logger.finest(String.format("FileSink: Wrote %d Bytes.", buffer.length));
+ }
+
}