fix: allow empty return types in kotlin

block · Nov 14, 2023 · ffa7f54 · ffa7f54
1 parent c72c615
commit ffa7f54
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...in-runtime/ftl-runtime/src/main/kotlin/xyz/block/ftl/schemaextractor/ExtractSchemaRule.kt b/...in-runtime/ftl-runtime/src/main/kotlin/xyz/block/ftl/schemaextractor/ExtractSchemaRule.kt
@@ -142,9 +142,11 @@ class SchemaExtractor(
       }
 
       // Validate return type
-      val respClass = verb.createTypeBindingForReturnType(bindingContext)?.type?.toClassDescriptor()
+      val respClass = verb.createTypeBindingForReturnType(bindingContext)?.type
         ?: throw IllegalStateException("Could not resolve ${verb.name} return type")
-      require(respClass.isData) { "Return type of ${verb.name} must be a data class" }
+      require(respClass.toClassDescriptor().isData || respClass.isEmptyClassTypeAlias()) {
+        "Return type of ${verb.name} must be a data class or typealias of Unit"
+      }
     }
   }