|
17 | 17 |
|
18 | 18 | import java.util.Collection;
|
19 | 19 | import java.util.List;
|
20 |
| -import java.util.Optional; |
21 | 20 | import java.util.function.Function;
|
22 | 21 | import java.util.function.ToIntFunction;
|
23 | 22 | import java.util.function.ToLongFunction;
|
24 | 23 | import java.util.function.UnaryOperator;
|
25 | 24 |
|
26 |
| -import org.jetbrains.annotations.NotNull; |
27 |
| -import org.jetbrains.annotations.Nullable; |
28 | 25 | import org.mybatis.dynamic.sql.BasicColumn;
|
29 | 26 | import org.mybatis.dynamic.sql.SqlBuilder;
|
30 | 27 | import org.mybatis.dynamic.sql.SqlTable;
|
|
53 | 50 | public class MyBatis3Utils {
|
54 | 51 | private MyBatis3Utils() {}
|
55 | 52 |
|
56 |
| - public static long count(ToLongFunction<SelectStatementProvider> mapper, |
| 53 | + public static SelectStatementProvider countFrom(SqlTable table, CountDSLCompleter completer) { |
| 54 | + return countFrom(SqlBuilder.countFrom(table), completer); |
| 55 | + } |
| 56 | + |
| 57 | + public static long countFrom(ToLongFunction<SelectStatementProvider> mapper, |
57 | 58 | SqlTable table, CountDSLCompleter completer) {
|
58 |
| - return count(mapper, SqlBuilder.countFrom(table), completer); |
| 59 | + return mapper.applyAsLong(countFrom(table, completer)); |
| 60 | + } |
| 61 | + |
| 62 | + public static SelectStatementProvider countFrom(CountDSL<SelectModel> start, CountDSLCompleter completer) { |
| 63 | + return completer.apply(start) |
| 64 | + .build() |
| 65 | + .render(RenderingStrategies.MYBATIS3); |
59 | 66 | }
|
60 | 67 |
|
61 |
| - public static long count(ToLongFunction<SelectStatementProvider> mapper, |
| 68 | + public static long countFrom(ToLongFunction<SelectStatementProvider> mapper, |
62 | 69 | CountDSL<SelectModel> start, CountDSLCompleter completer) {
|
63 |
| - return mapper.applyAsLong(completer.apply(start).build().render(RenderingStrategies.MYBATIS3)); |
| 70 | + return mapper.applyAsLong(countFrom(start, completer)); |
| 71 | + } |
| 72 | + |
| 73 | + public static DeleteStatementProvider deleteFrom(SqlTable table, DeleteDSLCompleter completer) { |
| 74 | + return completer.apply(SqlBuilder.deleteFrom(table)) |
| 75 | + .build() |
| 76 | + .render(RenderingStrategies.MYBATIS3); |
64 | 77 | }
|
65 | 78 |
|
66 | 79 | public static int deleteFrom(ToIntFunction<DeleteStatementProvider> mapper,
|
67 | 80 | SqlTable table, DeleteDSLCompleter completer) {
|
68 |
| - return mapper.applyAsInt( |
69 |
| - completer.apply(SqlBuilder.deleteFrom(table)) |
70 |
| - .build() |
71 |
| - .render(RenderingStrategies.MYBATIS3)); |
| 81 | + return mapper.applyAsInt(deleteFrom(table, completer)); |
72 | 82 | }
|
73 | 83 |
|
| 84 | + public static <R> InsertStatementProvider<R> insert(R record, SqlTable table, UnaryOperator<InsertDSL<R>> completer) { |
| 85 | + return completer.apply(SqlBuilder.insert(record).into(table)) |
| 86 | + .build() |
| 87 | + .render(RenderingStrategies.MYBATIS3); |
| 88 | + } |
| 89 | + |
74 | 90 | public static <R> int insert(ToIntFunction<InsertStatementProvider<R>> mapper, R record,
|
75 | 91 | SqlTable table, UnaryOperator<InsertDSL<R>> completer) {
|
76 |
| - return mapper.applyAsInt(completer.apply( |
77 |
| - SqlBuilder.insert(record).into(table)).build().render(RenderingStrategies.MYBATIS3)); |
| 92 | + return mapper.applyAsInt(insert(record, table, completer)); |
78 | 93 | }
|
79 | 94 |
|
| 95 | + public static <R> MultiRowInsertStatementProvider<R> insertMultiple(Collection<R> records, SqlTable table, |
| 96 | + UnaryOperator<MultiRowInsertDSL<R>> completer) { |
| 97 | + return completer.apply(SqlBuilder.insertMultiple(records).into(table)) |
| 98 | + .build() |
| 99 | + .render(RenderingStrategies.MYBATIS3); |
| 100 | + } |
| 101 | + |
80 | 102 | public static <R> int insertMultiple(ToIntFunction<MultiRowInsertStatementProvider<R>> mapper,
|
81 | 103 | Collection<R> records, SqlTable table, UnaryOperator<MultiRowInsertDSL<R>> completer) {
|
82 |
| - return mapper.applyAsInt(completer.apply( |
83 |
| - SqlBuilder.insertMultiple(records).into(table)).build().render(RenderingStrategies.MYBATIS3)); |
| 104 | + return mapper.applyAsInt(insertMultiple(records, table, completer)); |
84 | 105 | }
|
85 | 106 |
|
86 |
| - public static <R> List<R> selectDistinct(Function<SelectStatementProvider, List<R>> mapper, |
87 |
| - BasicColumn[] selectList, SqlTable table, SelectDSLCompleter completer) { |
88 |
| - return selectDistinct(mapper, SqlBuilder.selectDistinct(selectList).from(table), completer); |
| 107 | + public static SelectStatementProvider select(BasicColumn[] selectList, SqlTable table, |
| 108 | + SelectDSLCompleter completer) { |
| 109 | + return select(SqlBuilder.select(selectList).from(table), completer); |
| 110 | + } |
| 111 | + |
| 112 | + public static SelectStatementProvider select(QueryExpressionDSL<SelectModel> start, |
| 113 | + SelectDSLCompleter completer) { |
| 114 | + return completer.apply(start) |
| 115 | + .build() |
| 116 | + .render(RenderingStrategies.MYBATIS3); |
| 117 | + } |
| 118 | + |
| 119 | + public static SelectStatementProvider selectDistinct(BasicColumn[] selectList, SqlTable table, |
| 120 | + SelectDSLCompleter completer) { |
| 121 | + return select(SqlBuilder.selectDistinct(selectList).from(table), completer); |
89 | 122 | }
|
90 | 123 |
|
91 | 124 | public static <R> List<R> selectDistinct(Function<SelectStatementProvider, List<R>> mapper,
|
92 |
| - QueryExpressionDSL<SelectModel> start, SelectDSLCompleter completer) { |
93 |
| - return mapper.apply(completer.apply(start).build().render(RenderingStrategies.MYBATIS3)); |
| 125 | + BasicColumn[] selectList, SqlTable table, SelectDSLCompleter completer) { |
| 126 | + return mapper.apply(selectDistinct(selectList, table, completer)); |
94 | 127 | }
|
95 | 128 |
|
96 | 129 | public static <R> List<R> selectList(Function<SelectStatementProvider, List<R>> mapper,
|
97 | 130 | BasicColumn[] selectList, SqlTable table, SelectDSLCompleter completer) {
|
98 |
| - return selectList(mapper, SqlBuilder.select(selectList).from(table), completer); |
| 131 | + return mapper.apply(select(selectList, table, completer)); |
99 | 132 | }
|
100 | 133 |
|
101 | 134 | public static <R> List<R> selectList(Function<SelectStatementProvider, List<R>> mapper,
|
102 | 135 | QueryExpressionDSL<SelectModel> start, SelectDSLCompleter completer) {
|
103 |
| - return mapper.apply(completer.apply(start).build().render(RenderingStrategies.MYBATIS3)); |
| 136 | + return mapper.apply(select(start, completer)); |
104 | 137 | }
|
105 | 138 |
|
106 |
| - @Nullable |
107 | 139 | public static <R> R selectOne(Function<SelectStatementProvider, R> mapper,
|
108 | 140 | BasicColumn[] selectList, SqlTable table, SelectDSLCompleter completer) {
|
109 |
| - return selectOne(mapper, SqlBuilder.select(selectList).from(table), completer); |
| 141 | + return mapper.apply(select(selectList, table, completer)); |
110 | 142 | }
|
111 | 143 |
|
112 |
| - @Nullable |
113 | 144 | public static <R> R selectOne(Function<SelectStatementProvider, R> mapper,
|
114 |
| - QueryExpressionDSL<SelectModel> start, |
115 |
| - SelectDSLCompleter completer) { |
116 |
| - return mapper.apply(completer.apply(start).build().render(RenderingStrategies.MYBATIS3)); |
117 |
| - } |
118 |
| - |
119 |
| - @NotNull |
120 |
| - public static <R> Optional<R> selectOptional(Function<SelectStatementProvider, Optional<R>> mapper, |
121 |
| - BasicColumn[] selectList, SqlTable table, SelectDSLCompleter completer) { |
122 |
| - return selectOptional(mapper, SqlBuilder.select(selectList).from(table), completer); |
| 145 | + QueryExpressionDSL<SelectModel> start,SelectDSLCompleter completer) { |
| 146 | + return mapper.apply(select(start, completer)); |
123 | 147 | }
|
124 | 148 |
|
125 |
| - @NotNull |
126 |
| - public static <R> Optional<R> selectOptional(Function<SelectStatementProvider, Optional<R>> mapper, |
127 |
| - QueryExpressionDSL<SelectModel> start, SelectDSLCompleter completer) { |
128 |
| - return mapper.apply(completer.apply(start).build().render(RenderingStrategies.MYBATIS3)); |
| 149 | + public static UpdateStatementProvider update(SqlTable table, UpdateDSLCompleter completer) { |
| 150 | + return completer.apply(SqlBuilder.update(table)) |
| 151 | + .build() |
| 152 | + .render(RenderingStrategies.MYBATIS3); |
129 | 153 | }
|
130 | 154 |
|
131 | 155 | public static int update(ToIntFunction<UpdateStatementProvider> mapper,
|
132 | 156 | SqlTable table, UpdateDSLCompleter completer) {
|
133 |
| - return mapper.applyAsInt( |
134 |
| - completer.apply(SqlBuilder.update(table)) |
135 |
| - .build() |
136 |
| - .render(RenderingStrategies.MYBATIS3)); |
| 157 | + return mapper.applyAsInt(update(table, completer)); |
137 | 158 | }
|
138 | 159 | }
|
0 commit comments