Skip to content

Commit f7fb98c

Browse files
dkhalanskyjbdkhalanskyjb
committed
Simplify the code for Instant arithmetics
1 parent da397d0 commit f7fb98c

File tree

7 files changed

+59
-76
lines changed

7 files changed

+59
-76
lines changed

‎core/commonJs/src/internal/Platform.kt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -103,14 +103,14 @@ private object SystemTimeZone: TimeZone() {
103103
UtcOffset(minutes = -Date(instant.toEpochMilliseconds().toDouble()).getTimezoneOffset().toInt())
104104

105105
/* https://github.com/js-joda/js-joda/blob/8c1a7448db92ca014417346049fb64b55f7b1ac1/packages/core/src/zone/SystemDefaultZoneRules.js#L49-L55 */
106-
override fun atZone(dateTime: LocalDateTime, preferred: UtcOffset?): ZonedDateTime {
106+
override fun atZone(dateTime: LocalDateTime, preferred: UtcOffset?): LocalDateTimeWithOffset {
107107
val epochMilli = dateTime.toInstant(UTC).toEpochMilliseconds()
108108
val offsetInMinutesBeforePossibleTransition = Date(epochMilli.toDouble()).getTimezoneOffset().toInt()
109109
val epochMilliSystemZone = epochMilli +
110110
offsetInMinutesBeforePossibleTransition * SECONDS_PER_MINUTE * MILLIS_PER_ONE
111111
val offsetInMinutesAfterPossibleTransition = Date(epochMilliSystemZone.toDouble()).getTimezoneOffset().toInt()
112112
val offset = UtcOffset(minutes = -offsetInMinutesAfterPossibleTransition)
113-
return ZonedDateTime(dateTime, offset)
113+
return LocalDateTimeWithOffset(dateTime, offset)
114114
}
115115

116116
override fun equals(other: Any?): Boolean = other === this

‎core/commonKotlin/src/Instant.kt

Lines changed: 35 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -164,33 +164,39 @@ public actual class Instant internal constructor(public actual val epochSeconds:
164164

165165
}
166166

167-
private fun Instant.toZonedDateTimeFailing(zone: TimeZone): ZonedDateTime = try {
168-
toZonedDateTime(zone)
167+
private fun Instant.toLocalDateTimeFailing(offset: UtcOffset): LocalDateTime = try {
168+
toLocalDateTimeImpl(offset)
169169
} catch (e: IllegalArgumentException) {
170170
throw DateTimeArithmeticException("Can not convert instant $this to LocalDateTime to perform computations", e)
171171
}
172172

173-
/**
174-
* @throws IllegalArgumentException if the [Instant] exceeds the boundaries of [LocalDateTime]
175-
*/
176-
private fun Instant.toZonedDateTime(zone: TimeZone): ZonedDateTime {
177-
val currentOffset = zone.offsetAt(this)
178-
return ZonedDateTime(toLocalDateTimeImpl(currentOffset), currentOffset)
179-
}
180-
181-
/** Check that [Instant] fits in [ZonedDateTime].
173+
/** Check that [Instant] fits in [LocalDateTime].
182174
* This is done on the results of computations for consistency with other platforms.
183175
*/
184176
private fun Instant.check(zone: TimeZone): Instant = this@check.also {
185-
toZonedDateTimeFailing(zone)
177+
toLocalDateTimeFailing(offsetIn(zone))
186178
}
187179

188180
public actual fun Instant.plus(period: DateTimePeriod, timeZone: TimeZone): Instant = try {
189181
with(period) {
190-
val withDate = toZonedDateTimeFailing(timeZone)
191-
.run { if (totalMonths != 0L) timeZone.atZone(dateTime.plus(totalMonths, DateTimeUnit.MONTH), offset) else this }
192-
.run { if (days != 0) timeZone.atZone(dateTime.plus(days, DateTimeUnit.DAY), offset) else this }
193-
withDate.toInstant()
182+
val initialOffset = offsetIn(timeZone)
183+
val initialLdt = toLocalDateTimeFailing(initialOffset)
184+
val offsetAfterMonths: UtcOffset
185+
val ldtAfterMonths: LocalDateTime
186+
if (totalMonths != 0L) {
187+
val (ldt, offset) = timeZone.atZone(initialLdt.plus(totalMonths, DateTimeUnit.MONTH), initialOffset)
188+
offsetAfterMonths = offset
189+
ldtAfterMonths = ldt
190+
} else {
191+
offsetAfterMonths = initialOffset
192+
ldtAfterMonths = initialLdt
193+
}
194+
val instantAfterMonthsAndDays = if (days != 0) {
195+
timeZone.atZone(ldtAfterMonths.plus(days, DateTimeUnit.DAY), offsetAfterMonths).toInstant()
196+
} else {
197+
ldtAfterMonths.toInstant(offsetAfterMonths)
198+
}
199+
instantAfterMonthsAndDays
194200
.run { if (totalNanoseconds != 0L) plus(0, totalNanoseconds).check(timeZone) else this }
195201
}.check(timeZone)
196202
} catch (e: ArithmeticException) {
@@ -209,11 +215,9 @@ public actual fun Instant.minus(value: Int, unit: DateTimeUnit, timeZone: TimeZo
209215
public actual fun Instant.plus(value: Long, unit: DateTimeUnit, timeZone: TimeZone): Instant = try {
210216
when (unit) {
211217
is DateTimeUnit.DateBased -> {
212-
val toZonedDateTimeFailing = toZonedDateTimeFailing(timeZone)
213-
timeZone.atZone(
214-
toZonedDateTimeFailing.dateTime.plus(value, unit),
215-
toZonedDateTimeFailing.offset
216-
).toInstant()
218+
val preferredOffset = offsetIn(timeZone)
219+
val initialLdt = toLocalDateTimeFailing(preferredOffset)
220+
timeZone.atZone(initialLdt.plus(value, unit), preferredOffset).toInstant()
217221
}
218222
is DateTimeUnit.TimeBased ->
219223
check(timeZone).plus(value, unit).check(timeZone)
@@ -236,30 +240,23 @@ public actual fun Instant.plus(value: Long, unit: DateTimeUnit.TimeBased): Insta
236240
}
237241

238242
public actual fun Instant.periodUntil(other: Instant, timeZone: TimeZone): DateTimePeriod {
239-
var thisLdt = toZonedDateTimeFailing(timeZone)
240-
val otherLdt = other.toZonedDateTimeFailing(timeZone)
241-
242-
val months = thisLdt.dateTime.until(otherLdt.dateTime, DateTimeUnit.MONTH) // `until` on dates never fails
243-
thisLdt = timeZone.atZone(
244-
thisLdt.dateTime.plus(months, DateTimeUnit.MONTH),
245-
thisLdt.offset
246-
) // won't throw: thisLdt + months <= otherLdt, which is known to be valid
247-
val days =
248-
thisLdt.dateTime.until(otherLdt.dateTime, DateTimeUnit.DAY) // `until` on dates never fails
249-
thisLdt = timeZone.atZone(
250-
thisLdt.dateTime.plus(days, DateTimeUnit.DAY),
251-
thisLdt.offset
252-
) // won't throw: thisLdt + days <= otherLdt
253-
val nanoseconds =
254-
thisLdt.toInstant().until(otherLdt.toInstant(), DateTimeUnit.NANOSECOND) // |otherLdt - thisLdt| < 24h
243+
val thisOffset1 = offsetIn(timeZone)
244+
val thisLdt1 = toLocalDateTimeFailing(thisOffset1)
245+
val otherLdt = other.toLocalDateTimeFailing(other.offsetIn(timeZone))
246+
247+
val months = thisLdt1.until(otherLdt, DateTimeUnit.MONTH) // `until` on dates never fails
248+
val (thisLdt2, thisOffset2) = timeZone.atZone(thisLdt1.plus(months, DateTimeUnit.MONTH), thisOffset1) // won't throw: thisLdt + months <= otherLdt, which is known to be valid
249+
val days = thisLdt2.until(otherLdt, DateTimeUnit.DAY) // `until` on dates never fails
250+
val (thisLdt3, thisOffset3) = timeZone.atZone(thisLdt2.plus(days, DateTimeUnit.DAY), thisOffset2) // won't throw: thisLdt + days <= otherLdt
251+
val nanoseconds = thisLdt3.toInstant(thisOffset3).until(other, DateTimeUnit.NANOSECOND) // |otherLdt - thisLdt| < 24h
255252

256253
return buildDateTimePeriod(months, days.toInt(), nanoseconds)
257254
}
258255

259256
public actual fun Instant.until(other: Instant, unit: DateTimeUnit, timeZone: TimeZone): Long =
260257
when (unit) {
261258
is DateTimeUnit.DateBased ->
262-
toZonedDateTimeFailing(timeZone).dateTime.until(other.toZonedDateTimeFailing(timeZone).dateTime, unit)
259+
toLocalDateTimeFailing(offsetIn(timeZone)).until(other.toLocalDateTimeFailing(other.offsetIn(timeZone)), unit)
263260
.toLong()
264261
is DateTimeUnit.TimeBased -> {
265262
check(timeZone); other.check(timeZone)

‎core/commonKotlin/src/LocalDateTimeWithOffset.kt

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
/*
2+
* Copyright 2019-2020 JetBrains s.r.o.
3+
* Use of this source code is governed by the Apache 2.0 License that can be found in the LICENSE.txt file.
4+
*/
5+
/* Based on the ThreeTenBp project.
6+
* Copyright (c) 2007-present, Stephen Colebourne & Michael Nascimento Santos
7+
*/
8+
9+
package kotlinx.datetime
10+
11+
internal data class LocalDateTimeWithOffset(val dateTime: LocalDateTime, val offset: UtcOffset)
12+
13+
internal fun LocalDateTimeWithOffset.toInstant(): Instant =
14+
Instant(dateTime.toEpochSecond(offset), dateTime.nanosecond)

‎core/commonKotlin/src/TimeZone.kt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ public actual open class TimeZone internal constructor() {
9494
internal open fun localDateTimeToInstant(dateTime: LocalDateTime): Instant =
9595
atZone(dateTime).toInstant()
9696

97-
internal open fun atZone(dateTime: LocalDateTime, preferred: UtcOffset? = null): ZonedDateTime =
97+
internal open fun atZone(dateTime: LocalDateTime, preferred: UtcOffset? = null): LocalDateTimeWithOffset =
9898
error("Should be overridden")
9999

100100
actual override fun equals(other: Any?): Boolean =
@@ -118,8 +118,8 @@ public actual class FixedOffsetTimeZone internal constructor(public actual val o
118118

119119
override fun offsetAtImpl(instant: Instant): UtcOffset = offset
120120

121-
override fun atZone(dateTime: LocalDateTime, preferred: UtcOffset?): ZonedDateTime =
122-
ZonedDateTime(dateTime, offset)
121+
override fun atZone(dateTime: LocalDateTime, preferred: UtcOffset?): LocalDateTimeWithOffset =
122+
LocalDateTimeWithOffset(dateTime, offset)
123123

124124
override fun instantToLocalDateTime(instant: Instant): LocalDateTime = instant.toLocalDateTime(offset)
125125
override fun localDateTimeToInstant(dateTime: LocalDateTime): Instant = dateTime.toInstant(offset)

‎core/commonKotlin/src/ZonedDateTime.kt

Lines changed: 0 additions & 28 deletions
This file was deleted.

‎core/commonKotlin/src/internal/RegionTimeZone.kt

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,12 @@ internal class RegionTimeZone(private val tzid: TimeZoneRules, override val id:
1818
}
1919
}
2020

21-
override fun atZone(dateTime: LocalDateTime, preferred: UtcOffset?): ZonedDateTime =
21+
override fun atZone(dateTime: LocalDateTime, preferred: UtcOffset?): LocalDateTimeWithOffset =
2222
when (val info = tzid.infoAtDatetime(dateTime)) {
23-
is OffsetInfo.Regular -> ZonedDateTime(dateTime, info.offset)
23+
is OffsetInfo.Regular -> LocalDateTimeWithOffset(dateTime, info.offset)
2424
is OffsetInfo.Gap -> {
2525
try {
26-
ZonedDateTime(dateTime.plusSeconds(info.transitionDurationSeconds), info.offsetAfter)
26+
LocalDateTimeWithOffset(dateTime.plusSeconds(info.transitionDurationSeconds), info.offsetAfter)
2727
} catch (e: IllegalArgumentException) {
2828
throw DateTimeArithmeticException(
2929
"Overflow whet correcting the date-time to not be in the transition gap",
@@ -32,7 +32,7 @@ internal class RegionTimeZone(private val tzid: TimeZoneRules, override val id:
3232
}
3333
}
3434

35-
is OffsetInfo.Overlap -> ZonedDateTime(
35+
is OffsetInfo.Overlap -> LocalDateTimeWithOffset(
3636
dateTime,
3737
if (info.offsetAfter == preferred) info.offsetAfter else info.offsetBefore
3838
)

‎core/commonKotlin/test/ThreeTenBpTimeZoneTest.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class ThreeTenBpTimeZoneTest {
4040
fun overlappingLocalTime() {
4141
val t = LocalDateTime(2007, 10, 28, 2, 30, 0, 0)
4242
val zone = TimeZone.of("Europe/Paris")
43-
assertEquals(ZonedDateTime(
43+
assertEquals(LocalDateTimeWithOffset(
4444
LocalDateTime(2007, 10, 28, 2, 30, 0, 0),
4545
UtcOffset(seconds = 2 * 3600)
4646
), zone.atZone(t))

0 commit comments

Comments
 (0)