Add getTokenIDByNoteID method to ShareLinkDao

kahverengi · kahverengi · commit 116e36bfb9ba · 2021-05-04T22:57:08.000+03:00
diff --git a/src/main/kotlin/com/parnote/db/dao/ShareLinkDao.kt b/src/main/kotlin/com/parnote/db/dao/ShareLinkDao.kt
@@ -26,6 +26,12 @@ interface ShareLinkDao : Dao<ShareLink> {
         handler: (noteID: Int?, asyncResult: AsyncResult<*>) -> Unit
     )
 
+    fun getTokenIDByNoteID(
+        noteID: Int,
+        sqlConnection: SQLConnection,
+        handler: (tokenID: Int?, asyncResult: AsyncResult<*>) -> Unit
+    )
+
     fun isLinkExistsByNoteID(
         noteID: Int,
         sqlConnection: SQLConnection,
diff --git a/src/main/kotlin/com/parnote/db/entity/ShareLinkDaoImpl.kt b/src/main/kotlin/com/parnote/db/entity/ShareLinkDaoImpl.kt
@@ -85,6 +85,24 @@ class ShareLinkDaoImpl(override val tableName: String = "share_link") : DaoImpl(
         }
     }
 
+    override fun getTokenIDByNoteID(
+        noteID: Int,
+        sqlConnection: SQLConnection,
+        handler: (tokenID: Int?, asyncResult: AsyncResult<*>) -> Unit
+    ) {
+        val query =
+            "SELECT `token_id` FROM `${getTablePrefix() + tableName}` WHERE `note_id` = ?"
+
+        sqlConnection.queryWithParams(query, JsonArray().add(noteID)) { queryResult ->
+            if (queryResult.succeeded()) {
+                val row = queryResult.result().results[0]
+
+                handler.invoke(row.getInteger(0), queryResult)
+            } else
+                handler.invoke(null, queryResult)
+        }
+    }
+
     override fun isLinkExistsByNoteID(
         noteID: Int,
         sqlConnection: SQLConnection,
