kafka自定义partition

自定义分区设置
	class Partition implements Partitioner{
		//设置
		public void configure(Map<String, ?> configs) {

		}

		//分区逻辑
		public int partition(String topic, Object key, byte[] keyBytes, Object value, byte[] valueBytes){
			return 1;
		}

		//释放资源
		public void close(){

		}
	}

	{
		//配置生产者属性(指定多个参数)
		Properties prop = new Properties();

		//省略不写
		-----
		//消息在发送前必须序列化
		prop.put("key.serializer","org.apache.kafka.common.serialization.StringSerializer");
		prop.put("value.serializer","org.apache.kafka.common.serialization.StringSerializer");

		//分区指定
		prop.put("partitioner.class","com.terry.kafka.Partition");

		//2、实例化producer
		KafkaProducer<String, String> producer = new KafkaProducer<String, String>(prop);

		//3、发送消息
		for(int i = 0; i < 99; i++) {
			producer.send(new ProducerRecord<String, String>("test", i),new CallBack(){
				public void onCompletion(RecordMetadata metadata, Exception exception) {
					//如果metadata不为null,拿到当前的数据偏移量与分区
					if(metadata != null) {
						Sout(metadata.topic() + "------" + metadata.offset() + "------" + metadata.partition());
					}
				}
			});
		}

		//4、关闭资源
		producer.close();
	}

猜你喜欢

转载自blog.csdn.net/weixin_42898914/article/details/85062924