1 /* 2 * Licensed to the Apache Software Foundation (ASF) under one 3 * or more contributor license agreements. See the NOTICE file 4 * distributed with this work for additional information 5 * regarding copyright ownership. The ASF licenses this file 6 * to you under the Apache License, Version 2.0 (the 7 * "License"); you may not use this file except in compliance 8 * with the License. You may obtain a copy of the License at 9 * 10 * http://www.apache.org/licenses/LICENSE-2.0 11 * 12 * Unless required by applicable law or agreed to in writing, software 13 * distributed under the License is distributed on an "AS IS" BASIS, 14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 * See the License for the specific language governing permissions and 16 * limitations under the License. 17 */ 18 package org.apache.giraph.writable.kryo; 19 20 import java.io.DataInput; 21 import java.io.EOFException; 22 import java.io.IOException; 23 import java.io.InputStream; 24 25 /** 26 * Thin wrapper around DataInput so it can be used as an 27 * {@link java.io.InputStream} 28 * 29 * For use with {@link com.esotericsoftware.kryo.io.Input} 30 */ 31 public class DataInputWrapperStream extends InputStream { 32 /** Wrapped DataInput object */ 33 private DataInput in; 34 35 public void setDataInput(DataInput in) { 36 this.in = in; 37 } 38 39 @Override 40 public int read() throws IOException { 41 try { 42 return in.readByte() & 0xFF; 43 } catch (EOFException e) { 44 throw new RuntimeException( 45 "Chunked input should never read more than chunked output wrote", e); 46 } 47 } 48 49 @Override 50 public int read(byte[] b, int off, int len) throws IOException { 51 try { 52 in.readFully(b, off, len); 53 return len; 54 } catch (EOFException e) { 55 throw new RuntimeException( 56 "Chunked input should never read more than chunked output wrote", e); 57 } 58 } 59 }