复杂TSQL合并

我已经“继承”辉煌片的TSQL代码，这是否：复杂TSQL合并

循环行接一行上的光标。
光标包含表A中
需要被合并（Upserted）数据对于在光标存储的过程被称为每行循环。体proc：
- 如果表A中存在一个对应的行然后被更新
- 如果这样的行不存在，那么：
  - 插入一个单一的行中的一个不同的表B.
  - 抓取操作新生成的ID（称为它的IDB）
  - 在表A中插入一行。表A插入需要 IDB（该字段不为空，它应该只有来自表B的值，但没有FK约束到位）

显然，这很烂（性能&优雅的原因）！

问题起初，这看起来像一个MERGE用法的标准情况。我试图做：

MERGE [dbo].[TableA] AS Target 
USING <cursor data set as a select statement> as Src on target.IDA = Src.IDA 
WHEN MATCHED 
    //update 
WHEN NOT MATCHED 
//insert <------ Fails because obviously a new IDB is required

也试过各种方法，如一个nested select that sends IDB on the OUTPUT但它失败，因为IDB是一个PK。

其他种类的合并也未能如：

MERGE Table A with <cursor data set as a select statement> 
... 
MERGE Table A with Table B 
WHEN NOT MATCHED 
//insert on Table A 
WHEN NOT MATCHED 
// Update Table B

有没有人有这个想法？从本质上讲，我认为，如果我们概括出的问题是：

Can I insert and return the PK in one statement that can be nested in other statements

预先感谢任何答复

乔治

来源

2011-09-21 George

您可以合并到tableB中，并使用[技术这里]（http://stackoverflow.com/q/5365629/73226）我想。 –

谢谢，看到了。它其实是个好主意。本质上，这个人将新生成的ID存储在临时变量中。与我所寻找的不一样（在1合并中完成整个过程），但它仍然是一个有趣的方法。感谢Martin。 – George

使用单词'upserted'+1 ... :) –

如果您在TableB上有自动生成的PK，则可以使用与此类似的代码。否则，只需将INSERT更改为TableA即可首先从TableB中获取PK。

DECLARE @OldData CHAR(10) 
SET @OldData = 'Old' 
DECLARE @NewData CHAR(10) 
SET @NewData = 'New' 

CREATE TABLE #TableA 
(
    IDA INT IDENTITY(1,1) PRIMARY KEY, 
    IDB INT NOT NULL, 
    DataA CHAR(10) 
) 

CREATE TABLE #TableB 
(
    IDB INT IDENTITY(1,1) PRIMARY KEY, 
    DataB CHAR(10) 
) 

DECLARE @IDsToUpsert TABLE 
(
    ID INT 
) 

-- Add test values for existing rows 
INSERT INTO #TableB 
OUTPUT INSERTED.IDB, @OldData 
INTO #TableA 
SELECT @OldData UNION ALL 
SELECT @OldData UNION ALL 
SELECT @OldData UNION ALL 
SELECT @OldData 

-- Add test values for the rows to upsert 
INSERT INTO @IDsToUpsert 
SELECT 1 UNION -- exists 
SELECT 3 UNION -- exists 
SELECT 5 UNION -- does not exist 
SELECT 7 UNION -- does not exist 
SELECT 9  -- does not exist 

-- Data Before 
SELECT * From #TableA 
SELECT * From #TableB 

DECLARE rows_to_update CURSOR 
    FOR SELECT ID FROM @IDsToUpsert 

DECLARE @rowToUpdate INT 
DECLARE @existingIDB INT 

OPEN rows_to_update; 

FETCH NEXT FROM rows_to_update 
INTO @rowToUpdate; 

WHILE @@FETCH_STATUS = 0 
BEGIN 
    BEGIN TRANSACTION 

     IF NOT EXISTS 
     (
      SELECT 1 FROM #TableA WITH (UPDLOCK, ROWLOCK, HOLDLOCK) 
      WHERE IDA = @rowToUpdate    
     ) 
     BEGIN 
      -- Insert into B, then insert new val into A 
      INSERT INTO #TableB 
      OUTPUT INSERTED.IDB, INSERTED.DataB 
      INTO #TableA 
      SELECT @NewData 
      -- Change code here if PK on TableB is not autogenerated 
     END 
     ELSE 
     BEGIN 
      -- Update 
      UPDATE #TableA 
      SET DataA = @NewData 
      WHERE IDA = @rowToUpdate 
     END 

    COMMIT TRANSACTION 

    FETCH NEXT FROM rows_to_update 
    INTO @rowToUpdate; 
END 

CLOSE rows_to_update; 
DEALLOCATE rows_to_update; 

SELECT * FROM #TableA 
SELECT * FROM #TableB 

DROP TABLE #TableA 
DROP TABLE #TableB

来源

2011-09-21 19:16:25 GalacticJello

我认为它是一个非常好的答案;这<100点海报感谢你;-) – George

谢谢，糟糕的一天...对不起。 – GalacticJello

要回答你一般的问题 - “我可以插入和返回的PK中一个可以嵌套在其他陈述中的陈述' - 是的，绝对的。但这取决于创建PK的逻辑。在这种情况下，似乎要生成一个PK，您需要插入到不同的表中，然后从那里获取新生成的ID。这不是很有效（恕我直言），除非有非常具体的理由这样做。自动增量，GUID等往往比PK更好。如果你可以简化/改变背后的逻辑，并且你可以找到一个更简单的方法来实现它，所以PK'CAN'可以在一个语句/函数中生成，因此可以在其他语句中使用。

来源

2011-09-21 16:22:53

复杂TSQL合并

回答

相关问题