Skip to content

Commit b819973

Browse files
committed
Style updates and simplification
1 parent 205cd21 commit b819973

File tree

2 files changed

+31
-39
lines changed

2 files changed

+31
-39
lines changed

src/main/kotlin/org/mybatis/dynamic/sql/util/kotlin/model/ModelBuilderFunctions.kt

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -66,21 +66,21 @@ fun deleteFrom(table: SqlTable, tableAlias: String, completer: DeleteCompleter):
6666
KotlinDeleteBuilder(SqlBuilder.deleteFrom(table, tableAlias)).apply(completer).build()
6767

6868
fun <T> insert(row: T & Any, completer: KotlinInsertCompleter<T>): InsertModel<T> {
69-
val f : KotlinInsertBuilder<T> = KotlinInsertBuilder(row)
70-
return f.apply(completer).build()
69+
val builder : KotlinInsertBuilder<T> = KotlinInsertBuilder(row)
70+
return builder.apply(completer).build()
7171
}
7272

7373
fun <T> insertBatch(rows: Collection<T & Any>, completer: KotlinBatchInsertCompleter<T>): BatchInsertModel<T> {
74-
val f: KotlinBatchInsertBuilder<T> = KotlinBatchInsertBuilder(rows)
75-
return f.apply(completer).build()
74+
val builder: KotlinBatchInsertBuilder<T> = KotlinBatchInsertBuilder(rows)
75+
return builder.apply(completer).build()
7676
}
7777

7878
fun insertInto(table: SqlTable, completer: GeneralInsertCompleter): GeneralInsertModel =
7979
KotlinGeneralInsertBuilder(table).apply(completer).build()
8080

8181
fun <T> insertMultiple(rows: Collection<T & Any>, completer: KotlinMultiRowInsertCompleter<T>): MultiRowInsertModel<T> {
82-
val f: KotlinMultiRowInsertBuilder<T> = KotlinMultiRowInsertBuilder(rows)
83-
return f.apply(completer).build()
82+
val builder: KotlinMultiRowInsertBuilder<T> = KotlinMultiRowInsertBuilder(rows)
83+
return builder.apply(completer).build()
8484
}
8585

8686
fun insertSelect(table: SqlTable, completer: InsertSelectCompleter): InsertSelectModel =

src/main/kotlin/org/mybatis/dynamic/sql/util/kotlin/mybatis3/MapperSupportFunctions.kt

Lines changed: 25 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -42,41 +42,38 @@ fun count(
4242
table: SqlTable,
4343
completer: CountCompleter
4444
): Long =
45-
count(column) {
45+
mapper(count(column) {
4646
from(table)
4747
run(completer)
48-
}.run(mapper)
48+
})
4949

5050
fun countDistinct(
5151
mapper: (SelectStatementProvider) -> Long,
5252
column: BasicColumn,
5353
table: SqlTable,
5454
completer: CountCompleter
5555
): Long =
56-
countDistinct(column) {
56+
mapper(countDistinct(column) {
5757
from(table)
5858
run(completer)
59-
}.run(mapper)
59+
})
6060

6161
fun countFrom(mapper: (SelectStatementProvider) -> Long, table: SqlTable, completer: CountCompleter): Long =
62-
countFrom(table, completer).run(mapper)
62+
mapper(countFrom(table, completer))
6363

6464
fun deleteFrom(mapper: (DeleteStatementProvider) -> Int, table: SqlTable, completer: DeleteCompleter): Int =
65-
deleteFrom(table, completer).run(mapper)
65+
mapper(deleteFrom(table, completer))
6666

6767
fun <T> insert(
6868
mapper: (InsertStatementProvider<T>) -> Int,
6969
row: T & Any,
7070
table: SqlTable,
7171
completer: KotlinInsertCompleter<T>
72-
): Int {
73-
val f : InsertStatementProvider<T> = insert(row) {
72+
): Int =
73+
mapper(insert(row) {
7474
into(table)
7575
run(completer)
76-
}
77-
78-
return f.run(mapper)
79-
}
76+
})
8077

