org.hibernate.search.elasticsearch.util.impl.CountingOutputStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hibernate-search-elasticsearch Show documentation

Hibernate Search backend which has indexing operations forwarded to Elasticsearch

There is a newer version: 5.11.12.Final

Show newest version

/*
 * Hibernate Search, full-text search for your domain model
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or .
 */
package org.hibernate.search.elasticsearch.util.impl;

import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;

final class CountingOutputStream extends FilterOutputStream {

	private long bytesWritten = 0l;

	public CountingOutputStream(OutputStream out) {
		super( out );
	}

	@Override
	public void write(int b) throws IOException {
		out.write( b );
		count( 1 );
	}

	@Override
	public void write(byte[] b) throws IOException {
		write( b, 0, b.length );
	}

	@Override
	public void write(byte[] b, int off, int len) throws IOException {
		out.write( b, off, len );
		count( len );
	}

	protected void count(int written) {
		if ( written > 0 ) {
			bytesWritten += written;
		}
	}

	public long getBytesWritten() {
		return bytesWritten;
	}

}