View Javadoc

1   package org.apache.helix.alerts;
2   
3   /*
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *   http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.
20   */
21  
22  import java.util.ArrayList;
23  import java.util.Iterator;
24  import java.util.List;
25  
26  public class SumOperator extends Operator {
27  
28  	public SumOperator() {
29  		minInputTupleLists = 1;
30  		maxInputTupleLists = Integer.MAX_VALUE;
31  		inputOutputTupleListsCountsEqual = false;
32  		numOutputTupleLists = 1;
33  	}
34  
35  	
36  	public List<Iterator<Tuple<String>>> singleSetToIter(ArrayList<Tuple<String>> input) 
37  	{
38  		List out = new ArrayList();
39  		out.add(input.iterator());
40  		return out;
41  	}
42  	
43  	@Override
44  	public List<Iterator<Tuple<String>>> execute(List<Iterator<Tuple<String>>> input) {
45  		ArrayList<Tuple<String>> output = new ArrayList<Tuple<String>>();
46  		if (input == null || input.size() == 0) {
47  			return singleSetToIter(output);
48  		}
49  		while (true) { //loop through set of iters, return when 1 runs out (not completing the row in progress)
50  			Tuple<String> rowSum = null;
51  			for (Iterator<Tuple<String>> it : input) {
52  				if (!it.hasNext()) { //when any iterator runs out, we are done
53  					return singleSetToIter(output);
54  				}
55  				rowSum = sumTuples(rowSum, it.next());
56  			}
57  			output.add(rowSum);
58  		}
59  	}
60  }