Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Hocon config companions #562

Merged
merged 7 commits into from
Sep 30, 2024
Merged
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1,85 @@
package com.avsystem.commons
package hocon

import com.avsystem.commons.meta.MacroInstances
import com.avsystem.commons.serialization.GenCodec.ReadFailure
import com.avsystem.commons.serialization.{GenCodec, GenKeyCodec, GenObjectCodec}
import com.typesafe.config.{Config, ConfigFactory, ConfigObject}

import java.time.{Period, Duration as JDuration}
import scala.concurrent.duration.*

trait HoconGenCodecs {
implicit def configCodec: GenCodec[Config] = HoconGenCodecs.ConfigCodec
implicit def finiteDurationCodec: GenCodec[FiniteDuration] = HoconGenCodecs.FiniteDurationCodec
implicit def jDurationCodec: GenCodec[JDuration] = HoconGenCodecs.JavaDurationCodec
implicit def periodCodec: GenCodec[Period] = HoconGenCodecs.PeriodCodec
implicit def sizeInBytesCodec: GenCodec[SizeInBytes] = HoconGenCodecs.SizeInBytesCodec
implicit def classKeyCodec: GenKeyCodec[Class[?]] = HoconGenCodecs.ClassKeyCodec
implicit def classCodec: GenCodec[Class[?]] = HoconGenCodecs.ClassCodec
}
object HoconGenCodecs {
implicit final val ConfigCodec: GenCodec[Config] = GenCodec.nullable(
input =>
input.readCustom(ConfigValueMarker).fold(ConfigFactory.parseString(input.readSimple().readString())) {
case obj: ConfigObject => obj.toConfig
case v => throw new ReadFailure(s"expected a config OBJECT, got ${v.valueType}")
},
(output, value) =>
if (!output.writeCustom(ConfigValueMarker, value.root)) {
output.writeSimple().writeString(value.root.render)
},
)

implicit final val FiniteDurationCodec: GenCodec[FiniteDuration] = GenCodec.nullable(
input => input.readCustom(DurationMarker).fold(input.readSimple().readLong())(_.toMillis).millis,
(output, value) => output.writeSimple().writeLong(value.toMillis),
)

implicit val JavaDurationCodec: GenCodec[JDuration] = GenCodec.nullable(
input => input.readCustom(DurationMarker).getOrElse(JDuration.ofMillis(input.readSimple().readLong())),
(output, value) => output.writeSimple().writeLong(value.toMillis),
)

implicit final val PeriodCodec: GenCodec[Period] = GenCodec.nullable(
input => input.readCustom(PeriodMarker).getOrElse(Period.parse(input.readSimple().readString())),
(output, value) => output.writeSimple().writeString(value.toString),
)

implicit final val SizeInBytesCodec: GenCodec[SizeInBytes] = GenCodec.nonNull(
input => SizeInBytes(input.readCustom(SizeInBytesMarker).getOrElse(input.readSimple().readLong())),
(output, value) => output.writeSimple().writeLong(value.bytes),
)

implicit final val ClassKeyCodec: GenKeyCodec[Class[?]] =
GenKeyCodec.create(Class.forName, _.getName)

implicit final val ClassCodec: GenCodec[Class[?]] =
GenCodec.nullableString(Class.forName, _.getName)
}

object DefaultHoconGenCodecs extends HoconGenCodecs

trait ConfigObjectCodec[T] {
def objectCodec: GenObjectCodec[T]
}

abstract class AbstractConfigCompanion[Implicits <: HoconGenCodecs, T](
implicits: Implicits
)(implicit instances: MacroInstances[Implicits, ConfigObjectCodec[T]]
) {
implicit lazy val codec: GenCodec[T] = instances(implicits, this).objectCodec

final def read(config: Config): T = HoconInput.read[T](config)
}

/**
* Base class for companion objects of configuration case classes and sealed traits
* (typically deserialized from HOCON files).
*
* [[DefaultConfigCompanion]] is equivalent to [[com.avsystem.commons.serialization.HasGenCodec HasGenCodec]]
* except that it automatically imports codecs from [[HoconGenCodecs]] - codecs for third party types often used
* in configuration.
*/
abstract class DefaultConfigCompanion[T](implicit macroCodec: MacroInstances[HoconGenCodecs, ConfigObjectCodec[T]])
extends AbstractConfigCompanion[HoconGenCodecs, T](DefaultHoconGenCodecs)
13 changes: 13 additions & 0 deletions hocon/src/main/scala/com/avsystem/commons/hocon/SizeInBytes.scala
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
package com.avsystem.commons
package hocon

/**
* Use this type in data deserialized from HOCON to in order to read size in bytes represented with
* [[https://github.com/lightbend/config/blob/master/HOCON.md#size-in-bytes-format HOCON's nice representation]].
ddworak marked this conversation as resolved.
Show resolved Hide resolved
*/
final case class SizeInBytes(bytes: Long)
object SizeInBytes {
final val Zero = SizeInBytes(0)
final val `1KiB` = SizeInBytes(1024L)
final val `1MiB` = SizeInBytes(1024 * 1024L)
}
Original file line number Diff line number Diff line change
@@ -1,13 +1,30 @@
package com.avsystem.commons
package hocon

import java.time.{Duration, Period}

import com.avsystem.commons.serialization.json.JsonStringOutput
import com.avsystem.commons.serialization.{GenCodecRoundtripTest, Input, Output}
import com.typesafe.config.{ConfigFactory, ConfigMemorySize, ConfigValue, ConfigValueFactory, ConfigValueType}
import com.typesafe.config.*

import java.time.{Duration, Period}
import scala.concurrent.duration.*

object HoconInputTest {
case class CustomCodecsClass(
duration: FiniteDuration,
jDuration: Duration,
fileSize: SizeInBytes,
embeddedConfig: Config,
period: Period,
clazz: Class[?],
clazzMap: Map[Class[?], String],
)
object CustomCodecsClass extends DefaultConfigCompanion[CustomCodecsClass]
}

class HoconInputTest extends GenCodecRoundtripTest {
ddworak marked this conversation as resolved.
Show resolved Hide resolved

import HoconInputTest.*

type Raw = ConfigValue

def writeToOutput(write: Output => Unit): ConfigValue = {
Expand Down Expand Up @@ -56,4 +73,32 @@ class HoconInputTest extends GenCodecRoundtripTest {
test("number reading") {
assert(rawInput(42.0).readNumber().doubleValue == 42.0)
}

test("class reading") {
val config = ConfigFactory.parseString(
"""{
| duration = 1m
| jDuration = 5m
| fileSize = 1KiB
| embeddedConfig {
| something = "abc"
| }
| period = "7d"
| clazz = "com.avsystem.commons.hocon.HoconInputTest"
| clazzMap {
| "com.avsystem.commons.hocon.HoconInputTest" = "abc"
| }
|}""".stripMargin
)
val expected = CustomCodecsClass(
duration = 1.minute,
jDuration = Duration.ofMinutes(5),
fileSize = SizeInBytes.`1KiB`,
embeddedConfig = ConfigFactory.parseMap(JMap("something" -> "abc")),
period = Period.ofDays(7),
clazz = classOf[HoconInputTest],
clazzMap = Map(classOf[HoconInputTest] -> "abc"),
)
assert(CustomCodecsClass.read(config) == expected)
}
}
Loading