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

Fix addLastAll

Browse Source
This commit is contained in:
Him188 2020-02-12 20:47:30 +08:00
parent 20d7bdb4e4
commit 999ce68cec
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
mirai-core/src/commonMain/kotlin/net.mamoe.mirai/utils/LockFreeLinkedList.kt
View File

@ -142,13 +142,19 @@ open class LockFreeLinkedList<E> {
*/
@Suppress("DuplicatedCode")
open fun addLastAll(iterable: Iterable<E>) {
var firstNode: Node<E>? = null
var currentNode: Node<E>? = null
iterable.forEach {
val nextNode = it.asNode(tail)
if (firstNode == null) {
firstNode = nextNode
}
currentNode?.nextNode = nextNode
currentNode = nextNode
}
addLastNode(currentNode ?: error("iterable is empty"))
firstNode?.let { addLastNode(it) }
}
/**
@ -156,15 +162,19 @@ open class LockFreeLinkedList<E> {
*/
@Suppress("DuplicatedCode")
open fun addLastAll(iterable: Sequence<E>) {
var firstNode: Node<E>? = null
var currentNode: Node<E>? = null
iterable.forEach {
val nextNode = it.asNode(tail)
if (currentNode != null) { // do not use `?.` because atomicfu cannot transform properly: IllegalArgumentException: null passed
currentNode!!.nextNodeRef.value = nextNode
if (firstNode == null) {
firstNode = nextNode
}
currentNode?.nextNode = nextNode
currentNode = nextNode
}
addLastNode(currentNode ?: error("iterable is empty"))
firstNode?.let { addLastNode(it) }
}
open operator fun plusAssign(element: E) = this.addLast(element)