ai.platon.pulsar.common.collect.DelayCacheCollector.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pulsar-common Show documentation

There is a newer version: 2.1.0

package ai.platon.pulsar.common.collect

import ai.platon.pulsar.common.Priority13
import ai.platon.pulsar.common.collect.collector.AbstractPriorityDataCollector
import ai.platon.pulsar.common.urls.UrlAware
import java.util.*

open class DelayCacheCollector(
    val queue: Queue,
    priority: Int = Priority13.HIGHER.value
) : AbstractPriorityDataCollector(priority) {

    override var name = "DelayCacheC"

    override val size: Int
        get() = queue.size

    constructor(queue: Queue, priority: Priority13) : this(queue, priority.value)

    @Synchronized
    override fun hasMore() = queue.isNotEmpty()

    @Synchronized
    override fun collectTo(sink: MutableList): Int {
        beforeCollect()

        val count = queue.poll()?.takeIf { sink.add(it.url) }?.let { 1 } ?: 0

        return afterCollect(count)
    }

    @Synchronized
    override fun dump(): List {
        return queue.map { it.url.toString() + " delay: " + it.delay }
    }

    override fun clear() = queue.clear()
}