ftls.entity.ftl Maven / Gradle / Ivy

Go to download
package ${entityNamePackage}

import cn.hutool.core.util.RandomUtil
import cn.hutool.core.util.ReflectUtil
import cn.hutool.json.JSONObject
import com.github.jchanghong.kotlin.beanFillNullValues
import com.github.liaochong.myexcel.core.annotation.ExcelModel
import com.vladmihalcea.hibernate.type.json.JsonType
import org.hibernate.annotations.Type
import org.hibernate.annotations.TypeDef
import org.hibernate.annotations.TypeDefs
import org.hibernate.type.PostgresUUIDType
import org.springframework.stereotype.Component
import java.math.BigDecimal
import java.time.LocalDateTime
import java.util.*
import javax.persistence.*
import kotlin.reflect.KMutableProperty
import kotlin.reflect.full.createType
import kotlin.reflect.full.declaredMemberProperties
import kotlin.reflect.jvm.javaType
import java.io.Serializable

/**
* @description jpa层代码自动生成
* @author jiangchanghong
* @date ${today}
*/
@Entity
<#if pk_columns?size gt 1 >@IdClass(${entityName}PK::class)
@Table(name = "${table.tableName}"<#if table.schema?length!=0>, schema = "${table.schema}"<#if table.catalog?length!=0>, catalog = "${table.catalog}")
@TypeDefs(
    value = [
        TypeDef(name = "uuid", typeClass = PostgresUUIDType::class),
        TypeDef(name = "json", typeClass = JsonType::class),
        TypeDef(name = "jsonb", typeClass = JsonType::class)
    ]
)
@Component
@ExcelModel(includeAllField = true, useFieldNameAsTitle = true, dateFormat = "yyyy-MM-dd HH:mm:ss")
class ${entityName} :Serializable {
    <#list columns as column>
        <#if column.column.isPk==true>
            @field:Id
        
        <#if column.column.autoIncrement==true>
            @field:GeneratedValue(strategy = GenerationType.IDENTITY)
        
        <#if column.isUUID==true>
            @field:Type(type = "uuid")
        
        <#if column.isJsonB==true>
            @field:Type(type = "jsonb")
        
        <#if column.isJson==true>
            @field:Type(type = "json")
        
            @field:Column(name = "${column.column.name}", nullable = ${column.column.isNullable?string},columnDefinition = "${column.column.typeName}")
            var ${column.propertyName}: ${column.javaTypeName}? = null

    
<#if table.pkNames?size==0>
    @get:Id
    @get:Column(name = "auto_id", nullable = false,columnDefinition = "bigserial  primary key")
    @get:GeneratedValue(strategy=GenerationType.AUTO)
    var autoId: Long? = null


    override fun toString(): String =
        "Entity of type: ${r"$"}{javaClass.name} ( " +
                <#list columns as column>
                    "${column.propertyName} = ${r"$"}${column.propertyName} " +
                
                ")"

    // constant value returned to avoid entity inequality to itself before and after it's update/merge
    override fun hashCode(): Int = 42

    override fun equals(other: Any?): Boolean {
        if (this === other) return true
        if (javaClass != other?.javaClass) return false
        other as ${entityName}

       <#list columns as column>
           if (${column.propertyName} != other.${column.propertyName}) return false
       

        return true
    }

    companion object {
        fun newTestObject(): ${entityName} {
            val entity = ${entityName}().apply { this.beanFillNullValues() }
            for (kProperty1 in ${entityName}::class.declaredMemberProperties.mapNotNull { it as? KMutableProperty<*>? }) {
                if (kProperty1.returnType == String::class.createType(nullable = true)) {
                    kProperty1.setter.call(entity, RandomUtil.randomString(8))
                }
                if (kProperty1.returnType == Date::class.createType(nullable = true)) {
                    kProperty1.setter.call(entity, Date())
                }
                if (kProperty1.returnType == LocalDateTime::class.createType(nullable = true)) {
                    kProperty1.setter.call(entity, LocalDateTime.now())
                }
                if (kProperty1.returnType == JSONObject::class.createType(nullable = true)) {
                    kProperty1.setter.call(entity, JSONObject())
                }
                if (kProperty1.returnType == java.sql.Date::class.createType(nullable = true)) {
                    kProperty1.setter.call(entity, java.sql.Date(Date().time))
                }
                if (kProperty1.returnType == BigDecimal::class.createType(nullable = true)) {
                    kProperty1.setter.call(entity, BigDecimal("0.0"))
                }
                if (kProperty1.returnType == Boolean::class.createType(nullable = true)) {
                    kProperty1.setter.call(entity, false)
                }
                if (kProperty1.returnType == ByteArray::class.createType(nullable = true)) {
                    kProperty1.setter.call(entity, ByteArray(1))
                }
                if (kProperty1.returnType == UUID::class.createType(nullable = true)) {
                    kProperty1.setter.call(entity, UUID.randomUUID())
                }

            }
            return entity
        }
    }
}

<#if pk_columns?size gt 1>

    /**
    * @description jpa层代码自动生成
    * @author jiangchanghong
    * @date ${today}
    */
    @Component
    class ${entityName}PK :Serializable {
    <#list pk_columns as column>
        var ${column.propertyName}: ${column.javaTypeName}? = null
    

    override fun toString(): String =
    "Entity of type: ${r"$"}{javaClass.name} ( " +
    <#list pk_columns as column>
        "${column.propertyName} = ${r"$"}${column.propertyName} " +
    
    ")"

    // constant value returned to avoid entity inequality to itself before and after it's update/merge
    override fun hashCode(): Int = 42

    override fun equals(other: Any?): Boolean {
    if (this === other) return true
    if (javaClass != other?.javaClass) return false
    other as ${entityName}PK
    <#list pk_columns as column>
        if (${column.propertyName} != other.${column.propertyName}) return false
    
    return true
    }
    }