org.http4k.servirtium.InteractionStorage.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of http4k-testing-servirtium Show documentation

http4k incubator module

There is a newer version: 5.41.0.0

Show newest version

package org.http4k.servirtium

import java.io.File
import java.util.concurrent.atomic.AtomicReference
import java.util.function.Consumer
import java.util.function.Supplier

typealias StorageProvider = (String) -> InteractionStorage

/**
 * Provides storage for the recorded Servirtium interaction data.
 */
interface InteractionStorage : Supplier, Consumer {
    fun clean(): Boolean

    companion object {
        @JvmStatic
        fun Disk(root: File): StorageProvider = object : StorageProvider {
            override fun invoke(name: String): InteractionStorage {
                val file = fileFor(name)
                return object : InteractionStorage {
                    override fun get() = file.takeIf { it.exists() }?.readBytes() ?: ByteArray(0)
                    override fun accept(data: ByteArray) {
                        file.apply { parentFile.mkdirs() }.appendBytes(data)
                    }

                    override fun clean(): Boolean = fileFor(name).delete()
                }
            }

            private fun fileFor(name: String) = File(root, "$name.md")
        }

        @JvmStatic
        fun InMemory() = object : StorageProvider {
            private val created = mutableMapOf>()

            override fun invoke(name: String): InteractionStorage {
                val ref = created[name] ?: AtomicReference(ByteArray(0))
                created[name] = ref
                return object : InteractionStorage {
                    override fun get() = ref.get()
                    override fun accept(data: ByteArray) {
                        ref.set(ref.get() + data)
                    }

                    override fun clean() = created[name]?.let { it.set(ByteArray(0)); true } ?: false
                }
            }
        }
    }
}