mirror of
https://github.com/moparisthebest/Conversations
synced 2024-08-13 15:53:48 -04:00
115 lines
2.6 KiB
Java
115 lines
2.6 KiB
Java
|
package eu.siacs.conversations.xml;
|
||
|
|
||
|
import java.io.IOException;
|
||
|
import java.io.OutputStream;
|
||
|
import java.io.OutputStreamWriter;
|
||
|
import java.util.concurrent.LinkedBlockingQueue;
|
||
|
|
||
|
import eu.siacs.conversations.xmpp.stanzas.AbstractStanza;
|
||
|
|
||
|
public class TagWriter {
|
||
|
|
||
|
private OutputStream plainOutputStream;
|
||
|
private OutputStreamWriter outputStream;
|
||
|
private boolean finshed = false;
|
||
|
private LinkedBlockingQueue<AbstractStanza> writeQueue = new LinkedBlockingQueue<AbstractStanza>();
|
||
|
private Thread asyncStanzaWriter = new Thread() {
|
||
|
private boolean shouldStop = false;
|
||
|
|
||
|
@Override
|
||
|
public void run() {
|
||
|
while (!shouldStop) {
|
||
|
if ((finshed) && (writeQueue.size() == 0)) {
|
||
|
return;
|
||
|
}
|
||
|
try {
|
||
|
AbstractStanza output = writeQueue.take();
|
||
|
if (outputStream == null) {
|
||
|
shouldStop = true;
|
||
|
} else {
|
||
|
outputStream.write(output.toString());
|
||
|
outputStream.flush();
|
||
|
}
|
||
|
} catch (IOException e) {
|
||
|
shouldStop = true;
|
||
|
} catch (InterruptedException e) {
|
||
|
shouldStop = true;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
public TagWriter() {
|
||
|
}
|
||
|
|
||
|
public void setOutputStream(OutputStream out) throws IOException {
|
||
|
if (out == null) {
|
||
|
throw new IOException();
|
||
|
}
|
||
|
this.plainOutputStream = out;
|
||
|
this.outputStream = new OutputStreamWriter(out);
|
||
|
}
|
||
|
|
||
|
public OutputStream getOutputStream() throws IOException {
|
||
|
if (this.plainOutputStream == null) {
|
||
|
throw new IOException();
|
||
|
}
|
||
|
return this.plainOutputStream;
|
||
|
}
|
||
|
|
||
|
public TagWriter beginDocument() throws IOException {
|
||
|
if (outputStream == null) {
|
||
|
throw new IOException("output stream was null");
|
||
|
}
|
||
|
outputStream.write("<?xml version='1.0'?>");
|
||
|
outputStream.flush();
|
||
|
return this;
|
||
|
}
|
||
|
|
||
|
public TagWriter writeTag(Tag tag) throws IOException {
|
||
|
if (outputStream == null) {
|
||
|
throw new IOException("output stream was null");
|
||
|
}
|
||
|
outputStream.write(tag.toString());
|
||
|
outputStream.flush();
|
||
|
return this;
|
||
|
}
|
||
|
|
||
|
public TagWriter writeElement(Element element) throws IOException {
|
||
|
if (outputStream == null) {
|
||
|
throw new IOException("output stream was null");
|
||
|
}
|
||
|
outputStream.write(element.toString());
|
||
|
outputStream.flush();
|
||
|
return this;
|
||
|
}
|
||
|
|
||
|
public TagWriter writeStanzaAsync(AbstractStanza stanza) {
|
||
|
if (finshed) {
|
||
|
return this;
|
||
|
} else {
|
||
|
if (!asyncStanzaWriter.isAlive()) {
|
||
|
try {
|
||
|
asyncStanzaWriter.start();
|
||
|
} catch (IllegalThreadStateException e) {
|
||
|
// already started
|
||
|
}
|
||
|
}
|
||
|
writeQueue.add(stanza);
|
||
|
return this;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public void finish() {
|
||
|
this.finshed = true;
|
||
|
}
|
||
|
|
||
|
public boolean finished() {
|
||
|
return (this.writeQueue.size() == 0);
|
||
|
}
|
||
|
|
||
|
public boolean isActive() {
|
||
|
return outputStream != null;
|
||
|
}
|
||
|
}
|