Question

我一直试图在2个特定日期之间获得周末天数（周六和周日）。

我遇到了这个解决方案：http://crossedlogic.blogspot.ca/2008/09/using-sql-to-find-work-days-in-date.html

当我调用该函数时，它会在输出中返回OK作为消息，我希望它返回周末天数。

我做错了什么？

代码：

DROP FUNCTION IF EXISTS `fn_GET_WEEKEND_DAYS`;

DELIMITER $$
CREATE FUNCTION `fn_GET_WEEKEND_DAYS`(StartDate DATE, EndDate DATE) RETURNS INT
BEGIN
    # declare the variables.
    DECLARE varDays INT;

    # create the temprorary table to insert the data in.
    CREATE TEMPORARY TABLE temp(calendarDate DATE, isWeekend TINYINT(1));

    # insert the starting date.
    INSERT INTO temp VALUES(StartDate, NULL);

    # insert each day by increment of 1 day untill reached the end date.
    WHILE (SELECT MAX(CalendarDate) FROM temp) < EndDate DO
        INSERT INTO temp
        SELECT ADDDATE(MAX(CalendarDate), INTERVAL 1 DAY), NULL
        FROM temp;
    END WHILE;

    # update the is weekend field depending if the day of week of each row is 1 or 7. (saturday or sunday)
    UPDATE temp SET isWeekend = CASE WHEN DAYOFWEEK(calendarDate) IN (1, 7) THEN true ELSE false END;

    # count the date that are weekends.
    SELECT COUNT(calendarDate) INTO varDays FROM temp WHERE isWeekend = true;

    # drop the temp table.
    DROP TEMPORARY TABLE IF EXISTS temp;

    RETURN varDays;
END $$

TEST：

SELECT fn_GET_WEEKEND_DAYS(CURDATE(), ADDDATE(CURDATE(), INTERVAL 10 DAY)) AS TEST;

非常感谢任何帮助和建议。

Answer 1

快速修复您当前的解决方案： -

DROP FUNCTION IF EXISTS `fn_GET_WEEKEND_DAYS`;

DELIMITER $$
CREATE FUNCTION `fn_GET_WEEKEND_DAYS`(StartDate DATE, EndDate DATE) RETURNS INT
BEGIN
    # declare the variables.
    DECLARE varDays INT;
    DECLARE varDate DATE;

    IF (StartDate > EndDate) THEN
        SET varDate = StartDate;
        SET StartDate = EndDate;
        SET EndDate = varDate;
    END IF;

    # create the temprorary table to insert the data in.
    CREATE TEMPORARY TABLE temp1(calendarDate DATE, isWeekend TINYINT(1));

    # insert the starting date.
    INSERT INTO temp1 VALUES(StartDate, NULL);

    # insert each day by increment of 1 day untill reached the end date.
    SET varDate = StartDate;
    WHILE varDate < EndDate DO
        INSERT INTO temp1 VALUES(ADDDATE(varDate, INTERVAL 1 DAY), NULL);
        SELECT MAX(CalendarDate) INTO varDate FROM temp1;
    END WHILE;

    # update the is weekend field depending if the day of week of each row is 0 or 1. (saturday or sunday)
    UPDATE temp1 SET isWeekend = CASE WHEN DAYOFWEEK(calendarDate) = 7 OR DAYOFWEEK(calendarDate) = 1 THEN true ELSE false END;

    # count the date that are weekends.
    SELECT COUNT(calendarDate) INTO varDays FROM temp1 WHERE isWeekend = true;

    # drop the temp1 table.
    DROP TEMPORARY TABLE IF EXISTS temp1;

    RETURN varDays;
END $$

最大的问题是它正在使用临时表，并尝试根据临时表中的选择插入到临时表中，这是无法完成的。它会给出错误您不能在同一查询中多次引用TEMPORARY表。例如，以下内容不起作用：

如果您的日期范围有限，那么在单个SQL语句中可能很容易做到

编辑

使用单个SQL语句执行此操作的方法，复制的最大总日期范围为9999天： -

SELECT SUM(IF(DAYOFWEEK(DATE_ADD('2014-05-02', INTERVAL units.aCnt + tens.aCnt * 10 + hundreds.aCnt * 100 + thousands.aCnt * 1000 DAY) ) IN (1,7), 1, 0))
FROM (SELECT 1 AS aCnt UNION SELECT 2 UNION SELECT 3 UNION SELECT 4 UNION SELECT 5 UNION SELECT 6 UNION SELECT 7 UNION SELECT 8 UNION SELECT 9 UNION SELECT 0) units
CROSS JOIN (SELECT 1 AS aCnt UNION SELECT 2 UNION SELECT 3 UNION SELECT 4 UNION SELECT 5 UNION SELECT 6 UNION SELECT 7 UNION SELECT 8 UNION SELECT 9 UNION SELECT 0) tens
CROSS JOIN (SELECT 1 AS aCnt UNION SELECT 2 UNION SELECT 3 UNION SELECT 4 UNION SELECT 5 UNION SELECT 6 UNION SELECT 7 UNION SELECT 8 UNION SELECT 9 UNION SELECT 0) hundreds
CROSS JOIN (SELECT 1 AS aCnt UNION SELECT 2 UNION SELECT 3 UNION SELECT 4 UNION SELECT 5 UNION SELECT 6 UNION SELECT 7 UNION SELECT 8 UNION SELECT 9 UNION SELECT 0) thousands
WHERE '2017-03-01' >= DATE_ADD('2014-05-02', INTERVAL units.aCnt + tens.aCnt * 10 + hundreds.aCnt * 100 + thousands.aCnt * 1000 DAY)

MySQL函数返回周末的日子

1 个答案: