Sun, 20 Dec 2020 11:06:25 +0100
minimize footprint of flexmark - fixes #116
159 | 1 | /* |
2 | * Copyright 2020 Mike Becker. All rights reserved. | |
3 | * | |
4 | * Redistribution and use in source and binary forms, with or without | |
5 | * modification, are permitted provided that the following conditions are met: | |
6 | * | |
7 | * 1. Redistributions of source code must retain the above copyright | |
8 | * notice, this list of conditions and the following disclaimer. | |
9 | * | |
10 | * 2. Redistributions in binary form must reproduce the above copyright | |
11 | * notice, this list of conditions and the following disclaimer in the | |
12 | * documentation and/or other materials provided with the distribution. | |
13 | * | |
14 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" | |
15 | * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
16 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE | |
17 | * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE | |
18 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
19 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR | |
20 | * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER | |
21 | * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, | |
22 | * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
23 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
24 | * | |
25 | */ | |
26 | ||
27 | package de.uapcore.lightpit.dao | |
28 | ||
29 | import java.sql.PreparedStatement | |
30 | import java.sql.ResultSet | |
31 | import java.sql.Types | |
32 | ||
33 | abstract class AbstractDao<T> { | |
34 | ||
35 | abstract fun mapResult(rs: ResultSet): T | |
36 | ||
37 | protected fun list(stmt: PreparedStatement): List<T> { | |
38 | return sequence { | |
39 | stmt.executeQuery().use { result -> | |
40 | while (result.next()) yield(mapResult(result)) | |
41 | } | |
42 | }.toList() | |
43 | } | |
44 | ||
45 | protected fun find(stmt: PreparedStatement): T? { | |
46 | stmt.executeQuery().use { result -> | |
47 | return if (result.next()) { | |
48 | mapResult(result) | |
49 | } else { | |
50 | null | |
51 | } | |
52 | } | |
53 | } | |
54 | ||
55 | // TODO: create PreparedStatement abstraction that provides some features | |
56 | ||
57 | // TODO: remove the following legacy code helper function | |
58 | protected fun <T> setForeignKeyOrNull(stmt: PreparedStatement, index: Int, instance: T?, keyGetter: (obj: T) -> Int) { | |
59 | if (instance == null) { | |
60 | stmt.setNull(index, Types.INTEGER) | |
61 | } else { | |
62 | stmt.setInt(index, keyGetter(instance)) | |
63 | } | |
64 | } | |
65 | } |