EntitySpec Interface can return a schema

Author: alxmrs

java/arcs/core/storage/api/Entity.kt

@@ -13,6 +13,7 @@ package arcs.core.storage.api
 
 import arcs.core.common.Referencable
 import arcs.core.data.RawEntity
+import arcs.core.data.Schema
 import arcs.core.data.util.ReferencablePrimitive
 import kotlin.IllegalArgumentException
 import kotlin.reflect.KClass
@@ -38,6 +39,9 @@ interface EntitySpec<T : Entity> {
      * TODO: replace this with kotlinx.serialization
      */
     fun deserialize(data: RawEntity): T
+
+    /** Returns the corresponding [Schema] for the specified [Entity]. */
+    fun schema(): Schema
 }
 
 /**

src/tools/schema2kotlin.ts

@@ -315,6 +315,8 @@ ${this.opts.wasm ? '' : `\
             SchemaRegistry.register(schema)
         }
     }
+    
+    override fun schema() = schema
 `}
     override fun create() = ${name}()
     ${!this.opts.wasm ? `

src/tools/tests/goldens/generated-schemas.jvm.kt

@@ -87,6 +87,8 @@ class GoldInternal1_Spec() : EntitySpec<GoldInternal1> {
         }
     }
 
+    override fun schema() = schema
+
     override fun create() = GoldInternal1()
 
     override fun deserialize(data: RawEntity): GoldInternal1 {
@@ -215,6 +217,8 @@ class Gold_Data_Spec() : EntitySpec<Gold_Data> {
         }
     }
 
+    override fun schema() = schema
+
     override fun create() = Gold_Data()
 
     override fun deserialize(data: RawEntity): Gold_Data {