Fixing implementation and tests

Author: deusaquilus

controller-core/src/commonMain/kotlin/io/exoquery/controller/DecoderAny.kt

@@ -1,5 +1,6 @@
 package io.exoquery.controller
 
+import kotlin.jvm.JvmInline
 import kotlin.reflect.KClass
 
 open class DecoderAny<T: Any, Session, Row> @PublishedApi internal constructor(
@@ -45,11 +46,14 @@ open class DecoderAny<T: Any, Session, Row> @PublishedApi internal constructor(
    * Transforms this decoder into another decoder by applying the given function to the decoded value.
    * Alias for [map].
    */
-  inline fun <reified R: Any> transformInto(crossinline into: (T) -> R): DecoderAny<R, Session, Row> =
+  inline fun <reified R: Any> transformInto(crossinline into: MapContext<Session, Row>.(T) -> R): DecoderAny<R, Session, Row> =
     map(into)
 
-  inline fun <reified R: Any> map(crossinline into: (T) -> R): DecoderAny<R, Session, Row> =
-    DecoderAny<R, Session, Row>(R::class, this@DecoderAny.originalType, isNull) { ctx, index -> into(this.decode(ctx, index)) }
+  @JvmInline
+  final value class MapContext<Session, Row>(val ctx: DecodingContext<Session, Row>)
+
+  inline fun <reified R: Any> map(crossinline into: MapContext<Session, Row>.(T) -> R): DecoderAny<R, Session, Row> =
+    DecoderAny<R, Session, Row>(R::class, this@DecoderAny.originalType, isNull) { ctx, index -> into(MapContext(ctx), this.decode(ctx, index)) }
 
   override fun asNullable(): SqlDecoder<Session, Row, T?> =
     object: SqlDecoder<Session, Row, T?>() {

controller-core/src/commonMain/kotlin/io/exoquery/controller/EncoderAny.kt

@@ -1,5 +1,6 @@
 package io.exoquery.controller
 
+import kotlin.jvm.JvmInline
 import kotlin.reflect.KClass
 
 open class EncoderAny<T: Any, TypeId: Any, Session, Stmt> @PublishedApi internal constructor(
@@ -47,9 +48,12 @@ open class EncoderAny<T: Any, TypeId: Any, Session, Stmt> @PublishedApi internal
    * Transforms this encoder into another encoder by applying the given function to the value before encoding it.
    * Alias for [contramap].
    */
-  inline fun <reified R: Any> transformFrom(crossinline from: (R) -> T): EncoderAny<R, TypeId, Session, Stmt> =
+  inline fun <reified R: Any> transformFrom(crossinline from: ContrmapContext<Session, Stmt>.(R) -> T): EncoderAny<R, TypeId, Session, Stmt> =
     contramap(from)
 
-  inline fun <reified R: Any> contramap(crossinline from: (R) -> T): EncoderAny<R, TypeId, Session, Stmt> =
-    EncoderAny<R, TypeId, Session, Stmt>(this@EncoderAny.dataType, R::class, this@EncoderAny.originalType, this@EncoderAny.setNull) { ctx, value, i -> this.f(ctx, from(value), i) }
+  @JvmInline
+  final value class ContrmapContext<Session, Stmt>(val ctx: EncodingContext<Session, Stmt>)
+
+  inline fun <reified R: Any> contramap(crossinline from: ContrmapContext<Session, Stmt>.(R) -> T): EncoderAny<R, TypeId, Session, Stmt> =
+    EncoderAny<R, TypeId, Session, Stmt>(this@EncoderAny.dataType, R::class, this@EncoderAny.originalType, this@EncoderAny.setNull) { ctx, value, i -> this.f(ctx, from(ContrmapContext(ctx), value), i) }
 }

controller-jdbc/src/main/kotlin/io/exoquery/controller/jdbc/JdbcControllers.kt

@@ -5,6 +5,7 @@ import io.exoquery.controller.JavaTimeEncoding
 import io.exoquery.controller.JavaUuidEncoding
 import io.exoquery.controller.*
 import kotlinx.coroutines.flow.Flow
+import kotlinx.datetime.toJavaZoneId
 import java.sql.Connection
 import java.sql.PreparedStatement
 import java.sql.ResultSet
@@ -197,9 +198,22 @@ object JdbcControllers {
     override val encodingApi: JdbcSqlEncoding =
       object : JavaSqlEncoding<Connection, PreparedStatement, ResultSet>,
         BasicEncoding<Connection, PreparedStatement, ResultSet> by JdbcBasicEncoding,
-        JavaTimeEncoding<Connection, PreparedStatement, ResultSet> by JdbcTimeEncoding(),
+        JavaTimeEncoding<Connection, PreparedStatement, ResultSet> by SqlServerTimeEncoding,
         JavaUuidEncoding<Connection, PreparedStatement, ResultSet> by JdbcUuidStringEncoding {}
 
+    object SqlServerTimeEncoding: JdbcTimeEncoding() {
+      // Override java.util.Date encoding to use TIMESTAMP_WITH_TIMEZONE for DATETIMEOFFSET columns
+      override val JDateEncoder: JdbcEncoderAny<java.util.Date> = JdbcEncoderAny(Types.TIMESTAMP_WITH_TIMEZONE, java.util.Date::class) { ctx, v, i ->
+        val instant = v.toInstant()
+        val offsetDateTime = java.time.OffsetDateTime.ofInstant(instant, ctx.timeZone.toJavaZoneId())
+        ctx.stmt.setObject(i, offsetDateTime, Types.TIMESTAMP_WITH_TIMEZONE)
+      }
+      override val JDateDecoder: JdbcDecoderAny<java.util.Date> = JdbcDecoderAny(java.util.Date::class) { ctx, i ->
+        val offsetDateTime = ctx.row.getObject(i, java.time.OffsetDateTime::class.java)
+        java.util.Date.from(offsetDateTime.toInstant())
+      }
+    }
+
     override suspend fun <T> runActionReturningScoped(act: ControllerActionReturning<T>, options: JdbcExecutionOptions): Flow<T> =
       flowWithConnection(options) {
         val conn = localConnection()

controller-r2dbc/src/main/kotlin/io/exoquery/controller/r2dbc/R2dbcDecoders.kt

@@ -6,6 +6,8 @@ import io.r2dbc.spi.Connection
 import io.r2dbc.spi.Row
 import kotlin.reflect.KClass
 
+typealias R2dbcDecoder<T> = DecoderAny<T, Connection, Row>
+
 class R2dbcDecoderAny<T: Any>(
   override val type: KClass<T>,
   override val f: (R2dbcDecodingContext, Int) -> T?
@@ -21,7 +23,7 @@ class R2dbcDecoderAny<T: Any>(
 
 object R2dbcDecoders {
   @Suppress("UNCHECKED_CAST")
-  val decoders: Set<SqlDecoder<Connection, Row, out Any>> = setOf(
+  val decoders: Set<R2dbcDecoder<out Any>> = setOf(
     R2dbcBasicEncoding.BooleanDecoder,
     R2dbcBasicEncoding.ByteDecoder,
     R2dbcBasicEncoding.CharDecoder,