web-dev-qa-db-fra.com

Comment écrire des données sur deux objets Java.io.OutputStream simultanément?

Je cherche une classe Java magique qui me permettra de faire quelque chose comme ceci:

ByteArrayOutputStream byteStream = new ByteArrayOutputStream();
FileOutputStream fileStream = new FileOutputStream(new File("/tmp/somefile"));

MultiOutputStream outStream = new MultiOutputStream(byteStream, fileStream);

outStream.write("Hello world".getBytes());

En gros, je veux tee pour OutputStreams en Java. Des idées?

Merci!

javaiooutputstream
30
daveslab

Essayez le Apache Commons TeeOutputStream .

37
Dave Newton

Il suffit de rouler le vôtre. Il n'y a pas de magie du tout. En utilisant TeeOutputStream d'Apache , vous utiliseriez essentiellement le code ci-dessous. Bien sûr, en utilisant la bibliothèque d’E/S Apache Commons, vous pouvez utiliser d’autres classes, mais il est parfois agréable de en fait écrire quelque chose pour vous. :)

public final class TeeOutputStream extends OutputStream {

  private final OutputStream out;
  private final OutputStream tee;

  public TeeOutputStream(OutputStream out, OutputStream tee) {
    if (out == null)
      throw new NullPointerException();
    else if (tee == null)
      throw new NullPointerException();

    this.out = out;
    this.tee = tee;
  }

  @Override
  public void write(int b) throws IOException {
    out.write(b);
    tee.write(b);
  }

  @Override
  public void write(byte[] b) throws IOException {
    out.write(b);
    tee.write(b);
  }

  @Override
  public void write(byte[] b, int off, int len) throws IOException {
    out.write(b, off, len);
    tee.write(b, off, len);
  }

  @Override
  public void flush() throws IOException {
    out.flush();
    tee.flush();
  }

  @Override
  public void close() throws IOException {
    try {
      out.close();
    } finally {
      tee.close();
    }
  }
}

Test avec la classe ci-dessus avec ce qui suit

public static void main(String[] args) throws IOException {
  TeeOutputStream out = new TeeOutputStream(System.out, System.out);
  out.write("Hello world!".getBytes());
  out.flush();
  out.close();
}

imprimerait Hello World!Hello World!.

(Remarque: la close() surchargée pourrait prendre un peu de soin :)

19
Kohányi Róbert

Je viens de trouver ce fil parce que je devais faire face au même problème. Si quelqu'un veut voir ma solution (code Java7):

package Core;
import Java.io.IOException;
import Java.io.OutputStream;
import Java.util.Iterator;
import Java.util.LinkedList;
import Java.util.List;

public class MultiOutputStream extends OutputStream {

private List<OutputStream> out;

public MultiOutputStream(List<OutputStream> outStreams) {

    this.out = new LinkedList<OutputStream>();

    for (Iterator<OutputStream> i = outStreams.iterator(); i.hasNext();) {
        OutputStream outputStream = (OutputStream) i.next();

        if(outputStream == null){
            throw new NullPointerException();
        }
        this.out.add(outputStream);
    }
}

@Override
public void write(int arg0) throws IOException {

    for (Iterator<OutputStream> i = out.iterator(); i.hasNext();) {
        OutputStream var = (OutputStream) i.next();

        var.write(arg0);
    }
}

@Override
public void write(byte[] b) throws IOException{

    for (Iterator<OutputStream> i = out.iterator(); i.hasNext();) {
        OutputStream var = (OutputStream) i.next();

        var.write(b);
    }
}

@Override
public void write(byte[] b, int off, int len) throws IOException{

    for (Iterator<OutputStream> i = out.iterator(); i.hasNext();) {
        OutputStream var = (OutputStream) i.next();

        var.write(b, off, len);
    }
}

@Override
public void close() throws IOException{

    for (Iterator<OutputStream> i = out.iterator(); i.hasNext();) {
        OutputStream var = (OutputStream) i.next();

        var.close();
    }
}

@Override
public void flush() throws IOException{

    for (Iterator<OutputStream> i = out.iterator(); i.hasNext();) {
        OutputStream var = (OutputStream) i.next();

        var.flush();
    }
}

}

Fonctionne bien jusqu'à présent, vient de tester quelques opérations de base, par exemple. configuration d’un flux MultiOutputStream à partir du flux System.out et de deux flux d’écran, chacun écrivant dans un journal séparé. J’ai utilisé

System.setOut(multiOutputStream);

pour écrire sur mon écran de terminal et deux journaux qui ont fonctionné sans aucun problème.

3
Lejon Doering

Rouler le vôtre, c'est fondamentalement trivial. Utilisez un ArrayList<OutputStream> ou tout ce qui est populaire de nos jours pour stocker tous les flux souhaités et écrivez la méthode write pour les afficher en boucle. 

2
Kevin
final ByteArrayOutputStream byteStream = new ByteArrayOutputStream();
final FileOutputStream fileStream = new FileOutputStream(new File("/tmp/somefile"));
OutputStream outStream = new OutputStream() {

    public void write(int b) throws IOException {
        byteStream.write(b);
        fileStream.write(b);
    }
};
outStream.write("Hello world".getBytes());
0
Daniel De León