如何使用行组中的值根据另一个表更新一个表？

Question

我有两个表：

CREATE TABLE Employee (
   Site       ????         ????,
   WorkTypeId char(2)  NOT NULL,
   Emp_NO     int      NOT NULL,
   "Date"     ????     NOT NULL  
);

CREATE TABLE PTO (
   Site       ????         ????,
   WorkTypeId char(2)      NULL,
   Emp_NO     int      NOT NULL,
   "Date"     ????     NOT NULL  
);

我想更新 PTO 的 WorkTypeId 列中的值：

1. Employee 中的 EMP NO （查找表）和 PTO 应匹配。
2. 应仅从该月的第一次出现中选取单个 WorkTypeId 值。

例如，给定以下示例输入数据：

TABLE Employee:

Site	WorkTypeId	Emp_NO	Date
5015	MB	1005	2022-02-01
5015	MI	1005	2022-02-04
5015	PO	1005	2022 -02-04
5015	ME	2003	2022-01-01
5015	TT	2003	2022-01-10

表 PTO：

站点	WorkTypeId	Emp_NO	日期
5015		1005	2022-02-03
5015		1005	2022-02-14
5014		2003	2022-01-09

例如：

• 给定 Employee 和 Emp_NO = 1005...
  • ...Employee 表中该 Emp_NO 有 3 行，具有 3 个不同的 WorkTypeId 值，但 日期值。
  • 因此，选择最早的 Date (2022-02-01) 的 WorkTypeId 值，即 'MB'
  • 因此，Emp_NO 得到 WorkTypeId = 'MB'。
  • 并使用该单个值填充 PTO 表中 1005 的 WorkTypeId 单元格。
  • 还可以按月进行匹配。

因此，PTO 表中的预期输出为

Site	WorkTypeId	Emp_NO	Date
5015	MB	1005	2022-02-03
5015	MB	1005	2022-02-14
5014	ME	2003	2022-01-09

Answer 1

更新 2002-03-05

将其留在这里供后代使用，但我建议阅读 Dai 的精彩文章解决这个问题的不同方法。

---

尝试交叉应用来获取第一个具有匹配月份和年份的员工记录。

注意：使用 OUTER APPLY 始终返回所有 PTO 记录，即使没有找到匹配的 WorkTypeId。

SELECT p.Site
       , e.WorkTypeId
       , p.Emp_No
       , p.[Date]
FROM  PTO p CROSS APPLY 
        (
          SELECT TOP 1 WorkTypeId
          FROM   Employee e 
          WHERE  e.Emp_No = p.Emp_No
          AND    MONTH(e.[Date]) = MONTH(p.[Date])
          AND    YEAR(e.[Date]) = YEAR(p.[Date])
          ORDER BY [Date] ASC
        )e

结果：

Site | WorkTypeId | Emp_No | Date      
---: | :--------- | -----: | :---------
5015 | MB         |   1005 | 2022-02-03
5015 | MB         |   1005 | 2022-02-14
5014 | ME         |   2003 | 2022-01-09

db<>fiddle 此处

Answer 2

从与 GROUP BY 查询中的 MIN/MAX 表达式中使用的列不同获取值在 SQL 中仍然是一件令人惊讶的困难事情，虽然 SQL 语言（和 SQL Server）的现代版本使它变得更容易，但它们对大多数人来说完全不明显且违反直觉，因为它必然涉及更高级的主题，例如CTE，派生表（又名内部查询）、自连接和窗口函数，尽管查询在概念上很简单。

无论如何，与现代 SQL 一样，通常有 3 或 4 种不同的方法来完成相同的任务，但也有一些陷阱。

前言：

• 由于Site、Date、Year和Month都是T-SQL中的关键字，我用双引号对它们进行了转义，这是符合 ISO/ANSI SQL 标准的转义保留字的方法。

  • SQL Server 默认支持此功能。如果（出于某些不正当的原因）您设置了SET QUOTED IDENTIFIER OFF，则将双引号更改为方括号：[]

• 我假设 两个表中的 Site 列只是一个普通的“ol”数据列，如下所示：

  • 它不是 PRIMARY KEY 成员列。
  • 不应将其用作GROUP BY。
  • 不应在 JOIN 谓词中使用它。

• 以下所有方法均假定此数据库状态：

CREATE TABLE "Employee" (
    "Site"     int      NOT NULL,
    WorkTypeId char(2)  NOT NULL,
    Emp_NO     int      NOT NULL,
    "Date"     date     NOT NULL  
);

CREATE TABLE "PTO" (
    "Site"     int      NOT NULL,
    WorkTypeId char(2)      NULL,
    Emp_NO     int      NOT NULL,
    "Date"     date     NOT NULL  
);

GO

INSERT INTO "Employee" ( "Site", WorkTypeId, Emp_NO, "Date" )
VALUES
( 5015, 'MB', 1005, '2022-02-01' ),
( 5015, 'MI', 1005, '2022-02-04' ),
( 5015, 'PO', 1005, '2022-02-04' ),
( 5015, 'ME', 2003, '2022-01-01' ),
( 5015, 'TT', 2003, '2022-01-10' );

