package com.zdjizhi.utils.connections.kafka;

import org.apache.flink.api.common.serialization.SimpleStringSchema;
import org.apache.flink.streaming.connectors.kafka.FlinkKafkaConsumer;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Properties;


public class KafkaConsumer {

    public static FlinkKafkaConsumer<String> getKafkaConsumer(String topics, Properties Properties){


        List<String> topicList = new ArrayList<>();
        if (topics.contains(",")) {
            String[] words = topics.split(",");
            topicList.addAll(Arrays.asList(words));
        } else {
            topicList.add(topics);
        }
        return new FlinkKafkaConsumer<>(topicList, new SimpleStringSchema(), Properties);
    }

}