Sì, è possibile. Puoi rendere transitorio il tuo oggetto Connection in modo che non sia serializzato e crearlo una volta per bundle tramite startBundle
metodo. Una volta elaborati tutti gli elementi del bundle, la connessione può essere chiusa tramite il finishBundle
metodo.
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
}
}