dalbodeule/chibot-chzzk-bot
1
0
Fork 0
mirror of https://github.com/dalbodeule/chibot-chzzk-bot.git synced 2025-06-09 07:18:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

debug on eagerLoading (1x)

Browse Source
This commit is contained in:
dalbodeule 2024-08-28 21:01:35 +09:00
parent a7069046f6
commit 534aaecb6e
No known key found for this signature in database
GPG Key ID: EFA860D069C9FA65
3 changed files with 59 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
common/src/main/kotlin/space/mori/chzzk_bot/common/services/UserService.kt
View File

@ -1,5 +1,6 @@
package space.mori.chzzk_bot.common.services
import org.jetbrains.exposed.dao.load
import org.jetbrains.exposed.sql.SqlExpressionBuilder.eq
import org.jetbrains.exposed.sql.transactions.transaction
import space.mori.chzzk_bot.common.models.User
@ -27,13 +28,16 @@ object UserService {
fun updateUser(user: User, discordID: Long): User {
return transaction {
user.discord = discordID
user.load(User::subordinates, User::managers)
user
}
}
fun getUser(id: Int): User? {
return transaction {
User.findById(id)
val user = User.findById(id)
user?.load(User::subordinates, User::managers)
user
}
}
@ -41,7 +45,9 @@ object UserService {
return transaction {
val users = User.find(Users.discord eq discordID)
users.firstOrNull()
val user = users.firstOrNull()
user?.load(User::subordinates, User::managers)
user
}
}
@ -49,7 +55,9 @@ object UserService {
return transaction {
val users = User.find(Users.token eq chzzkID)
users.firstOrNull()
val user = users.firstOrNull()
user?.load(User::subordinates, User::managers)
user
}
}
@ -57,7 +65,9 @@ object UserService {
return transaction {
val users = User.find(Users.liveAlertGuild eq discordGuildId)
users.firstOrNull()
val user = users.firstOrNull()
user?.load(User::subordinates, User::managers)
user
}
}
@ -65,7 +75,9 @@ object UserService {
return transaction {
val users = User.find(Users.naverId eq naverId)
users.firstOrNull()
val user = users.firstOrNull()
user?.load(User::subordinates, User::managers)
user
}
}
@ -81,6 +93,8 @@ object UserService {
user.liveAlertChannel = channelId
user.liveAlertMessage = alertMessage ?: ""
user.load(User::subordinates, User::managers)
user
}
}

3
webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/routes/ApiRoutes.kt
View File

@ -18,6 +18,7 @@ import space.mori.chzzk_bot.common.services.UserService
import space.mori.chzzk_bot.common.utils.getStreamInfo
import space.mori.chzzk_bot.common.utils.getUserInfo
import space.mori.chzzk_bot.webserver.UserSession
import space.mori.chzzk_bot.webserver.utils.ChzzkUsercache
@Serializable
data class GetUserDTO(
@ -119,7 +120,7 @@ fun Routing.apiRoutes() {
val subordinates = user.subordinates
println(subordinates)
returnUsers.addAll(subordinates.map {
val subStatus = it.token?.let { it1 -> getStreamInfo(it1) }
val subStatus = it.token?.let { token -> ChzzkUsercache.getCachedUser(token) }
return@map if (it.token == null || subStatus?.content == null) {
null
} else {

38
webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/utils/ChzzkUserCache.kt Normal file
View File

@ -0,0 +1,38 @@
package space.mori.chzzk_bot.webserver.utils
import kotlinx.coroutines.sync.Mutex
import kotlinx.coroutines.sync.withLock
import org.slf4j.LoggerFactory
import space.mori.chzzk_bot.common.utils.IData
import space.mori.chzzk_bot.common.utils.IStreamInfo
import space.mori.chzzk_bot.common.utils.getStreamInfo
import java.time.Instant
import java.util.concurrent.ConcurrentHashMap
object ChzzkUsercache {
private val cache = ConcurrentHashMap<String, CachedUser>()
private const val EXP_SECONDS = 600L
private val mutex = Mutex()
private val logger = LoggerFactory.getLogger(this::class.java)
suspend fun getCachedUser(id: String): IData<IStreamInfo?>? {
val now = Instant.now()
var user = cache[id]
if(user == null || user.timestamp.plusSeconds(EXP_SECONDS).isBefore(now)) {
mutex.withLock {
if(user == null || user?.timestamp?.plusSeconds(EXP_SECONDS)?.isBefore(now) != false) {
user = CachedUser(getStreamInfo(id))
user?.let { cache[id] = user!! }
}
}
}
return cache[id]?.user
}
}
data class CachedUser(
val user: IData<IStreamInfo?>,
val timestamp: Instant = Instant.now(),
)