【代码分享-SQL】为DB中所有Table增加PK

辛辛苦苦将数据导入到DB中，却发现忘记创建脚本中忘记PK了。

好在Table都有规律，每个表有个[ID] int字段，PK是建立在其上的。

注：

1）为了代码的可读性，没有采用‘SQL拼接’方法，而是采用了‘先占位，后替换’的方法--看里面的Replace语句。推荐给大家

2）由于是执行DDL，出于谨慎考虑，‘生成’而非‘直接执行’SQL。若不然，读者可以将里面的EXEC语句去掉注释。

----这是代码---------------------------------------

 

1. SET ANSI_NULLS ON 
2. GO 
3. SET QUOTED_IDENTIFIER ON 
4. GO 
5.  
6. -- ============================================= 
7. -- Author:      Yew 
8. -- Create date: 2011-06-12 
9. -- Description: 为所有表增加PK （on ID） 
10.  
11. /* ---------History---------------------------- 
12.  
13. ---------------------------------------------*/ 
14. -- ============================================= 
15. CREATE PROCEDURE [TOOL].[sp_DB_AddPK] 
16. AS 
17. BEGIN 
18.     SET NOCOUNT ON; 
19.  
20.     DECLARE @vTable varchar(100) 
21.         ,@vSchema varchar(100) 
22.         ,@vSql  varchar(max) 
23.  
24.     DECLARE cur_ CURSOR FOR 
25.         SELECT [name], schema_name(schema_id) 
26.         FROM sys.objects 
27.         WHERE type in (N'U') 
28.         ORDER BY schema_id, name 
29.  
30.     OPEN cur_ 
31.     FETCH NEXT FROM cur_        INTO @vTable, @vSchema 
32.     WHILE @@FETCH_STATUS = 0 
33.     BEGIN 
34.         SET @vSQL = N' 
35. ALTER TABLE [#Schema#].[#Table#] ADD  CONSTRAINT [PK_#Table#] PRIMARY KEY CLUSTERED  
36. ( 
37.     [ID] ASC 
38. )WITH (SORT_IN_TEMPDB = OFF, IGNORE_DUP_KEY = ON, ONLINE = OFF) ON [PRIMARY] 
39. GO 
40.  
41. ' 
42.         SET @vSQL = Replace(@vSQL, '#Table#', @vTable) 
43.         SET @vSQL = Replace(@vSQL, '#Schema#', @vSchema) 
44. print @vSql 
45. --      EXEC (@vSql) 
46.  
47.         FETCH NEXT FROM cur_ INTO @vTable, @vSchema 
48.     END 
49.     CLOSE cur_ 
50.     DEALLOCATE cur_ 
51. END 

 

----这是代码--------------------------------------------

本文转自DavyYew 51CTO博客，原文链接：http://blog.51cto.com/davyyew/586877，如需转载请自行联系原作者