x

src/main/java/com/egolm/common/jdbc/dialect/ReverseUtil.java

@@ -37,13 +37,15 @@ public class ReverseUtil {
 		if(paramyKeys.size() > 0) {
 			sb.append("import javax.persistence.Id;" + "\n\n");
 		}
+		boolean is_date = false;
+		boolean is_decimal = false;
 		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");
-				break;
-			} else if (columnType.contains("decimal") || columnType.contains("numeric")) {
+				is_date = true;
+			} else if ((columnType.contains("decimal") || columnType.contains("numeric")) && !is_decimal) {
 				sb.append("import java.math.BigDecimal;" + "\n\n");
-				break;
+				is_decimal = true;
 			}
 		}