| /* |
| * 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 java.io; |
| |
| import java.util.Arrays; |
| import libcore.util.SneakyThrow; |
| |
| /** |
| * Wraps an existing {@link Writer} and <em>buffers</em> the output. Expensive |
| * interaction with the underlying reader is minimized, since most (smaller) |
| * requests can be satisfied by accessing the buffer alone. The drawback is that |
| * some extra space is required to hold the buffer and that copying takes place |
| * when filling that buffer, but this is usually outweighed by the performance |
| * benefits. |
| * |
| * <p/>A typical application pattern for the class looks like this:<p/> |
| * |
| * <pre> |
| * BufferedWriter buf = new BufferedWriter(new FileWriter("file.java")); |
| * </pre> |
| * |
| * @see BufferedReader |
| */ |
| public class BufferedWriter extends Writer { |
| |
| private Writer out; |
| |
| private char[] buf; |
| |
| private int pos; |
| |
| /** |
| * Constructs a new {@code BufferedWriter}, providing {@code out} with a buffer |
| * of 8192 chars. |
| * |
| * @param out the {@code Writer} the buffer writes to. |
| */ |
| public BufferedWriter(Writer out) { |
| this(out, 8192); |
| } |
| |
| /** |
| * Constructs a new {@code BufferedWriter}, providing {@code out} with {@code size} chars |
| * of buffer. |
| * |
| * @param out the {@code OutputStream} the buffer writes to. |
| * @param size the size of buffer in chars. |
| * @throws IllegalArgumentException if {@code size <= 0}. |
| */ |
| public BufferedWriter(Writer out, int size) { |
| super(out); |
| if (size <= 0) { |
| throw new IllegalArgumentException("size <= 0"); |
| } |
| this.out = out; |
| this.buf = new char[size]; |
| } |
| |
| /** |
| * Closes this writer. The contents of the buffer are flushed, the target |
| * writer is closed, and the buffer is released. Only the first invocation |
| * of close has any effect. |
| * |
| * @throws IOException |
| * if an error occurs while closing this writer. |
| */ |
| @Override |
| public void close() throws IOException { |
| synchronized (lock) { |
| if (isClosed()) { |
| return; |
| } |
| |
| Throwable thrown = null; |
| try { |
| flushInternal(); |
| } catch (Throwable e) { |
| thrown = e; |
| } |
| buf = null; |
| |
| try { |
| out.close(); |
| } catch (Throwable e) { |
| if (thrown == null) { |
| thrown = e; |
| } |
| } |
| out = null; |
| |
| if (thrown != null) { |
| SneakyThrow.sneakyThrow(thrown); |
| } |
| } |
| } |
| |
| /** |
| * Flushes this writer. The contents of the buffer are committed to the |
| * target writer and it is then flushed. |
| * |
| * @throws IOException |
| * if an error occurs while flushing this writer. |
| */ |
| @Override |
| public void flush() throws IOException { |
| synchronized (lock) { |
| checkNotClosed(); |
| flushInternal(); |
| out.flush(); |
| } |
| } |
| |
| private void checkNotClosed() throws IOException { |
| if (isClosed()) { |
| throw new IOException("BufferedWriter is closed"); |
| } |
| } |
| |
| /** |
| * Flushes the internal buffer. |
| */ |
| private void flushInternal() throws IOException { |
| if (pos > 0) { |
| out.write(buf, 0, pos); |
| } |
| pos = 0; |
| } |
| |
| /** |
| * Indicates whether this writer is closed. |
| * |
| * @return {@code true} if this writer is closed, {@code false} otherwise. |
| */ |
| private boolean isClosed() { |
| return out == null; |
| } |
| |
| /** |
| * Writes a newline to this writer. On Android, this is {@code "\n"}. |
| * The target writer may or may not be flushed when a newline is written. |
| * |
| * @throws IOException |
| * if an error occurs attempting to write to this writer. |
| */ |
| public void newLine() throws IOException { |
| write(System.lineSeparator()); |
| } |
| |
| /** |
| * Writes {@code count} characters starting at {@code offset} in |
| * {@code buffer} to this writer. If {@code count} is greater than this |
| * writer's buffer, then the buffer is flushed and the characters are |
| * written directly to the target writer. |
| * |
| * @param buffer |
| * the array containing characters to write. |
| * @param offset |
| * the start position in {@code buffer} for retrieving characters. |
| * @param count |
| * the maximum number of characters to write. |
| * @throws IndexOutOfBoundsException |
| * if {@code offset < 0} or {@code count < 0}, or if |
| * {@code offset + count} is greater than the size of |
| * {@code buffer}. |
| * @throws IOException |
| * if this writer is closed or another I/O error occurs. |
| */ |
| @Override |
| public void write(char[] buffer, int offset, int count) throws IOException { |
| synchronized (lock) { |
| checkNotClosed(); |
| if (buffer == null) { |
| throw new NullPointerException("buffer == null"); |
| } |
| Arrays.checkOffsetAndCount(buffer.length, offset, count); |
| if (pos == 0 && count >= this.buf.length) { |
| out.write(buffer, offset, count); |
| return; |
| } |
| int available = this.buf.length - pos; |
| if (count < available) { |
| available = count; |
| } |
| if (available > 0) { |
| System.arraycopy(buffer, offset, this.buf, pos, available); |
| pos += available; |
| } |
| if (pos == this.buf.length) { |
| out.write(this.buf, 0, this.buf.length); |
| pos = 0; |
| if (count > available) { |
| offset += available; |
| available = count - available; |
| if (available >= this.buf.length) { |
| out.write(buffer, offset, available); |
| return; |
| } |
| |
| System.arraycopy(buffer, offset, this.buf, pos, available); |
| pos += available; |
| } |
| } |
| } |
| } |
| |
| /** |
| * Writes the character {@code oneChar} to this writer. If the buffer |
| * gets full by writing this character, this writer is flushed. Only the |
| * lower two bytes of the integer {@code oneChar} are written. |
| * |
| * @param oneChar |
| * the character to write. |
| * @throws IOException |
| * if this writer is closed or another I/O error occurs. |
| */ |
| @Override |
| public void write(int oneChar) throws IOException { |
| synchronized (lock) { |
| checkNotClosed(); |
| if (pos >= buf.length) { |
| out.write(buf, 0, buf.length); |
| pos = 0; |
| } |
| buf[pos++] = (char) oneChar; |
| } |
| } |
| |
| /** |
| * Writes {@code count} characters starting at {@code offset} in {@code str} |
| * to this writer. If {@code count} is greater than this writer's buffer, |
| * then this writer is flushed and the remaining characters are written |
| * directly to the target writer. If count is negative no characters are |
| * written to the buffer. This differs from the behavior of the superclass. |
| * |
| * @param str |
| * the non-null String containing characters to write. |
| * @param offset |
| * the start position in {@code str} for retrieving characters. |
| * @param count |
| * maximum number of characters to write. |
| * @throws IOException |
| * if this writer has already been closed or another I/O error |
| * occurs. |
| * @throws IndexOutOfBoundsException |
| * if {@code offset < 0} or {@code offset + count} is greater |
| * than the length of {@code str}. |
| */ |
| @Override |
| public void write(String str, int offset, int count) throws IOException { |
| synchronized (lock) { |
| checkNotClosed(); |
| if (count <= 0) { |
| return; |
| } |
| if (offset < 0 || offset > str.length() - count) { |
| throw new StringIndexOutOfBoundsException(str, offset, count); |
| } |
| if (pos == 0 && count >= buf.length) { |
| char[] chars = new char[count]; |
| str.getChars(offset, offset + count, chars, 0); |
| out.write(chars, 0, count); |
| return; |
| } |
| int available = buf.length - pos; |
| if (count < available) { |
| available = count; |
| } |
| if (available > 0) { |
| str.getChars(offset, offset + available, buf, pos); |
| pos += available; |
| } |
| if (pos == buf.length) { |
| out.write(this.buf, 0, this.buf.length); |
| pos = 0; |
| if (count > available) { |
| offset += available; |
| available = count - available; |
| if (available >= buf.length) { |
| char[] chars = new char[count]; |
| str.getChars(offset, offset + available, chars, 0); |
| out.write(chars, 0, available); |
| return; |
| } |
| str.getChars(offset, offset + available, buf, pos); |
| pos += available; |
| } |
| } |
| } |
| } |
| } |