hive 计算两个工作日时间函数

Hive中没有自带的计算两个工作日时间的函数，但可以自己编写UDF来实现。

以下是一个简单的UDF示例，可以计算指定日期之后的第N个工作日：

import org.apache.hadoop.hive.ql.exec.UDF;

import java.text.SimpleDateFormat;
import java.util.Calendar;

public class NextWorkday extends UDF {

    // 计算指定日期之后的第N个工作日
    public String evaluate(String dateStr, int n) {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
        try {
            Calendar cal = Calendar.getInstance();
            cal.setTime(sdf.parse(dateStr));
            int count = 0;
            while (count < n) {
                cal.add(Calendar.DAY_OF_MONTH, 1);
                int dayOfWeek = cal.get(Calendar.DAY_OF_WEEK);
                if (dayOfWeek != Calendar.SATURDAY && dayOfWeek != Calendar.SUNDAY) {
                    count++;
                }
            }
            return sdf.format(cal.getTime());
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
}

编译打包后，在Hive中注册UDF并使用：

add jar /path/to/your/jar/next-workday.jar;
create temporary function next_workday as 'com.example.NextWorkday';

-- 计算2021-08-01之后的第3个工作日
select next_workday('2021-08-01', 3);