001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * Contributor(s):
025: *
026: * The Original Software is NetBeans. The Initial Developer of the Original
027: * Software is Sun Microsystems, Inc. Portions Copyright 1997-2007 Sun
028: * Microsystems, Inc. All Rights Reserved.
029: *
030: * If you wish your version of this file to be governed by only the CDDL
031: * or only the GPL Version 2, indicate your decision by adding
032: * "[Contributor] elects to include this software in this distribution
033: * under the [CDDL or GPL Version 2] license." If you do not indicate a
034: * single choice of license, a recipient has the option to distribute
035: * your version of this file under either the CDDL, the GPL Version 2 or
036: * to extend the choice of license to its licensees as provided above.
037: * However, if you add GPL Version 2 code and therefore, elected the GPL
038: * Version 2 license, then the option applies only if the new code is
039: * made subject to such option by the copyright holder.
040: */
041:
042: package org.netbeans.modules.server.output;
043:
044: import java.io.IOException;
045: import java.io.InputStream;
046: import java.nio.ByteBuffer;
047: import java.nio.channels.Channels;
048: import java.nio.channels.ReadableByteChannel;
049: import java.nio.charset.Charset;
050: import java.util.logging.Logger;
051: import org.netbeans.api.server.output.LineProcessor;
052: import org.netbeans.api.server.output.LineReader;
053: import org.openide.util.Parameters;
054:
055: /**
056: *
057: * This class is <i>NotThreadSafe</i>.
058: * @author Petr.Hejl
059: */
060: public class StreamLineReader implements LineReader {
061:
062: private static final Logger LOGGER = Logger
063: .getLogger(StreamLineReader.class.getName());
064:
065: private static final int BUFFER_SIZE = 128;
066:
067: private final LineParsingHelper helper = new LineParsingHelper();
068:
069: private final Charset charset;
070:
071: private final ReadableByteChannel channel;
072:
073: private final ByteBuffer buffer = ByteBuffer
074: .allocateDirect(BUFFER_SIZE);
075:
076: private boolean closed;
077:
078: public StreamLineReader(InputStream stream, Charset charset) {
079: Parameters.notNull("stream", stream);
080: Parameters.notNull("charset", charset);
081:
082: this .charset = charset;
083: this .channel = Channels.newChannel(stream);
084: }
085:
086: public int readLines(LineProcessor lineProcessor,
087: boolean allAvailable) throws IOException {
088: if (closed) {
089: throw new IllegalStateException("Already closed reader");
090: }
091:
092: int fetched = 0;
093:
094: buffer.clear();
095: int size = channel.read(buffer);
096: if (size > 0) {
097: buffer.position(0).limit(size);
098: String[] lines = helper.parse(buffer, charset);
099: fetched += lines.length;
100:
101: if (lineProcessor != null) {
102: for (String line : lines) {
103: lineProcessor.processLine(line);
104: }
105: }
106: }
107:
108: if (allAvailable) {
109: String line = helper.getTrailingLine(true);
110: if (line != null) {
111: if (lineProcessor != null) {
112: lineProcessor.processLine(line);
113: }
114: fetched += 1;
115: }
116: }
117:
118: return fetched;
119: }
120:
121: public void close() throws IOException {
122: closed = true;
123: channel.close();
124: }
125:
126: }
|