Scala中给定日期的当前和上一季度的第一个和最后一个日期

Question

我正在编写一个每月自动执行一次的代码，它将以计算机上的日期作为while循环执行的开始和结束日期。

val today = new DateTime().withZone(DateTimeZone.forID("Asia/Kolkata"))
var start =  today.toString(DateTimeFormat.forPattern("yyyy-MM-dd"))
val end = today.toString(DateTimeFormat.forPattern("yyyy-MM-dd"))
while(start<=end){
Data(start,end)
}
def Data (start: DateTime, end: DateTime) {
val start_temp = start
val end_temp = end
var start_temp_1 = start_temp.minusMonths(1).withDayOfMonth(1);
var start_date_monthly = start_temp_1.toString(DateTimeFormat.forPattern("yyyy-MM-dd"))
println("Last Month Start Date: " + start_date_monthly)
var end_temp_1 = end_temp.minusMonths(0).withDayOfMonth(1);
var end_temp_2 = end_temp_1.minusDays(1)
var end_date_monthly = end_temp_2.toString(DateTimeFormat.forPattern("yyyy-MM-dd"))
println("Last Month End Date: " + end_date_monthly)

}

变量today中的日期需要转换为的第一个和最后一个日期

1. 上月today
2. today当前和上一季度

我能够完成Data函数中显示的第一个。

假设今天的日期是2022-01-27T14:25:26.374+05:30上述函数返回

Last Month Start Date: 2021-12-01
Last Month End Date: 2021-12-31

我如何在第二个季度实现这一点——本季度和上一季度的第一个和最后一个日期？

Ex-today得到以下日期-2022-01-27T14:25:26.374+05:30我需要返回上一季度的2021-10-01和2021-12-31以及当前季度的CCD_ 9和CCD_。

某些解决方案建议使用Spark SQL&数据帧，但这不适用于这种情况。

有没有一种直接的方法可以像在月份的情况下那样做到这一点？还是udf是这里唯一的选项？

Answer 1

以下可以做到。如果必须将其应用于DF 中的每一行，则这可能是一个UDF

import java.time.temporal.IsoFields
import java.time.temporal.IsoFields.QUARTER_OF_YEAR
import java.time.{LocalDate, YearMonth}
def printQuarterBeginAndEnd(localDate: LocalDate): Unit = {
val currentQuarter = localDate.get(QUARTER_OF_YEAR)
val currentYear = localDate.getYear
val currentMonth = localDate.getMonth
val startOfQuarter = YearMonth.of(currentYear, (currentQuarter-1) * 3 + 1).`with`(QUARTER_OF_YEAR, currentQuarter).atDay(1)
val endOfQuarter = YearMonth.of(currentYear, currentQuarter * 3).`with`(QUARTER_OF_YEAR, currentQuarter).atEndOfMonth()
println(s"Start $startOfQuarter ends $endOfQuarter")
}
printQuarterBeginAndEnd(LocalDate.now())
printQuarterBeginAndEnd(LocalDate.now().minus(1, IsoFields.QUARTER_YEARS))

打印以下

import java.time.temporal.IsoFields
import java.time.temporal.IsoFields.QUARTER_OF_YEAR
import java.time.{LocalDate, YearMonth}
printQuarterBeginAndEnd: (localDate: java.time.LocalDate)Unit

Start 2022-01-01 ends 2022-03-31
Start 2021-10-01 ends 2021-12-31
scala> printQuarterBeginAndEnd(LocalDate.now().minus(1, IsoFields.QUARTER_YEARS)) Start 2021-10-01 ends 2021-12-31
scala> printQuarterBeginAndEnd(LocalDate.now()) Start 2022-01-01 ends 2022-03-31
scala> printQuarterBeginAndEnd(LocalDate.parse("2021-01-01")) Start 2021-01-01 ends 2021-03-31