File tree Expand file tree Collapse file tree 2 files changed +3
-6
lines changed
database-commons/src/main/java/io/cdap/plugin/db Expand file tree Collapse file tree 2 files changed +3
-6
lines changed Original file line number Diff line number Diff line change @@ -239,11 +239,9 @@ public void prepareRun(BatchSinkContext context) {
239239 configuration .set (ETLDBOutputFormat .COMMIT_BATCH_SIZE ,
240240 context .getArguments ().get (ETLDBOutputFormat .COMMIT_BATCH_SIZE ));
241241 }
242-
243- addOutputContext (context );
244-
245242 // set error details provider
246243 context .setErrorDetailsProvider (new ErrorDetailsProviderSpec (getErrorDetailsProviderClassName ()));
244+ addOutputContext (context );
247245 }
248246 protected void addOutputContext (BatchSinkContext context ) {
249247 context .addOutput (Output .of (dbSinkConfig .getReferenceName (),
Original file line number Diff line number Diff line change @@ -288,11 +288,10 @@ public void prepareRun(BatchSourceContext context) throws Exception {
288288 lineageRecorder .recordRead ("Read" , "Read from database plugin" ,
289289 schema .getFields ().stream ().map (Schema .Field ::getName ).collect (Collectors .toList ()));
290290 }
291- context .setInput (Input .of (sourceConfig .getReferenceName (), new SourceInputFormatProvider (
292- DataDrivenETLDBInputFormat .class , connectionConfigAccessor .getConfiguration ())));
293-
294291 // set error details provider
295292 context .setErrorDetailsProvider (new ErrorDetailsProviderSpec (getErrorDetailsProviderClassName ()));
293+ context .setInput (Input .of (sourceConfig .getReferenceName (), new SourceInputFormatProvider (
294+ DataDrivenETLDBInputFormat .class , connectionConfigAccessor .getConfiguration ())));
296295 }
297296
298297 /**
You can’t perform that action at this time.
0 commit comments