Fix changes made by previous commit

mpfaff · Apr 19, 2024 · 29718dd · 29718dd
1 parent 972d023
commit 29718dd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/java.base/share/classes/jdk/internal/classfile/impl/ChainedClassBuilder.java b/src/java.base/share/classes/jdk/internal/classfile/impl/ChainedClassBuilder.java
@@ -58,7 +58,7 @@ public Optional<ClassModel> original() {
 
     @Override
     public ClassBuilder withField(Utf8Entry name, Utf8Entry descriptor, Consumer<? super FieldBuilder> handler) {
-        return consumer.accept(new BufferedFieldBuilder(terminal.constantPool, terminal.context,
+        return with(new BufferedFieldBuilder(terminal.constantPool, terminal.context,
                                                         name, descriptor, null)
                                        .run(handler)
                                        .toModel());
@@ -70,13 +70,13 @@ public ClassBuilder transformField(FieldModel field, FieldTransform transform) {
                                                                 field.fieldName(), field.fieldType(),
                                                                 field);
         builder.transform(field, transform);
-        return consumer.accept(builder.toModel());
+        return with(builder.toModel());
     }
 
     @Override
     public ClassBuilder withMethod(Utf8Entry name, Utf8Entry descriptor, int flags,
                                    Consumer<? super MethodBuilder> handler) {
-        return consumer.accept(new BufferedMethodBuilder(terminal.constantPool, terminal.context,
+        return with(new BufferedMethodBuilder(terminal.constantPool, terminal.context,
                                                          name, descriptor, null)
                                        .run(handler)
                                        .toModel());
@@ -87,7 +87,7 @@ public ClassBuilder transformMethod(MethodModel method, MethodTransform transfor
         BufferedMethodBuilder builder = new BufferedMethodBuilder(terminal.constantPool, terminal.context,
                                                                   method.methodName(), method.methodType(), method);
         builder.transform(method, transform);
-        return consumer.accept(builder.toModel());
+        return with(builder.toModel());
     }
 
     @Override

diff --git a/src/java.base/share/classes/jdk/internal/classfile/impl/ChainedMethodBuilder.java b/src/java.base/share/classes/jdk/internal/classfile/impl/ChainedMethodBuilder.java
@@ -56,7 +56,7 @@ public MethodBuilder with(MethodElement element) {
 
     @Override
     public MethodBuilder withCode(Consumer<? super CodeBuilder> handler) {
-        return consumer.accept(terminal.bufferedCodeBuilder(null)
+        return with(terminal.bufferedCodeBuilder(null)
                                        .run(handler)
                                        .toModel());
     }
@@ -65,7 +65,7 @@ public MethodBuilder withCode(Consumer<? super CodeBuilder> handler) {
     public MethodBuilder transformCode(CodeModel code, CodeTransform transform) {
         BufferedCodeBuilder builder = terminal.bufferedCodeBuilder(code);
         builder.transform(code, transform);
-        return consumer.accept(builder.toModel());
+        return with(builder.toModel());
     }
 
     @Override