Tak to mozliwe. Możesz sprawić, że Twój obiekt Connection stanie się przejściowy, aby nie był serializowany, i utworzyć go raz na pakiet za pomocą startBundle
metoda. Po przetworzeniu wszystkich elementów w pakiecie połączenie można zamknąć za pomocą finishBundle
metoda.
class MyDoFn extends DoFn<X, Y> {
private transient Connection jdbc;
@Setup
public void setup(Context c) {
jdbc = // Create connection
}
@ProcessElement
public void processElement(ProcessContext c) {
// query database
}
@Teardown
public void tearDown(Context c) {
// close connection
}
}