| /* |
| |
| Derby - Class org.apache.derbyTesting.functionTests.harness.BackgroundStreamDrainer |
| |
| Licensed to the Apache Software Foundation (ASF) under one or more |
| contributor license agreements. See the NOTICE file distributed with |
| this work for additional information regarding copyright ownership. |
| The ASF licenses this file to You under the Apache License, Version 2.0 |
| (the "License"); you may not use this file except in compliance with |
| the License. You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| |
| */ |
| |
| package org.apache.derbyTesting.functionTests.harness; |
| |
| import java.io.*; |
| import java.sql.Timestamp; |
| |
| public class BackgroundStreamDrainer implements Runnable |
| { |
| |
| protected ByteArrayOutputStream data; |
| protected InputStream in; |
| protected boolean finished; |
| protected IOException ioe; |
| protected long startTime; |
| protected Thread myThread; |
| protected int timeout; |
| |
| public BackgroundStreamDrainer(InputStream in, String timemin) |
| { |
| data = new ByteArrayOutputStream(); |
| this.in = in; |
| this.startTime = System.currentTimeMillis();; |
| if (timemin != null) |
| { |
| Integer i = Integer.valueOf(timemin); |
| timeout = i.intValue(); |
| } |
| else |
| timeout = 0; |
| //System.out.println("timeout set to: " + timeout); |
| |
| myThread = new Thread(this, getClass().getName()); |
| myThread.setPriority(Thread.MIN_PRIORITY); |
| myThread.start(); |
| } |
| |
| public void run() |
| { |
| if ( in == null ) |
| { |
| System.out.println("The inputstream is null"); |
| System.exit(1); |
| } |
| |
| try |
| { |
| byte[] ca = new byte[1024]; |
| int valid; |
| while ((valid = in.read(ca, 0, ca.length)) != -1) |
| { |
| if (timeout > 0) |
| { |
| long millis = System.currentTimeMillis(); |
| |
| long diff = millis - startTime; |
| |
| int mins = (int) (diff / (1000 * 60)); |
| |
| if (mins > timeout) { |
| |
| System.out.println("kill stderr thread..."); |
| synchronized (this) |
| { |
| finished = true; |
| break; |
| } |
| } |
| } |
| //System.out.println("Bytes read to write data: " + valid); |
| data.write(ca, 0, valid); |
| } |
| } |
| catch (IOException ioe) |
| { |
| this.ioe = ioe; |
| System.out.println(ioe.getMessage()); |
| } |
| |
| synchronized (this) |
| { |
| finished = true; |
| notifyAll(); |
| } |
| } |
| |
| public InputStream getData() throws IOException |
| { |
| // FIXME: On Netware, the last read throws an IOException, |
| // which prevents the test output from getting written |
| //if (ioe != null) |
| //{ |
| //throw ioe; |
| //} |
| |
| synchronized (this) |
| { |
| try |
| { |
| while (!finished) |
| { |
| wait(); |
| } |
| } catch (InterruptedException ie) |
| { |
| System.out.println("IOException: " + ie); |
| throw new IOException(ie.toString()); |
| } |
| } |
| return new ByteArrayInputStream(data.toByteArray()); |
| } |
| } |