2626import org .openrewrite .java .AddImport ;
2727import org .openrewrite .java .JavaTemplate ;
2828import org .openrewrite .java .JavaVisitor ;
29- import org .openrewrite .java .RemoveImport ;
3029import org .openrewrite .java .tree .Expression ;
3130import org .openrewrite .java .tree .J ;
3231import org .openrewrite .java .tree .JavaType ;
@@ -64,10 +63,7 @@ public J visitNewClass(J.NewClass newClass, ExecutionContext executionContext) {
6463 }
6564
6665 if (type .isAssignableFrom (COMPETE_MPU ) && newClass .getArguments ().size () == 4 ) {
67- addV2S3ModelImport ("CompleteMultipartUploadRequest" );
6866 addV2S3ModelImport ("CompletedMultipartUpload" );
69- removeV1S3ModelImport ("CompleteMultipartUploadRequest" );
70-
7167 List <Expression > params = newClass .getArguments ();
7268 String v2Builder = "CompleteMultipartUploadRequest.builder().bucket(#{any()}).key(#{any()}).uploadId(#{any()})"
7369 + ".multipartUpload(CompletedMultipartUpload.builder().parts(#{any()}).build()).build()" ;
@@ -76,18 +72,12 @@ public J visitNewClass(J.NewClass newClass, ExecutionContext executionContext) {
7672 params .get (0 ), params .get (1 ), params .get (2 ), params .get (3 ));
7773 }
7874 if (type .isAssignableFrom (OBJECT_TAGGING ) && newClass .getArguments ().size () == 1 ) {
79- addV2S3ModelImport ("Tagging" );
80- removeV1S3ModelImport ("ObjectTagging" );
81-
8275 String v2Builder = "Tagging.builder().tagSet(#{any()}).build();" ;
8376 return JavaTemplate .builder (v2Builder )
8477 .build ().apply (getCursor (), newClass .getCoordinates ().replace (),
8578 newClass .getArguments ().get (0 ));
8679 }
8780 if (type .isAssignableFrom (GET_OBJECT_TAGGING_RESULT ) && newClass .getArguments ().size () == 1 ) {
88- addV2S3ModelImport ("GetObjectTaggingResponse" );
89- removeV1S3ModelImport ("GetObjectTaggingResponse" );
90-
9181 String v2Builder = "GetObjectTaggingResponse.builder().tagSet(#{any()}).build();" ;
9282 return JavaTemplate .builder (v2Builder )
9383 .build ().apply (getCursor (), newClass .getCoordinates ().replace (),
@@ -97,10 +87,6 @@ public J visitNewClass(J.NewClass newClass, ExecutionContext executionContext) {
9787 return super .visitNewClass (newClass , executionContext );
9888 }
9989
100- private void removeV1S3ModelImport (String className ) {
101- doAfterVisit (new RemoveImport <>(V1_S3_MODEL_PKG + className , true ));
102- }
103-
10490 private void addV2S3ModelImport (String className ) {
10591 doAfterVisit (new AddImport <>(V2_S3_MODEL_PKG + className , null , false ));
10692 }
0 commit comments