Handle version properties for Kotlin code generation (#15)

Author: nakamura-to

codegen/src/main/java/org/seasar/doma/gradle/codegen/desc/EntityPropertyDesc.java

@@ -8,6 +8,8 @@ public class EntityPropertyDesc {
 
   protected String propertyClassName;
 
+  protected String languageClassName;
+
   protected boolean id;
 
   protected GenerationType generationType;
@@ -58,6 +60,14 @@ public String getPropertyClassSimpleName() {
     return ClassUtil.getSimpleName(propertyClassName);
   }
 
+  public String getLanguageClassSimpleName() {
+    return ClassUtil.getSimpleName(languageClassName);
+  }
+
+  public void setLanguageClassName(String languageClassName) {
+    this.languageClassName = languageClassName;
+  }
+
   public void setId(boolean id) {
     this.id = id;
   }

codegen/src/main/java/org/seasar/doma/gradle/codegen/desc/EntityPropertyDescFactory.java

@@ -132,8 +132,9 @@ protected void handlePropertyClass(
               columnMeta.getSqlType()));
       className = String.class.getName();
     }
+    propertyDesc.setPropertyClassName(className);
     LanguageClass languageClass = languageClassResolver.resolve(className, columnMeta);
-    propertyDesc.setPropertyClassName(languageClass.getClassName());
+    propertyDesc.setLanguageClassName(languageClass.getClassName());
     propertyDesc.setDefaultValue(languageClass.getDefaultValue());
     if (columnMeta.isNullable() || "null".equals(languageClass.getDefaultValue())) {
       propertyDesc.setNullable(true);

codegen/src/main/resources/org/seasar/doma/gradle/codegen/template/kotlin/dao.ftl

@@ -27,7 +27,7 @@ interface ${simpleName} {
      * @return the <#if entityDesc.entityPrefix??>${entityDesc.entityPrefix}</#if>${entityDesc.simpleName}<#if entityDesc.entitySuffix??>${entityDesc.entitySuffix}</#if> entity
      */
     @Select
-    fun selectById(<#list entityDesc.idEntityPropertyDescs as property>${property.name}: ${property.propertyClassSimpleName}<#if property_has_next>, </#if></#list>): <#if entityDesc.entityPrefix??>${entityDesc.entityPrefix}</#if>${entityDesc.simpleName}<#if entityDesc.entitySuffix??>${entityDesc.entitySuffix}</#if>
+    fun selectById(<#list entityDesc.idEntityPropertyDescs as property>${property.name}: ${property.languageClassSimpleName}<#if property_has_next>, </#if></#list>): <#if entityDesc.entityPrefix??>${entityDesc.entityPrefix}</#if>${entityDesc.simpleName}<#if entityDesc.entitySuffix??>${entityDesc.entitySuffix}</#if>
 
 </#if>
 <#if entityDesc.idEntityPropertyDescs?size gt 0 && entityDesc.versionEntityPropertyDesc??>
@@ -39,7 +39,7 @@ interface ${simpleName} {
      * @return the <#if entityDesc.entityPrefix??>${entityDesc.entityPrefix}</#if>${entityDesc.simpleName}<#if entityDesc.entitySuffix??>${entityDesc.entitySuffix}</#if> entity
      */
     @Select(ensureResult = true)
-    fun selectByIdAndVersion(<#list entityDesc.idEntityPropertyDescs as property>${property.name}: ${property.propertyClassSimpleName}, </#list>${entityDesc.versionEntityPropertyDesc.name}: ${entityDesc.versionEntityPropertyDesc.propertyClassSimpleName}): <#if entityDesc.entityPrefix??>${entityDesc.entityPrefix}</#if>${entityDesc.simpleName}<#if entityDesc.entitySuffix??>${entityDesc.entitySuffix}</#if>
+    fun selectByIdAndVersion(<#list entityDesc.idEntityPropertyDescs as property>${property.name}: ${property.languageClassSimpleName}, </#list>${entityDesc.versionEntityPropertyDesc.name}: ${entityDesc.versionEntityPropertyDesc.languageClassSimpleName}): <#if entityDesc.entityPrefix??>${entityDesc.entityPrefix}</#if>${entityDesc.simpleName}<#if entityDesc.entitySuffix??>${entityDesc.entitySuffix}</#if>
 
 </#if>
     /**

codegen/src/main/resources/org/seasar/doma/gradle/codegen/template/kotlin/entity.ftl

@@ -48,7 +48,7 @@ class <#if entityPrefix??>${entityPrefix}</#if>${simpleName}<#if entitySuffix??>
   <#if property.showColumnName && property.columnName??>
     @Column(name = "${property.columnName}")
   </#if>
-    var ${property.name}: ${property.propertyClassSimpleName}<#if property.nullable>?</#if> = ${property.defaultValue}
+    var ${property.name}: ${property.languageClassSimpleName}<#if property.nullable>?</#if> = ${property.defaultValue}
 </#list>
 <#if originalStatesPropertyName??>
 

codegen/src/test/resources/org/seasar/doma/gradle/codegen/KotlinGeneratorTest_DefaultConfigDao.txt

@@ -19,6 +19,14 @@ interface HogeDao {
     @Select
     fun selectById(id: Int): Hoge
 
+    /**
+     * @param id
+     * @param version
+     * @return the Hoge entity
+     */
+    @Select(ensureResult = true)
+    fun selectByIdAndVersion(id: Int, version: Int): Hoge
+
     /**
      * @param entity
      * @return affected rows

codegen/src/test/resources/org/seasar/doma/gradle/codegen/KotlinGeneratorTest_Metamodel.txt

@@ -5,6 +5,7 @@ import org.seasar.doma.Entity
 import org.seasar.doma.Id
 import org.seasar.doma.Metamodel
 import org.seasar.doma.Table
+import org.seasar.doma.Version
 
 /**
  * COMMENT for HOGE
@@ -23,6 +24,7 @@ class Hoge {
     var empName: String? = null
 
     /** COMMENT for VERSION */
+    @Version
     @Column(name = "VERSION")
     var version: Int = -1
 }

codegen/src/test/resources/org/seasar/doma/gradle/codegen/KotlinGeneratorTest_OriginalStates.txt

@@ -5,6 +5,7 @@ import org.seasar.doma.Entity
 import org.seasar.doma.Id
 import org.seasar.doma.OriginalStates
 import org.seasar.doma.Table
+import org.seasar.doma.Version
 
 /**
  * COMMENT for HOGE
@@ -23,6 +24,7 @@ class Hoge {
     var empName: String? = null
 
     /** COMMENT for VERSION */
+    @Version
     @Column(name = "VERSION")
     var version: Int = -1
 

codegen/src/test/resources/org/seasar/doma/gradle/codegen/KotlinGeneratorTest_SimpleDao.txt

@@ -20,6 +20,14 @@ interface HogeDao {
     @Select
     fun selectById(id: Int): Hoge
 
+    /**
+     * @param id
+     * @param version
+     * @return the Hoge entity
+     */
+    @Select(ensureResult = true)
+    fun selectByIdAndVersion(id: Int, version: Int): Hoge
+
     /**
      * @param entity
      * @return affected rows

codegen/src/test/resources/org/seasar/doma/gradle/codegen/KotlinGeneratorTest_SimpleDao_with_prefix.txt

@@ -20,6 +20,14 @@ interface THogeDao {
     @Select
     fun selectById(id: Int): THoge
 
+    /**
+     * @param id
+     * @param version
+     * @return the THoge entity
+     */
+    @Select(ensureResult = true)
+    fun selectByIdAndVersion(id: Int, version: Int): THoge
+
     /**
      * @param entity
      * @return affected rows

codegen/src/test/resources/org/seasar/doma/gradle/codegen/KotlinGeneratorTest_SimpleDao_with_prefix_and_suffix.txt

@@ -20,6 +20,14 @@ interface THogeEntityDao {
     @Select
     fun selectById(id: Int): THogeEntity
 
+    /**
+     * @param id
+     * @param version
+     * @return the THogeEntity entity
+     */
+    @Select(ensureResult = true)
+    fun selectByIdAndVersion(id: Int, version: Int): THogeEntity
+
     /**
      * @param entity
      * @return affected rows