Author: richard.opalka(a)jboss.com Date: 2008-09-25 08:33:29 -0400 (Thu, 25 Sep 2008) New Revision: 8224 Added: common/trunk/src/main/java/org/jboss/wsf/common/concurrent/ common/trunk/src/main/java/org/jboss/wsf/common/concurrent/CopyJob.java Log: [JBWS-2322] adding helper class Added: common/trunk/src/main/java/org/jboss/wsf/common/concurrent/CopyJob.java =================================================================== --- common/trunk/src/main/java/org/jboss/wsf/common/concurrent/CopyJob.java (rev 0) +++ common/trunk/src/main/java/org/jboss/wsf/common/concurrent/CopyJob.java 2008-09-25 12:33:29 UTC (rev 8224) @@ -0,0 +1,142 @@ +/* + * JBoss, Home of Professional Open Source. + * Copyright 2008, Red Hat Middleware LLC, and individual contributors + * as indicated by the @author tags. See the copyright.txt file in the + * distribution for a full listing of individual contributors. + * + * This is free software; you can redistribute it and/or modify it + * under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1 of + * the License, or (at your option) any later version. + * + * This software is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this software; if not, write to the Free + * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA + * 02110-1301 USA, or see the FSF site: http://www.fsf.org. + */ +package org.jboss.wsf.common.concurrent; + +import java.io.IOException; +import java.io.InputStream; +import java.io.PrintStream; + +/** + * Sample usage: + * + * <blockquote><pre> + * CopyJob copyJob = new CopyJob( inputStream, printStream ); + * new Thread( copyJob ).start(); + * try + * { + * // do some other staff + * ... + * } + * finally + * { + * copyJob.kill(); + * } + * </pre></blockquote> + * + * @author richard.opalka(a)jboss.com + */ +public final class CopyJob implements Runnable +{ + + /** + * Input stream to data read from. + */ + private final InputStream is; + /** + * Output stream to write data to. + */ + private final PrintStream os; + /** + * Whether this job is terminated. + */ + private boolean terminated; + + /** + * Constructor. + * @param is input stream to read data from + * @param os output stream to write data to + */ + public CopyJob( InputStream is, PrintStream os ) + { + super(); + + if ( ( is == null ) || ( os == null ) ) + { + throw new IllegalArgumentException( "Constructor parameters can't be null" ); + } + + this.is = is; + this.os = os; + } + + /** + * Copies all data from <b>input stream</b> to <b>output stream</b> (both passed to constructor) until job is killed + */ + public final void run() + { + try + { + copy( this.is, this.os ); + } + catch ( IOException ioe ) + { + ioe.printStackTrace( this.os ); + } + finally + { + try { this.is.close(); } catch ( IOException ioe ) { ioe.printStackTrace( this.os ); } + } + } + + /** + * Copies all data from <b>is</b> to <b>os</b> until job is killed + * @param is input stream to read data from + * @param os output stream to write data to + * @throws IOException if I/O error occurs + */ + private void copy( final InputStream is, final PrintStream os ) throws IOException + { + final byte[] buffer = new byte[ 512 ]; + int countOfBytes = -1; + + while ( !this.terminated ) + { + while ( is.available() <= 0 ) + { + synchronized( this ) + { + try + { + if ( this.terminated ) return; + this.wait( 10 ); // guard + } + catch ( InterruptedException ie ) + { + ie.printStackTrace( os ); + } + } + } + + countOfBytes = is.read( buffer, 0, buffer.length ); + os.write( buffer, 0, countOfBytes ); + } + } + + /** + * Kills this job. Calling this method also ensures that input stream passed to the constructor will be closed properly + */ + public final void kill() + { + this.terminated = true; + } + +} \ No newline at end of file