Skip to content

Commit 1c3b3c7

Browse files
committed
simplified some date conversions for arrow
1 parent d7eed68 commit 1c3b3c7

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

dataframe-arrow/src/main/kotlin/org/jetbrains/kotlinx/dataframe/io/ArrowWriterImpl.kt

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ internal class ArrowWriterImpl(
229229
is DateDayVector ->
230230
column.convertToLocalDate()
231231
.forEachIndexed { i, value ->
232-
value?.also { vector.set(i, value.toJavaLocalDate().toEpochDay().toInt()) }
232+
value?.also { vector.set(i, value.toEpochDays()) }
233233
?: vector.setNull(i)
234234
}
235235

@@ -243,29 +243,29 @@ internal class ArrowWriterImpl(
243243
is TimeNanoVector ->
244244
column.convertToLocalTime()
245245
.forEachIndexed { i, value ->
246-
value?.also { vector.set(i, value.toJavaLocalTime().toNanoOfDay()) }
246+
value?.also { vector.set(i, value.toNanosecondOfDay()) }
247247
?: vector.setNull(i)
248248
}
249249

250250
is TimeMicroVector ->
251251
column.convertToLocalTime()
252252
.forEachIndexed { i, value ->
253-
value?.also { vector.set(i, value.toJavaLocalTime().toNanoOfDay() / 1000) }
253+
value?.also { vector.set(i, value.toNanosecondOfDay() / 1000) }
254254
?: vector.setNull(i)
255255
}
256256

257257
is TimeMilliVector ->
258258
column.convertToLocalTime()
259259
.forEachIndexed { i, value ->
260-
value?.also { vector.set(i, (value.toJavaLocalTime().toNanoOfDay() / 1000 / 1000).toInt()) }
260+
value?.also { vector.set(i, (value.toNanosecondOfDay() / 1000 / 1000).toInt()) }
261261
?: vector.setNull(i)
262262
}
263263

264264
is TimeSecVector ->
265265
column.convertToLocalTime()
266266
.forEachIndexed { i, value ->
267267
value?.also {
268-
vector.set(i, (value.toJavaLocalTime().toNanoOfDay() / 1000 / 1000 / 1000).toInt())
268+
vector.set(i, (value.toNanosecondOfDay() / 1000 / 1000 / 1000).toInt())
269269
} ?: vector.setNull(i)
270270
}
271271

0 commit comments

Comments
 (0)