在運行時切換數據庫上下文

Question

我正在使用下面的代碼將db上下文切換爲主並創建過程並設置啓動腳本。

BEGIN TRY 
DECLARE @dbName NVARCHAR(100) 
SET @dbName = DB_NAME() 
USE MASTER 
IF NOT EXISTS (
      SELECT name 
      FROM sys.objects 
      WHERE object_id = OBJECT_ID('spSetTrustWorthyOn') 
      ) 
     EXEC (
       'CREATE PROCEDURE spSetTrustWorthyOn 
     AS 
     BEGIN 
      ALTER DATABASE [' + @dbName + '] SET TRUSTWORTHY ON 
     END' 
       ) 

    EXECUTE sp_procoption 'spSetTrustWorthyOn' 
     ,'startup' 
     ,'ON' 
    END TRY 
    BEGIN CATCH 
    END CATCH 
    GO 

現在問題是當我想切換回現有的數據庫。我找不到任何方式回到我原來的數據庫。

我也不能硬編碼數據庫，因爲這是動態查詢，我們有多個數據庫。

任何幫助將不勝感激。

感謝

Answer 1

代替主數據庫USE語句的，有資格的目錄視圖，並使用EXEC sp_executesql的語句與限定主數據庫。這將避免更改外部腳本中的數據庫上下文。

DECLARE 
    @dbName sysname = DB_NAME() 
    ,@sql nvarchar(MAX); 

BEGIN TRY 

    IF NOT EXISTS (
     SELECT * 
     FROM master.sys.objects 
     WHERE object_id = OBJECT_ID(N'spSetTrustWorthyOn') 
     ) 
    BEGIN 
     SET @sql = N'CREATE PROCEDURE spSetTrustWorthyOn 
      AS 
      BEGIN 
       ALTER DATABASE ' + QUOTENAME(@dbName) + ' SET TRUSTWORTHY ON; 
      END;'; 
     EXECUTE master..sp_executesql @sql; 
     EXECUTE sp_procoption 
      'spSetTrustWorthyOn' 
      ,'startup' 
      ,'ON'; 
    END; 
END TRY 
BEGIN CATCH 
    THROW; 
END CATCH; 
GO