|
15 | 15 | */
|
16 | 16 | package org.mybatis.dynamic.sql.util.kotlin.spring
|
17 | 17 |
|
| 18 | +import org.mybatis.dynamic.sql.BasicColumn |
| 19 | +import org.mybatis.dynamic.sql.SqlBuilder |
| 20 | +import org.mybatis.dynamic.sql.SqlTable |
18 | 21 | import org.mybatis.dynamic.sql.delete.render.DeleteStatementProvider
|
19 | 22 | import org.mybatis.dynamic.sql.insert.render.InsertStatementProvider
|
20 | 23 | import org.mybatis.dynamic.sql.select.render.SelectStatementProvider
|
21 | 24 | import org.mybatis.dynamic.sql.update.render.UpdateStatementProvider
|
| 25 | +import org.mybatis.dynamic.sql.util.kotlin.* |
22 | 26 | import org.springframework.jdbc.core.namedparam.BeanPropertySqlParameterSource
|
23 | 27 | import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate
|
24 | 28 | import java.sql.ResultSet
|
25 | 29 |
|
26 | 30 | fun NamedParameterJdbcTemplate.count(selectStatement: SelectStatementProvider) =
|
27 | 31 | queryForObject(selectStatement.selectStatement, selectStatement.parameters, Long::class.java)!!
|
28 | 32 |
|
| 33 | +fun NamedParameterJdbcTemplate.count(table: SqlTable, completer: CountCompleter) = |
| 34 | + count(org.mybatis.dynamic.sql.util.kotlin.spring.count(table, completer)) |
| 35 | + |
29 | 36 | fun NamedParameterJdbcTemplate.delete(deleteStatement: DeleteStatementProvider) =
|
30 | 37 | update(deleteStatement.deleteStatement, deleteStatement.parameters)
|
31 | 38 |
|
| 39 | +fun NamedParameterJdbcTemplate.deleteFrom(table: SqlTable, completer: DeleteCompleter) = |
| 40 | + delete(org.mybatis.dynamic.sql.util.kotlin.spring.deleteFrom(table, completer)) |
| 41 | + |
32 | 42 | fun <T> NamedParameterJdbcTemplate.insert(insertStatement: InsertStatementProvider<T>) =
|
33 | 43 | update(insertStatement.insertStatement, BeanPropertySqlParameterSource(insertStatement.record))
|
34 | 44 |
|
35 |
| -fun <T> NamedParameterJdbcTemplate.selectMany(selectStatement: SelectStatementProvider, rowMapper: (rs: ResultSet, rowNum: Int) -> T): List<T> = |
| 45 | +fun <T> NamedParameterJdbcTemplate.insert(record: T, table: SqlTable, completer: InsertCompleter<T>) = |
| 46 | + insert(org.mybatis.dynamic.sql.util.kotlin.spring.insert(record, table, completer)) |
| 47 | + |
| 48 | +fun NamedParameterJdbcTemplate.select(vararg selectList: BasicColumn) = |
| 49 | + FromGatherer(selectList.toList(), this) |
| 50 | + |
| 51 | +fun NamedParameterJdbcTemplate.selectDistinct(vararg selectList: BasicColumn) = |
| 52 | + DistinctFromGatherer(selectList.toList(), this) |
| 53 | + |
| 54 | +fun <T> NamedParameterJdbcTemplate.selectList(selectStatement: SelectStatementProvider, rowMapper: (rs: ResultSet, rowNum: Int) -> T): List<T> = |
36 | 55 | query(selectStatement.selectStatement, selectStatement.parameters, rowMapper)
|
37 | 56 |
|
38 | 57 | fun <T> NamedParameterJdbcTemplate.selectOne(selectStatement: SelectStatementProvider, rowMapper: (rs: ResultSet, rowNum: Int) -> T): T? =
|
39 | 58 | queryForObject(selectStatement.selectStatement, selectStatement.parameters, rowMapper)
|
40 | 59 |
|
41 | 60 | fun NamedParameterJdbcTemplate.update(updateStatement: UpdateStatementProvider) =
|
42 | 61 | update(updateStatement.updateStatement, updateStatement.parameters)
|
| 62 | + |
| 63 | +fun NamedParameterJdbcTemplate.update(table: SqlTable, completer: UpdateCompleter) = |
| 64 | + update(org.mybatis.dynamic.sql.util.kotlin.spring.update(table, completer)) |
| 65 | + |
| 66 | +// support classes for select DSL |
| 67 | +class FromGatherer(private val selectList: List<BasicColumn>, private val template: NamedParameterJdbcTemplate) { |
| 68 | + fun from(table: SqlTable, completer: SelectCompleter) = |
| 69 | + MapperGatherer(SqlBuilder.select(selectList).from(table, completer), template) |
| 70 | + |
| 71 | + fun from(table: SqlTable, alias: String, completer: SelectCompleter) = |
| 72 | + MapperGatherer(SqlBuilder.select(selectList).from(table, alias, completer), template) |
| 73 | +} |
| 74 | + |
| 75 | +class DistinctFromGatherer(private val selectList: List<BasicColumn>, private val template: NamedParameterJdbcTemplate) { |
| 76 | + fun from(table: SqlTable, completer: SelectCompleter) = |
| 77 | + MapperGatherer(SqlBuilder.selectDistinct(selectList).from(table, completer), template) |
| 78 | + |
| 79 | + fun from(table: SqlTable, alias: String, completer: SelectCompleter) = |
| 80 | + MapperGatherer(SqlBuilder.selectDistinct(selectList).from(table, alias, completer), template) |
| 81 | +} |
| 82 | + |
| 83 | +class MapperGatherer(private val selectStatement: SelectStatementProvider, private val template: NamedParameterJdbcTemplate) { |
| 84 | + fun <T> withRowMapper(rowMapper: (rs: ResultSet, rowNum: Int) -> T) = |
| 85 | + template.selectList(selectStatement, rowMapper) |
| 86 | +} |
0 commit comments