Call done on when mapping transformation end

2016-03-14 19:10:36 +03:00
parent 4f77b02a58
commit 3eb6749377
1 changed files with 5 additions and 1 deletions
@@ -92,8 +92,12 @@ class WhenMappingTransformer(
        assert(clinit.name == "<clinit>", { "When mapping should contains only <clinit> method, but contains '${clinit.name}'" })

        var transformedClinit = cutOtherMappings(clinit)
-        val result = classBuilder.visitor.visitMethod(transformedClinit.access, transformedClinit.name, transformedClinit.desc, transformedClinit.signature, transformedClinit.exceptions.toTypedArray())
+        val result = classBuilder.newMethod(
+                JvmDeclarationOrigin.NO_ORIGIN, transformedClinit.access, transformedClinit.name, transformedClinit.desc,
+                transformedClinit.signature, transformedClinit.exceptions.toTypedArray()
+        )
        transformedClinit.accept(result)
+        classBuilder.done()

        return transformationResult
    }