使用go实现删除sql里面的注释和字符串功能(demo)

wx_DdQG9

项目里面有一个需求，要对sql进行简单的语法分析
为了避免sql里面的字符串和注释对语法分析做干扰，我写了一个java函数，对sql进行修剪，删除里面字符串和注释，用空格代替
周末闲着没事，我用go重新实现了这个功能，感觉应该会有后来人可以用上
说明：
sql里面的注释有两种单行注释和多行注释，其中单行注释以--开头，以\n结尾，多行注释以/开头，以/结尾
sql字符串是以'开头，'结尾，但特别的地方是连续两个单引号是代表一个单引号而不是字符串结束标志
关键函数如下：

1. `
2. /**
4. 将字节数组里面注释和字符串，用空格替换 rangeBeg和rangeEnd是数组元素起始位置 左闭右开
5. */
6. func TrimSqlByteArray(sql []byte, rangeBeg int, rangeEnd int) []byte {
7. sqlLength := rangeEnd - rangeBeg - 1;
8. //删除注释或者字符串后 用空格填充 必免因删除导致粘连改变sql语义
9. const chPad = ' '
11. //结果切片，预分配空间为入参sql长度一半
12. result := make([] byte, 0, sqlLength / 2)
14. //本字符类型
15. var charType int = NORMAL;
16. for i := rangeBeg; i < rangeEnd; i++ {
17. /*
18. *utf8编码不影响判断
19. //跳过非英文字符
20. if sql[i] & 0x80 != 0 {
21. //utf8编码：UTF-8是一种变长字节编码方式。对于某一个字符的UTF-8编码，如果只有一个字节则其最高二进制位为0；
22. //如果是多字节，其第一个字节从最高位开始，连续的二进制位值为1的个数决定了其编码的位数，其余各字节均以10开头。
23. //UTF-8最多可用到6个字节。 这里不考虑异常，因为go的字符串基本都是标准utf8编码
24. i += getPreNotZeroCount(sql[i]) - 1
25. continue;
26. }
27. */

复制代码

1. //本字符类型 预设为普通字符
2. charType = NORMAL
3. ch := sql[i]
5. //下一个字符
6. var chNext byte;
7. chNext = getCharSafe(sql, rangeEnd, i + 1)
9. //非有效sql内容结束位置
10. endPos := 0
12. if ch == '-' && chNext == '-' {
13.         //单行注释
14.         charType = LINE
15.         //下标移到非有效字符的最后
16.         endPos = seekToNext(sql, i + 2, rangeEnd, charType)
17. } else if ch == '/' && chNext == '*' {
18.         //多行注释
19.         charType = MULTI
20.         //下标移到非有效字符的最后
21.         endPos = seekToNext(sql, i + 2, rangeEnd, charType)
22. } else if ch == '\'' {
23.         //字符串
24.         charType = STRING
25.         //下标移到非有效字符的最后
26.         endPos = seekToNext(sql, i + 1, rangeEnd, charType)
27. }
29. //如果字符是非有效字符 则用空格代替 否则保持原样
30. if charType == NORMAL {
31.         result = append(result, ch)
32. } else {
33.         result = append(result, chPad)
34.         i = endPos - 1
35. }

复制代码

1. }
3. return result;
4. }
6. /**

复制代码

获取字符串或者注释的右边界位置（不包含）

1. rangeEnd是数组边界
2. */
3. func seekToNext(sql []byte, begPos int, rangeEnd int, charType int) int {
5. result := begPos;
7. switch charType {
8. case MULTI:
9. for ; result < rangeEnd; result++ {
10. ch := sql[result]
11. chNext := getCharSafe(sql, rangeEnd, result+ 1)

复制代码

1. if ch == '*' && chNext == '/' {
2.                 result = result + 1;
3.                 break;
4.         }
5. }
7. break

复制代码

1. case LINE:
2. for ; result < rangeEnd; result++ {
3. ch := sql[result]

复制代码

1. if ch == '\n' {
2.                 break;
3.         }
4. }
6. break

复制代码

1. case STRING:
2. for ; result < rangeEnd; result++ {
3. ch := sql[result]
4. chNext := getCharSafe(sql, rangeEnd, result + 1)

复制代码

1. //sql字符串里面连续的单引号被认为是' 则不是字符串结束标志
2.         if ch == '\'' && chNext == '\'' {
3.                 result = result + 1;
4.                 continue;
5.         } else if ch == '\'' {
6.                 break;
7.         }
8. }
10. break

复制代码

1. default:
2. break;
3. }
5. result++;
7. return result;
8. }

复制代码

完整代码及单元测试已上传 https://github.com/kingstarer/kingstarer.git
到此这篇关于使用go实现删除sql里面的注释和字符串功能的文章就介绍到这了,更多相关go删除sql注释和字符串内容请搜索咔叽论坛以前的文章或继续浏览下面的相关文章希望大家以后多多支持咔叽论坛！

原文地址：https://www.jb51.net/article/200950.htm