[+] Mai2 music ranking

src/main/java/icu/samnyan/aqua/sega/maimai2/Maimai2ServletController.kt

@@ -33,6 +33,7 @@ class Maimai2ServletController(
     val getUserFavoriteItem: GetUserFavoriteItemHandler,
     val getUserRivalMusic: GetUserRivalMusicHandler,
     val getUserCharacter: GetUserCharacterHandler,
+    val getGameRanking: GetGameRankingHandler,
     val repos: Mai2Repos
 ) {
     companion object {
@@ -222,7 +223,6 @@ class Maimai2ServletController(
     val getUserIntimate = UserReqHandler { _, uid -> mapOf("userId" to uid, "length" to 0, "userIntimateList" to empty) }
     val getTransferFriend = UserReqHandler { _, uid -> mapOf("userId" to uid, "transferFriendList" to empty) }
     val getGameNgMusicId = BaseHandler { mapOf("length" to 0, "musicIdList" to empty) }
-    val getGameRanking = BaseHandler { mapOf("type" to it["type"].toString(), "gameRankingList" to empty) }
     val getGameTournamentInfo = BaseHandler { mapOf("length" to 0, "gameTournamentInfoList" to empty) }
     val getGameKaleidxScope = BaseHandler { mapOf("gameKaleidxScopeList" to empty) }
     val getUserKaleidxScope = UserReqHandler { _, uid -> mapOf("userId" to uid, "userKaleidxScopeList" to empty) }

src/main/java/icu/samnyan/aqua/sega/maimai2/handler/GetGameRankingHandler.kt

@@ -0,0 +1,69 @@
+package icu.samnyan.aqua.sega.maimai2.handler
+
+import com.querydsl.jpa.impl.JPAQueryFactory
+import ext.logger
+import ext.thread
+import icu.samnyan.aqua.sega.general.BaseHandler
+import icu.samnyan.aqua.sega.maimai2.model.userdata.QMai2UserPlaylog
+import org.springframework.scheduling.annotation.Scheduled
+import org.springframework.stereotype.Component
+import java.time.LocalDateTime
+import java.time.format.DateTimeFormatter
+import kotlin.concurrent.Volatile
+
+/**
+ * @author samnyan (privateamusement@protonmail.com)
+ */
+@Component("Maimai2GetGameRankingHandler")
+class GetGameRankingHandler(
+    private val queryFactory: JPAQueryFactory
+) : BaseHandler {
+    private data class MusicRankingItem(val id: Int, val point: Long, val userName: String = "")
+
+    @Volatile
+    private var musicRankingCache: List<MusicRankingItem> = emptyList()
+
+    init {
+        // To make sure the cache is initialized before the first request,
+        // not using `initialDelay = 0` in `@Scheduled`.
+        thread { refreshMusicRankingCache() }
+    }
+
+    @Scheduled(fixedDelay = 3600_000)
+    private fun refreshMusicRankingCache() {
+        // Get the play count of each music in the last N days
+        val queryAfter = LocalDateTime.now().minusDays(LOOK_BACK_DAYS)
+        val formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")
+        val queryAfterStr = queryAfter.format(formatter)
+
+        val qPlaylog = QMai2UserPlaylog.mai2UserPlaylog
+        val cMusicId = qPlaylog.musicId
+        val cUserCount = qPlaylog.user.id.countDistinct()
+        musicRankingCache = queryFactory
+            .select(cMusicId, cUserCount)
+            .from(qPlaylog)
+            .where(qPlaylog.userPlayDate.stringValue().goe(queryAfterStr))
+            .groupBy(cMusicId)
+            .orderBy(cUserCount.desc())
+            .limit(QUERY_LIMIT)
+            .fetch()
+            .map { MusicRankingItem(it.get(cMusicId)!!, it.get(cUserCount)!!) }
+
+        log.info("Refreshed music ranking cache: ${musicRankingCache.size} items")
+    }
+
+    override fun handle(request: Map<String, Any>): Any = mapOf(
+        "type" to request["type"],
+        "gameRankingList" to when(request["type"]) {
+            1 -> musicRankingCache
+            else -> emptyList()
+        }
+    )
+
+    companion object {
+        val log = logger()
+        
+        const val LOOK_BACK_DAYS: Long = 7
+        const val QUERY_LIMIT: Long = 50
+    }
+}

src/main/java/icu/samnyan/aqua/spring/QuerydslConfig.kt

@@ -0,0 +1,18 @@
+package icu.samnyan.aqua.spring
+
+import com.querydsl.jpa.impl.JPAQueryFactory
+import jakarta.persistence.EntityManager
+import jakarta.persistence.PersistenceContext
+import org.springframework.context.annotation.Bean
+import org.springframework.context.annotation.Configuration
+
+@Configuration
+class QuerydslConfig {
+    @PersistenceContext
+    private lateinit var entityManager: EntityManager
+
+    @Bean
+    fun jpaQueryFactory(): JPAQueryFactory {
+        return JPAQueryFactory(entityManager)
+    }
+}