Как добавить приемник из одного потока данных в разные пути в зависимости от конкретного ключа в json?

#java #json #apache-flink #flink-streaming #sink

#java #json #apache-flink #flink-streaming #приемник

Вопрос:

У меня есть json, такие как,

 {
  "name":"someone",
  "job":"doctor",
  "etc":"etc"
}
  

в каждом json есть разные значения для «задания», такие как doctor, pilot, driver, watchman и т.д.
я хочу разделить каждый json на основе значения «job» и сохранить его в разных местах, таких как /home/doctor , /home/pilot , /home/driver и т.д.

я пробовал функцию SplitStream для этого, но я должен указать эти значения, чтобы соответствовать условию.

 public class MyFlinkJob {   
    private static JsonParser jsonParser = new JsonParser();
    private static String key_1 = "doctor";
    private static String key_2 = "driver";
    private static String key_3 = "pilot";
    private static String key_default = "default";

    public static void main(String args[]) throws Exception {
        Properties prop = new Properties();
        StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();

        Properties props = new Properties();
        props.setProperty("bootstrap.servers", kafka);
        props.setProperty("group.id", "myjob");

        FlinkKafkaConsumer<String> myConsumer = new FlinkKafkaConsumer<>("topic", new SimpleStringSchema(), props);
        DataStream<String> record = env.addSource(myConsumer).rebalance()

        SplitStream<String> split = record.split(new OutputSelector<String>() {
            @Override
            public Iterable<String> select(String val) {
                JsonObject json = (JsonObject)jsonParser.parse(val);
                String jsonValue = CommonFields.getFieldValue(json, "job");
                List<String> output = new ArrayList<String>();

                if (key_1.equalsIgnoreCase(jsonValue)) {
                    output.add("doctor");
                } else if (key_2.equalsIgnoreCase(jsonValue)) {
                    output.add("driver");
                } else if (key_3.equalsIgnoreCase(jsonValue)) {
                    output.add("pilot");
                } else {
                    output.add("default");
                }
                return output;
            }});

        DataStream<String> doctor = split.select("doctor");
        DataStream<String> driver = split.select("driver");
        DataStream<String> pilot = split.select("pilot");
        DataStream<String> default1 = split.select("default");
        doctor.addSink(getBucketingSink(batchSize, prop, key_1));
        driver.addSink(getBucketingSink(batchSize, prop, key_2));
        pilot.addSink(getBucketingSink(batchSize, prop, key_3));
        default1.addSink(getBucketingSink(batchSize, prop, key_default));
        env.execute("myjob");
    } catch (IOException ex) {
        ex.printStackTrace();
    } finally {
        if (input != null) {
            try {
                input.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

public static BucketingSink<String> getBucketingSink(Long BatchSize, Properties prop, String key) {
    BucketingSink<String> sink = new BucketingSink<String>("hdfs://*/home/" key);
    Configuration conf = new Configuration();
    conf.set("hadoop.job.ugi", "hdfs");
    sink.setFSConfig(conf);
    sink.setBucketer(new DateTimeBucketer<String>(prop.getProperty("DateTimeBucketer")));
    return sink;
}
}
  

предположим, что если в «job» входит какое-либо другое значение, например engineer или что-то еще, и я не указал в class, то оно отправляется в папку по умолчанию, есть ли какой-либо способ автоматически разделить эти события json на основе значения «job», не указывая его, и создать путь, содержащий имя значения, например /home / enginerr?

Ответ №1:

Вы хотите использовать BucketingSink, который поддерживает запись записей в отдельные сегменты на основе значения поля. Вероятно, у меня есть функция map, которая принимает строку JSON, анализирует ее и выдает Tuple2<String, String> , где первый элемент — это значение job поля в JSON, а второй элемент — полная строка JSON.

Комментарии:

1. спасибо за ответ @kkrugler, как вы сказали, теперь я создал функцию, которая выдает Tuple2<String,строка>, но я не понимаю, как использовать это с bucketingSink

2. JavaDocs для BucketingSink, на который я ссылался в своем ответе, показывают общую форму для его использования. Для вашего конкретного случая вам понадобится класс, который реализует интерфейс Bucketer и создает путь к корзине на основе .f0 поля в элементе, которому передается его getBucketPath() метод.