Fixed inline call and added query selector

Astra-Interactive · Aug 6, 2022 · d72012d · d72012d
1 parent 6e80c2e
commit d72012d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/main/kotlin/com/astrainteractive/astratemplate/api/DatabaseApi.kt b/src/main/kotlin/com/astrainteractive/astratemplate/api/DatabaseApi.kt
@@ -20,11 +20,11 @@ object DatabaseApi {
      * @return null or T
      */
     suspend fun createUserTable() = catching {
-        SQLDatabase.createTable(User::class.java)
+        SQLDatabase.createTable<User>()
     }
 
     suspend fun createRatingTable() = catching {
-        SQLDatabase.createTable(RatingRelation::class.java)
+        SQLDatabase.createTable<RatingRelation>()
     }
 
     /**

diff --git a/src/main/kotlin/com/astrainteractive/astratemplate/sqldatabase/lib/DatabaseCore.kt b/src/main/kotlin/com/astrainteractive/astratemplate/sqldatabase/lib/DatabaseCore.kt
@@ -28,6 +28,8 @@ abstract class DatabaseCore() {
     inline fun <reified T> select(where: String = "") = select(T::class.java, where)
     inline fun <reified T> insert(instance: T) = insert(T::class.java, instance)
     inline fun <reified T> createTable() = createTable(T::class.java)
+    inline fun <reified T> querySelect(query: String, noinline rsBuilder: (ResultSet) -> T?) =
+        querySelect(T::class.java, query, rsBuilder)
     fun <T> update(clazz: Class<out T>, instance: T): Boolean? {
         val info = AnnotationUtils.EntityInfo.create(clazz, instance)
         val primaryKeyInfo =
@@ -64,6 +66,14 @@ abstract class DatabaseCore() {
     }
 
 
+
+    fun <T> querySelect(clazz: Class<out T>, query: String = "", rsBuilder: (ResultSet) -> T?): List<T>? {
+        return connection?.createStatement()?.executeQuery(query)?.mapNotNull { rs ->
+            rsBuilder.invoke(rs)
+        }
+    }
+
+
     fun <T> insert(clazz: Class<out T>, instance: T): Long? {
         val info = AnnotationUtils.EntityInfo.create(clazz, instance)
         val keys = info?.columns?.mapNotNull {