雜七雜八——使用LINQ檢索重複值

雜七雜八——使用LINQ檢索重複值

 

日常工作的時候,我們經常遇到需要檢索一組數據中是否有重複值的情況,再根據具體情況進行相應的操作。如果不使用LINQ,那我們就得使用一層一層的foreach循環來做,不但麻煩、工作效率低(因爲你得繞啊~~繞啊~~~),而且容易出bug。使用LINQ就方便得多。

 

舉個我前天在工作中遇到的例子吧。當時是這樣的——我需要查看一下用戶新插入的值是否已經存在於數據庫當中,如果有,就提醒客戶不能插入這個值(實際上這個值是PK)。因爲在進入頁面的時候,我已經把目前數據庫裏所有的值都讀出來、放在一個DataTable裏了,所以在用戶保存的時候,我沒必要連接數據庫進行查詢,只需要檢索當前這個DataTable就OK了。

 

假設這個DataTable有3列,分別是ID、Name、Age,ID是PK。使用foreach的笨辦法,是這樣:

 

  1. foreach (DataRow r1 in table.Rows)
  2. {
  3.     foreach (DataRow r2 in table.Rows)
  4.     {
  5.         if (r2 != r1 && r2["ID"].ToString() == r1["ID"].ToString())
  6.         {
  7.             Console.WriteLine("Warning!");
  8.             return false// 檢驗失敗
  9.         }
  10.     }
  11. }

如果寫成這樣,需要注意兩點:

  • 必需要有r2 != r1這個條件,不然當一個DataRow“自己遇到自己”的時候,無論如何都會return false的
  • 必需把r1["ID"]和r2["ID"]轉換成string(如果你確定它是int,那轉成int也行)再進行比較,不然r1["ID"]和r2["ID"]是兩個object,調用==操作符,比較的是這兩個對象是否是同一個對象——當然不是!所以,永遠也不會return false

上面這種笨辦法適用於兩種情況:

  1. 需要比較簡單、只要有重複值立刻撤退的情況
  2. 公司把代碼行數與程序工資掛鉤的情況

使用LINQ可以更簡單地完成上面的任務,而且還可以衍生出很多附加功能:

 

我們把需求稍微改動一下,改成檢驗Age有沒有重複的,如果有、每個值有幾個重複。如果用foreach循環,那我們就要在循環內部加上一個Dictionay,以Age爲Key,並對每個Key進行計數了。

 

使用LINQ中的GroupBy操作,就能輕鬆解決這個問題。源碼如下:

 

  1. // 水之真諦
  2. // http://blog.csdn.net/FantasiaX
  3. using System;
  4. using System.Collections.Generic;
  5. using System.Linq;
  6. using System.Text;
  7. using System.Data;
  8. namespace ConsoleApplication1
  9. {
  10.     class Program
  11.     {
  12.         static void Main(string[] args)
  13.         {
  14.             DataTable table = new DataTable();
  15.             table.Columns.Add(new DataColumn("ID"typeof(int)));
  16.             table.Columns.Add(new DataColumn("Name"typeof(string)));
  17.             table.Columns.Add(new DataColumn("Age"typeof(int)));
  18.             int[] ids = new int[] { 1, 2, 3, 4, 5, 6 };
  19.             string[] names = new string[] { "Tim""Yan""Xiao Chen""Miao""Big Ma""Little Ma" };
  20.             int[] ages = new int[] { 28, 24, 28, 22, 27, 22 };
  21.             for (int i = 0; i < 6; i++)
  22.             {
  23.                 DataRow row = table.NewRow();
  24.                 row["ID"] = ids[i];
  25.                 row["Name"] = names[i];
  26.                 row["Age"] = ages[i];
  27.                 table.Rows.Add(row);
  28.             }
  29.             // 使用foreach
  30.             Dictionary<intint> dic = new Dictionary<intint>();
  31.             foreach (DataRow row in table.Rows)
  32.             {
  33.                 if (!dic.Keys.Contains(Convert.ToInt32(row["Age"])))
  34.                 {
  35.                     dic.Add(Convert.ToInt32(row["Age"]), 1);
  36.                 }
  37.                 else
  38.                 {
  39.                     dic[Convert.ToInt32(row["Age"])]++;
  40.                 }
  41.             }
  42.             foreach (var item in dic)
  43.             {
  44.                 Console.WriteLine("{0}, {1}", item.Key, item.Value);
  45.             }
  46.             Console.WriteLine("=========================================");
  47.             // 使用LINQ (隱式數據類型)
  48.             var ageGroups = table.Rows.Cast<DataRow>().GroupBy(row => Convert.ToInt32(row["Age"]));
  49.             foreach (var group in ageGroups)
  50.             {
  51.                 Console.WriteLine("{0}, {1}", group.Key.ToString(), group.Count().ToString());
  52.             }
  53.             Console.WriteLine("=========================================");
  54.             // 使用LINQ (顯式數據類型)
  55.             IEnumerable<IGrouping<int, DataRow>> ageGroups2 = table.Rows.Cast<DataRow>().GroupBy(row => Convert.ToInt32(row["Age"]));
  56.             foreach (IGrouping<int, DataRow> group in ageGroups2)
  57.             {
  58.                 Console.WriteLine("{0}, {1}", group.Key.ToString(), group.Count().ToString());
  59.             }
  60.             Console.WriteLine("=========================================");
  61.             // 很酷的寫法
  62.             var ageGroups3 = from row in table.Rows.Cast<DataRow>() group row by Convert.ToInt32(row["Age"]) into resultCollection select resultCollection;
  63.             foreach (var group in ageGroups3)
  64.             {
  65.                 Console.WriteLine("{0}, {1}", group.Key.ToString(), group.Count().ToString());
  66.             }
  67.             Console.WriteLine("=========================================");
  68.             // 比較BT的寫法
  69.             foreach (var group in from row in table.Rows.Cast<DataRow>() group row by Convert.ToInt32(row["Age"]) into resultCollection select resultCollection)
  70.             {
  71.                 Console.WriteLine("{0}, {1}", group.Key.ToString(), group.Count().ToString());
  72.             }
  73.         }
  74.     }
  75. }

其中最核心的一句是:var ageGroups = table.Rows.Cast<DataRow>().GroupBy(row => Convert.ToInt32(row["Age"]));

意思是告訴LINQ解析器說“請以Convert.ToInt32(row["Age"])爲Key,對row們進行分組,並把這些組放在名爲ageGroups的集合中去”。爲了不讓程序記憶太多的數據類型,C# 3.0提供了var隱式數據類型語法——程序員可以不知道是什麼類型、但編譯器對類型卻清清楚楚。實際上,GroupBy()操作後的結果是一個以IGrouping<int, DataRow> 爲元素的IEnumerable<IGrouping<int, DataRow>> 集合。

 

因爲GroupBy()操作正好對應有LINQ關鍵字,所以纔有最近兩種改寫。並不是每個LINQ操作都有對應的LINQ關鍵字,不知道C# 4.0會不會有所擴展。

 

BTW,因爲LINQ操作只能應用在可枚舉的集合類型上,而DataTable.Rows集合是個普通集合、不具有可枚舉性,所以需要Cast一下。

 

如果這時候客戶的需求再改成:列出Age相同的人的Name,那麼使用foreach循環的複雜度就有點兒失控了(估計程序員的情緒也比較失控)……而使用LINQ則只需要對每個group進行一下枚舉。

 

LINQ語法適合與以下兩種情況:

  • 懶人,就像我一樣
  • 工資與效率掛鉤

OVER

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章