org.apache.nifi.processors.kafka.pubsub

Class PublisherLease

java.lang.Object

org.apache.nifi.processors.kafka.pubsub.PublisherLease

All Implemented Interfaces:

Closeable, AutoCloseable

public class PublisherLease
extends Object
implements Closeable

Field Summary

Fields

Modifier and Type	Field and Description
private boolean	activeTransaction
private Pattern	attributeNameRegex
private Charset	headerCharacterSet
private ComponentLog	logger
private long	maxAckWaitMillis
private int	maxMessageSize
private AtomicLong	messagesSent
private boolean	poisoned
private org.apache.kafka.clients.producer.Producer<byte[],byte[]>	producer
private InFlightMessageTracker	tracker
private boolean	transactionsInitialized
private boolean	useTransactions

Constructor Summary

Constructors

Constructor and Description
PublisherLease(org.apache.kafka.clients.producer.Producer<byte[],byte[]> producer, int maxMessageSize, long maxAckWaitMillis, ComponentLog logger, boolean useTransactions, Pattern attributeNameRegex, Charset headerCharacterSet)

Method Summary

Modifier and Type	Method and Description
private void	addHeaders(FlowFile flowFile, Map<String,String> additionalAttributes, org.apache.kafka.clients.producer.ProducerRecord<?,?> record)
(package private) void	beginTransaction()
void	close()
PublishResult	complete()
(package private) void	fail(FlowFile flowFile, Exception cause)
InFlightMessageTracker	getTracker()
boolean	isPoisoned()
protected void	poison()
protected void	publish(FlowFile flowFile, byte[] messageKey, byte[] messageContent, String topic, InFlightMessageTracker tracker, Integer partition)
(package private) void	publish(FlowFile flowFile, InputStream flowFileContent, byte[] messageKey, byte[] demarcatorBytes, String topic, Integer partition)
protected void	publish(FlowFile flowFile, Map<String,String> additionalAttributes, byte[] messageKey, byte[] messageContent, String topic, InFlightMessageTracker tracker, Integer partition)
(package private) void	publish(FlowFile flowFile, RecordSet recordSet, RecordSetWriterFactory writerFactory, RecordSchema schema, String messageKeyField, String topic, Function<Record,Integer> partitioner)
(package private) void	rollback()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

logger

private final ComponentLog logger

producer

private final org.apache.kafka.clients.producer.Producer<byte[],byte[]> producer

maxMessageSize

private final int maxMessageSize

maxAckWaitMillis

private final long maxAckWaitMillis

useTransactions

private final boolean useTransactions

attributeNameRegex

private final Pattern attributeNameRegex

headerCharacterSet

private final Charset headerCharacterSet

poisoned

private volatile boolean poisoned

messagesSent

private final AtomicLong messagesSent

transactionsInitialized

private volatile boolean transactionsInitialized

activeTransaction

private volatile boolean activeTransaction

tracker

private InFlightMessageTracker tracker

Constructor Detail

PublisherLease

public PublisherLease(org.apache.kafka.clients.producer.Producer<byte[],byte[]> producer,
                      int maxMessageSize,
                      long maxAckWaitMillis,
                      ComponentLog logger,
                      boolean useTransactions,
                      Pattern attributeNameRegex,
                      Charset headerCharacterSet)

Method Detail

poison

protected void poison()

isPoisoned

public boolean isPoisoned()

beginTransaction

void beginTransaction()

rollback

void rollback()

fail

void fail(FlowFile flowFile,
          Exception cause)

publish

void publish(FlowFile flowFile,
             InputStream flowFileContent,
             byte[] messageKey,
             byte[] demarcatorBytes,
             String topic,
             Integer partition)
      throws IOException

Throws:

IOException

publish

void publish(FlowFile flowFile,
             RecordSet recordSet,
             RecordSetWriterFactory writerFactory,
             RecordSchema schema,
             String messageKeyField,
             String topic,
             Function<Record,Integer> partitioner)
      throws IOException

Throws:

IOException

addHeaders

private void addHeaders(FlowFile flowFile,
                        Map<String,String> additionalAttributes,
                        org.apache.kafka.clients.producer.ProducerRecord<?,?> record)

publish

protected void publish(FlowFile flowFile,
                       byte[] messageKey,
                       byte[] messageContent,
                       String topic,
                       InFlightMessageTracker tracker,
                       Integer partition)

publish

protected void publish(FlowFile flowFile,
                       Map<String,String> additionalAttributes,
                       byte[] messageKey,
                       byte[] messageContent,
                       String topic,
                       InFlightMessageTracker tracker,
                       Integer partition)

complete

public PublishResult complete()

close

public void close()

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable

getTracker

public InFlightMessageTracker getTracker()