如何更新插入到表中？

StackOverflow https://stackoverflow.com/questions/19089

  •  09-06-2019
  •  | 
  •  

题

我有一个视图，其中包含作业列表，其中包含诸如分配给谁以及所处阶段等数据。我需要编写一个存储过程，返回每个人在每个阶段有多少工作。

到目前为止我有这个（简化的）：

DECLARE @ResultTable table 
(
  StaffName nvarchar(100),
  Stage1Count int,
  Stage2Count int
)

INSERT INTO @ResultTable (StaffName, Stage1Count)
  SELECT StaffName, COUNT(*) FROM ViewJob
  WHERE InStage1 = 1
  GROUP BY StaffName

INSERT INTO @ResultTable (StaffName, Stage2Count)
  SELECT StaffName, COUNT(*) FROM ViewJob
  WHERE InStage2 = 1
  GROUP BY StaffName

问题在于行不合并。因此，如果一名员工在 stage1 和 stage2 都有工作，@ResultTable 中就会有两行。我真正想做的是更新该行（如果工作人员存在该行）并插入一个新行（如果不存在）。

有谁知道如何做到这一点，或者可以建议不同的方法？我真的很想避免使用游标来迭代用户列表（但这是我的后备选项）。

我正在使用 SQL Server 2005。

编辑：@李： 不幸的是，InStage1 = 1 是一种简化。它实际上更像是 WHERE DateStarted IS NOT NULL 和 DateFinished IS NULL。

编辑：@BCS： 我喜欢先插入所有员工的想法，所以我每次只需要进行更新。但我正在努力让这些 UPDATE 语句正确。

有帮助吗？

解决方案

IIRC 有某种“On Duplicate”（名称可能是错误的）语法，可以让您更新行是否存在（MySQL）

或者某种形式：

INSERT INTO @ResultTable (StaffName, Stage1Count, Stage2Count)
  SELECT StaffName,0,0 FROM ViewJob
  GROUP BY StaffName

UPDATE @ResultTable Stage1Count= (
  SELECT COUNT(*) AS count FROM ViewJob
  WHERE InStage1 = 1
  @ResultTable.StaffName = StaffName)

UPDATE @ResultTable Stage2Count= (
  SELECT COUNT(*) AS count FROM ViewJob
  WHERE InStage2 = 1
  @ResultTable.StaffName = StaffName)

其他提示

事实上，我认为你让事情变得比实际困难得多。这段代码不适合您想要做的事情吗？

SELECT StaffName, SUM(InStage1) AS 'JobsAtStage1', SUM(InStage2) AS 'JobsAtStage2'
  FROM ViewJob
GROUP BY StaffName

您只需检查是否存在并使用适当的命令即可。我相信这确实在幕后使用了光标，但这是您可能得到的最好的： 

IF (EXISTS (SELECT * FROM MyTable WHERE StaffName = @StaffName))
begin
    UPDATE MyTable SET ... WHERE StaffName = @StaffName
end
else
begin
    INSERT MyTable ...
end

SQL2008 有一个新的 MERGE 功能，这很酷，但不是 2005 年的。

要获得真正的“upsert”类型的查询，您需要使用 if 存在...类型的东西，不幸的是这意味着使用光标。

但是，您可以运行两个查询，一个查询在存在现有行的情况下进行更新，然后插入新行。我认为这种基于集合的方法会更好，除非您只处理少量的行。

结果表上的以下查询应再次合并行。这是假设 InStage1 和 InStage2 永远不会同时为“1”。

select distinct(rt1.StaffName), rt2.Stage1Count, rt3.Stage2Count
from @ResultTable rt1
left join @ResultTable rt2 on rt1.StaffName=rt2.StaffName and rt2.Stage1Count is not null
left join @ResultTable rt3 on rt1.StaffName=rt2.StaffName and rt3.Stage2Count is not null

我设法让它与 BCS 答案的变体一起工作。但它不允许我使用表变量，所以我必须创建一个临时表。

CREATE TABLE #ResultTable
(
  StaffName nvarchar(100),
  Stage1Count int,
  Stage2Count int
)

INSERT INTO #ResultTable (StaffName)
  SELECT StaffName FROM ViewJob
  GROUP BY StaffName

UPDATE #ResultTable SET 
  Stage1Count= (
    SELECT COUNT(*) FROM ViewJob V
    WHERE InStage1 = 1 AND 
        V.StaffName = @ResultTable.StaffName COLLATE Latin1_General_CI_AS
    GROUP BY V.StaffName),
  Stage2Count= (
    SELECT COUNT(*) FROM ViewJob V
    WHERE InStage2 = 1 AND 
        V.StaffName = @ResultTable.StaffName COLLATE Latin1_General_CI_AS
    GROUP BY V.StaffName)

SELECT StaffName, Stage1Count, Stage2Count FROM #ResultTable

DROP TABLE #ResultTable
许可以下： CC-BY-SA 和 归因
不隶属于 StackOverflow
scroll top