简化代码

This commit is contained in:
yulichang 2024-07-21 21:13:51 +08:00
parent ac635a91db
commit a3de7ec409
2 changed files with 16 additions and 18 deletions

View File

@ -1,10 +1,20 @@
package com.github.yulichang.apt; package com.github.yulichang.apt;
import lombok.AccessLevel;
import lombok.Getter;
import lombok.NoArgsConstructor;
import java.io.Serializable; import java.io.Serializable;
public interface BaseColumn<T> extends Serializable { /**
* @author yulichang
* @since 1.5.0
*/
@Getter
@NoArgsConstructor(access = AccessLevel.PROTECTED)
public abstract class BaseColumn<T> implements Serializable {
Class<T> getColumnClass(); protected String alias;
String getAlias(); abstract public Class<T> getColumnClass();
} }

View File

@ -124,7 +124,6 @@ public class EntityProcessor extends AbstractProcessor {
.addClass(tableInfo.getClassComment(), tableInfo.getTagClassName(), .addClass(tableInfo.getClassComment(), tableInfo.getTagClassName(),
BaseColumn.class.getSimpleName() + "<" + tableInfo.getSimpleClassName() + ">", BaseColumn.class.getSimpleName() + "<" + tableInfo.getSimpleClassName() + ">",
c -> c c -> c
.addDefaultField()
.addConstructor(tableInfo) .addConstructor(tableInfo)
.addFields(tableInfo) .addFields(tableInfo)
.addMethod(tableInfo) .addMethod(tableInfo)
@ -188,7 +187,7 @@ public class EntityProcessor extends AbstractProcessor {
addComment("", comment); addComment("", comment);
sb.append("public class ").append(tagClassName); sb.append("public class ").append(tagClassName);
if (StringUtil.isNotEmpty(impl)) { if (StringUtil.isNotEmpty(impl)) {
sb.append(" implements ").append(impl); sb.append(" extends ").append(impl);
} }
sb.append(" {\n"); sb.append(" {\n");
consumer.accept(this); consumer.accept(this);
@ -198,11 +197,12 @@ public class EntityProcessor extends AbstractProcessor {
public StringBuilderHelper addConstructor(TableInfo tableInfo) { public StringBuilderHelper addConstructor(TableInfo tableInfo) {
// 无参构造 // 无参构造
newLine();
sb.append(String.format("\tpublic %s() {\n\t}\n", tableInfo.getTagClassName())); sb.append(String.format("\tpublic %s() {\n\t}\n", tableInfo.getTagClassName()));
newLine(); newLine();
//有参构造 //有参构造
sb.append("\tpublic ").append(tableInfo.getTagClassName()).append("(String alias) {\n" + sb.append("\tpublic ").append(tableInfo.getTagClassName()).append("(String alias) {\n" +
"\t\tthis._alias_q2Gv$ = alias;\n" + "\t\tsuper.alias = alias;\n" +
"\t}\n"); "\t}\n");
newLine(); newLine();
return this; return this;
@ -239,24 +239,12 @@ public class EntityProcessor extends AbstractProcessor {
return this; return this;
} }
public StringBuilderHelper addDefaultField() {
newLine();
sb.append("\tprivate String _alias_q2Gv$;\n");
newLine();
return this;
}
public StringBuilderHelper addMethod(TableInfo tableInfo) { public StringBuilderHelper addMethod(TableInfo tableInfo) {
sb.append("\t@Override\n" + sb.append("\t@Override\n" +
"\tpublic Class<").append(tableInfo.getSimpleClassName()).append("> getColumnClass() {\n") "\tpublic Class<").append(tableInfo.getSimpleClassName()).append("> getColumnClass() {\n")
.append("\t\treturn ").append(tableInfo.getSimpleClassName()).append(".class;\n") .append("\t\treturn ").append(tableInfo.getSimpleClassName()).append(".class;\n")
.append("\t}\n"); .append("\t}\n");
newLine(); newLine();
sb.append("\t@Override\n" +
"\tpublic String getAlias() {\n" +
"\t\treturn this._alias_q2Gv$;\n" +
"\t}\n");
newLine();
return this; return this;
} }