com.regnosys.rosetta.common.util.TeeReader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rosetta-common Show documentation

Rune Common is a java library that is utilised by Rosetta Code Generators and models expressed in the Rosetta DSL.

There is a newer version: 11.27.1

package com.regnosys.rosetta.common.util;

/*-
 * ==============
 * Rune Common
 * ==============
 * Copyright (C) 2018 - 2024 REGnosys
 * ==============
 * 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.
 * ==============
 */

import java.io.IOException;
import java.io.Reader;
import java.util.concurrent.Phaser;

public class TeeReader {
	
	private Reader in;
	
	private char[] buffer = new char[1024];
	private int endPointer;
	private boolean ended=false;
	
	private Phaser barrier;
	
	private TeedReader[] tees;


	public TeeReader(Reader in) {
		this.in = in;
		readMoreUnderlying();
	}
	
	public void readerDeath() throws IOException {
		barrier.arriveAndDeregister();
		if (barrier.getRegisteredParties()==0) {
			in.close();
		}
	}
	
	public Reader[] splitInto(int count) {
		tees = new TeedReader[count];
		for (int i=0;i0) {
				System.arraycopy(buffer, pointer, cbuf, off, toRead);
				pointer+=toRead;
				return toRead;
			}
			else {
				boolean sucess=readMore();
				if (sucess) {
					return read(cbuf, off, len);
				}
			}
			return -1;
		}

		@Override
		public void close() throws IOException {
			readerDeath();
		}
		
	}
}