修改表主键字段数据类型(V2.0)


这存储过程up_ModifyPrimaryColumn会生成转换的执行代码,可以使用Copy方式拿执行代码到查询分析器中检查和执行。


 

Use Test
Go
if object_ID('[up_ModifyPrimaryColumn]'is not null
    
Drop Procedure [up_ModifyPrimaryColumn]
Go
Create Proc up_ModifyPrimaryColumn
(
    
@objectname sysname=null,
    
@columnname sysname=null,
    
@type_new sysname=null
)
As
/*
修改表主鍵字段數據類型(V2.0) Andy 2009-3-6 

說明:在原版腳本的基礎上,不刪除原表,以確保原來的數據可日后還原。

Step:
1. Drop Constraint and Index 
 
2. Insert Into tempTables
 
3. Update tmpTables
 
4. Create Table Tables_new
 
5. Insert Into Tables_new
 
6. Drop Table Tables_old and Tables_temp
 
7. rename Tables_new
 
8. Create Constraint/Index
 
*/
    
Set Nocount On
    
Declare    @objectid int,
            
@columnid int,
            
@type_old nvarchar(512),
            
@typeid_old int,
            
@typeid_new int,
            
@Enter nvarchar(10),
            
@Tab nvarchar(10),
            
@Spacebar nvarchar(10),
            
@Date nvarchar(8)
 
    
Select    @Enter=Char(13)+Char(10),
            
@Tab=Char(9),
            
@Spacebar=Char(32),
            
@Date=Convert(char(8),Getdate(),112)
 
    
Select @objectid=object_id,@objectname=name
        
From sys.objects As a
        
Where name=@objectname And type='U' And is_ms_shipped=0 And
        
Not Exists(Select 1 
                        
From sys.extended_properties 
                        
Where major_id=a.Object_id 
                            
And minor_id=0 
                            
And class=1 
                            
And Name = N'microsoft_database_tools_support'
                    )
 
    
If @objectid Is null
    
Begin
        
Raiserror 50001 N'無效的表名!'
        
Return 
    
End
    
    
If not Exists(Select 1 From sys.columns Where object_id=@objectid And name=@columnname)
    
Begin
        
Raiserror 50001 N'無效的字段名!'
        
Return
    
End
    
    
If Not Exists(Select 1 
                    
From sys.indexes As a  
                        
Inner Join sys.index_columns As b On b.object_id=a.object_id 
                            
And b.index_id=a.index_id  
                            
And b.index_id=1  
                            
And a.is_primary_key=1
                        
Inner Join sys.columns As c On c.object_id=b.object_id 
                            
And c.column_id=b.column_id
                    
Where a.object_id=@objectid  
                            
And c.name=@columnname
                 )
    
Begin
        
Raiserror 50001 N'非主鍵字段,不需要調用當前存儲過程來修改數據類型.'
        
Return
    
End

    
    
Exec('Declare @x '+@type_new)
    
If @@Error>0
    
Begin
        
Print N'數據類型定義錯誤!'
        
Return
    
End
 
    
Set @typeid_new=type_id(Left(@type_new+'(',Charindex('(',@type_new+'(')-1))
 
    
If Object_id('tempdb..#ScriptTable'Is Not null
        
Drop Table #ScriptTable
    
Create Table #ScriptTable (id int Identity(1,1Primary Key,[content] nvarchar(4000))
 
    
/* 1. Drop Constraint and Index */
 
    
If Object_id('tempdb..#Tables'Is Not null
        
Drop Table #Tables
    
Create Table #Tables
    (
        objectid 
int Primary Key,
        objectname sysname,
        columnid 
int ,
        columnname sysname,
        typeid_new 
int,
        type sysname,
        TypeDefinition_new 
nvarchar(512),
        Flag 
bit,
        IsIdentity_new 
bit,
        IsRowGuidCol_new 
bit
    )
    
    
Insert Into #Tables
        
Select    a.object_id,object_name(a.object_id),a.column_id,a.name,a.user_type_id,b.name,
                
@type_new +
                    
Case 
                        
When @typeid_new In(48,52,56,59,62,106,108,127And a.object_id=@objectid Then ' Not null Identity(1,1)'
                        
When @typeid_new =36 And a.object_id=@objectid Then ' Rowguidcol Not null '
                        
Else ''
                    
End
                
Case a.object_id When @objectid Then 1 Else 0 End,
                
Case When @typeid_new In(48,52,56,59,62,106,108,127Then 1 Else 0 End,
                
Case When @typeid_new =36 Then 1 Else 0 End
            
From sys.columns As a
                
Inner Join sys.types As b On b.user_type_id=a.user_type_id
            
Where (a.object_id=@objectid And a.name=@columnnameOr 
                    
Exists    (Select 1 
                                
From sys.foreign_key_columns 
                                
Where parent_object_id=a.object_id  And
                                        Referenced_object_id
=@objectid And parent_column_id=a.column_id
                            )
 
    
Insert Into #ScriptTable
        
Select 'Use '+Quotename(db_name())+@Enter+'Go'
 
    
Insert Into #ScriptTable
    
Select 'If object_id('''+Quotename(a.name)+''') Is Not Null Alter Table '+
            
Quotename(object_name(a.parent_object_id))+' Drop Constraint '+Quotename(a.name)+@Enter+'Go'
        
From sys.objects As a
            
Left Outer Join #Tables As b On b.objectid=a.Parent_object_id
        
Where a.type In('PK','','','','UQ'And 
                (    
Exists(Select 1 From #Tables Where objectid=a.Parent_object_id) Or 
                    
Exists(Select 1 
                                
From sys.foreign_keys as x 
                                    
Inner Join #Tables As y On y.objectid=x.referenced_object_id 
                                        
And x.object_id=a.object_id
                          )
                )
        
Order By b.Flag ,
            
Case a.type 
                    
When '' Then 1 
                    
When '' Then 2
                    
When 'UQ' Then 3
                    
When '' Then 4
                    
When 'PK' Then 5
                    
Else 6 
                
End
    
        

    
Insert Into #ScriptTable
    
Select 'If Exists(Select 1 From sys.indexes Where object_id=object_id('''+b.objectname+''') And name='''+
            a.name
+''') Drop Index '+Quotename(b.objectname)+'.'+Quotename(a.name)
        
From sys.indexes As a
            
Inner Join #Tables As b On b.objectid=a.object_id
        
Where a.name Is Not null
        
Order By b.Flag
 
    
Insert Into #ScriptTable Select 'Go'
    
    
/* 2. Insert Into tempTables */
    
Insert Into #ScriptTable
    
Select 'If Object_id(''Tempdb..[#'+objectname+']'') Is Not Null Drop Table [#'+objectname+']'+@Enter+@Tab+
            
Case Flag
                
When 1 Then 
                    
Case 
                        
When IsRowGuidCol_new =1 Then 
                            
'Select *,Newid() As ['+columnname+'_new] Into [#'+objectname+'] From '+Quotename(objectname)
                        
When IsIdentity_new =1 Then 
                            
'Select *,Row_number() Over(Order By '+Quotename(columnname)+') As ['+columnname+'_new] Into [#'+objectname+'] From '+
                                
Quotename(objectname)
                        
Else ''
                    
End
                
Else 'Select *,Convert('+TypeDefinition_new+',null) As ['+columnname+'_new] Into [#'+objectname+'] From '+
                            
Quotename(objectname)
            
End+@Enter
        
From #Tables
    
Insert Into #ScriptTable Select 'Go'
    
    
/* 3. Update tmpTables */
    
Insert Into #ScriptTable
    
Select 'Update a Set a.['+columnname+'_new]=b.['+@columnname+'_new] From [#'+objectname+'] As a Inner Join [#'+
                
@objectname+'] As b On b.'+Quotename(@columnname)+'=a.'+Quotename(columnname)
        
From #Tables 
        
Where Flag=0
    
Insert Into #ScriptTable Select 'Go'
    
 
 
    
/* 4. Create Table Tables_new */
 
    
If Object_id('tempdb..#ObjectDefinition'Is Not null
        
Drop Table #ObjectDefinition
 
    
Select  a.object_id As objectid,a.column_id As columnid,a.name As columnname,
        
--Column_name
        Case 
            
When a.column_id=e.columnid Then Quotename(a.name)+@Spacebar+e.TypeDefinition_new 
            
Else
                
Quotename(a.name)+@Spacebar+b.name+
                
--data_length
                Case 
                    
When a.user_type_id In (127,104,61,34,56,60,99,59,58,52,122,98,35,189,48,36,241Then ''
                    
When a.user_type_id In (106,108Then '('+Rtrim(a.precision)+','+Rtrim(a.scale)+')'
                    
When a.user_type_id In (239,231Then '('+Rtrim(a.max_length/2)+')'
                    
When a.max_length<>b.max_length Then  '('+Rtrim(a.max_length)+')'
                    
Else ''
                
End
        
End +@Spacebar+
        
--Rowguidcol
        Case a.is_rowguidcol When 1 Then 'Rowguidcol'+@Spacebar Else '' End +
        
--Identity
        Case a.is_identity When 1 Then 'Identity('+Convert(nvarchar(10),c.seed_value)+','+
                
Convert(nvarchar(10),c.increment_value)+')'+@Spacebar Else '' End +
        
--Collate
        --Isnull('Collate '+a.collation_name+@Spacebar,'')+
        --Null
        Case a.is_nullable When 1 Then 'Null' Else 'Not Null' End +@Spacebar
        
--Default
        As column_definition
        
Into #ObjectDefinition
    
From sys.columns As a
        
Inner Join sys.types As b On b.user_type_id=a.user_type_id
        
Left Outer Join sys.Identity_columns As c On c.object_id=a.object_id And c.column_id=a.column_id
        
Inner Join #Tables As e On e.objectid=a.object_id
        
Order By a.object_id,a.column_id

    
Create Clustered Index IX_#ObjectDefinition_ObjectidColumnid On #ObjectDefinition(objectid Asc,columnid Asc)
 
    
Insert Into #ScriptTable
    
Select 'If Object_id(''['+objectname+'_new]'') Is Not Null Drop Table ['+objectname +'_new]'+@Enter+
            
'Create Table ['+objectname +'_new]'+@Enter+'('+@Enter+@Tab+b.column_definition+@Enter+')'
        
From #Tables As a
        
Outer Apply(Select column_definition=
                        
Stuff((Select ','+Case  
                                            
When a.Flag=1 And columnid=a.columnid Then 
                                                a.columnname
+@Spacebar+TypeDefinition_new 
                                            
Else column_definition 
                                          
End+Char(10)+@Tab 
                                    
From #ObjectDefinition 
                                    
Where objectid=a.objectid For Xml Path(''))
                            ,
1,1,'')
                    ) 
As b
    
Insert Into #ScriptTable Select 'Go'
 
    
    
/* 5. Insert Into Tables_new */
    
Insert Into #ScriptTable
    
Select 
            
Case 
                
When Flag=1 And IsIdentity_new = 1 Then 'Set Identity_insert ['+objectname+'_new] On '+@Enter 
                
When Flag=0 And Exists(Select 1 
                                            
From sys.columns 
                                            
Where object_id=a.objectid And 
                                                Is_Identity
=1
                                        ) 
Then 'Set Identity_insert ['+objectname+'_new] On '+@Enter
                
Else ''
            
End+
            
'Insert Into ['+objectname+'_new] ('+columnListA+') Select '+columnListB+' From [#'+objectname+']'++@Enter+
            
Case 
                
When Flag=1 And IsIdentity_new = 1 Then 'Set Identity_insert ['+objectname+'_new] Off ' 
                
When Flag=0 And Exists(Select 1 
                                            
From sys.columns 
                                            
Where object_id=a.objectid 
                                                
And Is_Identity=1
                                        ) 
Then 'Set Identity_insert ['+objectname+'_new] Off '
                
Else ''
            
End
        
From #Tables As a
        
Outer Apply(Select columnListA=Stuff((Select ','+Quotename(columnname) 
                                                    
From #ObjectDefinition 
                                                    
Where objectid=a.objectid 
                                                    
For Xml Path('')),1,1,'')
                    ) 
As b
        
Outer Apply(Select columnListB=Stuff((Select ','+Quotename(columnname+Case columnid When a.columnid Then '_new' Else '' End
                                                    
From #ObjectDefinition 
                                                    
Where objectid=a.objectid 
                                                    
For Xml Path('')),1,1,'')
                    ) 
As c
    
Insert Into #ScriptTable Select 'Go'
 
 
    
/* 6. Drop Table Tables_old and Tables_temp */
    
Insert Into #ScriptTable
    
Select 'If object_id('''+Quotename(objectname)+''') Is Not Null '+@Enter+@Tab+'Drop Table '+
--            Quotename(objectname)+@Enter+'Drop Table [#'+objectname+']'
            '[#'+objectname+']'
        
From #Tables
        
Order By Flag 
    
Insert Into #ScriptTable Select 'Go'
 
    
/* 7. rename Tables_new */
    
Insert Into #ScriptTable
    
Select N'Begin Try' Union All Select Char(9+N'Begin Transaction'
    
    
Insert Into #ScriptTable
    
Select 'If object_id(''['+objectname+']'') Is Not Null And object_id(''['+objectname+@Date+']'') Is Null Exec sp_rename '''+objectname+''','''+objectname+@Date+''',''OBJECT'''
        
From #Tables
        
Order By Flag 

    
Insert Into #ScriptTable
    
Select 'If object_id(''['+objectname+'_new]'') Is Not Null And object_id(''['+objectname+']'') Is Null  Exec sp_rename '''+objectname+'_new'','''+objectname+''',''OBJECT'''
        
From #Tables
        
Order By Flag 
 
    
Insert Into #ScriptTable
        
Select Char(9)+N'Commit Transaction' Union All 
        
Select N'End Try' Union All 
        
Select N'Begin Catch' Union All 
        
Select N'Print N''重命名表的時候發生錯誤!''' Union All 
        
Select N'Rollback Transaction' Union All 
        
Select N'End Catch' Union All 
        
Select 'Go'

 
    
/* 8. Create Constraint/Index */
    
    
If Object_id('tempdb..#indexes'Is Not null
        
Drop Table #indexes
 
    
Select  id=Identity(int,1,1),
            a.
object_id As objectid,
            a.name 
As Index_name,a.Index_id,
            a.type_desc Collate Chinese_Taiwan_Stroke_CI_AS 
As type_desc, --當應用在不同的語言下,這里需要修改.
            Case a.is_unique When 1 Then 'Unique' Else '' End As is_unique,
            
null As data_space,--e.name As data_space,
            Case a.is_primary_key When 1 Then N'Primary Key'  Else null End As is_primary_key,
            
Case a.is_unique_constraint When 1 Then N'Unique'  Else null End As is_unique_constraint,
            a.fill_factor,a.is_padded,a.is_disabled,a.is_hypothetical,a.allow_row_locks,a.allow_page_locks,
            b.index_column_id,
Case b.is_descending_key When 0 Then N'Asc' Else N'Desc' End As descending,b.is_included_column,
            
Quotename(c.name) As column_name
            
Into #indexes
        
From sys.indexes As a
            
Inner Join sys.index_columns As b On b.object_id=a.object_id And b.index_id=a.index_id
            
Inner Join sys.columns As c On c.object_id=b.object_id And c.column_id=b.column_id
            
Inner Join sys.data_spaces As d On d.data_space_id=a.data_space_id
            
Inner Join sys.filegroups As e On e.data_space_id=d.data_space_id
        
Where
            
Exists(Select 1 From  #Tables  Where objectid=a.object_id)
 
    
    
Insert Into #ScriptTable
    
Select 
            
--Default
            Case When c1.name Is null Then '' Else 'If object_id('''+Quotename(a.name)+''') Is Null Alter Table '+
                    
Quotename(object_name(a.parent_object_id))+' Add Constraint '+Quotename(c1.name)+
                    
' Default '+c1.definition+' For '+Quotename(c2.name)+@Enter End+
            
--Check
            Case When d.name Is null Then '' Else 'If object_id('''+Quotename(a.name)+''') Is Null Alter Table '+
                    
Quotename(object_name(a.parent_object_id))+' Add Constraint '+Quotename(d.name)+
                    
' Check '+d.definition+@Enter End+
            
--Unique Key/Primary Key
            Case 
                
When e1.name Is null Then '' 
                
Else 'If object_id('''+Quotename(a.name)+''') Is Null Alter Table '+
                    
Quotename(object_name(a.parent_object_id))+' Add Constraint '+
                    
Quotename(a.Name)+@Spacebar+Isnull(e2.is_primary_key,e2.is_unique_constraint)+@Spacebar+
                    e2.type_desc
+@Spacebar+'('+e3.column_constraint+')'+@Spacebar+Isnull('On '+Quotename(e2.data_space)+@Enter,''
            
End
            
        
From sys.objects As a
            
Inner Join #Tables As b On b.objectid=a.parent_object_id
            
Left Outer Join sys.Default_constraints As c1 On c1.object_id=a.object_id And a.Type='D' And c1.Parent_column_id<>b.columnid
            
Left Outer Join sys.columns As c2 On c2.object_id=c1.parent_object_id And c2.column_id=c1.parent_column_id
            
Left Outer Join sys.check_constraints As d On d.object_id=a.object_id And a.Type='C' And c1.Parent_column_id<>b.columnid
            
Left Outer Join sys.key_constraints As e1 On e1.object_id=a.object_id And a.Type In('UQ','PK')
            
Left Outer Join #indexes As e2 On e2.index_id=e1.unique_index_id 
                    
And e2.objectid=e1.parent_object_id 
                    
And e2.id=(Select Max(id) From #indexes Where  objectid=e2.objectid And index_id=e2.index_id)
            
Outer Apply(Select column_constraint=Stuff((Select ','+column_name+@Spacebar+descending    
                                
From #indexes 
                                
Where objectid=e2.objectid And index_id=e2.index_id 
                                
For xml Path('')),1,1,'')
                        ) 
As e3
        
Where a.Type In('','','UQ','PK')
        
Order By b.Flag Desc,
                
Case a.type 
                    
When '' Then 1 
                    
When '' Then 2
                    
When 'UQ' Then 3
                    
When 'PK' Then 4
                    
Else 5 
                
End

    
--Foreign Key
    Insert Into #ScriptTable
    
Select 
            
Case When f1.name Is null Then '' Else 'If object_id('''+Quotename(a.name)+''') Is Null Alter Table '+
                    
Quotename(object_name(a.parent_object_id))++' With '+
                    
Case f1.is_disabled When 0 Then 'Check ' Else 'Nocheck ' End+
                    
'Add Constraint '+Quotename(a.name)+' Foreign Key '+'('+f2.Foreign_column+') References '+
                    
Quotename(object_name(f1.referenced_object_id))+'('+f3.referenced_column+')'+' On Delete '+
                    
Replace(f1.delete_referential_action_desc,'_',' ')+' On Update '+
                    
Replace(f1.update_referential_action_desc,'_',' ')+@Enter 
            
End 
        
From sys.objects As a
            
Left Outer Join #Tables As b On b.objectid=a.parent_object_id
            
Inner Join sys.foreign_keys As f1 On f1.object_id=a.object_id
            
Outer Apply(Select Foreign_column=Stuff((Select ','+Quotename(y.name) 
                                
From sys.foreign_key_columns x  
                                    
Inner Join sys.columns As y On y.object_id=x.parent_object_id 
                                        
And y.column_id=x.Parent_column_id 
                                
Where x.constraint_object_id=a.object_id 
                                
For xml Path('')),1,1,'')
                        ) 
As f2

            
Outer Apply(Select referenced_column=Stuff((Select ','+Quotename(y.name) 
                                
From sys.foreign_key_columns x  
                                    
Inner Join sys.columns As y On y.object_id=x.referenced_object_id 
                                        
And y.column_id=x.referenced_column_id 
                                
Where x.constraint_object_id=a.object_id 
                                
For xml Path('')),1,1,'')
                        ) 
As f3
        
Where  a.Type='' And 
                    (    
Exists(Select 1 From #Tables Where objectid=a.Parent_object_id) Or 
                        
Exists(Select 1 
                                    
From sys.foreign_keys as x 
                                        
Inner Join #Tables As y On y.objectid=x.referenced_object_id 
                                            
And x.object_id=a.object_id
                               )
                    )
        
Order By b.Flag Desc
    
Insert Into #ScriptTable Select 'Go'
    
--Index
    Insert Into #ScriptTable
    
Select    Distinct 
            
'Create '+is_unique+@Spacebar+a.type_desc + @Spacebar+
            
'Index '+Quotename(a.index_name)+' On '+Quotename(object_name(a.objectid))+@Spacebar+
            
'('+b.[column]+')'+@Spacebar+
            
Isnull('Include ('+c.Clustered_column+')'+@Spacebar,'')+
            
Isnull('On '+Quotename(a.data_space),'')
 
        
From #indexes As a 
            
Outer Apply(Select [column]=Stuff((Select ','+column_name 
                                                    
From #indexes 
                                                    
Where objectid=a.objectid 
                                                        
And index_id=a.index_id 
                                                        
And is_included_column=0 
                                                    
For Xml Path('')),1,1,'')
                        ) 
As b
            
Outer Apply(Select Clustered_column=Stuff((Select ','+column_name 
                                                            
From #indexes 
                                                            
Where objectid=a.objectid 
                                                                
And index_id=a.index_id 
                                                                
And is_included_column=1 
                                                            
For Xml Path('')),1,1,'')
                        ) 
As c
        
Where Not Exists(Select 1 
                            
From sys.key_constraints As x 
                                
Inner Join #Tables As y On y.objectid=x.parent_object_id 
                                        
And x.parent_object_id=a.objectid 
                                        
And x.unique_index_id=a.index_id
                        )
 
    
Insert Into #ScriptTable Select 'Go'
 
    
--Print
    Declare    @Print nvarchar(4000),
            
@i int
     
    
Select @i=1,@Print=[content] From #ScriptTable Where id=1
    
While @Print>''
    
Begin
        
Print @Print
        
Set @i=@i+1
        
Set @Print=(Select [content] From #ScriptTable Where id=@i)
    
End
 
    
Drop Table    #Tables,
                #ObjectDefinition,
                #indexes,
                #ScriptTable
    
/*
    e.g:
    Use Test
    Go
    Exec up_ModifyPrimaryColumn
            @objectname='Test1',
            @columnname='id',
            @type_new='uniqueidentifier'
    
*/
 
Go

 

 

 

posted @ 2009-06-02 23:02  ok_008  阅读(1373)  评论(1编辑  收藏  举报
给我写信