- **程式碼示例 - 自定義分割槽**
```java
public class PhonenumPartitioner implements Partitioner{
    @Override
    public void configure(Map<string> configs) {
        // TODO nothing
    }
    @Override
    public int partition(String topic, Object key, byte[] keyBytes, Object value, byte[] valueBytes, Cluster cluster) {
        // 得到 topic 的 partitions 資訊
        List<partitioninfo> partitions = cluster.partitionsForTopic(topic);
        int numPartitions = partitions.size();
        // 模擬客服
        if(key.toString().equals("10000") || key.toString().equals("11111")) {
            // 放到最後一個分割槽中
            return numPartitions - 1;
        }
        String phoneNum = key.toString();
        return phoneNum.substring(0, 2).hashCode() % (numPartitions - 1);
    }
    @Override
    public void close() {
        // TODO nothing
    }
}
```

- **自定義分割槽 - 使用**
```java
public class PartitionerProducer {
    private static final String[] PHONE_NUMS = new String[]{
            "10000", "10000", "11111", "13700000003", "13700000004",
            "10000", "15500000006", "11111", "15500000008",
            "17600000009", "10000", "17600000011"
    };
    public static void main(String[] args) throws Exception {
        Properties props = new Properties();
        props.put("bootstrap.servers", "host1:9092,host2.89:9093,host3:9094");
        // 設定分割槽器
        props.put("partitioner.class", "com.bonc.rdpe.kafka110.partitioner.PhonenumPartitioner");
        props.put("key.serializer", "org.apache.kafka.common.serialization.StringSerializer");
        props.put("value.serializer", "org.apache.kafka.common.serialization.StringSerializer");
        Producer<string string=""> producer = new KafkaProducer<>(props);
        int count = 0;
        int length = PHONE_NUMS.length;
        while(count < 10) {
            Random rand = new Random();
            String phoneNum = PHONE_NUMS[rand.nextInt(length)];
            ProducerRecord<string string=""> record = new ProducerRecord<>("dev3-test-topic001", phoneNum, phoneNum);
            RecordMetadata metadata = producer.send(record).get();
            String result = "phonenum [" + record.value() + "] has been sent to partition " + metadata.partition();
            System.out.println(result);
            Thread.sleep(500);
            count++;
        }
        producer.close();
    }
}
```</string></string></partitioninfo></string>

Kafka - 自定義分割槽器

相關文章