diff --git a/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/query/LambdaQueryChainWrapper.java b/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/query/LambdaQueryChainWrapper.java index bf2316378..cdcc6f034 100644 --- a/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/query/LambdaQueryChainWrapper.java +++ b/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/query/LambdaQueryChainWrapper.java @@ -61,6 +61,12 @@ public LambdaQueryChainWrapper(BaseMapper baseMapper, Class entityClass) { super.wrapperChildren = new LambdaQueryWrapper<>(entityClass); } + public LambdaQueryChainWrapper(BaseMapper baseMapper, LambdaQueryWrapper wrapperChildren) { + super(); + this.baseMapper = baseMapper; + super.wrapperChildren = wrapperChildren; + } + @Override public LambdaQueryChainWrapper select(boolean condition, List> columns) { return doSelect(condition, columns); diff --git a/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/query/QueryChainWrapper.java b/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/query/QueryChainWrapper.java index 8a2169a8b..bdcb3ae78 100644 --- a/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/query/QueryChainWrapper.java +++ b/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/query/QueryChainWrapper.java @@ -83,4 +83,13 @@ public BaseMapper getBaseMapper() { public Class getEntityClass() { return entityClass; } + + + public LambdaQueryChainWrapper lambda(){ + return new LambdaQueryChainWrapper<>( + baseMapper, + wrapperChildren.lambda() + ); + } + } diff --git a/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/update/LambdaUpdateChainWrapper.java b/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/update/LambdaUpdateChainWrapper.java index 9ba500978..e8bfc744d 100644 --- a/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/update/LambdaUpdateChainWrapper.java +++ b/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/update/LambdaUpdateChainWrapper.java @@ -44,6 +44,12 @@ public LambdaUpdateChainWrapper(Class entityClass) { super.wrapperChildren = new LambdaUpdateWrapper<>(entityClass); } + public LambdaUpdateChainWrapper(BaseMapper baseMapper, LambdaUpdateWrapper wrapperChildren) { + super(); + this.baseMapper = baseMapper; + super.wrapperChildren = wrapperChildren; + } + @Override public LambdaUpdateChainWrapper set(boolean condition, SFunction column, Object val, String mapping) { wrapperChildren.set(condition, column, val, mapping); diff --git a/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/update/UpdateChainWrapper.java b/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/update/UpdateChainWrapper.java index 5353ebe2a..68578a6e2 100644 --- a/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/update/UpdateChainWrapper.java +++ b/mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/conditions/update/UpdateChainWrapper.java @@ -74,4 +74,11 @@ public BaseMapper getBaseMapper() { public Class getEntityClass() { return entityClass; } + + public LambdaUpdateChainWrapper lambda(){ + return new LambdaUpdateChainWrapper<>( + baseMapper, + wrapperChildren.lambda() + ); + } }