INSERT INTO "PTO" ( "Site", WorkTypeId, Emp_NO, "Date" )
VALUES
( 5015, NULL, 1005, '2022-02-03' ),
( 5015, NULL, 1005, '2022-02-14' ),
( 5014, NULL, 2003, '2022-01-09' );

• 这两种方法都定义扩展 Employee 和 PTO 的 CTE e 和 p分别添加计算的 "Year" 和 "Month" 列，从而避免重复使用 YEAR( "Date" ) AS "Year"在GROUP BY中并且JOIN 表达式。
  • 如果可以的话，我建议您将它们添加为基表中的计算列，因为无论如何它们通常都会有用。不要忘记对它们进行适当的索引。

方法 1：使用基本聚合组合 CTE，然后更新：

WITH
-- Step 1: Extend both the `Employee` and `PTO` tables with YEAR and MONTH columns (this simplifies things later on):
e AS (
    SELECT
        Emp_No,
        "Site",
        WorkTypeId,
        "Date",

        YEAR( "Date" ) AS "Year",
        MONTH( "Date" ) AS "Month"
    FROM
        Employee
),
p AS (
    SELECT
        Emp_No,
        "Site",
        WorkTypeId,
        "Date",

        YEAR( "Date" ) AS "Year",
        MONTH( "Date" ) AS "Month"
    FROM
        PTO
),
-- Step 2: Get the MIN( "Date" ) value for each group:
minDatesForEachEmployeeMonthYearGroup AS (
    SELECT
        e.Emp_No,
        e."Year",
        e."Month",

        MIN( "Date" ) AS "FirstDate"
    FROM
        e
    GROUP BY
        e.Emp_No,
        e."Year",
        e."Month"
),
-- Step 3: INNER JOIN back on `e` to get the first WorkTypeId in each group:
firstWorkTypeIdForEachEmployeeMonthYearGroup AS (
    /* WARNING: This query will fail if multiple rows (for the same Emp_NO, Year and Month) have the same "Date" value. This can be papered-over with GROUP BY and MIN, but I don't think that's a good idea at all). */
    SELECT
        e.Emp_No,
        e."Year",
        e."Month",

        e.WorkTypeId AS FirstWorkTypeId
    FROM
        e
        INNER JOIN minDatesForEachEmployeeMonthYearGroup AS q ON
            e.Emp_NO = q.Emp_NO
            AND
            e."Date" = q.FirstDate
)
-- Step 4: Do the UPDATE.
-- *Yes*, you can UPDATE a CTE (provided the CTE is "simple" and has a 1:1 mapping back to source rows on-disk).
UPDATE
    p
SET
    p.WorkTypeId = f.FirstWorkTypeId
FROM
    p
    INNER JOIN firstWorkTypeIdForEachEmployeeMonthYearGroup AS f ON
        p.Emp_No = f.Emp_No
        AND
        p."Year" = f."Year"
        AND
        p."Month" = f."Month"
WHERE
    p.WorkTypeId IS NULL;

这是 SSMS 的屏幕截图，显示了上述查询运行之前和之后 PTO 表的内容：

< a href="https://i.sstatic.net/5eQxN.png" rel="nofollow noreferrer">

方法 2：使用 FIRST_VALUE 跳过自JOIN：

此方法提供了更短、稍微简单的查询，但需要SQL Server 2012 或更高版本（并且您的数据库在兼容级别 110 或更高版本中运行）。

令人惊讶的是，您不能在以下位置使用 FIRST_VALUE GROUP BY 查询，尽管它与 MIN 明显相似，但可以使用 SELECT DISTINCT 构建等效查询

WITH
-- Step 1: Extend the `Employee` table with YEAR and MONTH columns:
e AS (
    SELECT
        Emp_No,
        "Site",
        WorkTypeId,
        "Date",

        YEAR( "Date" ) AS "Year",
        MONTH( "Date" ) AS "Month"
    FROM
        Employee
),
firstWorkTypeIdForEachEmployeeMonthYearGroup AS (

    SELECT
        DISTINCT
        e.Emp_No,
        e."Year",
        e."Month",
        FIRST_VALUE( WorkTypeId ) OVER (
            PARTITION BY
                Emp_No,
                e."Year",
                e."Month"
            ORDER BY
                "Date" ASC
        ) AS FirstWorkTypeId
    FROM
        e
)
-- Step 3: UPDATE PTO:
UPDATE
    p
SET
    p.WorkTypeId = f.FirstWorkTypeId
FROM
    PTO AS p
    INNER JOIN firstWorkTypeIdForEachEmployeeMonthYearGroup AS f ON
        p.Emp_No = f.Emp_No
        AND
        YEAR( p."Date" ) = f."Year"
        AND
        MONTH( p."Date" ) = f."Month"
WHERE
    p.WorkTypeId IS NULL;

：运行后 SELECT * FROM PTO 给出了与方法 2 完全相同的输出。

方法 2b，但更短：

这样 @SOS 就不会觉得自己太自鸣得意了。 SQL 比我的要短得多