3333import io .cdap .cdap .api .annotation .Plugin ;
3434import io .cdap .cdap .api .data .schema .Schema ;
3535import io .cdap .cdap .api .plugin .PluginConfig ;
36- import io .cdap .cdap .etl .api .Arguments ;
3736import io .cdap .cdap .etl .api .FailureCollector ;
3837import io .cdap .cdap .etl .api .PipelineConfigurer ;
3938import io .cdap .cdap .etl .api .StageMetrics ;
5049import io .cdap .plugin .format .plugin .FileSinkProperties ;
5150import io .cdap .plugin .gcp .common .CmekUtils ;
5251import io .cdap .plugin .gcp .common .GCPConnectorConfig ;
53- import io .cdap .plugin .gcp .common .GCPReferenceSinkConfig ;
5452import io .cdap .plugin .gcp .common .GCPUtils ;
5553import io .cdap .plugin .gcp .gcs .Formats ;
5654import io .cdap .plugin .gcp .gcs .GCSPath ;
@@ -87,7 +85,6 @@ public class GCSBatchSink extends AbstractFileSink<GCSBatchSink.GCSBatchSinkConf
8785 private final GCSBatchSinkConfig config ;
8886 private String outputPath ;
8987
90-
9188 public GCSBatchSink (GCSBatchSinkConfig config ) {
9289 super (config );
9390 this .config = config ;
@@ -138,6 +135,7 @@ public void prepareRun(BatchSinkContext context) throws Exception {
138135 if (bucket == null ) {
139136 GCPUtils .createBucket (storage , config .getBucket (), config .getLocation (), cmekKeyName );
140137 }
138+ this .outputPath = getOutputDir (context );
141139 }
142140
143141 @ Override
@@ -156,12 +154,6 @@ protected Map<String, String> getFileSystemProperties(BatchSinkContext context)
156154 return properties ;
157155 }
158156
159- @ Override
160- protected String getOutputDir (long logicalStartTime ) {
161- this .outputPath = super .getOutputDir (logicalStartTime );
162- return this .outputPath ;
163- }
164-
165157 @ Override
166158 protected void recordLineage (LineageRecorder lineageRecorder , List <String > outputFields ) {
167159 lineageRecorder .recordWrite ("Write" , "Wrote to Google Cloud Storage." , outputFields );
0 commit comments