MS SQL數據庫備份和恢復存儲過程(加強版本)

2024-07-21 02:32:03

字體：大中小

來源：轉載

供稿：網友

先將代碼發布出來，大家共享。
如有發現BUG，請大家指教或EMAIL：aierong@vip.sina.com
/**//*備份數據庫*/
create PRoc pr_backup_db
@flag varchar(20) out,
@backup_db_name varchar(128),
@filename varchar(1000) --路徑＋文件名字
as
declare @sql nvarchar(4000),@par nvarchar(1000)
if not exists(
select * from master..sysdatabases
where name=@backup_db_name
)
begin
select @flag='db not exist' /**//*數據庫不存在*/
return
end
else
begin
if right(@filename,1)<>'' and charindex('',@filename)<>0
begin
select @par='@filename varchar(1000)'
select @sql='BACKUP DATABASE '+@backup_db_name
+' to disk=@filename with init'
execute sp_executesql @sql,@par,@filename
select @flag='ok'
return
end
else
begin
select @flag='file type error' /**//*參數@filename輸入格式錯誤*/
return
end
end

/**//*創建函數,得到文件得路徑*/
create function fn_GetFilePath(@filename nvarchar(260))
returns nvarchar(260)
as
begin
declare @file_path nvarchar(260)
declare @filename_reverse nvarchar(260)
select @filename_reverse=reverse(@filename)
select @file_path=substring(@filename,1,len(@filename)+1-charindex('',@filename_reverse))
return @file_path
end

/**//*恢復數據庫*/
CREATE proc pr_restore_db
/**//*
------------------------------------------------
Create Time:    2004-03-20
Update Time:    2004-03-29 11:05
Author:         aierong
Remark:         恢復數據庫

------------------------------------------------
*/
/**//*過程運行的狀態標志,是輸入參數*/
@flag varchar(20) out,
/**//*要恢復的數據名字*/
@restore_db_name nvarchar(128),
/**//*備份文件存放的路徑+備份文件名字*/
@filename nvarchar(260)
as
/**//*返回系統存儲過程xp_cmdshell運行結果*/
declare @proc_result tinyint
/**//*循環次數*/
declare @loop_time smallint
/**//*@tem表的ids列最大數*/
declare @max_ids smallint
/**//*原數據庫存放路徑*/
declare @file_bak_path nvarchar(260)
/**//*文件存放標志*/
declare @flag_file bit
/**//*數據庫master文件路徑*/
declare @master_path nvarchar(260)
declare @sql nvarchar(4000),@par nvarchar(1000)
declare @sql_sub nvarchar(4000)
declare @sql_cmd nvarchar(100)
declare @sql_kill nvarchar(100)
/**//*
判斷參數@filename文件格式合法性，以防止用戶輸入類似d: 或者 c:a 等非法文件名
參數@filename里面必須有''并且不以''結尾
*/
if right(@filename,1)<>'' and charindex('',@filename)<>0
begin
select @sql_cmd='dir '+@filename
EXEC @proc_result = master..xp_cmdshell @sql_cmd,no_output
/**//*系統存儲過程xp_cmdshell返回代碼值:0(成功）或1（失敗）*/
IF (@proc_result<>0)
begin
/**//*備份文件不存在*/
select @flag='not exist'
/**//*退出過程*/
return
end
/**//*創建臨時表,保存由備份集內包含的數據庫和日志文件列表組成的結果集*/
create table #tem(
     /**//*文件的邏輯名稱*/
     LogicalName nvarchar(128),
     /**//*文件的物理名稱或操作系統名稱*/
     PhysicalName nvarchar(260) ,
            /**//*數據文件 (D) 或日志文件 (L)*/
     Type char(1),
     /**//*包含文件的文件組名稱*/
     FileGroupName nvarchar(128),
                          /**//*當前大小（以字節為單位）*/
     [Size] numeric(20,0),
     /**//*允許的最大大小（以字節為單位）*/
     [MaxSize] numeric(20,0)
   )
