/**
* Copyright (C) 2014-2016 LinkedIn Corp. (pinot-core@linkedin.com)
*
* 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 com.linkedin.pinot.core.data.partition;
import com.google.common.base.Preconditions;
/**
* Modulo operation based partition function, where:
* <ul>
* <li> partitionId = value % {@link #_numPartitions}</li>
* </ul>
*
*/
public class ModuloPartitionFunction implements PartitionFunction {
private static final String NAME = "Modulo";
private final int _numPartitions;
/**
* Constructor for the class.
* @param numPartitions Number of partitions.
*/
public ModuloPartitionFunction(int numPartitions) {
Preconditions.checkArgument(numPartitions > 0, "Number of partitions must be > 0, specified", numPartitions);
_numPartitions = numPartitions;
}
@Override
public int getPartition(Object value) {
if (value instanceof Integer) {
return ((Integer) value) % _numPartitions;
} else {
throw new IllegalArgumentException(
"Illegal argument for partitioning, expected Integer, got: " + value.getClass().getName());
}
}
@Override
public int getNumPartitions() {
return _numPartitions;
}
@Override
public String toString() {
return NAME;
}
}