if exists (select * from dbo.sysobjects where id = object_id(N'[dbo].[p_RestoreDb]') and OBJECTPROPERTY(id, N'IsProcedure') = 1)
drop procedure [dbo].[p_RestoreDb]
GO

/*--恢復數據庫的通用存儲過程

--鄒建 2003.10--*/

/*--調用示例
--完整恢復數據庫
exec p_RestoreDb @bkfile='c:\db_20031015_db.bak',@dbname='db'

--差異備份恢復
exec p_RestoreDb @bkfile='c:\db_20031015_db.bak',@dbname='db',@retype='DBNOR'
exec p_RestoreDb @bkfile='c:\db_20031015_df.bak',@dbname='db',@retype='DF'

--日志備份恢復
exec p_RestoreDb @bkfile='c:\db_20031015_db.bak',@dbname='db',@retype='DBNOR'
exec p_RestoreDb @bkfile='c:\db_20031015_log.bak',@dbname='db',@retype='LOG'

--*/

create proc p_RestoreDb
@bkfile nvarchar(1000),		--定義要恢復的備份文件名(帶路徑)
@dbname sysname='',    		--定義恢復后的數據庫名,默認為備份的文件名
@dbpath nvarchar(260)='',	--恢復后的數據庫存放目錄,不指定則為SQL的默認數據目錄
@retype nvarchar(10)='DB',	--恢復類型:'DB'完事恢復數據庫,'DBNOR' 為差異恢復,日志恢復進行完整恢復,'DF' 差異備份的恢復,'LOG' 日志恢復
@filenumber int=1,			--恢復的文件號
@overexist bit=1,	      	--是否覆蓋已經存在的數據庫,僅@retype為'DB'/'DBNOR'是有效
@killuser bit=1,		   	--是否關閉用戶使用進程,僅@overexist=1時有效
@password nvarchar(20)=''	--備份文件的密碼(僅sql2000支持),如果備份時設置了密碼,必須提供此密碼
as
declare @sql varchar(8000)

--得到恢復后的數據庫名
if isnull(@dbname,'')=''
	select @sql=reverse(@bkfile)
		,@sql=case when charindex('.',@sql)=0 then @sql
			else substring(@sql,charindex('.',@sql)+1,1000) end
		,@sql=case when charindex('\',@sql)=0 then @sql
			else left(@sql,charindex('\',@sql)-1) end
		,@dbname=reverse(@sql)

--得到恢復后的數據庫存放目錄
if isnull(@dbpath,'')=''
begin
	select @dbpath=rtrim(reverse(filename)) from master..sysfiles where name='master'
	select @dbpath=reverse(substring(@dbpath,charindex('\',@dbpath),4000))
end

--生成數據庫恢復語句
set @sql='restore '+case @retype when 'LOG' then 'log ' else 'database ' end+@dbname
	+' from disk='''+@bkfile+''''
	+' with file='+cast(@filenumber as varchar)
	+case when @overexist=1 and @retype in('DB','DBNOR') then ',replace' else '' end
	+case @retype when 'DBNOR' then ',NORECOVERY' else ',RECOVERY' end
	+case isnull(@password,'') when '' then '' else ',PASSWORD='''+@password+'''' end

--添加移動邏輯文件的處理
if @retype='DB' or @retype='DBNOR'
begin
	--從備份文件中獲取邏輯文件名
	declare @lfn nvarchar(128),@tp char(1),@i int,@s varchar(1000)

	--創建臨時表,保存獲取的信息
	create table #tb(ln nvarchar(128),pn nvarchar(260),tp char(1),fgn nvarchar(128),sz numeric(20,0),Msz numeric(20,0))
	--從備份文件中獲取信息
	set @s='restore filelistonly from disk='''+@bkfile+''''
		++case isnull(@password,'') when '' then '' else ' with PASSWORD='''+@password+'''' end
	insert into #tb exec(@s)
	declare #f cursor for select ln,tp from #tb
	open #f
	fetch next from #f into @lfn,@tp
	set @i=0
	while @@fetch_status=0
	begin
		select @sql=@sql+',move '''+@lfn+''' to '''+@dbpath+@dbname+cast(@i as varchar)
			+case @tp when 'D' then '.mdf''' else '.ldf''' end
			,@i=@i+1
		fetch next from #f into @lfn,@tp
	end
	close #f
	deallocate #f
end

--關閉用戶進程處理
if @overexist=1 and @killuser=1
begin
	declare hCForEach cursor for
	select s='kill '+cast(spid as varchar) from master..sysprocesses 
	where dbid=db_id(@dbname)
	exec sp_msforeach_worker '?'
end

--恢復數據庫
exec(@sql)
go