首页 > 编程开发 > Objective-C编程 >
-
如何剔掉 sql 语句中的尾巴,我用 C# 苦思了五种办法
一:背景
1. 讲故事
这几天都在修复bug真的太忙了,期间也遇到了一个挺有趣bug,和大家分享一下,这是一块sql挺复杂的报表相关业务,不知道哪一位大佬在错综复杂的 嵌套 + 平行
if判断中sql拼接在某些UI组合下出问题了,最终的 sql 架构类似这样的。
var sql = "select 1 union all select 2 union all select 3 union all";
这种sql到数据库去肯定是报错的,有些朋友可能想说这还不简单,在相关的 if 判断中不要追加这个 union all
就好了,这确实是一个根治的办法,但现实情况这一块的业务太复杂了,也不太敢改里面的代码,改的没问题还好,改出问题你得兜着走,所以最保险的办法就是怎么去掉 union all
这个大尾巴,所以我干脆思考了一会,想出了如下五种办法。
二:剔除 union all 的五大方式
1. 最原始的 for 循环
最简单的办法就是通过 for 循环搞定,我可以倒序判断最后几个字符是不是关键词 union all
就可以了,如下代码所示:
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
var isall = true;
int i = 0;
for (i = 1; i <= keyword.Length; i++)
{
if (keyword[keyword.Length - i] != sql[sql.Length - i])
{
isall = false;
break;
}
}
if (isall)
{
var query = sql.Substring(0, sql.Length - i + 1);
Console.WriteLine(query);
}
}
从代码中可以看出,只要在倒序的过程中,有一个字符和 keyword 中的不符,那就直接跳出,否则就是全匹配,拿到最后的 i 进行 Substring 截取即可。
2. 使用 Substring 搞定
第一种方式确实可以实现,但实现的并不轻松,毕竟大家都是用 C# 写代码而不是 C,为了这点小功能写了这么多代码,显得太 low 了,所以尽量能用类库的方法就用类库的方法吧,改进措施很简单,可以从 sql 尾部切出 keyword.length
个字符,也就是: start:sql.length - keyword.length
,然后判断一下是否和 keyword 相等即可,代码修改如下:
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
var isSucc = sql.Substring(sql.Length - keyword.Length) == keyword;
if (isSucc)
{
var query = sql.Substring(0, sql.Length - keyword.Length);
Console.WriteLine(query);
}
}
3. 使用 LastIndexOf
第二种方式写出来的代码确实比较简洁,但大家有没有发现一个问题,我为了获取最后的 string 做了两次 substring 操作,也就是说在托管堆中生成了两个 string 对象,那能不能免掉一个 substring 呢?给 gc 减轻一些负担,这就可以用到 LastIndexOf 方法了,代码如下:
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
var index = sql.LastIndexOf(keyword);
if (sql.Length - index == keyword.Length)
{
var query = sql.Substring(0, index);
Console.WriteLine(query);
}
}
思想很简单,就是判断最后出现的 union all
的位置到尾部的距离 是否恰好和 keyword.length 一致,如果是的话那 keyword 就是 sql 的大尾巴,这里的 if 写的有点难懂,其实还可以使用 EndsWith 再优化一下代码:
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
if (sql.EndsWith(keyword))
{
var query = sql.Substring(0, sql.Length - keyword.Length);
Console.WriteLine(query);
}
}
4. 使用 Split 切割
前面几种方式都是在 string 上做文章,要么 substring,要么 LastIndexOf,要么 EndsWith,其实也可以跳出这个定势思维,转换成数组进行处理,用 union all
作为分隔符切割字符串,如果数组的最后一个元素为 string.Empty,那就表明 sql 尾巴就是 keyword, 对吧,代码修改如下:
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
var arr = sql.Split("union all");
if (string.IsNullOrEmpty(arr[arr.Length - 1]))
{
var query = string.Join(keyword, arr.Take(arr.Length - 1));
Console.WriteLine(query);
}
}
5. 使用 TrimEnd
相信很多朋友用这个方法的场景大多在于剔除尾部的空格,哈哈,其实它还有一个隐藏功能,不仅可以剔除空格,还可以剔除任意多个指定的字符,这就