-
Notifications
You must be signed in to change notification settings - Fork 9
/
UnionDoubleSummarySketch.java
117 lines (100 loc) · 4 KB
/
UnionDoubleSummarySketch.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
/*
* Copyright 2016, Yahoo! Inc.
* Licensed under the terms of the Apache License 2.0. See LICENSE file at the project root for terms.
*/
package com.yahoo.sketches.pig.tuple;
import org.apache.pig.Algebraic;
import com.yahoo.sketches.tuple.DoubleSummary;
import com.yahoo.sketches.tuple.DoubleSummaryDeserializer;
import com.yahoo.sketches.tuple.DoubleSummaryFactory;
import com.yahoo.sketches.tuple.DoubleSummarySetOperations;
import com.yahoo.sketches.tuple.SummarySetOperations;
/**
* This is to union Sketch<DoubleSummary>.
* It supports all three ways: exec(), Accumulator and Algebraic
*/
public class UnionDoubleSummarySketch extends UnionSketch<DoubleSummary> implements Algebraic {
/**
* Constructor with default sketch size and default mode (sum)
*/
public UnionDoubleSummarySketch() {
super(new DoubleSummarySetOperations(), new DoubleSummaryDeserializer());
}
/**
* Constructor with default mode (sum)
* @param sketchSize String representation of sketch size
*/
public UnionDoubleSummarySketch(final String sketchSize) {
super(Integer.parseInt(sketchSize), new DoubleSummarySetOperations(), new DoubleSummaryDeserializer());
}
/**
* Constructor
* @param sketchSize String representation of sketch size
* @param summaryMode String representation of mode (sum, min or max)
*/
public UnionDoubleSummarySketch(final String sketchSize, final String summaryMode) {
super(Integer.parseInt(sketchSize),
new DoubleSummarySetOperations(DoubleSummary.Mode.valueOf(summaryMode)),
new DoubleSummaryDeserializer());
}
@Override
public String getInitial() {
return Initial.class.getName();
}
@Override
public String getIntermed() {
return IntermediateFinal.class.getName();
}
@Override
public String getFinal() {
return IntermediateFinal.class.getName();
}
public static class Initial extends AlgebraicInitial {
/**
* Constructor for the initial pass of an Algebraic function.
* Default sketch size and default mode.
*/
public Initial() {}
/**
* Constructor for the initial pass of an Algebraic function. This will be passed the same
* constructor arguments as the original UDF.
* @param sketchSize String representation of sketch size
*/
public Initial(final String sketchSize) {}
/**
* Constructor for the initial pass of an Algebraic function. This will be passed the same
* constructor arguments as the original UDF.
* @param sketchSize String representation of sketch size
* @param summaryMode String representation of mode (sum, min or max)
*/
public Initial(final String sketchSize, final String summaryMode) {}
}
public static class IntermediateFinal extends UnionSketchAlgebraicIntermediateFinal<DoubleSummary> {
/**
* Constructor for the intermediate and final passes of an Algebraic function.
* Default sketch size and default mode.
*/
public IntermediateFinal() {
super(new DoubleSummarySetOperations(), new DoubleSummaryDeserializer());
}
/**
* Constructor for the intermediate and final passes of an Algebraic function. This will be
* passed the same constructor arguments as the original UDF.
* @param sketchSize String representation of sketch size
*/
public IntermediateFinal(final String sketchSize) {
super(Integer.parseInt(sketchSize), new DoubleSummarySetOperations(), new DoubleSummaryDeserializer());
}
/**
* Constructor for the intermediate and final passes of an Algebraic function. This will be
* passed the same constructor arguments as the original UDF.
* @param sketchSize String representation of sketch size
* @param summaryMode String representation of mode (sum, min or max)
*/
public IntermediateFinal(final String sketchSize, final String summaryMode) {
super(Integer.parseInt(sketchSize),
new DoubleSummarySetOperations(DoubleSummary.Mode.valueOf(summaryMode)),
new DoubleSummaryDeserializer());
}
}
}