1. SQL Server 批量造数与字符串截取:从 While 循环到游标逐行处理
如果你正在做数据初始化、批量补数、日志清洗这类活儿,SQL Server 里的WHILE循环、CURSOR游标,加上LEFT/RIGHT/SUBSTRING字符串截取,基本就是一套绕不开的组合拳。这篇内容聚焦的就是这个场景:用WHILE循环配合游标逐行处理并插入数据,再用字符串函数把字段里的编码、时间戳、优先级拆出来,最后通过行数比对和结果集校验确认加工结果对不对。
适合谁看?适合已经会写基础INSERT、SELECT,但一遇到“按行加工再回写”就容易写乱的人。比如要给一张表补 30 天日期数据、要把3000000q20201010090559这种拼接串拆成优先级和时间戳、要遍历结果集逐条更新,这些都能直接套用下面的骨架。
我试过在几百万行的表上无脑用游标,结果跑了十几分钟,所以后面也会讲清楚游标该在什么数据量下用、什么时候该换成集合操作。整篇按“问题场景 → 环境准备 → 可复制脚本 → 验证结果 → 排错 → 工具衔接”的顺序展开,脚本都能直接贴进 SSMS 执行。
2. 场景拆解:为什么 While + 游标 + 字符串截取总是一起出现
先说清楚这三者为什么会凑到一块。WHILE解决的是“重复执行 N 次”的问题,比如造 30 天数据、循环 61 天判断工作日;游标解决的是“逐行拿到结果集里的每一行”的问题,比如遍历UserName LIKE '%tian%'的用户逐个更新;字符串截取解决的是“字段里塞了复合信息,需要拆开”的问题,比如sort字段里同时存了优先级和时间戳。
单独用都不难,难的是组合起来不出错。常见的坑有这么几个:游标忘记CLOSE/DEALLOCATE导致资源不释放;@@FETCH_STATUS判断写反导致多处理一行或漏一行;SUBSTRING的起始位置从 1 开始而不是 0,写惯了其他语言的人特别容易错;CHARINDEX找不到分隔符时返回 0,直接拿去当长度参数会算出负数。
所以这篇不是单纯罗列语法,而是给一套能跑通的完整流程:先建表,再用WHILE造数,然后用游标遍历,中间穿插字符串截取,最后用行数和结果集双重校验。你可以把它当成一个可复用的模板,改改表名和字段就能用到自己的库上。
3. 前置准备:建表、造数、确认环境
在写循环之前,先把目标表和测试数据准备好。下面这套脚本建了一张Category表用来演示WHILE插入,建了一张User表用来演示游标遍历,还建了一张trans_queue用来演示字符串截取。你可以按需取用。
-- 演示 WHILE 循环插入的目标表 IF OBJECT_ID('dbo.Category','U') IS NOT NULL DROP TABLE dbo.Category; CREATE TABLE dbo.Category( Id INT IDENTITY(1,1) PRIMARY KEY, LanguageId INT, Title NVARCHAR(100), CreatedDate DATETIME ); -- 演示游标遍历的目标表 IF OBJECT_ID('dbo.[User]','U') IS NOT NULL DROP TABLE dbo.[User]; CREATE TABLE dbo.[User]( Id INT IDENTITY(1,1) PRIMARY KEY, UserName NVARCHAR(100), CreatedDate DATETIME ); INSERT INTO dbo.[User](UserName,CreatedDate) VALUES ('tian001',GETDATE()),('tian002',GETDATE()),('other003',GETDATE()); -- 演示字符串截取的队列表 IF OBJECT_ID('dbo.trans_queue','U') IS NOT NULL DROP TABLE dbo.trans_queue; CREATE TABLE dbo.trans_queue( queue_code INT, sort NVARCHAR(200) ); INSERT INTO dbo.trans_queue(queue_code,sort) VALUES (17,'3000000q20201010090559');建完之后先SELECT COUNT(*)记一下每张表的初始行数,后面验证要用。这一步别省,很多人跑完循环发现数据不对,回头连原始行数是多少都说不清。
注意:
User是 SQL Server 的保留字,建表和查询时用方括号[User]包起来,否则会报语法错误。这是新手最常踩的坑之一。
4. 可复制配置:While 循环插入 + 游标遍历 + 字符串截取
4.1 While 循环插入 30 天数据
最基础的WHILE造数骨架,用@num当计数器,DATEADD生成递增日期,CONVERT把数字拼进标题。这段可以直接跑。
DECLARE @num INT = 1; DECLARE @dt DATETIME = '2000-01-01 00:00:00'; WHILE (@num <= 30) BEGIN INSERT INTO dbo.Category(LanguageId,Title,CreatedDate) VALUES (1, 'test' + CONVERT(VARCHAR(100),@num), DATEADD(DAY,@num,@dt)); SET @num = @num + 1; END GO执行完SELECT COUNT(*) FROM dbo.Category应该是 30 行。这里有个细节:@@IDENTITY在批量插入场景下容易拿到触发器产生的 ID,如果你需要回写自增主键,建议用SCOPE_IDENTITY()替代,作用域更干净。
4.2 游标遍历结果集逐行更新
游标的标准四步:声明、打开、FETCH循环、关闭释放。下面这段遍历UserName LIKE '%tian%'的用户,把 Id 追加到用户名后面。
DECLARE @id INT; DECLARE @name NVARCHAR(50); DECLARE cursor1 CURSOR FOR SELECT Id, UserName FROM dbo.[User] WHERE UserName LIKE '%tian%'; OPEN cursor1; FETCH NEXT FROM cursor1 INTO @id, @name; WHILE @@FETCH_STATUS = 0 BEGIN UPDATE dbo.[User] SET UserName = UserName + CONVERT(VARCHAR(100),@id) WHERE Id = @id; FETCH NEXT FROM cursor1 INTO @id, @name; END CLOSE cursor1; DEALLOCATE cursor1; GO关键点在于FETCH NEXT要写两次:循环外先取第一行,循环内处理完再取下一行。@@FETCH_STATUS = 0表示取数成功,等于 -1 表示已到末尾,等于 -2 表示行被删除。判断条件写错就会出现死循环或者漏处理最后一行。
4.3 LEFT / RIGHT / SUBSTRING 字符串截取
三个函数的分工很明确:LEFT从左边取 N 个字符,RIGHT从右边取 N 个字符,SUBSTRING从指定位置取指定长度。先看基础输出。
SELECT LEFT('Welcome to China',7); -- Welcome SELECT RIGHT('Welcome to China',5); -- China SELECT SUBSTRING('Welcome to China',1,7); -- Welcome SELECT SUBSTRING('Welcome to China',12,LEN('Welcome')); -- China真正有用的是配合CHARINDEX做动态截取。比如从3000000q20201010090559里拆出q前面的优先级和后面的时间戳。
DECLARE @sort NVARCHAR(200) = '3000000q20201010090559'; DECLARE @priority VARCHAR(1); DECLARE @time_stamp NVARCHAR(200); SELECT @priority = LEFT(@sort,1), @time_stamp = REVERSE(SUBSTRING(REVERSE(@sort),1,CHARINDEX('q',REVERSE(@sort)) - 1)); SELECT @priority AS priority, @time_stamp AS time_stamp;这里用了一个反向截取技巧:先REVERSE把字符串倒过来,CHARINDEX('q',...)找到q在倒序串里的位置,再SUBSTRING取前面的部分,最后再REVERSE回来。这样不管q后面有多长,都能稳定拿到时间戳。如果直接正向写,q的位置是变的,长度不好固定。
再看一个按分隔符拆分的例子,把100/200拆成两段:
SELECT SUBSTRING('100/200', 0, CHARINDEX('/', '100/200')); -- 100 SELECT SUBSTRING('100/200', CHARINDEX('/', '100/200') + 1, 5); -- 200注意第一个SUBSTRING的起始位置写的是 0,SQL Server 里起始位置小于 1 时会自动按 1 处理,所以结果和写 1 一样。但为了可读性,建议统一从 1 开始写。
4.4 游标 + 字符串截取组合:逐行拆解并回写
把上面两块拼起来,遍历trans_queue,把sort拆成优先级和时间戳后插入到一张结果表。先建结果表:
IF OBJECT_ID('dbo.trans_queue_parsed','U') IS NOT NULL DROP TABLE dbo.trans_queue_parsed; CREATE TABLE dbo.trans_queue_parsed( queue_code INT, priority VARCHAR(1), time_stamp NVARCHAR(200) );然后游标遍历:
DECLARE @qcode INT; DECLARE @sort NVARCHAR(200); DECLARE @priority VARCHAR(1); DECLARE @time_stamp NVARCHAR(200); DECLARE cur_queue CURSOR FOR SELECT queue_code, sort FROM dbo.trans_queue; OPEN cur_queue; FETCH NEXT FROM cur_queue INTO @qcode, @sort; WHILE @@FETCH_STATUS = 0 BEGIN SET @priority = LEFT(@sort,1); SET @time_stamp = REVERSE(SUBSTRING(REVERSE(@sort),1,CHARINDEX('q',REVERSE(@sort)) - 1)); INSERT INTO dbo.trans_queue_parsed(queue_code,priority,time_stamp) VALUES (@qcode,@priority,@time_stamp); FETCH NEXT FROM cur_queue INTO @qcode, @sort; END CLOSE cur_queue; DEALLOCATE cur_queue; GO跑完SELECT * FROM dbo.trans_queue_parsed应该看到17 | 3 | 20201010090559。这就是游标加字符串截取的完整闭环。
5. 验证请求与成功结果:行数比对 + 结果集校验
脚本跑完不能只看“执行成功”,要做两层验证。
第一层是行数比对。执行前记下Category是 0 行,执行后应该是 30 行;trans_queue是 1 行,trans_queue_parsed执行后也应该是 1 行。用下面这句一次性看:
SELECT 'Category' AS tbl, COUNT(*) AS cnt FROM dbo.Category UNION ALL SELECT 'trans_queue', COUNT(*) FROM dbo.trans_queue UNION ALL SELECT 'trans_queue_parsed', COUNT(*) FROM dbo.trans_queue_parsed;第二层是结果集校验,重点看截取出来的值对不对:
SELECT queue_code, priority, time_stamp, LEN(time_stamp) AS ts_len FROM dbo.trans_queue_parsed;time_stamp应该是 14 位(20201010090559),priority是 1 位。如果长度不对,说明CHARINDEX或SUBSTRING的参数有问题,回到 4.3 节检查。
再校验一下游标更新的结果:
SELECT Id, UserName FROM dbo.[User] WHERE UserName LIKE '%tian%';应该看到tian0011、tian0022这种形式,Id 被追加到了用户名末尾。如果用户名没变,多半是WHERE Id = @id没匹配上,或者游标FETCH没取到值。
6. 本篇常见错排查
报错一:Must declare the scalar variable "@num"。这是把@num用在了没声明的作用域里。比如在游标循环里PRINT DATEADD(day,@num,@dt),但@num只在另一个批次声明过。GO会切断变量作用域,跨GO的变量不共享。解决办法是把相关逻辑放在同一个批次里,或者重新声明。
报错二:游标死循环。十有八九是FETCH NEXT只写了一次,或者@@FETCH_STATUS判断写成了= 1。正确写法是循环外一次、循环内一次,判断用= 0。
报错三:SUBSTRING返回空或乱码。检查CHARINDEX的返回值。如果分隔符不存在,CHARINDEX返回 0,SUBSTRING(@str, 0, ...)会从开头取,结果就不是你想要的。稳妥做法是先判断:
IF CHARINDEX('q', @sort) > 0 BEGIN -- 执行截取 END报错四:User表名报语法错误。保留字问题,统一加方括号[User]。
报错五:插入后行数不对。检查WHILE的边界条件。@num <= 30从 1 开始会插入 30 行,如果写成@num < 30就只有 29 行。另外确认没有在循环里误加BREAK或CONTINUE。
性能提醒:游标是逐行处理,数据量上万行就会明显变慢。如果只是简单的批量插入或更新,优先用INSERT INTO ... SELECT这种集合操作。比如把一张表的数据整体搬到另一张表:
INSERT INTO dbo.trans_queue_parsed(queue_code,priority,time_stamp) SELECT queue_code, LEFT(sort,1), REVERSE(SUBSTRING(REVERSE(sort),1,CHARINDEX('q',REVERSE(sort)) - 1)) FROM dbo.trans_queue;一行顶游标几十行,而且快得多。游标只在“每行处理逻辑不同、需要条件分支”时才值得用。
7. 把脚本跑通之后:用工具链管理你的 SQL 与模型调用
上面这套WHILE+ 游标 + 字符串截取的脚本,本质上是数据加工流水线的一环。如果你在做的是 AI 应用相关的数据准备,比如给模型调用日志做清洗、把拼接字段拆成结构化参数,那这类 SQL 加工会经常出现。跑通之后,下一步往往是把加工好的数据接到模型接口上做验证。
这时候可以用 TaoToken 的模型对话能力快速试一下拆出来的字段是否符合预期,地址是 https://taotoken.net/api ,对话入口在 https://taotoken.net/api?utm_source=taotoken_aicg_blog_end&utm_medium=csdn&utm_campaign=rewrite&utm_content=model-chat 。如果你是要长期跑编码任务或者搭 Agent 做批量处理,Coding Plan 会更合适:https://taotoken.net/api?utm_source=taotoken_aicg_blog_end&utm_medium=csdn&utm_campaign=rewrite&utm_content=coding-plan 。接入前先在控制台把 API Key 建好,https://taotoken.net/api?utm_source=taotoken_aicg_blog_end&utm_medium=csdn&utm_campaign=rewrite&utm_content=console ,密钥管理页在 https://taotoken.net/api?utm_source=taotoken_aicg_blog_end&utm_medium=csdn&utm_campaign=rewrite&utm_content=api-keys ,接入文档看 https://taotoken.net/api?utm_source=taotoken_aicg_blog_end&utm_medium=csdn&utm_campaign=rewrite&utm_content=doc 。官网入口是 https://taotoken.net/?utm_source=taotoken_aicg_blog_end&utm_medium=csdn&utm_campaign=rewrite&utm_content= 。
最后留一个实用习惯:每次写完循环脚本,先在测试库跑一遍,用SELECT COUNT(*)比对执行前后行数,再抽查几条截取结果。这个动作花不了一分钟,但能省掉很多回头排查的时间。