github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-24 23:20:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Him188 2020-01-17 23:29:22 +08:00
parent ccf8545646
commit 1dee5fe075
1 changed files with 14 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
mirai-api-http/src/main/kotlin/net.mamoe.mirai.api.http/Session.kt
View File

@ -1,13 +1,11 @@
package net.mamoe.mirai.api.http package net.mamoe.mirai.api.http
import kotlinx.coroutines.CompletableJob
import kotlinx.coroutines.CoroutineScope import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.SupervisorJob import kotlinx.coroutines.SupervisorJob
import java.lang.StringBuilder
import kotlin.coroutines.CoroutineContext import kotlin.coroutines.CoroutineContext
import kotlin.coroutines.EmptyCoroutineContext import kotlin.coroutines.EmptyCoroutineContext
tailrec fun generateSessionKey():String{ tailrec fun generateSessionKey(): String {
fun generateRandomSessionKey(): String { fun generateRandomSessionKey(): String {
val all = "QWERTYUIOPASDFGHJKLZXCVBNM1234567890qwertyuiopasdfghjklzxcvbnm" val all = "QWERTYUIOPASDFGHJKLZXCVBNM1234567890qwertyuiopasdfghjklzxcvbnm"
return buildString(capacity = 8) { return buildString(capacity = 8) {
@ -18,7 +16,7 @@ tailrec fun generateSessionKey():String{
} }
val key = generateRandomSessionKey() val key = generateRandomSessionKey()
if(!SessionManager.allSession.containsKey(key)){ if (!SessionManager.allSession.containsKey(key)) {
return key return key
} }
@ -27,43 +25,42 @@ tailrec fun generateSessionKey():String{
object SessionManager { object SessionManager {
val allSession:MutableMap<String,Session> = mutableMapOf() val allSession: MutableMap<String, Session> = mutableMapOf()
fun createTempSession():TempSession = TempSession(EmptyCoroutineContext).also { allSession[it.key] = it } fun createTempSession(): TempSession = TempSession(EmptyCoroutineContext).also { allSession[it.key] = it }
fun closeSession(sessionKey: String) = allSession.remove(sessionKey)?.also {it.close() } fun closeSession(sessionKey: String) = allSession.remove(sessionKey)?.also { it.close() }
fun closeSession(session: Session) = closeSession(session.key) fun closeSession(session: Session) = closeSession(session.key)
} }
/** /**
* @author NaturalHG * @author NaturalHG
* 这个用于管理不同Client与Mirai HTTP的会话 * 这个用于管理不同Client与Mirai HTTP的会话
*/ */
abstract class Session internal constructor( abstract class Session internal constructor(
coroutineContext: CoroutineContext
) : CoroutineScope {
private val supervisorJob = SupervisorJob()
final override val coroutineContext: CoroutineContext = supervisorJob + coroutineContext
): CoroutineScope { val key: String = generateSessionKey()
private val sessionJob = SupervisorJob()
val key:String = generateSessionKey()
internal fun close(){ internal fun close() {
sessionJob.cancel() supervisorJob.cancel()
} }
} }
/** /**
* 任何新链接建立后分配一个[TempSession] * 任何新链接建立后分配一个[TempSession]
* *
* TempSession在建立180s内没有转变为[AuthedSession]应被清除 * TempSession在建立180s内没有转变为[AuthedSession]应被清除
*/ */
class TempSession internal constructor(override val coroutineContext: CoroutineContext) : Session() { class TempSession internal constructor(coroutineContext: CoroutineContext) : Session(coroutineContext) {
} }
@ -71,7 +68,7 @@ class TempSession internal constructor(override val coroutineContext: CoroutineC
* 任何[TempSession]认证后转化为一个[AuthedSession] * 任何[TempSession]认证后转化为一个[AuthedSession]
* 在这一步[AuthedSession]应该已经有assigned的bot * 在这一步[AuthedSession]应该已经有assigned的bot
*/ */
class AuthedSession internal constructor(botNumber:Int, override val coroutineContext: CoroutineContext):Session(){ class AuthedSession internal constructor(botNumber: Int, coroutineContext: CoroutineContext) : Session(coroutineContext) {
} }