diff --git a/src/main/java/com/github/yulichang/query/MPJQueryWrapper.java b/src/main/java/com/github/yulichang/query/MPJQueryWrapper.java index a4294d1..da94388 100644 --- a/src/main/java/com/github/yulichang/query/MPJQueryWrapper.java +++ b/src/main/java/com/github/yulichang/query/MPJQueryWrapper.java @@ -7,14 +7,10 @@ import com.baomidou.mybatisplus.core.conditions.segments.MergeSegments; import com.baomidou.mybatisplus.core.metadata.TableFieldInfo; import com.baomidou.mybatisplus.core.metadata.TableInfo; import com.baomidou.mybatisplus.core.metadata.TableInfoHelper; -import com.baomidou.mybatisplus.core.toolkit.ArrayUtils; -import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; -import com.baomidou.mybatisplus.core.toolkit.StringPool; -import com.baomidou.mybatisplus.core.toolkit.StringUtils; +import com.baomidou.mybatisplus.core.toolkit.*; import com.github.yulichang.query.interfaces.MPJJoin; import com.github.yulichang.toolkit.Constant; import com.github.yulichang.toolkit.MPJWrappers; -import org.springframework.util.Assert; import java.util.ArrayList; import java.util.Arrays; @@ -48,7 +44,7 @@ public class MPJQueryWrapper extends AbstractWrapper extends AbstractWrapper extends AbstractWrapper - Constant.TABLE_ALIAS + StringPool.DOT + c.getColumn()).collect(Collectors.toList())); + alias + StringPool.DOT + c.getColumn()).collect(Collectors.toList())); return typedThis; } @@ -134,7 +130,7 @@ public class MPJQueryWrapper extends AbstractWrapper selectAll(Class clazz) { - selectAll(clazz, Constant.TABLE_ALIAS); + selectAll(clazz, alias); return typedThis; } @@ -172,7 +168,23 @@ public class MPJQueryWrapper extends AbstractWrapper + * 正例 new QueryWrapper().setAlias("a").selectAll(UserDO.class).... + * 反例 new QueryWrapper().selectAll(UserDO.class).setAlias("a").... + *
+     *
+     * @param alias 主表别名
+     */
+    public MPJQueryWrapper setAlias(String alias) {
+        Assert.isTrue(StringUtils.isNotBlank(alias), "别名不能为空");
+        this.alias = alias;
+        return this;
     }
 
     /**