Есть ли какая-нибудь функция udf для add_months в scala

Question

Можете ли вы помочь, если в scala есть какой-либо udf, эквивалентный add_months в sql.

 add_months(startDate: date, numMonths: Int)

Answer 1

Какой тип даты?Если вы используете spark и DataFrame, уже есть функция, вам не нужно определять свой собственный udf:

def add_months(startDate: org.apache.spark.sql.Column,numMonths: Int): org.apache.spark.sql.Column

, который можно использовать следующим образом:

df.withColumn(add_months($"field_with_date"), number)

Еслиэто стандартный Date класс, который вы можете использовать DateUtils следующим образом:

import org.apache.commons.lang.time.DateUtils
DateUtils.addMonths(startDate, 1);