diff options
Diffstat (limited to 'src/java_tools/singlejar/java/com/google/devtools/build/zip/CountingOutputStream.java')
-rw-r--r-- | src/java_tools/singlejar/java/com/google/devtools/build/zip/CountingOutputStream.java | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/src/java_tools/singlejar/java/com/google/devtools/build/zip/CountingOutputStream.java b/src/java_tools/singlejar/java/com/google/devtools/build/zip/CountingOutputStream.java new file mode 100644 index 0000000000..a08159ff14 --- /dev/null +++ b/src/java_tools/singlejar/java/com/google/devtools/build/zip/CountingOutputStream.java @@ -0,0 +1,54 @@ +// Copyright 2015 Google Inc. All rights reserved. +// +// Licensed 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 com.google.devtools.build.zip; + +import java.io.FilterOutputStream; +import java.io.IOException; +import java.io.OutputStream; + +/** An OutputStream that counts the number of bytes written. */ +final class CountingOutputStream extends FilterOutputStream { + + private long count; + + /** + * Wraps another output stream, counting the number of bytes written. + * + * @param out the output stream to be wrapped + */ + public CountingOutputStream(OutputStream out) { + super(out); + } + + /** Returns the number of bytes written. */ + public long getCount() { + return count; + } + + @Override public void write(int b) throws IOException { + out.write(b); + count++; + } + + @Override public void write(byte[] b) throws IOException { + out.write(b); + count += b.length; + } + + @Override public void write(byte[] b, int off, int len) throws IOException { + out.write(b, off, len); + count += len; + } +}
\ No newline at end of file |