|
17 | 17 |
|
18 | 18 | import java.util.Objects;
|
19 | 19 | import java.util.concurrent.atomic.AtomicInteger;
|
20 |
| -import java.util.function.Consumer; |
21 | 20 |
|
22 | 21 | import org.mybatis.dynamic.sql.delete.DeleteModel;
|
23 | 22 | import org.mybatis.dynamic.sql.render.RenderingStrategy;
|
|
28 | 27 |
|
29 | 28 | public class DeleteRenderer {
|
30 | 29 | private DeleteModel deleteModel;
|
| 30 | + private RenderingStrategy renderingStrategy; |
31 | 31 |
|
32 |
| - private DeleteRenderer(DeleteModel deleteModel) { |
33 |
| - this.deleteModel = Objects.requireNonNull(deleteModel); |
| 32 | + private DeleteRenderer(Builder builder) { |
| 33 | + deleteModel = Objects.requireNonNull(builder.deleteModel); |
| 34 | + renderingStrategy = Objects.requireNonNull(builder.renderingStrategy); |
34 | 35 | }
|
35 | 36 |
|
36 |
| - public DeleteSupport render(RenderingStrategy renderingStrategy) { |
37 |
| - DeleteSupport.Builder builder = new DeleteSupport.Builder() |
38 |
| - .withTableName(deleteModel.table().name()); |
39 |
| - |
40 |
| - deleteModel.whereModel().ifPresent(applyWhere(builder, renderingStrategy)); |
41 |
| - |
42 |
| - return builder.build(); |
43 |
| - } |
44 |
| - |
45 |
| - private Consumer<WhereModel> applyWhere(DeleteSupport.Builder builder, RenderingStrategy renderingStrategy) { |
46 |
| - return whereModel -> applyWhere(builder, renderingStrategy, whereModel); |
| 37 | + public DeleteSupport render() { |
| 38 | + return new DeleteSupport.Builder() |
| 39 | + .withTableName(deleteModel.table().name()) |
| 40 | + .withWhereSupport(deleteModel.whereModel().map(this::renderWhereModel)) |
| 41 | + .build(); |
47 | 42 | }
|
48 | 43 |
|
49 |
| - private void applyWhere(DeleteSupport.Builder builder, RenderingStrategy renderingStrategy, WhereModel whereModel) { |
50 |
| - WhereSupport whereSupport = new WhereRenderer.Builder() |
| 44 | + private WhereSupport renderWhereModel(WhereModel whereModel) { |
| 45 | + return new WhereRenderer.Builder() |
51 | 46 | .withWhereModel(whereModel)
|
52 | 47 | .withRenderingStrategy(renderingStrategy)
|
53 | 48 | .withSequence(new AtomicInteger(1))
|
54 | 49 | .withTableAliasCalculator(TableAliasCalculator.empty())
|
55 | 50 | .build()
|
56 | 51 | .render();
|
57 |
| - |
58 |
| - builder.withWhereClause(whereSupport.getWhereClause()); |
59 |
| - builder.withParameters(whereSupport.getParameters()); |
60 | 52 | }
|
61 | 53 |
|
62 |
| - public static DeleteRenderer of(DeleteModel deleteModel) { |
63 |
| - return new DeleteRenderer(deleteModel); |
| 54 | + public static class Builder { |
| 55 | + private DeleteModel deleteModel; |
| 56 | + private RenderingStrategy renderingStrategy; |
| 57 | + |
| 58 | + public Builder withDeleteModer(DeleteModel deleteModel) { |
| 59 | + this.deleteModel = deleteModel; |
| 60 | + return this; |
| 61 | + } |
| 62 | + |
| 63 | + public Builder withRenderingStrategy(RenderingStrategy renderingStrategy) { |
| 64 | + this.renderingStrategy = renderingStrategy; |
| 65 | + return this; |
| 66 | + } |
| 67 | + |
| 68 | + public DeleteRenderer build() { |
| 69 | + return new DeleteRenderer(this); |
| 70 | + } |
64 | 71 | }
|
65 | 72 | }
|
0 commit comments