Question

我创建了这个接受参数@UpdType char(1)的存储过程。无论输入什么参数，该过程都会执行。然而;如果参数是R或U，则存储过程仅完成其功能。

有没有办法约束存储过程只接受参数R或U作为输入？

CREATE PROCEDURE UPDATECATS @UPDTYPE CHAR(1)
AS
BEGIN
    IF @UPDTYPE = 'U' --UPDATE TO NEW CATEGORIES IF PARAMETER IS R
    BEGIN
        UPDATE CATLEVEL1 
        SET CAT1DESC = 'LABORATORY EQUIPMENT' 
        WHERE CAT1DESC = 'MEDICAL & ALLIED EQUIPMENT'

        UPDATE CATLEVEL1 
        SET CAT1DESC = 'MACHINERY & CAMPUS EQUIPMENT' 
        WHERE CAT1DESC = 'OTHER MACHINERY & EQUIPMENT'
    END

    IF @UPDTYPE = 'R' --REVERSE TO OLD CATEGORIES IF THE PARAMETER IS U
    BEGIN
        UPDATE CATLEVEL1 
        SET CAT1DESC = 'MEDICAL & ALLIED EQUIPMENT' 
        WHERE CAT1DESC = 'LABORATORY EQUIPMENT'

        UPDATE CATLEVEL1 
        SET CAT1DESC = 'OTHER MACHINERY & EQUIPMENT' 
        WHERE CAT1DESC = 'MACHINERY & CAMPUS EQUIPMENT'
    END

    IF @UPDTYPE = 'R' OR @UPDTYPE = 'U' --UPDATE THE CATEGORIES IN THE ASSETS TABLE
    BEGIN
        UPDATE ASSETS 
        SET ASSETS.CAT3 = B.CATLEVEL3LINK 
        FROM ASSETS A 
        LEFT JOIN ASSETDESC B ON A.ASDES = B.ASSETDESCRIPTION

        UPDATE ASSETS 
        SET ASSETS.CAT2 = B.CATLEVEL2LINK 
        FROM ASSETS A 
        LEFT JOIN CATLEVEL3 B ON A.CAT3 = B.CAT3DESC

        UPDATE ASSETS 
        SET ASSETS.CAT1 = B.CATLEVEL1LINK 
        FROM ASSETS A 
        LEFT JOIN CATLEVEL2 B ON A.CAT2 = B.CAT2DESC
    END
END

Answer 1

一种方法是添加简单的验证：

CREATE PROCEDURE UPDATECATS @UPDTYPE CHAR(1)
AS
BEGIN
IF @UPDTYPE IN ('R', 'U')
  BEGIN
    -- your code
  END
ELSE
  BEGIN
    -- RAISERROR/THROW
  END
END

或：

CREATE PROCEDURE UPDATECATS @UPDTYPE CHAR(1)
AS
BEGIN
  IF (@UPDTYPE NOT IN ('R', 'U')) OR (@UPDTYPE IS NULL)
    RAISERROR/THROW ..

  --rest your code
END

Answer 2

你有一些选择。

选项1：从存储过程返回状态，并检查输入上的有效值。换句话说，就像这样：

if (@UPDTYPE NOT IN ('R', 'D'))
begin
    print 'Invalid argument';
    return(-1);
end;

选项2：使用标志代替IsRUpd。您可以将其声明为bit not null，将其限制为两个值。

选项3：创建您自己的用户定义类型，只接受这两个值。

在我看来，第一种方法是最简单的实现方法。并且，存储过程中的正确错误处理具有许多长期的好处。

SQL - 限制存储过程输入参数

2 个答案: