Например, реализация bigqueryio go очень проста. Он считывает результаты запроса в цикле и последовательно выдает записи. Поэтому поток данных не масштабирует количество воркеров и обнаруживает отстающего.
Псевдокод такого пользовательского DoFn в Java:
Код: Выделить всё
@ProcessElement
public void processElement(@Element byte[] input, OutputReceiver output) throws Exception {
// ..
result = bigquery.query(queryConfig);
for (FieldValueList row : result.iterateAll()) {
output.output(mapRowToType(row));
}
}
Пример преобразования вышеуказанного источника в Java:
Код: Выделить всё
var bigqueryRows = pipeline.apply("ReadFromBigQuery", BigQueryIO.read(//...));
var mutations = bigqueryRows.apply("ProcessRows", ParDo.of(/**/ {
@ProcessElement
public void processElement(@Element ItemRow row, OutputReceiver out) {
String rowKey = row.getId();
BigtableIO.Write.Mutation mutation = BigtableIO.Write.Mutation.create(rowKey);
// ...
out.output(mutation);
}
}));
Есть ли стратегия, которую можно позже добавить в конвейер, чтобы он начал распараллеливаться обработка нескольким работникам?
*ПРИМЕЧАНИЕ:
Хотя моя конкретная проблема связана с go sdk и bigqueryio, мой вопрос касается такого сценария в целом. Не стесняйтесь давать общий ответ на любом знакомом SDK языка луча.
Подробнее здесь: https://stackoverflow.com/questions/792 ... aflow-beam
Мобильная версия