io.github.databob.DateTimeGenerator.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of databob.kotlin Show documentation
Kill all builders!
There is a newer version: 1.9.0
package io.github.databob

import io.github.databob.Databob
import io.github.databob.Generator
import io.github.databob.Generators
import java.lang.reflect.Type
import java.sql.Timestamp
import java.time.*
import java.util.*

class DateTimeGenerator : Generator {

    /**
     * Pre-packed generator instances
     */
    object instances {
        private val defaults = CompositeGenerator(
                Generators.ofType { d -> ZoneId.systemDefault() },
                Generators.ofType { d -> LocalDateTime.of(d.mk(), d.mk()) },
                Generators.ofType { d -> ZonedDateTime.of(d.mk(), d.mk()) },
                Generators.ofType { d -> java.sql.Date(d.mk().time) },
                Generators.ofType { d -> Timestamp.valueOf(d.mk()) },
                Generators.ofType { d -> Duration.ofMillis(d.mk()) },
                Generators.ofType { d -> Period.between(d.mk(), d.mk()) }
        )

        val random = CompositeGenerator(
                Generators.ofType { d -> Date(d.mk()) },
                Generators.ofType { d -> LocalTime.ofNanoOfDay(d.mk().toLong()) },
                Generators.ofType { d -> LocalDate.ofEpochDay(d.mk().toLong()) }
        ).with(defaults)

        val now = CompositeGenerator(
                Generators.ofType { -> Date() },
                Generators.ofType { d -> LocalTime.now(d.mk()) },
                Generators.ofType { d -> LocalDate.now(d.mk()) }
        ).with(defaults)

        val epoch = CompositeGenerator(
                Generators.ofType { -> Date(0) },
                Generators.ofType { -> LocalTime.ofSecondOfDay(0) },
                Generators.ofType { -> LocalDate.ofEpochDay(0) }
        ).with(defaults)
    }

    private val generator = CompositeGenerator(
            Generators.ofType { d -> ZoneId.systemDefault() },
            Generators.ofType { d -> Date(d.mk()) },
            Generators.ofType { d -> LocalTime.ofNanoOfDay(d.mk().toLong()) },
            Generators.ofType { d -> LocalDate.ofEpochDay(d.mk().toLong()) },
            Generators.ofType { d -> LocalDateTime.of(d.mk(), d.mk()) },
            Generators.ofType { d -> ZonedDateTime.of(d.mk(), d.mk()) },
            Generators.ofType { d -> java.sql.Date(d.mk().time) },
            Generators.ofType { d -> Timestamp.valueOf(d.mk()) },
            Generators.ofType { d -> Duration.ofMillis(d.mk()) },
            Generators.ofType { d -> Period.between(d.mk(), d.mk()) }
    )

    override fun mk(type: Type, databob: Databob): Any? = generator.mk(type, databob)
}