/***********************************************************************************************************************
*
* Copyright (C) 2010-2014 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.pact.runtime.test.util;
import java.io.IOException;
import eu.stratosphere.pact.runtime.test.util.types.StringPair;
import eu.stratosphere.util.MutableObjectIterator;
public class UniformStringPairGenerator implements MutableObjectIterator<StringPair> {
final int numKeys;
final int numVals;
int keyCnt = 0;
int valCnt = 0;
boolean repeatKey;
public UniformStringPairGenerator(int numKeys, int numVals, boolean repeatKey) {
this.numKeys = numKeys;
this.numVals = numVals;
this.repeatKey = repeatKey;
}
@Override
public StringPair next(StringPair target) throws IOException {
if(!repeatKey) {
if(valCnt >= numVals) {
return null;
}
target.setKey(Integer.toString(keyCnt++));
target.setValue(Integer.toBinaryString(valCnt));
if(keyCnt == numKeys) {
keyCnt = 0;
valCnt++;
}
} else {
if(keyCnt >= numKeys) {
return null;
}
target.setKey(Integer.toString(keyCnt));
target.setValue(Integer.toBinaryString(valCnt++));
if(valCnt == numVals) {
valCnt = 0;
keyCnt++;
}
}
return target;
}
}