From 4ccef7a90a023c77b063860013086939c8049721 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=80=E6=BA=90=E6=B5=B7=E5=93=A5?= Date: Sun, 21 May 2023 18:16:37 +0800 Subject: [PATCH] optimize all_columns field in apt --- .../com/mybatisflex/processor/QueryEntityProcessor.java | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java b/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java index d5433030..47af6dae 100644 --- a/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java +++ b/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java @@ -100,7 +100,8 @@ public class QueryEntityProcessor extends AbstractProcessor { private static final String columnsTemplate = " public QueryColumn @property = new QueryColumn(this, \"@columnName\");\n"; private static final String defaultColumnsTemplate = "\n public QueryColumn[] DEFAULT_COLUMNS = new QueryColumn[]{@allColumns};\n"; - private static final String allColumnsTemplate = " public QueryColumn[] ALL_COLUMNS = new QueryColumn[]{@allColumns};\n\n"; +// private static final String allColumnsTemplate = " public QueryColumn[] ALL_COLUMNS = new QueryColumn[]{@allColumns};\n\n"; + private static final String allColumnsTemplate = " public QueryColumn ALL_COLUMNS = new QueryColumn(this, \"*\");\n\n"; private Filer filer; private Elements elementUtils; @@ -321,8 +322,10 @@ public class QueryEntityProcessor extends AbstractProcessor { StringJoiner allColumns = new StringJoiner(", "); propertyAndColumns.forEach((property, column) -> allColumns.add(buildName(property, tablesNameStyle))); - String allColumnsString = allColumnsTemplate.replace("@allColumns", allColumns.toString()) - .replace("ALL_COLUMNS", buildName("allColumns", tablesNameStyle)); +// String allColumnsString = allColumnsTemplate.replace("@allColumns", allColumns.toString()) +// .replace("ALL_COLUMNS", buildName("allColumns", tablesNameStyle)); + + String allColumnsString = allColumnsTemplate; StringJoiner defaultColumnStringJoiner = new StringJoiner(", ");