selenium 인스턴스 풀 관리

RemoteWebDriverPool.kt

import kotlinx.coroutines.sync.Semaphore
import java.util.concurrent.ConcurrentLinkedQueue

open class ObjectPool<T : Any>(
    capacity: Int,
    private val instanceFactory: () -> T,
) {
    private val semaphore = Semaphore(permits = capacity)
    private val queue = ConcurrentLinkedQueue<T>()

    suspend fun borrow(): T {
        semaphore.acquire()
        try {
            return queue.poll() ?: instanceFactory()
        } catch (e: Exception) {
            semaphore.release()
            throw e
        }
    }

    fun recycle(instance: T) {
        queue.add(instance)
        semaphore.release()
    }

    open fun dispose(instance: T) {
        semaphore.release()
    }
}

suspend inline fun <T : Any, R> ObjectPool<T>.useInstance(block: (T) -> R): R {
    val instance = borrow()
    try {
        return block(instance).also {
            recycle(instance)
        }
    } catch (e: Exception) {
        dispose(instance)
        throw e
    }
}

///////////////////////////////////////////////////////
///////////////////////////////////////////////////////
///////////////////////////////////////////////////////

import org.openqa.selenium.remote.RemoteWebDriver

open class RemoteWebDriverPool(
    capacity: Int,
    instanceFactory: () -> RemoteWebDriver,
): ObjectPool<RemoteWebDriver>(capacity, instanceFactory) {
    override fun dispose(instance: RemoteWebDriver) {
        instance.close()
        super.dispose(instance)
    }
}