1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
|
12 | |
|
13 | |
|
14 | |
|
15 | |
|
16 | |
|
17 | |
package datafu.pig.bags.sets; |
18 | |
|
19 | |
import org.apache.pig.EvalFunc; |
20 | |
import org.apache.pig.data.DataBag; |
21 | |
import org.apache.pig.data.DataType; |
22 | |
import org.apache.pig.impl.logicalLayer.schema.Schema; |
23 | |
|
24 | 633 | public abstract class SetOperationsBase extends EvalFunc<DataBag> |
25 | |
{ |
26 | |
@Override |
27 | |
public Schema outputSchema(Schema input) |
28 | |
{ |
29 | |
try { |
30 | 160 | for (Schema.FieldSchema fieldSchema : input.getFields()) |
31 | |
{ |
32 | 320 | if (fieldSchema.type != DataType.BAG) |
33 | |
{ |
34 | 0 | throw new RuntimeException("Expected a bag but got: " + DataType.findTypeName(fieldSchema.type)); |
35 | |
} |
36 | |
} |
37 | |
|
38 | 160 | Schema bagSchema = input.getField(0).schema; |
39 | |
|
40 | 160 | Schema outputSchema = new Schema(new Schema.FieldSchema(getSchemaName(this.getClass() |
41 | |
.getName() |
42 | |
.toLowerCase(), input), |
43 | |
bagSchema, |
44 | |
DataType.BAG)); |
45 | |
|
46 | 160 | return outputSchema; |
47 | |
} |
48 | 0 | catch (Exception e) { |
49 | 0 | return null; |
50 | |
} |
51 | |
} |
52 | |
} |