diff --git a/src/main/java/com/baomidou/config/po/TableInfo.java b/src/main/java/com/baomidou/config/po/TableInfo.java index 5cba13f2f325af023e18f20188b40ed6a54a064b..fce0e8c4decaad25c8da43cdfc8b8a191ac44a4a 100644 --- a/src/main/java/com/baomidou/config/po/TableInfo.java +++ b/src/main/java/com/baomidou/config/po/TableInfo.java @@ -1,6 +1,5 @@ package com.baomidou.config.po; - import org.apache.commons.lang.StringUtils; import java.util.List; @@ -26,6 +25,7 @@ public class TableInfo { private List fields; private String fieldNames; private boolean hasDate; + private boolean hasDecimal; public String getName() { return name; @@ -84,14 +84,14 @@ public class TableInfo { } public String getControllerName() { - return controllerName; - } + return controllerName; + } - public void setControllerName(String controllerName) { - this.controllerName = controllerName; - } + public void setControllerName(String controllerName) { + this.controllerName = controllerName; + } - public List getFields() { + public List getFields() { return fields; } @@ -135,6 +135,16 @@ public class TableInfo { return hasDate; } + public boolean isHasDecimal() { + for (TableField fieldInfo : fields) { + if (fieldInfo.getPropertyType().equals("BigDecimal")) { + hasDecimal = true; + break; + } + } + return hasDecimal; + } + /** * mapper xml中的字字段添加as * diff --git a/src/main/resources/template/entity.java.vm b/src/main/resources/template/entity.java.vm index 69eee6e321ada421ffa17fa8383a73ae6b135835..aded641504e6cd70bf8a2aa00d0a943199acf0df 100644 --- a/src/main/resources/template/entity.java.vm +++ b/src/main/resources/template/entity.java.vm @@ -4,7 +4,7 @@ package ${package.Entity}; import com.baomidou.mybatisplus.activerecord.Model; #end #if(${idGenType}!="ID_WORKER") -import com.baomidou.mybatisplus.annotations.IdType; +import com.baomidou.mybatisplus.enums.IdType; import com.baomidou.mybatisplus.annotations.TableId; #end import com.baomidou.mybatisplus.annotations.TableField; @@ -15,6 +15,9 @@ import java.io.Serializable; #if(${table.hasDate}) import java.util.Date; #end +#if(${table.hasDecimal}) +import java.math.BigDecimal; +#end /** *