|
11 | 11 | import java.io.IOException;
|
12 | 12 | import java.math.BigDecimal;
|
13 | 13 | import java.util.ArrayList;
|
| 14 | +import java.util.Arrays; |
14 | 15 | import java.util.Date;
|
15 | 16 | import java.util.List;
|
| 17 | +import java.util.concurrent.atomic.AtomicInteger; |
16 | 18 | import java.util.regex.Matcher;
|
17 | 19 | import java.util.regex.Pattern;
|
18 | 20 |
|
@@ -421,4 +423,61 @@ public static List<FieldInfo> processJsonObjectToFieldList(JSONObject jsonObject
|
421 | 423 | }
|
422 | 424 | return fieldList;
|
423 | 425 | }
|
| 426 | + |
| 427 | + public static ClassInfo processInsertSqlToClassInfo(ParamInfo paramInfo) { |
| 428 | + // field List |
| 429 | + List<FieldInfo> fieldList = new ArrayList<FieldInfo>(); |
| 430 | + //return classInfo |
| 431 | + ClassInfo codeJavaInfo = new ClassInfo(); |
| 432 | + |
| 433 | + //get origin sql |
| 434 | + String fieldSqlStr = paramInfo.getTableSql().toLowerCase().trim(); |
| 435 | + fieldSqlStr=fieldSqlStr.replaceAll(" "," ").replaceAll("\\\\n`","") |
| 436 | + .replaceAll("\\+","").replaceAll("``","`").replaceAll("\\\\",""); |
| 437 | + String valueStr = fieldSqlStr.substring(fieldSqlStr.lastIndexOf("values")+6).replaceAll(" ","").replaceAll("\\(","").replaceAll("\\)",""); |
| 438 | + //get the string between insert into and values |
| 439 | + fieldSqlStr=fieldSqlStr.substring(0,fieldSqlStr.lastIndexOf("values")); |
| 440 | + |
| 441 | + System.out.println(fieldSqlStr); |
| 442 | + |
| 443 | + String insertSqlPattenStr = "insert into (?<tableName>.*) \\((?<columnsSQL>.*)\\)"; |
| 444 | + //String DDL_PATTEN_STR="\\s*create\\s+table\\s+(?<tableName>\\S+)[^\\(]*\\((?<columnsSQL>[\\s\\S]+)\\)[^\\)]+?(comment\\s*(=|on\\s+table)\\s*'(?<tableComment>.*?)'\\s*;?)?$"; |
| 445 | + |
| 446 | + Matcher matcher1 = Pattern.compile(insertSqlPattenStr).matcher(fieldSqlStr); |
| 447 | + while(matcher1.find()){ |
| 448 | + |
| 449 | + String tableName = matcher1.group("tableName"); |
| 450 | + //System.out.println("tableName:"+tableName); |
| 451 | + codeJavaInfo.setClassName(tableName); |
| 452 | + codeJavaInfo.setTableName(tableName); |
| 453 | + |
| 454 | + String columnsSQL = matcher1.group("columnsSQL"); |
| 455 | + //System.out.println("columnsSQL:"+columnsSQL); |
| 456 | + |
| 457 | + List<String> valueList = new ArrayList<>(); |
| 458 | + //add values as comment |
| 459 | + Arrays.stream(valueStr.split(",")).forEach(column->{ |
| 460 | + valueList.add(column); |
| 461 | + }); |
| 462 | + AtomicInteger n= new AtomicInteger(0); |
| 463 | + //add column to fleldList |
| 464 | + Arrays.stream(columnsSQL.replaceAll(" ", "").split(",")).forEach(column->{ |
| 465 | + FieldInfo fieldInfo2 = new FieldInfo(); |
| 466 | + fieldInfo2.setFieldName(column); |
| 467 | + fieldInfo2.setColumnName(column); |
| 468 | + fieldInfo2.setFieldClass(String.class.getSimpleName()); |
| 469 | + if(n.get()<valueList.size()){ |
| 470 | + fieldInfo2.setFieldComment(column+" , eg."+valueList.get(n.get())); |
| 471 | + } |
| 472 | + fieldList.add(fieldInfo2); |
| 473 | + n.getAndIncrement(); |
| 474 | + }); |
| 475 | + |
| 476 | + } |
| 477 | + if(fieldList.size()<1){ |
| 478 | + throw new CodeGenerateException("INSERT SQL解析失败"); |
| 479 | + } |
| 480 | + codeJavaInfo.setFieldList(fieldList); |
| 481 | + return codeJavaInfo; |
| 482 | + } |
424 | 483 | }
|
0 commit comments