一:背景
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
置信很多敌人用这个办法的场景大多在于剔除尾部的空格,哈哈,其实它还有一个暗藏性能,不仅能够剔除空格,还能够剔除任意多个指定的字符,这就????????了,不信的话能够看看 TrimEnd 办法签名即可:
public sealed class String { public string TrimEnd(params char[] trimChars) { if (trimChars == null || trimChars.Length == 0) { return TrimHelper(1); } return TrimHelper(trimChars, 1); } }
能够看到 trimChars 是一个字符数组,你能够灌入你想剔除的任意多个字符,有了这个思维,我能够将 keyword 转成 char[] 再灌入到 TrimEnd 即可,代码如下:
static void Main(string[] args) { var sql = "select 1 union all select 2 union all select 3 union all"; var keyword = "union all"; var query = sql.TrimEnd(keyword.ToArray()); Console.WriteLine(query); }
能够看出已胜利剔除,此时我很好奇底层到底是怎么实现的,源码如下:
private string TrimHelper(char[] trimChars, int trimType) { int num = Length - 1; int i = 0; if (trimType != 1) { for (i = 0; i < Length; i++) { int num2 = 0; char c = this[i]; for (num2 = 0; num2 < trimChars.Length && trimChars[num2] != c; num2++) { } if (num2 == trimChars.Length) { break; } } } if (trimType != 0) { for (num = Length - 1; num >= i; num--) { int num3 = 0; char c2 = this[num]; for (num3 = 0; num3 < trimChars.Length && trimChars[num3] != c2; num3++) { } if (num3 == trimChars.Length) { break; } } } return CreateTrimmedString(i, num); } private string CreateTrimmedString(int start, int end) { int num = end - start + 1; if (num == Length) { return this; } if (num == 0) { return Empty; } return InternalSubString(start, num); }
从源码中能够看出,它的代码逻辑是从后往前筛查,始终到遇见的字符不在 char[] 中为止,这里就有一个潜在的坑,大家要留神了,比方上面sql的后果:
static void Main(string[] args) { var sql = "select * from english_l union all"; var keyword = "union all"; var query = sql.TrimEnd(keyword.ToArray()); Console.WriteLine(query); }
能够看到: english_l
中的 l 也被意外剔除,这就不是我想看到的后果。
此坑须要感激公号昵称为:神仙氏族
的情谊揭示 ????
四:总结
五种形式各有利弊,不论是简略粗犷的,基于性能的,灵便奇妙的,都能达到最终的目标,临时就想到这5种,脑仁曾经疼了????????????, 更多好玩的写法,欢送大家留言探讨哈!
更多高质量干货:参见我的 GitHub: dotnetfly