Extract generateEncodeMethod()

1 files changed, 27 insertions, 15 deletions

diff --git a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatransient0/JavaTransient0Generator.java b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatransient0/JavaTransient0Generator.java
index 6d0a6c9..23f699b 100644
--- a/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatransient0/JavaTransient0Generator.java
+++ b/src/main/java/org/zwobble/hobgoblin/compiler/output/generators/javatransient0/JavaTransient0Generator.java
@@ -126,19 +126,9 @@ public class JavaTransient0Generator implements Generator {
                 var structJavaTypeRef = this.javaGenerator.generateTypeRef(structDefinition.type());
 
                 yield Stream.of(
-                    new JavaMethodDeclaration(
-                        encodeMethodName(structDefinition.type()),
-                        JavaVisibility.PUBLIC,
-                        JavaMethodKind.STATIC,
-                        JavaTypeRef.VOID,
-                        List.of(
-                            new JavaParam(structJavaTypeRef, VALUE_NAME),
-                            new JavaParam(
-                                JavaTypeRef.OUTPUT_STREAM,
-                                OUTPUT_STREAM_NAME
-                            )
-                        ),
-                        List.of(JavaTypeRef.IO_EXCEPTION),
+                    generateEncodeMethod(
+                        structDefinition.type(),
+                        structJavaTypeRef,
                         new JavaBlock(
                             structDefinition.fields().stream()
                                 .flatMap(field -> generateEncode(
@@ -189,6 +179,28 @@ public class JavaTransient0Generator implements Generator {
         };
     }
 
+    private JavaMethodDeclaration generateEncodeMethod(
+        Type type,
+        JavaTypeRef javaTypeRef,
+        JavaBlock body
+    ) {
+        return new JavaMethodDeclaration(
+            encodeMethodName(type),
+            JavaVisibility.PUBLIC,
+            JavaMethodKind.STATIC,
+            JavaTypeRef.VOID,
+            List.of(
+                new JavaParam(javaTypeRef, VALUE_NAME),
+                new JavaParam(
+                    JavaTypeRef.OUTPUT_STREAM,
+                    OUTPUT_STREAM_NAME
+                )
+            ),
+            List.of(JavaTypeRef.IO_EXCEPTION),
+            body
+        );
+    }
+
     private Stream<JavaBlockStatement> generateEncode(JavaExpression value, Type type) {
         return switch (type) {
             case ConstructedNativeType constructedNativeType -> {
@@ -415,11 +427,11 @@ public class JavaTransient0Generator implements Generator {
             .map(x -> x * 8);
     }
 
-    private JavaIdentifier encodeMethodName(StructType type) {
+    private JavaIdentifier encodeMethodName(Type type) {
         return this.javaGenerator.generateFieldName("encode" + type.name());
     }
 
-    private JavaIdentifier decodeMethodName(StructType type) {
+    private JavaIdentifier decodeMethodName(Type type) {
         return this.javaGenerator.generateFieldName("decode" + type.name());
     }