/**//*
創建表變量，表結構與臨時表基本一樣
就是多了兩列,
列ids（自增編號列）,
列file_path,存放文件的路徑
*/
declare @tem table(
     /**//*自增編號列*/
     ids smallint identity,
     LogicalName nvarchar(128),
     PhysicalName nvarchar(260),
     File_path nvarchar(260),
     Type char(1),
     FileGroupName nvarchar(128)
   )
insert into #tem
execute('restore filelistonly from disk='''+@filename+'''')
/**//*將臨時表導入表變量中,并且計算出相應得路徑*/ http://www.survivalescaperooms.com/
insert into @tem(LogicalName,PhysicalName,File_path,Type,FileGroupName)
select LogicalName,PhysicalName,dbo.fn_GetFilePath(PhysicalName),Type,FileGroupName
   from #tem
if @@rowcount>0
begin
drop table #tem
end
select @loop_time=1
/**//*@tem表的ids列最大數*/
select @max_ids=max(ids)
from @tem
while @loop_time<=@max_ids
begin
select @file_bak_path=file_path
   from @tem where ids=@loop_time
select @sql_cmd='dir '+@file_bak_path
EXEC @proc_result = master..xp_cmdshell @sql_cmd,no_output
/**//*系統存儲過程xp_cmdshell返回代碼值:0(成功）或1（失敗）*/
IF (@proc_result<>0)
   select @loop_time=@loop_time+1
else
   /**//*沒有找到備份前數據文件原有存放路徑，退出循環*/
   BREAK
end
select @master_path=''
if @loop_time>@max_ids
/**//*備份前數據文件原有存放路徑存在*/
select @flag_file=1
else
begin
/**//*備份前數據文件原有存放路徑不存在*/
select @flag_file=0
select @master_path=dbo.fn_GetFilePath(filename)
   from master..sysdatabases
   where name='master'
end
select @sql_sub=''
/**//*type='d'是數據文件,type='l'是日志文件 */
/**//*@flag_file=1時新的數據庫文件還是存放在原來路徑，否則存放路徑和master數據庫路徑一樣*/
select @sql_sub=@sql_sub+'move '''+LogicalName+''' to '''
   +case type
         when 'd' then case @flag_file
             when 1 then File_path
      else @master_path
          end
         when 'l' then case @flag_file
      when 1 then File_path
      else @master_path
          end
   end
   +case type
    when 'd' then @restore_db_name
           +'_DATA'
           /**//*給文件編號*/
           +convert(sysname,ids)
           +'.'
           /**//*給文件加入后綴名,mdf or ndf*/
           +right(PhysicalName,3)
           +''','
    when 'l' then @restore_db_name
           +'_LOG'
           /**//*給文件編號*/
           +convert(sysname,ids)
           +'.'
           /**//*給文件加入后綴名,mdf or ndf*/
           +right(PhysicalName,3)
           +''','
    end
   from @tem
select @sql='RESTORE DATABASE @db_name '
                        +'FROM DISK=@filename with '
select @sql=@sql+@sql_sub+'replace'
select @par='@db_name nvarchar(128),@filename nvarchar(260)'
/**//*關閉相關進程，把相應進程狀況導入臨時表中*/
select identity(int,1,1) ids, spid
into #temp
from master..sysprocesses
where dbid=db_id(@restore_db_name)
/**//*找到相應進程*/
if @@rowcount>0
begin
select @max_ids=max(ids)
   from #temp
select @loop_time=1
while @loop_time<=@max_ids
begin
   select @sql_kill='kill '+convert(nvarchar(20),spid)
    from #temp
    where ids=@loop_time
   execute sp_executesql @sql_kill
   select @loop_time=@loop_time+1
end
end
drop table #temp
execute sp_executesql @sql,
                                      @par,
                                      @db_name=@restore_db_name,
                                      @filename=@filename
/**//*操作成功*/
select @flag='ok'
end
else
begin
/**//*參數@filename輸入格式錯誤*/
SELECT @flag='file type error'
end