package com.facebook.hive.orc;
import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.hadoop.io.BytesWritable;
import org.apache.hadoop.io.Text;
public class OrcTestUtils {
public static class InnerStruct {
Integer int1;
Text string1 = new Text();
InnerStruct(Integer int1, String string1) {
this.int1 = int1;
if (string1 == null) {
this.string1 = null;
} else {
this.string1.set(string1);
}
}
}
public static class StringStruct {
Text string1 = new Text();
StringStruct(String string1) {
if (string1 == null) {
this.string1 = null;
} else {
this.string1.set(string1);
}
}
}
public static class IntStruct {
Integer int1;
IntStruct(Integer int1) {
this.int1 = int1;
}
}
public static class MiddleStruct {
List<InnerStruct> list = null;
public MiddleStruct(InnerStruct... items) {
list = new ArrayList<InnerStruct>(items.length);
for(InnerStruct item: items) {
list.add(item);
}
}
}
/**
*
* ReallyBigRow.
*
* A BigRow with additional String and numeric columns,
* useful for testing both dictionary and directly encoded columns
*
* Unfortunately this cannot simply extend BigRow in order to make the reflection
* ObjectInspector work.
*/
public static class ReallyBigRow {
Boolean boolean1;
Byte byte1;
Short short1;
Integer int1;
Long long1;
Short short2;
Integer int2;
Long long2;
Short short3;
Integer int3;
Long long3;
Float float1;
Double double1;
BytesWritable bytes1;
Text string1;
Text string2;
Text string3;
MiddleStruct middle;
List<InnerStruct> list = null;
Map<Text, InnerStruct> map = null;
public ReallyBigRow(Boolean b1, Byte b2, Short s1, Integer i1, Long l1, Short s2, Integer i2,
Long l2, Short s3, Integer i3, Long l3, Float f1, Double d1, BytesWritable b3, String s4,
String s5, String s6, MiddleStruct m1, List<InnerStruct> l4, Map<Text, InnerStruct> m2) {
this.boolean1 = b1;
this.byte1 = b2;
this.short1 = s1;
this.int1 = i1;
this.long1 = l1;
this.short2 = s2;
this.int2 = i2;
this.long2 = l2;
this.short3 = s3;
this.int3 = i3;
this.long3 = l3;
this.float1 = f1;
this.double1 = d1;
this.bytes1 = b3;
if (s4 == null) {
this.string1 = null;
} else {
this.string1 = new Text(s4);
}
if (s5 == null) {
this.string2 = null;
} else {
this.string2 = new Text(s5);
}
if (s6 == null) {
this.string3 = null;
} else {
this.string3 = new Text(s6);
}
this.middle = m1;
this.list = l4;
this.map = m2;
}
}
public static class DoubleRow {
Double double1;
public DoubleRow(double double1) {
this.double1 = double1;
}
}
public static class BigRow {
Boolean boolean1;
Byte byte1;
Short short1;
Integer int1;
Long long1;
Float float1;
Double double1;
BytesWritable bytes1;
Text string1;
MiddleStruct middle;
List<InnerStruct> list = null;
Map<Text, InnerStruct> map = null;
public BigRow(Boolean b1, Byte b2, Short s1, Integer i1, Long l1, Float f1,
Double d1,
BytesWritable b3, String s2, MiddleStruct m1,
List<InnerStruct> l2, Map<Text, InnerStruct> m2) {
this.boolean1 = b1;
this.byte1 = b2;
this.short1 = s1;
this.int1 = i1;
this.long1 = l1;
this.float1 = f1;
this.double1 = d1;
this.bytes1 = b3;
if (s2 == null) {
this.string1 = null;
} else {
this.string1 = new Text(s2);
}
this.middle = m1;
this.list = l2;
this.map = m2;
}
}
public static InnerStruct inner(int i, String s) {
return new InnerStruct(i, s);
}
public static Map<Text, InnerStruct> map(InnerStruct... items) {
Map<Text, InnerStruct> result = new HashMap<Text, InnerStruct>(items.length);
for(InnerStruct i: items) {
result.put(i == null || i.string1 == null ? null : new Text(i.string1), i);
}
return result;
}
public static List<InnerStruct> list(InnerStruct... items) {
List<InnerStruct> result = new ArrayList<InnerStruct>(items.length);
for(InnerStruct s: items) {
result.add(s);
}
return result;
}
public static BytesWritable bytes(int... items) {
BytesWritable result = new BytesWritable();
result.setSize(items.length);
for(int i=0; i < items.length; ++i) {
result.getBytes()[i] = (byte) items[i];
}
return result;
}
public static ByteBuffer byteBuf(int... items) {
ByteBuffer result = ByteBuffer.allocate(items.length);
for(int item: items) {
result.put((byte) item);
}
return result;
}
public static class StringListWithId {
Integer id;
List<String> list = null;
public StringListWithId(Integer id, List<String> l1) {
this.id = id;
this.list = l1;
}
}
}