我想迭代给定Joda间隔内的所有日期:
val interval = Interval(DateTime.now().minusDays(42), DateTime.now())
如何在Kotlin做到这一点?
以下扩展函数给出了来自给定Sequence
的LocalDate
的Interval
对象,可用于迭代这些日期。
fun Interval.toLocalDates(): Sequence<LocalDate> = generateSequence(start) { d ->
d.plusDays(1).takeIf { it < end }
}.map(DateTime::toLocalDate)
用法:
val interval = Interval(DateTime.now().minusDays(42), DateTime.now())
interval.toLocalDates().forEach {
println(it)
}
在这个解决方案中,最后一天,DateTime.now()
不包含在Sequence
中,因为这也是Interval
的实现方式:
“时间间隔表示两个时刻之间的一段时间。间隔包括开始时刻和结束时的排除。”
如果出于任何原因,你想让它包括最后一天,只需将takeIf
条件更改为it <= end
。
我想如果你不止一次需要它,最好重载rangeTo运算符以允许这种语法
for (i in LocalDate.now() .. LocalDate.now().plusWeeks(1)) {
System.out.print(i) // 2018-08-30 2018-08-31 2018-09-01
}
以下是运营商扩展的代码:
operator fun LocalDate.rangeTo(other: LocalDate): LocalDateRange {
return LocalDateRange(this, other)
}
必要的课程:
class LocalDateRange(override val start: LocalDate, override val endInclusive: LocalDate)
: ClosedRange<LocalDate>, Iterable<LocalDate> {
override fun iterator(): Iterator<LocalDate> {
return DateIterator(start, endInclusive)
}
}
class DateIterator(start: LocalDate, private val endInclusive: LocalDate)
: Iterator<LocalDate> {
private var current = start
override fun hasNext(): Boolean {
return current <= endInclusive
}
override fun next(): LocalDate {
current = current.plusDays(1)
return current
}
}
您当前的解决方案深受启发:
fun Interval.toDateTimes() = generateSequence(start) { it.plusDays(1) }
.takeWhile(::contains)
用法:
interval.toDateTimes()
.forEach { println(it) }
如果您需要LocalDate
,您仍然可以执行以下操作:
interval.toDateTimes()
.map(DateTime::toLocalDate)
.forEach { println(it) }
或再次作为Interval
的扩展函数:
fun Interval.toLocalDates() = toDateTimes().map(DateTime::toLocalDate)
如果您希望结束日期是包容性的,请改用takeWhile { it <= end }
。
LocalDate
现在是首选,所以我们可以简单地用day作为数字迭代:
for (day in minDate.toEpochDay()..maxDate.toEpochDay()) {
// ...
}
要么:
(minDate.toEpochDay()..maxDate.toEpochDay()).forEach {
// ...
}
以日期作为日期迭代:
generateSequence(minDate) { it.plusDays(1) }.takeWhile { it < maxDate }.forEach {
// it ...
}
要么:
var day = minDate;
while (day < maxDate) {
day = day.plusDays(1);
// ...
}