8178
/**
8279
* This function simply inserts all rows using the supplied mapper. It is up
@@ -91,90 +88,85 @@ fun <T> insertBatch(
9188
table: SqlTable,
9289
completer: KotlinBatchInsertCompleter<T>
9390
): List<Int> {
94-
val f: BatchInsert<T> = insertBatch(records) {
91+
val batchInsert: BatchInsert<T> = insertBatch(records) {
9592
into(table)
9693
run(completer)
9794
}
9895

99-
return f.insertStatements().map(mapper)
96+
return batchInsert.insertStatements().map(mapper)
10097
}
10198

10299
fun insertInto(
103100
mapper: (GeneralInsertStatementProvider) -> Int,
104101
table: SqlTable,
105102
completer: GeneralInsertCompleter
106103
): Int =
107-
insertInto(table, completer).run(mapper)
104+
mapper(insertInto(table, completer))
108105

109106
fun <T> insertMultiple(
110107
mapper: (MultiRowInsertStatementProvider<T>) -> Int,
111108
records: Collection<T & Any>,
112109
table: SqlTable,
113110
completer: KotlinMultiRowInsertCompleter<T>
114-
): Int {
115-
val f: MultiRowInsertStatementProvider<T> = insertMultiple(records) {
111+
): Int =
112+
mapper(insertMultiple(records) {
116113
into(table)
117114
run(completer)
118-
}
119-
120-
return f.run(mapper)
121-
}
115+
})
122116

123117
fun <T> insertMultipleWithGeneratedKeys(
124118
mapper: (String, List<T>) -> Int,
125119
records: Collection<T & Any>,
126120
table: SqlTable,
127121
completer: KotlinMultiRowInsertCompleter<T>
128122
): Int {
129-
val f: MultiRowInsertStatementProvider<T> = insertMultiple(records) {
123+
val provider: MultiRowInsertStatementProvider<T> = insertMultiple(records) {
130124
into(table)
131125
run(completer)
132126
}
133127

134-
return f.run {
135-
mapper(insertStatement, this.records)
136-
}
128+
return mapper(provider.insertStatement, provider.records)
137129
}
138130

139131
fun insertSelect(
140132
mapper: (InsertSelectStatementProvider) -> Int,
141133
table: SqlTable,
142134
completer: InsertSelectCompleter
143135
): Int =
144-
insertSelect(table, completer).run(mapper)
136+
mapper(insertSelect(table, completer))
145137

146138
fun <T> selectDistinct(
147139
mapper: (SelectStatementProvider) -> List<T>,
148140
selectList: List<BasicColumn>,
149141
table: SqlTable,
150142
completer: SelectCompleter
151143
): List<T> =
152-
selectDistinct(selectList) {
144+
mapper(selectDistinct(selectList) {
153145
from(table)
154146
run(completer)
155-
}.run(mapper)
147+
})
156148

157149
fun <T> selectList(
158150
mapper: (SelectStatementProvider) -> List<T>,
159151
selectList: List<BasicColumn>,
160152
table: SqlTable,
161153
completer: SelectCompleter
162154
): List<T> =
163-
select(selectList) {
155+
mapper(select(selectList) {
164156
from(table)
165157
run(completer)
166-
}.run(mapper)
158+
})
167159

168160
fun <T> selectOne(
169161
mapper: (SelectStatementProvider) -> T?,
170162
selectList: List<BasicColumn>,
171163
table: SqlTable,
172164
completer: SelectCompleter
173165
): T? =
174-
select(selectList) {
166+
mapper(select(selectList) {
175167
from(table)
176168
run(completer)
177-
}.run(mapper)
169+
})
178170

179171
fun update(mapper: (UpdateStatementProvider) -> Int, table: SqlTable, completer: UpdateCompleter): Int =
180-
update(table, completer).run(mapper)
172+
mapper(update(table, completer))

0 commit comments

Comments
 (0)