1 /*******************************************************************************
2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
10 * http://www.apache.org/licenses/LICENSE-2.0
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
17 * ============LICENSE_END=========================================================
19 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
21 *******************************************************************************/
22 package com.att.dmf.mr.backends.kafka;
24 import java.util.ArrayList;
26 import org.apache.kafka.clients.consumer.ConsumerRecords;
30 import com.att.eelf.configuration.EELFLogger;
31 import com.att.eelf.configuration.EELFManager;
34 * A consumer Util class for force polling when a rebalance issue is anticipated
39 public class Kafka011ConsumerUtil {
40 private static final EELFLogger log = EELFManager.getInstance().getLogger(Kafka011ConsumerUtil.class);
43 * @param fconsumercache
49 public static boolean forcePollOnConsumer(final String fTopic, final String fGroup, final String fId) {
51 Thread forcepollThread = new Thread(new Runnable() {
55 ArrayList<Kafka011Consumer> kcsList = null;
57 kcsList = KafkaConsumerCache.getInstance().getConsumerListForCG(fTopic + "::" + fGroup + "::", fId);
58 if (null != kcsList) {
59 for (int counter = 0; counter < kcsList.size(); counter++) {
61 Kafka011Consumer kc1 = kcsList.get(counter);
64 ConsumerRecords<String, String> recs = kc1.getConsumer().poll(0);
65 log.info("soft poll on " + kc1);
66 } catch (java.util.ConcurrentModificationException e) {
67 log.error("Error occurs for " + e);
74 } catch (Exception e) {
75 log.error("Failed and go to Exception block for " + fGroup + " " + e.getMessage());
81 forcepollThread.start();
88 * @param fconsumercache
92 public static boolean forcePollOnConsumer(final String group) {
94 Thread forcepollThread = new Thread(new Runnable() {
97 ArrayList<Kafka011Consumer> kcsList = new ArrayList<Kafka011Consumer>();
98 kcsList = KafkaConsumerCache.getInstance().getConsumerListForCG(group);
100 if (null != kcsList) {
102 for (int counter = 0; counter < kcsList.size(); counter++) {
104 Kafka011Consumer kc1 = kcsList.get(counter);
105 log.info("soft poll on remote nodes " + kc1);
106 ConsumerRecords<String, String> recs = kc1.getConsumer().poll(0);
111 } catch (java.util.ConcurrentModificationException e) {
112 log.error("Error occurs for " + e);
113 } catch (Exception e) {
114 log.error("Failed and go to Exception block for " + group + " " + e.getMessage());
120 forcepollThread.start();