1bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor/* 21d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Copyright (C) 2008 The Guava Authors 3bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * 4bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * Licensed under the Apache License, Version 2.0 (the "License"); 5bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * you may not use this file except in compliance with the License. 6bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * You may obtain a copy of the License at 7bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * 8bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * http://www.apache.org/licenses/LICENSE-2.0 9bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * 10bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * Unless required by applicable law or agreed to in writing, software 11bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * distributed under the License is distributed on an "AS IS" BASIS, 12bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * See the License for the specific language governing permissions and 14bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * limitations under the License. 15bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor */ 16bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 17bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnorpackage com.google.common.io; 18bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 19bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnorimport com.google.common.base.Preconditions; 20bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 21bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnorimport java.io.IOException; 221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.io.Reader; 23bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnorimport java.util.Iterator; 24bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 25bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor/** 261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * A {@link Reader} that concatenates multiple readers. 27bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * 28bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * @author Bin Zhu 291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @since 1.0 30bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor */ 31bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnorclass MultiReader extends Reader { 32bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor private final Iterator<? extends InputSupplier<? extends Reader>> it; 33bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor private Reader current; 34bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 35bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor MultiReader(Iterator<? extends InputSupplier<? extends Reader>> readers) 36bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor throws IOException { 37bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor this.it = readers; 38bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor advance(); 39bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 40bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 41bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor /** 42bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor * Closes the current reader and opens the next one, if any. 43bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor */ 44bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor private void advance() throws IOException { 45bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor close(); 46bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor if (it.hasNext()) { 47bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor current = it.next().getInput(); 48bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 49bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 50bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 51bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor @Override public int read(char cbuf[], int off, int len) throws IOException { 52bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor if (current == null) { 53bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor return -1; 54bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 55bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor int result = current.read(cbuf, off, len); 56bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor if (result == -1) { 57bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor advance(); 58bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor return read(cbuf, off, len); 59bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 60bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor return result; 61bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 62bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 63bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor @Override public long skip(long n) throws IOException { 64bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor Preconditions.checkArgument(n >= 0, "n is negative"); 65bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor if (n > 0) { 66bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor while (current != null) { 67bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor long result = current.skip(n); 68bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor if (result > 0) { 69bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor return result; 70bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 71bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor advance(); 72bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 73bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 74bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor return 0; 75bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 76bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 77bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor @Override public boolean ready() throws IOException { 78bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor return (current != null) && current.ready(); 79bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 80bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor 81bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor @Override public void close() throws IOException { 82bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor if (current != null) { 83bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor try { 84bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor current.close(); 85bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } finally { 86bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor current = null; 87bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 88bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 89bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor } 90bfe2dd089341dcb4c1fb65a5b6b077ad0ebbf6dcDan Egnor} 91