Java ProducerRecord.partition方法代码示例

import org.apache.kafka.clients.producer.ProducerRecord; //导入方法依赖的package包/类
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    switch (method.getName()) {
        case SEND_METHOD_NAME: {
            ProducerRecord record = (ProducerRecord) args[0];

            args[0] = new ProducerRecord<>(
                    BaseIntegrationTest.adjustTopicNameForTest(record.topic()),
                    record.partition(),
                    record.timestamp(),
                    record.key(),
                    record.value()
            );
            break;
        }
        case PARTITIONS_FOR_METHOD_NAME: {
            args[0] = BaseIntegrationTest.adjustTopicNameForTest((String) args[0]);
            break;
        }
    }
    return method.invoke(producer, args);
}
 

import org.apache.kafka.clients.producer.ProducerRecord; //导入方法依赖的package包/类
/**
 * Called before record is sent by producer
 */
static <K, V> void onSend(ProducerRecord<K, V> record, Span span) {
  Tags.COMPONENT.set(span, COMPONENT_NAME);
  Tags.MESSAGE_BUS_DESTINATION.set(span, record.topic());
  if (record.partition() != null) {
    span.setTag("partition", record.partition());
  }
}
 

import org.apache.kafka.clients.producer.ProducerRecord; //导入方法依赖的package包/类
@Override
@SuppressWarnings("unchecked")
public boolean matches(Object argument) {
    if (!(argument instanceof ProducerRecord))
        return false;
    ProducerRecord<byte[], byte[]> other = (ProducerRecord<byte[], byte[]>) argument;
    return record.topic().equals(other.topic()) &&
            record.partition() != null ? record.partition().equals(other.partition()) : other.partition() == null &&
            record.key() != null ? Arrays.equals(record.key(), other.key()) : other.key() == null &&
            record.value() != null ? Arrays.equals(record.value(), other.value()) : other.value() == null;
}
 

import org.apache.kafka.clients.producer.ProducerRecord; //导入方法依赖的package包/类
/**
 * Read the next record from the given topic. These records were output by the topology during the previous calls to
 * {@link #process(String, byte[], byte[])}.
 *
 * @param topic the name of the topic
 * @param keyDeserializer the deserializer for the key type
 * @param valueDeserializer the deserializer for the value type
 * @return the next record on that topic, or null if there is no record available
 */
public <K, V> ProducerRecord<K, V> readOutput(final String topic,
                                              final Deserializer<K> keyDeserializer,
                                              final Deserializer<V> valueDeserializer) {
    final ProducerRecord<byte[], byte[]> record = readOutput(topic);
    if (record == null) {
        return null;
    }
    final K key = keyDeserializer.deserialize(record.topic(), record.key());
    final V value = valueDeserializer.deserialize(record.topic(), record.value());
    return new ProducerRecord<>(record.topic(), record.partition(), record.timestamp(), key, value);
}
 

import org.apache.kafka.clients.producer.ProducerRecord; //导入方法依赖的package包/类
@Override
public ProducerRecord<Integer, String> onSend(ProducerRecord<Integer, String> record) {
    onSendCount++;
    if (throwExceptionOnSend)
        throw new KafkaException("Injected exception in AppendProducerInterceptor.onSend");

    return new ProducerRecord<>(
            record.topic(), record.partition(), record.key(), record.value().concat(appendStr));
}
 

import org.apache.kafka.clients.producer.ProducerRecord; //导入方法依赖的package包/类
public static TopicPartition getSendPartition(ProducerRecord<String, byte[]> record) {
    return new TopicPartition(record.topic(), record.partition());
}
 

import org.apache.kafka.clients.producer.ProducerRecord; //导入方法依赖的package包/类
@Override
public ProducerRecord<String, String> onSend(ProducerRecord<String, String> record) {
    ONSEND_COUNT.incrementAndGet();
    return new ProducerRecord<>(
            record.topic(), record.partition(), record.key(), record.value().concat(appendStr));
}