lzc256
/
AquaDX
mirror of https://github.com/hykilpikonna/AquaDX
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

[O] Better pagination

matching
Azalea 2024-12-27 23:31:22 -05:00
parent 04a178eda6
commit 8d23b262c7
1 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/main/java/icu/samnyan/aqua/sega/general/BaseHandler.kt
View File

@ -11,20 +11,21 @@ fun interface BaseHandler {
fun handle(request: Map<String, Any>): Any? fun handle(request: Map<String, Any>): Any?
} }
const val MAX_COUNT = Int.MAX_VALUE / 2
class RequestContext( class RequestContext(
val req: HttpServletRequest, val req: HttpServletRequest,
val data: Map<String, Any>, val data: MutableMap<String, Any>,
) { ) {
val uid by lazy { parsing { data["userId"]!!.long } } val uid by lazy { parsing { data["userId"]!!.long } }
val nextIndex by lazy { parsing { data["nextIndex"]?.int ?: -1 } } val nextIndex by lazy { parsing { data["nextIndex"]?.int ?: -1 } }
val maxCount by lazy { parsing { data["maxCount"]?.int ?: Int.MAX_VALUE } } val maxCount by lazy { parsing { data["maxCount"]?.int ?: MAX_COUNT } }
} }
typealias SpecialHandler = RequestContext.() -> Any? typealias SpecialHandler = RequestContext.() -> Any?
fun BaseHandler.toSpecial() = { ctx: RequestContext -> handle(ctx.data) } fun BaseHandler.toSpecial() = { ctx: RequestContext -> handle(ctx.data) }
typealias PagedHandler = RequestContext.() -> List<Any> typealias PagedHandler = RequestContext.() -> List<Any>
typealias AddFn = RequestContext.() -> Map<String, Any> typealias AddFn = RequestContext.() -> Pair<Map<String, Any>?, PagedHandler>
// A very :3 way of declaring APIs // A very :3 way of declaring APIs
abstract class MeowApi(val serialize: (String, Any?) -> String) { abstract class MeowApi(val serialize: (String, Any?) -> String) {
@ -35,30 +36,33 @@ abstract class MeowApi(val serialize: (String, Any?) -> String) {
// Page Cache: {cache key: (timestamp, full list)} // Page Cache: {cache key: (timestamp, full list)}
val pageCache = mutableMapOf<String, Pair<Long, List<Any>>>() val pageCache = mutableMapOf<String, Pair<Long, List<Any>>>()
private fun String.pagedHelper(key: String, fn: PagedHandler, addFn: AddFn?) = this api@ { fun String.pagedWithKind(key: String, addFn: AddFn) = this api@ {
val add = addFn?.invoke(this) ?: emptyMap() val (maybeAdd, fn) = addFn()
val add = maybeAdd ?: emptyMap()
if (nextIndex == -1) return@api fn().let { if (nextIndex == -1) return@api fn().let { mapOf("userId" to uid, "length" to it.size, key to it) + add }
mapOf("userId" to uid, "length" to it.size, "nextIndex" to -1, key to it) + add
}
// Try to get cache // Try to get cache
val cacheKey = "$this:$uid:$add" val cacheKey = "$this:$uid:$add"
val list = pageCache.getOrPut(cacheKey) { millis() to fn() }.r val list = pageCache.getOrPut(cacheKey) { fn().let {
if (it.size > maxCount) millis() to it
else return@api mapOf("userId" to uid, "length" to it.size, "nextIndex" to -1, key to it) + add
} }.r
// Get sublist and next index // Get sublist and next index
val iAfter = (nextIndex + maxCount).coerceAtMost(list.size) var iAfter = (nextIndex + maxCount).coerceAtMost(list.size)
val lst = list.slice(nextIndex until iAfter) val lst = list.slice(nextIndex until iAfter)
// Update cache if needed // Update cache if needed
if (iAfter == list.size) pageCache.remove(cacheKey) if (iAfter == list.size) {
pageCache.remove(cacheKey)
iAfter = -1
}
else pageCache[cacheKey] = millis() to list else pageCache[cacheKey] = millis() to list
mapOf("userId" to uid, "length" to lst.size, "nextIndex" to iAfter, key to lst) + add mapOf("userId" to uid, "length" to lst.size, "nextIndex" to iAfter, key to lst) + add
} }
fun String.paged(key: String, fn: PagedHandler) = pagedHelper(key, fn, null) fun String.paged(key: String, fn: PagedHandler) = pagedWithKind(key) { null to fn }
fun String.pagedWith(key: String, fn: PagedHandler) = this to key to fn
infix fun Pair<String, Pair<String, PagedHandler>>.with(addFn: AddFn) = l.pagedHelper(r.l, r.r, addFn)
// Page cache cleanup every minute // Page cache cleanup every minute
@Scheduled(fixedRate = (1000 * 60)) @Scheduled(fixedRate = (1000 * 60))