【GaussDB】在UTF8的库上按GBK的字节数截取字符串
【GaussDB】在UTF8的库上按GBK的字节数截取字符串
背景
在国产数据库迁移过程中,可能会有这样一个场景:源库ORACLE用的是ZHS16GBK字符集,目标库GaussDB用的是UTF8字符集,如果源库是VARCHAR2(10),且字段内容里包含汉字,那么目标库字段需要增长,假设按照常规的一个汉字在GBK占2字节,到UTF8占3个字节,也就是按原长度的1.5倍,即VARCHAR2(15)。
然而如果迁移后,国产库里产生的新数据需要回流到ORACLE,可能会存在ORACLE中长度不足的问题,比如新数据为 “咕咕嘎嘎123” ,在UTF8是15个字节,在ZHS16GBK是11个字节。
所以为了确保数据能回去,只能进行截断(别说为什么前端不控制,总会有各种各样的原因),然而如果用在UTF8库上按字节数进行截断,可能又会截多了,比如源库VARCHAR2(10) “咕咕嘎嘎啊” ,在UTF8上substrb截10个字节,会变成"咕咕嘎"+一个字节,截断尾部乱码后就只剩三个汉字了,少了两个字。
由此产生了一个需求:
在GaussDB的UTF8库里按照GBK编码的字节数实现一个类似substrb的自定义函数,输入字符串、开始位置、保留字符数,返回截取后的字符串,以确保尽可能保留ORACLE中的数据。
我的方案
createorreplacefunctionuf_gbk_substrb(input_strvarchar,pos number,len number)returnsvarcharRETURNSNULLONNULLinput IMMUTABLEas$$declarel_tmp bytea;beginiflength(input_str)=0thenreturnnull;endif;l_tmp:=convert_to(input_str::text,'GBK');foriin1..3loopbeginreturnconvert_from(substr(l_tmp,greatest(pos-1,0),least(len-i+1,length(l_tmp))),'GBK');exceptionwhenothersthennull;end;endloop;RAISE EXCEPTION'invalid byte sequence for encoding "GBK": %',l_tmp::text;end;$$languageplpgsql;基本用法和普通的subtrb一样,不过我没去写反向截取(即负数入参)的逻辑。下面就是按GBK10个字节去截的效果。
gaussdb=>selectuf_gbk_substrb('咕咕嘎嘎123',1,10);uf_gbk_substrb----------------咕咕嘎嘎12(1row)gaussdb=>selectuf_gbk_substrb('123咕咕嘎嘎',1,10);uf_gbk_substrb----------------123咕咕嘎(1row)gaussdb=>selectuf_gbk_substrb('咕咕嘎嘎啊',1,10);uf_gbk_substrb----------------咕咕嘎嘎啊(1row)注意到第2个测试用例,截完后GBK占9个字节,并没有填满10个字节,是因为PG系对不完整字符的支持都不好,所以我只能尽可能保留UTF8能识别的字符,尾部的半个汉字需要截掉。
接下来逐行解释一下这个函数的设计思路
NULLONNULLinput如果输入参数为空,则直接返回空
iflength(input_str)=0thenreturnnull;endif;如果输入字符串长度为0,则直接返回空(有人可能认为这个已经有了null on null input就够了,但其实null和0长度字符串是不一样的,哪怕是在A兼容模式下)
l_tmp:=convert_to(input_str::text,'GBK');使用临时变量暂存输入字符串的GBK二进制数据
foriin1..3loopbeginreturnconvert_from(substr(l_tmp,greatest(pos-1,0),least(len-i+1,length(l_tmp))),'GBK');exceptionwhenothersthennull;end;endloop;对这个临时二进制变量按字节截取(注意当入参是bytea类型时,substr是按字节截取,且最小开始位置是0,而不是1)。
substr截完后还是二进制,再转回字符串。此时如果尾部存在不完整字符,则可能报错,因此尝试再少截一个字符进行重试,最多3次。
RAISE EXCEPTION'invalid byte sequence for encoding "GBK": %',l_tmp::text;三次都转换报错,则函数报错无效的字节序,并打印对应的GBK二进制数据。
这函数我纯手搓的,因为我经常与字符集打交道,这个逻辑对我来说算是挺简单的了。
AI会怎么做?
然后我突发奇想,如果把这个需求丢给AI ,AI会怎么做?
于是我把本文开头的背景说明直接给到了mimo-v2.5-pro,ta生成了一个逻辑比我的方案复杂得多的函数(AI生成的代码里带了parallel,GaussDB不支持,我手动删掉了)
-- ============================================================-- gbk_substrb: 在 UTF8 数据库中按 GBK 编码字节数实现 substrb-- 用途:GaussDB(UTF8) → Oracle(ZHS16GBK) 数据回流时的字段截断---- 与 Oracle SUBSTRB 行为对齐:-- start_pos=0 → 等效于 1-- start_pos<0 → 从字符串末尾倒数-- 截取跨越某字符的 GBK 字节边界时,该字符被跳过(不截断半个字符)---- 参数:-- src_str - 输入字符串(UTF8)-- start_pos - 起始位置(GBK 字节数,从 1 开始)-- sub_length - 截取长度(GBK 字节数,可选,默认取到末尾)-- ============================================================CREATEORREPLACEFUNCTIONgbk_substrb(src_strTEXT,start_posINTEGER,sub_lengthINTEGERDEFAULTNULL)RETURNSTEXTLANGUAGEplpgsql IMMUTABLEAS$$DECLAREv_lenINTEGER;v_charTEXT;v_cpINTEGER;v_blenINTEGER;v_eff_startINTEGER;v_real_posINTEGER:=0;-- 当前 GBK 字节位置(从 1 开始)v_resultTEXT:='';v_beginINTEGER:=0;-- 截取开始的 GBK 字节位置v_endINTEGER;-- 截取结束的 GBK 字节位置v_startedBOOLEAN:=FALSE;BEGINIFsrc_strISNULLORstart_posISNULLTHENRETURNNULL;ENDIF;v_len :=char_length(src_str);IFv_len=0THENRETURN'';ENDIF;-- Oracle 行为:start_pos=0 等效于 1IFstart_pos=0THENv_eff_start :=1;ELSEv_eff_start :=start_pos;ENDIF;-- 计算截取结束位置(GBK 字节数)IFsub_lengthISNULLTHENv_end :=2147483647;-- 取到末尾ELSEv_end :=v_eff_start+sub_length-1;ENDIF;-- 负数 start_pos:从末尾倒数IFv_eff_start<0THEN-- 先算出整个字符串的 GBK 字节总长DECLAREv_totalINTEGER:=0;v_chTEXT;v_cINTEGER;BEGINFORiIN1..v_lenLOOPv_ch :=substr(src_str,i,1);v_c :=ascii(v_ch);IFv_c<=127THENv_total :=v_total+1;ELSIF v_cBETWEEN192AND223THENv_total :=v_total+1;ELSEv_total :=v_total+2;ENDIF;ENDLOOP;v_eff_start :=v_total+v_eff_start+1;v_end :=v_eff_start+COALESCE(sub_length,1)-1;END;ENDIF;IFv_eff_start<=0THENRETURNNULL;ENDIF;-- 遍历每个字符,按 GBK 字节数做截取FORiIN1..v_lenLOOPv_char :=substr(src_str,i,1);v_cp :=ascii(v_char);-- 计算该字符在 GBK 编码中的字节数IFv_cp<=127THEN-- ASCII(含英文、数字、符号)v_blen :=1;ELSIF v_cpBETWEEN192AND223THEN-- U+0080 ~ U+07FF:Latin 扩展、希腊语、阿拉伯语等-- GBK 通常 1 字节(直通映射或 ?)v_blen :=1;ELSE-- U+0800+:CJK 等,在 GBK 中占 2 字节v_blen :=2;ENDIF;v_real_pos :=v_real_pos+v_blen;-- 定位截取起始点IFNOTv_startedANDv_real_pos>=v_eff_startTHENv_started :=TRUE;v_begin :=v_real_pos-v_blen+1;ENDIF;-- 截取结束(字符跨越 end 边界则跳过)IFv_startedANDv_real_pos>v_endTHENEXIT;ENDIF;IFv_startedTHENv_result :=v_result||v_char;ENDIF;ENDLOOP;RETURNv_result;END;$$;COMMENTONFUNCTIONgbk_substrb(TEXT,INTEGER,INTEGER)IS'在 UTF8 库中按 GBK 字节数截取字符串,用于 GaussDB→Oracle(ZHS16GBK) 数据回流截断';大概的逻辑就是从前到后逐字符遍历一个一个字符截取,根据每一个字符的ascii码大小范围判断它是1个字节还是多个字节的(多字节都按2字节算),视为GBK字节长度,然后把一个一个字符拼起来,拼的同时GBK字节数也累加,直到计数器达到目标长度就跳出循环,返回拼接的字符串。
方案对比
AI写的更像一个真正的程序,虽然代码长逻辑复杂,但这个处理逻辑拿到任何开发语言里都是可以用的,不像我写的那个依赖数据库自己的转码函数去判断能不能转码成功。
先造一些用例对比下两个方案返回结果是否一致
selectgbk_substrb('咕咕嘎嘎123',1,10)=uf_gbk_substrb('咕咕嘎嘎123',1,10)b;selectgbk_substrb('123咕咕嘎嘎',1,10)=uf_gbk_substrb('123咕咕嘎嘎',1,10)b;selectgbk_substrb('咕咕嘎嘎啊',1,10)=uf_gbk_substrb('咕咕嘎嘎啊',1,10)b;这三个例子验证都为true
然后对比一下性能
createtablet_text(idint,atext);insertintot_textselectid,casemod(id,3)when0then'咕咕嘎嘎123'when1then'123咕咕嘎嘎'when2then'咕咕嘎嘎啊'endfrompg_catalog.generate_series(1,10000)id;insertintot_textselectid+10000,casemod(id,2)when0then'null'elseencode('\x','hex')endfrompg_catalog.generate_series(1,10000)id;explainanalyzeselectid,gbk_substrb(a,1,10)fromt_text;--Total runtime: 611.122 msexplainanalyzeselectid,uf_gbk_substrb(a,1,10)fromt_text;--Total runtime: 1918.647 ms我的方案耗时是AI的三倍还多!
AI的字符遍历可能对于数据越长,性能就越差,于是我又弄了个用例
createtablet_text2(idint,atext);insertintot_text2selectid,casemod(id,3)when0thenrepeat('咕咕嘎嘎123',200)when1thenrepeat('123咕咕嘎嘎',200)when2thenrepeat('咕咕嘎嘎啊',200)endfrompg_catalog.generate_series(1,10000)id;explainanalyzeselectid,gbk_substrb(a,1,10)fromt_text2;--Total runtime: 1014.254 msexplainanalyzeselectid,uf_gbk_substrb(a,1,10)fromt_text2;--Total runtime: 1849.356 ms果然输入字符串的长度对性能有影响,但是尽管加长到3K字节,AI的方案性能依旧更好。现在也就一万行,如果真实数据上千万行甚至有多个字段都需要截取,中间的差异时间就会更大了。
AI的额外启发
AI其实还同时生成了另一个计算gbk字节数的函数
-- ============================================================-- gbk_bytelen: 计算字符串在 GBK 编码下的字节长度-- 用途:截断前预判是否需要截断-- ============================================================CREATEORREPLACEFUNCTIONgbk_bytelen(src_strTEXT)RETURNSINTEGERLANGUAGEplpgsql IMMUTABLE PARALLEL SAFEAS$$DECLAREv_lenINTEGER;v_totalINTEGER:=0;v_charTEXT;v_cpINTEGER;BEGINIFsrc_strISNULLTHENRETURNNULL;ENDIF;v_len :=char_length(src_str);FORiIN1..v_lenLOOPv_char :=substr(src_str,i,1);v_cp :=ascii(v_char);IFv_cp<=127THENv_total :=v_total+1;ELSIF v_cpBETWEEN192AND223THENv_total :=v_total+1;ELSEv_total :=v_total+2;ENDIF;ENDLOOP;RETURNv_total;END;$$;COMMENTONFUNCTIONgbk_bytelen(TEXT)IS'计算字符串在 GBK 编码下的字节数(ASCII=1, CJK=2)';这个可以放到数据入库的前置校验逻辑中去,这样库里的数据就不需要再截断了。
另外,我也尝试了清空所有上下文,只给AI一句最简单的指令,看看ta能生成什么
“写一个GaussDB的自定义函数,在UTF8的库上按GBK的字节数截取字符串”
然后AI给了我3个方案
-- 方法1:PL/pgSQL 函数(推荐,兼容性好)CREATEORREPLACEFUNCTIONgbk_substr(p_strTEXT,p_max_bytesINTEGER)RETURNSTEXTLANGUAGEplpgsql IMMUTABLEAS$$DECLAREv_resultTEXT:='';v_current_bytesINTEGER:=0;v_charTEXT;v_char_bytesINTEGER;v_lenINTEGER;v_iINTEGER;BEGIN-- 参数校验IFp_strISNULLORp_max_bytes<=0THENRETURN'';ENDIF;v_len :=length(p_str);FORv_iIN1..v_lenLOOPv_char :=substr(p_str,v_i,1);-- 计算该字符在 GBK 编码下的字节数-- ASCII 字符 (0-127) 在 GBK 中占 1 字节-- 中文字符在 GBK 中占 2 字节IFascii(v_char)<=127THENv_char_bytes :=1;ELSE-- 非 ASCII 字符(中文等)在 GBK 中占 2 字节v_char_bytes :=2;ENDIF;-- 检查是否超出限制IFv_current_bytes+v_char_bytes>p_max_bytesTHENEXIT;ENDIF;v_current_bytes :=v_current_bytes+v_char_bytes;v_result :=v_result||v_char;ENDLOOP;RETURNv_result;END;$$;COMMENTONFUNCTIONgbk_substr(TEXT,INTEGER)IS'在 UTF8 数据库上按 GBK 字节数截取字符串。 参数: p_str - 输入字符串(UTF8 编码) p_max_bytes - 最大 GBK 字节数 返回:截取后的字符串,确保转为 GBK 后字节数不超过 p_max_bytes 示例:SELECT gbk_substr('你好世界Hello', 8); -- 返回 "你好" (2*2=4字节) 或 "你好H" (4+1=5字节)';-- 测试用例-- SELECT gbk_substr('你好世界Hello', 8); -- 预期:你好He (2+2+2+1+1=8字节)-- SELECT gbk_substr('你好世界', 5); -- 预期:你好 (4字节,再加"世"会超)-- SELECT gbk_substr('Hello', 3); -- 预期:Hel-- SELECT gbk_substr(NULL, 10); -- 预期:空字符串-- SELECT gbk_substr('abc', 0); -- 预期:空字符串-- 方法2:SQL 函数(更简洁,但依赖 ascii() 函数可用性)-- 如果 GaussDB 支持 ascii() 函数,可以用这个更简洁的版本CREATEORREPLACEFUNCTIONgbk_substr_sql(p_strTEXT,p_max_bytesINTEGER)RETURNSTEXTLANGUAGEsqlIMMUTABLEAS$$WITHcharsAS(SELECTc,CASEWHENascii(c)<=127THEN1ELSE2ENDASbyte_len,ROW_NUMBER()OVER()ASrnFROMunnest(string_to_array(p_str,NULL))ASc),cumulativeAS(SELECTc,byte_len,SUM(byte_len)OVER(ORDERBYrn)AScumulative_bytesFROMchars)SELECTCOALESCE(string_agg(c,ORDERBYcumulative_bytes),'')FROMcumulativeWHEREcumulative_bytes<=p_max_bytes;$$;COMMENTONFUNCTIONgbk_substr_sql(TEXT,INTEGER)IS'SQL 版本的 GBK 字节截取函数(更简洁)';-- 方法3:如果需要处理 GBK 编码中的特殊字符(如一些罕用字可能是 4 字节)-- 使用 convert_to 函数进行精确计算(需要 PostgreSQL 9.x+ 或 GaussDB 支持)CREATEORREPLACEFUNCTIONgbk_substr_exact(p_strTEXT,p_max_bytesINTEGER)RETURNSTEXTLANGUAGEplpgsql IMMUTABLEAS$$DECLAREv_resultTEXT:='';v_current_bytesINTEGER:=0;v_charTEXT;v_char_bytesINTEGER;v_lenINTEGER;v_iINTEGER;v_gbk_bytes BYTEA;BEGINIFp_strISNULLORp_max_bytes<=0THENRETURN'';ENDIF;v_len :=length(p_str);FORv_iIN1..v_lenLOOPv_char :=substr(p_str,v_i,1);-- 使用 convert_to 精确计算 GBK 字节数BEGINv_gbk_bytes :=convert_to(v_char,'GBK');v_char_bytes :=length(v_gbk_bytes);EXCEPTIONWHENOTHERSTHEN-- 转换失败时按 2 字节估算v_char_bytes :=2;END;IFv_current_bytes+v_char_bytes>p_max_bytesTHENEXIT;ENDIF;v_current_bytes :=v_current_bytes+v_char_bytes;v_result :=v_result||v_char;ENDLOOP;RETURNv_result;END;$$;COMMENTONFUNCTIONgbk_substr_exact(TEXT,INTEGER)IS'精确版本:使用 convert_to 计算 GBK 字节数(性能较低但最准确)';三个方案都是遍历字符,然后拼接,但区别是:
- 第一个方案和AI前面的方案原理一样,循环截取,然后按ascii大小判断GBK长度
- 第二个方案是SQL语言函数,将字符串按每个字符切分转成一个数组,然后再按ascii大小计算字节长度
- 第三个方案考虑了生僻字,使用了内置函数convert_to来精确识别字节数
总结
不是为自己找补,我的方案和AI第一版的方案相比,虽然AI的性能更好,但是如果需要把方案给到一个对字符集不是很理解的开发人员,考虑到函数可能还存在一些边界场景的BUG(比如生僻字),我可能仍然还会提供我自己的方案。因为AI的方案代码量稍大,有些特殊场景也没有考虑,我的方案里是有数据库自己的内置函数来进行未知情况的兜底。
不过,如果能有真实大量的数据测试一下AI这个函数,识别到可能的边界场景,让AI来完善这个函数的处理,并且自动化进行性能测试分析和调整,这可能是更符合AI-AGENT时代的做法了。
另一方面,不能忽视的是,虽然AI这次写得挺好,但前提是我能准确地描述这个问题,即本文中的背景说明,这一段并非我收到的原始描述,而是我根据开发人员的无法准确描述的未知担忧逐步拆解分析出来的。AI的使用者依旧需要具有清晰的逻辑思考能力以及语言表达能力,才能让AI完成想让ta完成的事。
但是,描述得越详细,有时候也可能会扼杀AI的创造力,比如我第二次给AI提的一句话需求的回答中,其实AI反而考虑到了前面长提示词的回答没有考虑到的生僻字场景。
最后我问了AI一句
我:
你能搜一下网上有没有类似的文章或者案例么
AI:
搜了一圈,没有找到直接对标的案例。搜索结果主要分两类:
同编码截取:Oracle/GaussDB 的 SUBSTRB 按当前库编码的字节数截取,比如 UTF8 库按 UTF8 字节数、GBK 库按 GBK 字节数
乱码处理:SUBSTRING 截取中文时因编码不一致导致的乱码问题
你这个场景比较特殊:UTF8 库存储,但要按 GBK 字节数限制输出。典型场景是对接老系统(接口字段长度按 GBK 算),或者写文件时控制 GBK 编码的行宽。所以前面写的函数确实是个"空白地带",市面上没有现成方案。你这个需求挺实用的,要是写成博客应该有人看 👀
- 本文作者:DarkAthena
- 本文链接:https://www.darkathena.top/archives/gaussdb-utf8-gbk-substrb
- 版权声明:本博客所有文章除特别声明外,均采用 CC BY-NC-SA 3.0 许可协议。转载请注明出处