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

Add contracts to buildMessageChain

Browse Source
This commit is contained in:
Him188 2020-12-08 12:07:02 +08:00
parent 9b6f386f1c
commit 7df5dc65d3
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mirai-core-api/src/commonMain/kotlin/message/data/MessageChainBuilder.kt
View File

@ -13,9 +13,8 @@
package net.mamoe.mirai.message.data package net.mamoe.mirai.message.data
import kotlin.jvm.JvmMultifileClass import kotlin.contracts.InvocationKind.EXACTLY_ONCE
import kotlin.jvm.JvmName import kotlin.contracts.contract
import kotlin.jvm.JvmSynthetic
/** /**
* 构建一个 [MessageChain] * 构建一个 [MessageChain]
@ -24,6 +23,7 @@ import kotlin.jvm.JvmSynthetic
*/ */
@JvmSynthetic @JvmSynthetic
public inline fun buildMessageChain(block: MessageChainBuilder.() -> Unit): MessageChain { public inline fun buildMessageChain(block: MessageChainBuilder.() -> Unit): MessageChain {
contract { callsInPlace(block, EXACTLY_ONCE) }
return MessageChainBuilder().apply(block).asMessageChain() return MessageChainBuilder().apply(block).asMessageChain()
} }
@ -34,6 +34,7 @@ public inline fun buildMessageChain(block: MessageChainBuilder.() -> Unit): Mess
*/ */
@JvmSynthetic @JvmSynthetic
public inline fun buildMessageChain(initialSize: Int, block: MessageChainBuilder.() -> Unit): MessageChain { public inline fun buildMessageChain(initialSize: Int, block: MessageChainBuilder.() -> Unit): MessageChain {
contract { callsInPlace(block, EXACTLY_ONCE) }
return MessageChainBuilder(initialSize).apply(block).asMessageChain() return MessageChainBuilder(initialSize).apply(block).asMessageChain()
} }
@ -64,7 +65,7 @@ public open class MessageChainBuilder private constructor(
flushCache() flushCache()
@Suppress("UNCHECKED_CAST") @Suppress("UNCHECKED_CAST")
return when (element) { return when (element) {
is ConstrainSingle<*> -> addAndCheckConstrainSingle(element) is ConstrainSingle -> addAndCheckConstrainSingle(element)
is SingleMessage -> container.add(element) // no need to constrain is SingleMessage -> container.add(element) // no need to constrain
is Iterable<*> -> this.addAll(element.flatten()) is Iterable<*> -> this.addAll(element.flatten())
else -> error("stub") else -> error("stub")
@ -214,14 +215,14 @@ public open class MessageChainBuilder private constructor(
private var firstConstrainSingleIndex = -1 private var firstConstrainSingleIndex = -1
private fun addAndCheckConstrainSingle(element: SingleMessage): Boolean { private fun addAndCheckConstrainSingle(element: SingleMessage): Boolean {
if (element is ConstrainSingle<*>) { if (element is ConstrainSingle) {
if (firstConstrainSingleIndex == -1) { if (firstConstrainSingleIndex == -1) {
firstConstrainSingleIndex = container.size firstConstrainSingleIndex = container.size
return container.add(element) return container.add(element)
} }
val key = element.key val key = element.key
val index = container.indexOfFirst(firstConstrainSingleIndex) { it is ConstrainSingle<*> && it.key == key } val index = container.indexOfFirst(firstConstrainSingleIndex) { it is ConstrainSingle && it.key == key }
if (index != -1) { if (index != -1) {
container[index] = element container[index] = element
} else { } else {