local time support

Tinkoff · Jul 31, 2023 · 5a8b79a · 5a8b79a
1 parent 7d40104
commit 5a8b79a
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/...in/java/ru/tinkoff/kora/database/annotation/processor/cassandra/CassandraNativeTypes.java b/...in/java/ru/tinkoff/kora/database/annotation/processor/cassandra/CassandraNativeTypes.java
@@ -10,6 +10,7 @@
 import java.time.Instant;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
+import java.time.LocalTime;
 import java.util.List;
 import java.util.Objects;
 
@@ -71,6 +72,11 @@ public class CassandraNativeTypes {
             (rsName, i) -> CodeBlock.of("$L.getInstant($L)", rsName, i),
             (stmt, var, i) -> CodeBlock.of("$L.setInstant($L, $L)", stmt, i, var)
         );
+        var localTime = CassandraNativeType.of(
+            TypeName.get(LocalTime.class),
+            (rsName, i) -> CodeBlock.of("$L.getLocalTime($L)", rsName, i),
+            (stmt, var, i) -> CodeBlock.of("$L.setLocalTime($L, $L)", stmt, i, var)
+        );
 
         nativeTypes = List.of(
             booleanPrimitive,
@@ -86,7 +92,8 @@ public class CassandraNativeTypes {
             byteBuffer,
             localDateTime,
             localDate,
-            instant
+            instant,
+            localTime
         );
     }
 

diff --git a/.../database-cassandra/src/main/java/ru/tinkoff/kora/database/cassandra/CassandraModule.java b/.../database-cassandra/src/main/java/ru/tinkoff/kora/database/cassandra/CassandraModule.java
@@ -100,7 +100,7 @@ default CassandraRowMapper<ZonedDateTime> zonedDateTimeCassandraRowMapper() {
     default CassandraRowMapper<Instant> instantCassandraRowMapper() {
         return row -> row.isNull(0)
             ? null
-            : row.get(0, Instant.class);
+            : row.getInstant(0);
     }
 
     default CassandraReactiveResultSetMapper<Void, Mono<Void>> voidMonoCassandraReactiveResultSetMapper() {

diff --git a/...c/main/kotlin/ru/tinkoff/kora/database/symbol/processor/cassandra/CassandraNativeTypes.kt b/...c/main/kotlin/ru/tinkoff/kora/database/symbol/processor/cassandra/CassandraNativeTypes.kt
@@ -8,6 +8,7 @@ import java.nio.ByteBuffer
 import java.time.Instant
 import java.time.LocalDate
 import java.time.LocalDateTime
+import java.time.LocalTime
 
 object CassandraNativeTypes {
     private val nativeTypes = listOf(
@@ -111,6 +112,16 @@ object CassandraNativeTypes {
             { rsName, i -> CodeBlock.of("%N.getInstant(%L)", rsName, i) },
             { stmt, variableName, idx -> CodeBlock.of("%N.setInstant(%L, %L)", stmt, idx, variableName) },
         ),
+        CassandraNativeType.of(
+            LocalTime::class.asTypeName(),
+            { rsName, i -> CodeBlock.of("%N.getLocalTime(%L)", rsName, i) },
+            { stmt, variableName, idx -> CodeBlock.of("%N.setLocalTime(%L, %L)", stmt, idx, variableName) },
+        ),
+        CassandraNativeType.of(
+            LocalTime::class.asTypeName().copy(true),
+            { rsName, i -> CodeBlock.of("%N.getLocalTime(%L)", rsName, i) },
+            { stmt, variableName, idx -> CodeBlock.of("%N.setLocalTime(%L, %L)", stmt, idx, variableName) },
+        ),
     )
 
     fun findNativeType(typeName: TypeName): CassandraNativeType? {