hive 计算两个时间中工作日天数udf

以下是一个简单的UDF示例，可以计算两个时间中间的工作日天数：

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

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

public class WorkdayBetween extends UDF {

    // 计算两个时间中间的工作日天数
    public int evaluate(String startDateStr, String endDateStr) {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
        try {
            Calendar startCal = Calendar.getInstance();
            startCal.setTime(sdf.parse(startDateStr));
            Calendar endCal = Calendar.getInstance();
            endCal.setTime(sdf.parse(endDateStr));
            int count = 0;
            while (startCal.getTimeInMillis() < endCal.getTimeInMillis()) {
                int dayOfWeek = startCal.get(Calendar.DAY_OF_WEEK);
                if (dayOfWeek != Calendar.SATURDAY && dayOfWeek != Calendar.SUNDAY) {
                    count++;
                }
                startCal.add(Calendar.DAY_OF_MONTH, 1);
            }
            return count;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return 0;
    }
}

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

add jar /path/to/your/jar/workday-between.jar;
create temporary function workday_between as 'com.example.WorkdayBetween';

-- 计算2021-08-01到2021-08-05之间的工作日天数
select workday_between('2021-08-01', '2021-08-05');