Merge pull request #57 from netheril96/main

Properly close resources after writing.
keiji · Mar 23, 2024 · 89fa0d9 · 89fa0d9
2 parents 4667e25 + 3b4fb9d
commit 89fa0d9
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 36 deletions.
diff --git a/tlv-ksp/src/main/java/dev/keiji/tlv/BerTlvDecoderProcessor.kt b/tlv-ksp/src/main/java/dev/keiji/tlv/BerTlvDecoderProcessor.kt
@@ -125,17 +125,19 @@ fun ${classDeclaration.simpleName.asString()}.readFrom(
 
         val onItemDetected = generateOnItemDetected(annotatedProperties, logger)
 
-        file.appendText("package $packageName")
-            .appendText("")
-            .appendText(imports)
-            .appendText("")
-            .appendText(classTemplate0)
-            .appendText("")
-            .appendText(classTemplate1)
-            .appendText("")
-            .appendText(onItemDetected)
-            .appendText("")
-            .appendText(classTemplate2)
+        file.use {
+            it.appendText("package $packageName")
+                .appendText("")
+                .appendText(imports)
+                .appendText("")
+                .appendText(classTemplate0)
+                .appendText("")
+                .appendText(classTemplate1)
+                .appendText("")
+                .appendText(onItemDetected)
+                .appendText("")
+                .appendText(classTemplate2)
+        }
     }
 
     @Suppress("MaxLineLength")

diff --git a/tlv-ksp/src/main/java/dev/keiji/tlv/BerTlvEncoderProcessor.kt b/tlv-ksp/src/main/java/dev/keiji/tlv/BerTlvEncoderProcessor.kt
@@ -127,13 +127,15 @@ fun ${classDeclaration.simpleName.asString()}.writeTo(outputStream: OutputStream
 
             val writeTo = generateWriteTo(annotatedProperties)
 
-            file.appendText("package $packageName")
-                .appendText("")
-                .appendText(imports)
-                .appendText("")
-                .appendText(classTemplate1)
-                .appendText(writeTo)
-                .appendText(classTemplate2)
+            file.use {
+                it.appendText("package $packageName")
+                    .appendText("")
+                    .appendText(imports)
+                    .appendText("")
+                    .appendText(classTemplate1)
+                    .appendText(writeTo)
+                    .appendText(classTemplate2)
+            }
         }
 
         @Suppress("MaxLineLength")

diff --git a/tlv-ksp/src/main/java/dev/keiji/tlv/CompactTlvDecoderProcessor.kt b/tlv-ksp/src/main/java/dev/keiji/tlv/CompactTlvDecoderProcessor.kt
@@ -111,17 +111,19 @@ fun ${classDeclaration.simpleName.asString()}.readFrom(
 
         val onItemDetected = generateOnItemDetected(annotatedProperties, logger)
 
-        file.appendText("package $packageName")
-            .appendText("")
-            .appendText(imports)
-            .appendText("")
-            .appendText(classTemplate0)
-            .appendText("")
-            .appendText(classTemplate1)
-            .appendText("")
-            .appendText(onItemDetected)
-            .appendText("")
-            .appendText(classTemplate2)
+        file.use {
+            it.appendText("package $packageName")
+                .appendText("")
+                .appendText(imports)
+                .appendText("")
+                .appendText(classTemplate0)
+                .appendText("")
+                .appendText(classTemplate1)
+                .appendText("")
+                .appendText(onItemDetected)
+                .appendText("")
+                .appendText(classTemplate2)
+        }
     }
 
     @Suppress("MaxLineLength")

diff --git a/tlv-ksp/src/main/java/dev/keiji/tlv/CompactTlvEncoderProcessor.kt b/tlv-ksp/src/main/java/dev/keiji/tlv/CompactTlvEncoderProcessor.kt
@@ -124,13 +124,15 @@ fun ${classDeclaration.simpleName.asString()}.writeTo(outputStream: OutputStream
 
             val writeTo = generateWriteTo(annotatedProperties)
 
-            file.appendText("package $packageName")
-                .appendText("")
-                .appendText(imports)
-                .appendText("")
-                .appendText(classTemplate1)
-                .appendText(writeTo)
-                .appendText(classTemplate2)
+            file.use {
+                it.appendText("package $packageName")
+                    .appendText("")
+                    .appendText(imports)
+                    .appendText("")
+                    .appendText(classTemplate1)
+                    .appendText(writeTo)
+                    .appendText(classTemplate2)
+            }
         }
 
         @Suppress("MaxLineLength")