|
@@ -20,7 +20,6 @@ import com.baomidou.mybatisplus.annotation.TableField;
|
|
|
import com.baomidou.mybatisplus.annotation.TableId;
|
|
|
import com.baomidou.mybatisplus.core.handlers.AnnotationHandler;
|
|
|
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
|
|
|
-import com.baomidou.mybatisplus.core.toolkit.GlobalConfigUtils;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
|
|
import com.baomidou.mybatisplus.generator.IFill;
|
|
|
import com.baomidou.mybatisplus.generator.ITemplate;
|
|
@@ -54,6 +53,8 @@ import java.util.stream.Collectors;
|
|
|
*/
|
|
|
public class Entity implements ITemplate {
|
|
|
|
|
|
+ private final AnnotationHandler annotationHandler = new AnnotationHandler(){};
|
|
|
+
|
|
|
private static final Logger LOGGER = LoggerFactory.getLogger(Entity.class);
|
|
|
|
|
|
private Entity() {
|
|
@@ -196,8 +197,6 @@ public class Entity implements ITemplate {
|
|
|
* @param clazz 实体父类 Class
|
|
|
*/
|
|
|
public void convertSuperEntityColumns(Class<?> clazz) {
|
|
|
- com.baomidou.mybatisplus.core.metadata.TableInfo tableInfo = TableInfoHelper.getTableInfo(clazz);
|
|
|
- AnnotationHandler annotationHandler = GlobalConfigUtils.getGlobalConfig(tableInfo.getConfiguration()).getAnnotationHandler();
|
|
|
List<Field> fields = TableInfoHelper.getAllFields(clazz, annotationHandler);
|
|
|
this.superEntityColumns.addAll(fields.stream().map(field -> {
|
|
|
TableId tableId = annotationHandler.getAnnotation(field, TableId.class);
|