/***********************************************************************************************************************
* Copyright (C) 2010-2013 by the Stratosphere project (http://stratosphere.eu)
*
* 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 eu.stratosphere.nephele.types;
import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.Arrays;
import eu.stratosphere.core.io.IOReadableWritable;
import eu.stratosphere.core.io.StringRecord;
public class FileRecord implements IOReadableWritable {
private String fileName;
private static final byte[] EMPTY_BYTES = new byte[0];
private byte[] bytes;
public FileRecord() {
this.bytes = EMPTY_BYTES;
fileName = "empty";
}
public FileRecord(final String fileName) {
this.bytes = EMPTY_BYTES;
this.fileName = fileName;
}
public void setFileName(final String fileName) {
this.fileName = fileName;
}
public String getFileName() {
return this.fileName;
}
public byte[] getDataBuffer() {
return this.bytes;
}
/**
* Append a range of bytes to the end of the given data.
*
* @param data
* the data to copy from
* @param start
* the first position to append from data
* @param len
* the number of bytes to append
*/
public void append(final byte[] data, final int start, final int len) {
final int oldLength = this.bytes.length;
setCapacity(this.bytes.length + len, true);
System.arraycopy(data, start, this.bytes, oldLength, len);
}
private void setCapacity(final int len, final boolean keepData) {
if (this.bytes == null || this.bytes.length < len) {
final byte[] newBytes = new byte[len];
if (this.bytes != null && keepData) {
System.arraycopy(this.bytes, 0, newBytes, 0, this.bytes.length);
}
this.bytes = newBytes;
}
}
@Override
public void read(final DataInput in) throws IOException {
this.fileName = StringRecord.readString(in);
final int newLength = in.readInt();
this.bytes = new byte[newLength];
in.readFully(this.bytes, 0, newLength);
}
@Override
public void write(final DataOutput out) throws IOException {
StringRecord.writeString(out, fileName);
out.writeInt(this.bytes.length);
out.write(this.bytes, 0, this.bytes.length);
}
@Override
public boolean equals(final Object obj) {
if (!(obj instanceof FileRecord)) {
return false;
}
final FileRecord fr = (FileRecord) obj;
if (this.bytes.length != fr.bytes.length) {
return false;
}
return Arrays.equals(this.bytes, fr.bytes);
}
@Override
public int hashCode() {
return (int) ((11L * this.bytes.length) % Integer.MAX_VALUE);
}
}