com.sitewhere.microservice.kafka.DirectKafkaConsumer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sitewhere-microservice Show documentation

SiteWhere Microservice Components Library

The newest version!

/**
 * Copyright © 2014-2021 The SiteWhere Authors
 *
 * 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.sitewhere.microservice.kafka;

import java.util.List;
import java.util.Map;

import org.apache.kafka.clients.consumer.ConsumerRecord;
import org.apache.kafka.clients.consumer.OffsetAndMetadata;
import org.apache.kafka.clients.consumer.OffsetCommitCallback;
import org.apache.kafka.common.TopicPartition;

import com.sitewhere.spi.SiteWhereException;

public abstract class DirectKafkaConsumer extends MicroserviceKafkaConsumer {

    /*
     * @see
     * com.sitewhere.spi.microservice.kafka.IMicroserviceKafkaConsumer#process(org.
     * apache.kafka.common.TopicPartition, java.util.List)
     */
    @Override
    public void process(TopicPartition topicPartition, List> records) {
	try {
	    attemptToProcess(topicPartition, records);
	    getConsumer().commitAsync(new OffsetCommitCallback() {
		public void onComplete(Map offsets, Exception e) {
		    if (e != null) {
			getLogger().error("Commit failed for offsets " + offsets, e);
		    }
		}
	    });
	} catch (SiteWhereException e) {
	    getLogger().error("Exception in consumer processing.", e);
	} catch (Throwable e) {
	    getLogger().error("Unhandled exception in consumer processing.", e);
	}
    }

    /**
     * Attempts to process a batch of records, throwing an exception if processing
     * fails.
     * 
     * @param topicPartition
     * @param records
     * @throws SiteWhereException
     */
    public abstract void attemptToProcess(TopicPartition topicPartition, List> records)
	    throws SiteWhereException;
}