Skip to content

C
common
Commit 2122516a authored by Quxl's avatar Quxl
Browse files
Options

x

parent 56f067ec
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 4 deletions
src/main/java/com/egolm/common/jdbc/dialect/ReverseUtil.java
View file @ 2122516a
...@@ -37,13 +37,15 @@ public class ReverseUtil { ...@@ -37,13 +37,15 @@ public class ReverseUtil {
if(paramyKeys.size() > 0) { if(paramyKeys.size() > 0) {
sb.append("import javax.persistence.Id;" + "\n\n"); sb.append("import javax.persistence.Id;" + "\n\n");
} }
boolean is_date = false;
boolean is_decimal = false;
for(String columnType : colTypes) { for(String columnType : colTypes) {
if (columnType.equalsIgnoreCase("datetime") || columnType.equalsIgnoreCase("date")) { if ((columnType.equalsIgnoreCase("datetime") || columnType.equalsIgnoreCase("date")) && !is_date) {
sb.append("import java.util.Date;" + "\n\n"); sb.append("import java.util.Date;" + "\n\n");
break; is_date = true;
} else if (columnType.contains("decimal") || columnType.contains("numeric")) { } else if ((columnType.contains("decimal") || columnType.contains("numeric")) && !is_decimal) {
sb.append("import java.math.BigDecimal;" + "\n\n"); sb.append("import java.math.BigDecimal;" + "\n\n");
break; is_decimal = true;
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment