基于用户输入国家代码实现表ID列分组自增的存储过程编写问题

CREATE PROCEDURE [dbo].[sp_Incremental]
    @code varchar(50)
AS
BEGIN
    SET NOCOUNT ON; -- 关闭影响行数返回，避免额外输出
    DECLARE @ID bigint;
    
    BEGIN TRY
        -- 仅统计当前传入国家代码的最大ID，无数据时默认返回0，加1后即为起始ID
        SELECT @ID = ISNULL(MAX(ID), 0) + 1 
        FROM [Incremental_tbl]
        WHERE CountryCode = @code;
        
        INSERT INTO [Incremental_tbl] (ID, CountryCode) 
        VALUES (@ID, @code);
    END TRY
    BEGIN CATCH
        -- 可按需添加异常处理逻辑，比如抛出错误信息
        THROW;
    END CATCH
END
GO

核心改动说明 #

• 新增WHERE CountryCode = @code过滤条件：统计ID时仅针对当前传入的国家代码，实现分组自增的核心逻辑
• 把count(ID)+1替换为ISNULL(MAX(ID), 0) + 1：避免表中存在该国家的历史数据被删除时，count统计得到的序号和原有ID序列不匹配的问题，无对应国家数据时默认从1开始编号
• 补全DECLARE @ID bigint的语法：原代码中变量名和类型之间缺少空格，会触发语法错误
• 补充完整的TRY-CATCH结构：原代码只有TRY块没有对应的CATCH块，会导致语法异常，新增THROW语句可以在插入出错时抛出原始错误信息便于排查
• 显式指定插入列名：插入语句中明确指定要插入的列，避免表结构变更时插入逻辑出错
• 新增SET NOCOUNT ON：减少不必要的输出信息，提升存储过程执行效率

并发场景优化 #

如果存在多并发同时插入同国家代码数据的场景，可在查询语句的表名后添加WITH (UPDLOCK, HOLDLOCK)锁提示，避免并发导致的ID重复问题：

SELECT @ID = ISNULL(MAX(ID), 0) + 1 
FROM [Incremental_tbl] WITH (UPDLOCK, HOLDLOCK)
WHERE CountryCode = @code;

内容的提问来源于stack exchange，提问作者user17551512