代码优化

framework-database/src/main/java/com/chelvc/framework/database/support/DefaultEnhanceService.java

@@ -7,6 +7,7 @@ import java.util.List;
 
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
+import com.baomidou.mybatisplus.extension.conditions.query.QueryChainWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.chelvc.framework.common.util.ObjectUtils;
 import com.chelvc.framework.database.listener.AddListener;
@@ -39,6 +40,13 @@ public class DefaultEnhanceService<M extends BaseMapper<T>, T> extends ServiceIm
         applicationContext.getBeansOfType(OperateListener.class).values().forEach(this::register);
     }
 
+    @Override
+    public QueryChainWrapper<T> query() {
+        QueryChainWrapper<T> query = super.query();
+        query.setEntityClass(this.getEntityClass());
+        return query;
+    }
+
     @Override
     public LambdaQueryChainWrapper<T> lambdaQuery() {
         LambdaQueryChainWrapper<T> query = super.lambdaQuery();