001/* 002 * Logback: the reliable, generic, fast and flexible logging framework. 003 * Copyright (C) 1999-2022, QOS.ch. All rights reserved. 004 * 005 * This program and the accompanying materials are dual-licensed under 006 * either the terms of the Eclipse Public License v1.0 as published by 007 * the Eclipse Foundation 008 * 009 * or (per the licensee's choosing) 010 * 011 * under the terms of the GNU Lesser General Public License version 2.1 012 * as published by the Free Software Foundation. 013 */ 014package ch.qos.logback.core.testUtil; 015 016import java.io.ByteArrayOutputStream; 017import java.io.IOException; 018import java.io.OutputStream; 019import java.io.PrintStream; 020 021/** 022 * This stream writes its output to the target PrintStream supplied to its 023 * constructor. At the same time, all the available bytes are collected and 024 * returned by the toString() method. 025 * 026 * @author Ceki Gulcu 027 */ 028public class TeeOutputStream extends OutputStream { 029 030 final PrintStream targetPS; 031 public final ByteArrayOutputStream baos = new ByteArrayOutputStream(); 032 033 public TeeOutputStream(PrintStream targetPS) { 034 // allow for null arguments 035 this.targetPS = targetPS; 036 } 037 038 public void write(int b) throws IOException { 039 baos.write(b); 040 if (targetPS != null) { 041 targetPS.write(b); 042 } 043 } 044 045 public String toString() { 046 return baos.toString(); 047 } 048 049 public byte[] toByteArray() { 050 return baos.toByteArray(); 051 } 052}