项目里面有一个需求,要对sql进行简单的语法分析
为了避免sql里面的字符串和注释对语法分析做干扰,我写了一个java函数,对sql进行修剪,删除里面字符串和注释,用空格代替
周末闲着没事,我用go重新实现了这个功能,感觉应该会有后来人可以用上
说明:
sql里面的注释有两种单行注释和多行注释,其中单行注释以--开头,以\n结尾,多行注释以/开头,以/结尾
sql字符串是以'开头,'结尾,但特别的地方是连续两个单引号是代表一个单引号而不是字符串结束标志
关键函数如下:
` /** 将字节数组里面注释和字符串,用空格替换 rangeBeg和rangeEnd是数组元素起始位置 左闭右开 */ func TrimSqlByteArray(sql []byte, rangeBeg int, rangeEnd int) []byte { sqlLength := rangeEnd - rangeBeg - 1; //删除注释或者字符串后 用空格填充 必免因删除导致粘连改变sql语义 const chPad = ' ' //结果切片,预分配空间为入参sql长度一半 result := make([] byte, 0, sqlLength / 2) //本字符类型 var charType int = NORMAL; for i := rangeBeg; i
//本字符类型 预设为普通字符 charType = NORMAL ch := sql[i] //下一个字符 var chNext byte; chNext = getCharSafe(sql, rangeEnd, i + 1) //非有效sql内容结束位置 endPos := 0 if ch == '-' && chNext == '-' { //单行注释 charType = LINE //下标移到非有效字符的最后 endPos = seekToNext(sql, i + 2, rangeEnd, charType) } else if ch == '/' && chNext == '*' { //多行注释 charType = MULTI //下标移到非有效字符的最后 endPos = seekToNext(sql, i + 2, rangeEnd, charType) } else if ch == '\'' { //字符串 charType = STRING //下标移到非有效字符的最后 endPos = seekToNext(sql, i + 1, rangeEnd, charType) } //如果字符是非有效字符 则用空格代替 否则保持原样 if charType == NORMAL { result = append(result, ch) } else { result = append(result, chPad) i = endPos - 1 }
} return result; } /**
获取字符串或者注释的右边界位置(不包含)
rangeEnd是数组边界 */ func seekToNext(sql []byte, begPos int, rangeEnd int, charType int) int { result := begPos; switch charType { case MULTI: for ; result
if ch == '*' && chNext == '/' { result = result + 1; break; } } break
case LINE: for ; result
if ch == '\n' { break; } } break
case STRING: for ; result
//sql字符串里面连续的单引号被认为是' 则不是字符串结束标志 if ch == '\'' && chNext == '\'' { result = result + 1; continue; } else if ch == '\'' { break; } } break
default: break; } result++; return result; }
完整代码及单元测试已上传 https://github.com/kingstarer/kingstarer.git