Dbml檔案提取建表TSql-CodeSmith

weixin_33763244發表於2017-11-15
SQLMIT

在昨天一個大學師弟，他問我能不能將LinqToSql檔案轉化為建立表的TSql語句，他是剛開始學習.NET，所以在網上下些示例看，但苦於沒有資料庫。所以就有了這一篇部落格，作為我的Code生成技術的CodeSimth的最後一篇示例。在下一步Code 生成技術將轉到Microsoft的T4模板，Code生成技術目前完成的有CodeDom,CodeSmith模板，高手請不要拍磚，請直接跳過。

     在Linq2Sql的Dbml檔案其實就是一個Xml檔案，記錄著資料庫與生成Linq2SqlCode的資料資訊，所以轉化為TSql沒有什麼說的。我們需要提取其中的資料庫資訊，在轉化為我們的Tsql，在這裡建立了DBTable、DBColumn、DBAssociation三個實體類：程式碼



  1.  1 using System;   
    2. 

  2.   2 using System.Collections.Generic;   
    3. 

  3.   3 using System.Linq;   
    4. 

  4.   4 using System.Text;   
    5. 

  5.   5   
    6. 

  6.   6 namespace DbmlToTable   
    7. 

  7.   7 {   
    8. 

  8. 
  8     public class DBTable   

    9. 

  9.   9     {   
    10. 

  10.  10   
    11. 

  11. 
 11         public DBTable()   

    12. 

  12.  12         {   
    13. 

  13.  13             Columns = new List<DBColumn>();   
    14. 

  14.  14             this.Associations = new List<DBAssociation>();   
    15. 

  15.  15         }   
    16. 

  16.  16   
    17. 

  17. 
 17         public string TableName   

    18. 

  18.  18         {   
    19. 

  19.  19             get;   
    20. 

  20. 
 20             set;   

    21. 

  21.  21         }   
    22. 

  22.  22   
    23. 

  23. 
 23         public List<DBColumn> Columns   

    24. 

  24.  24         {   
    25. 

  25.  25             get;   
    26. 

  26. 
 26             set;   

    27. 

  27.  27         }   
    28. 

  28.  28   
    29. 

  29. 
 29         public List<DBAssociation> Associations   

    30. 

  30.  30         {   
    31. 

  31.  31             get;   
    32. 

  32. 
 32             set;   

    33. 

  33.  33         }   
    34. 

  34.  34   
    35. 

  35.  35     }   
    36. 

  36.  36   
    37. 

  37. 
 37     public class DBColumn   

    38. 

  38.  38     {   
    39. 

  39. 
 39         public string Name   

    40. 

  40.  40         {   
    41. 

  41.  41             get;   
    42. 

  42. 
 42             set;   

    43. 

  43.  43         }   
    44. 

  44.  44   
    45. 

  45. 
 45         public string DBType   

    46. 

  46.  46         {   
    47. 

  47.  47             get;   
    48. 

  48. 
 48             set;   

    49. 

  49.  49         }   
    50. 

  50.  50   
    51. 

  51. 
 51         public bool IsPrimaryKey   

    52. 

  52.  52         {   
    53. 

  53.  53             get;   
    54. 

  54. 
 54             set;   

    55. 

  55.  55         }   
    56. 

  56.  56   
    57. 

  57. 
 57         public bool IsDbGenerated   

    58. 

  58.  58         {   
    59. 

  59.  59             get;   
    60. 

  60. 
 60             set;   

    61. 

  61.  61         }   
    62. 

  62.  62   
    63. 

  63. 
 63         public bool CanBeNull   

    64. 

  64.  64         {   
    65. 

  65.  65             get;   
    66. 

  66. 
 66             set;   

    67. 

  67.  67         }   
    68. 

  68.  68     }   
    69. 

  69.  69   
    70. 

  70. 
 70     public class DBAssociation   

    71. 

  71.  71     {   
    72. 

  72. 
 72         public string Name   

    73. 

  73.  73         {   
    74. 

  74.  74             get;   
    75. 

  75. 
 75             set;   

    76. 

  76.  76         }   
    77. 

  77.  77   
    78. 

  78. 
 78         public string ThisKey   

    79. 

  79.  79         {   
    80. 

  80.  80             get;   
    81. 

  81. 
 81             set;   

    82. 

  82.  82         }   
    83. 

  83.  83   
    84. 

  84. 
 84         public string OtherKey   

    85. 

  85.  85         {   
    86. 

  86.  86             get;   
    87. 

  87. 
 87             set;   

    88. 

  88.  88         }   
    89. 

  89.  89   
    90. 

  90. 
 90         public bool IsForeignKey   

    91. 

  91.  91         {   
    92. 

  92.  92             get;   
    93. 

  93. 
 93             set;   

    94. 

  94.  94         }   
    95. 

  95.  95     }   
    96. 

  96.  96   
    97. 

  97. 
 97     public class DBTableHlper   

    98. 

  98.  98     {   
    99. 

  99. 
 99         public static DBTable GetAssociationTable(List<DBTable> collection,string assName)   

    100. 

  100. 100         {   
    101. 

  101. 101   
    102. 

  102. 
102             return collection.Find(t => t.Associations.Find(a => !a.IsForeignKey && a.Name == assName) != null);   

    103. 

  103. 103         }   
    104. 

  104. 104     }   
    105. 

  105. 105 }  
    106. 

  106. 106   
    107. 

  107. 107   
    108. 

  108. 複製程式碼 
    109. 



   其中DBTableHlper是由於我的Codesimth是2.0版本的，不能用lamdam表示式，所以我將它編譯在程式集裡面。


   建立了一個 將我們的dbml檔案xml Document轉化為實體類輔助類：




  1. 程式碼   
    2. 

  2.  
    3. 

  3.  1 using System;   
    4. 

  4.  2 using System.Collections.Generic;   
    5. 

  5.  3 using System.Linq;   
    6. 

  6.  4 using System.Text;   
    7. 

  7.  5 using System.Xml;   
    8. 

  8.  6 using System.Xml.Linq;   
    9. 

  9.  7   
    10. 

  10.  8 namespace DbmlToTable   
    11. 

  11.  9 {   
    12. 

  12. 10   
    13. 

  13. 
11     public interface IDbTableCollectionHelper   

    14. 

  14. 12     {   
    15. 

  15. 13         List<DBTable> Transport(XElement element);   
    16. 

  16. 14     }   
    17. 

  17. 15   
    18. 

  18. 
16     public class DbTableCollectionHelper : IDbTableCollectionHelper   

    19. 

  19. 17     {   
    20. 

  20. 18         #region IDbTableCollectionHelper 成員   
    21. 

  21. 19   
    22. 

  22. 
20         public List<DBTable> Transport(XElement element)   

    23. 

  23. 21         {   
    24. 

  24. 22             List<DBTable> collection = new List<DBTable>();   
    25. 

  25. 
23             var tables = element.Elements(XName.Get("Table", "http://schemas.microsoft.com/linqtosql/dbml/2007%22));   

    26. 

  26. 
24             foreach (var tab in tables)   

    27. 

  27. 25             {   
    28. 

  28. 
26                 DBTable t = new DBTable() { TableName = tab.Attribute("Name").Value };   

    29. 

  29. 
27                 var cols = tab.Element(XName.Get("Type", "http://schemas.microsoft.com/linqtosql/dbml/2007%22)).Elements(XName.Get(%22Column%22, "http://schemas.microsoft.com/linqtosql/dbml/2007%22));   

    30. 

  30. 
28                 foreach (var col in cols)   

    31. 

  31. 29                 {   
    32. 

  32. 30                     DBColumn c = new DBColumn()   
    33. 

  33. 31                     {   
    34. 

  34. 
32                         CanBeNull = col.Attribute("CanBeNull") != null ? col.Attribute("CanBeNull").Value.ToLower() == "true" : false,   

    35. 

  35. 
33                         DBType = col.Attribute("DbType") != null ? col.Attribute("DbType").Value : "",   

    36. 

  36. 
34                         IsDbGenerated = col.Attribute("IsDbGenerated") != null ? col.Attribute("IsDbGenerated").Value.ToLower() == "true" : false,   

    37. 

  37. 
35                         IsPrimaryKey = col.Attribute("IsPrimaryKey") != null ? col.Attribute("IsPrimaryKey").Value.ToLower() == "true" : false,   

    38. 

  38. 
36                         Name = col.Attribute("Name") != null ? col.Attribute("Name").Value : ""   

    39. 

  39. 37                     };   
    40. 

  40. 
38                     t.Columns.Add(c);   

    41. 

  41. 39                 }   
    42. 

  42. 40   
    43. 

  43. 
41                 var ass = tab.Element(XName.Get("Type", "http://schemas.microsoft.com/linqtosql/dbml/2007%22)).Elements(XName.Get(%22Association%22, "http://schemas.microsoft.com/linqtosql/dbml/2007%22));   

    44. 

  44. 
42                 foreach (var item in ass)   

    45. 

  45. 43                 {   
    46. 

  46. 44                     DBAssociation a = new DBAssociation()   
    47. 

  47. 45                     {   
    48. 

  48. 
46                         Name = item.Attribute("Name") != null ? item.Attribute("Name").Value : "",   

    49. 

  49. 
47                         OtherKey = item.Attribute("OtherKey") != null ? item.Attribute("OtherKey").Value : "",   

    50. 

  50. 
48                         ThisKey = item.Attribute("ThisKey") != null ? item.Attribute("ThisKey").Value : "",   

    51. 

  51. 
49                         IsForeignKey = item.Attribute("IsForeignKey") != null ? item.Attribute("IsForeignKey").Value.ToLower() == "true" : false   

    52. 

  52. 50                     };   
    53. 

  53. 
51                     t.Associations.Add(a);   

    54. 

  54. 52                 }   
    55. 

  55. 
53                 collection.Add(t);   

    56. 

  56. 54             }   
    57. 

  57. 
55             return collection;   

    58. 

  58. 56         }   
    59. 

  59. 57   
    60. 

  60. 58         #endregion   
    61. 

  61. 59     }   
    62. 

  62. 60 }  
    63. 

  63. 61   
    64. 

  64. 62   
    65. 

  65. 複製程式碼 
    66. 

  在轉化為我們的實體類，我們剩下的就是編寫我們的CodeSmith模板了（更多知識可以參考CodeSmith模板）：




  1. 程式碼   
    2. 

  2.  
    3. 

  3. 
 1 <%@ CodeTemplate Language="C#" TargetLanguage="Text" Src="" Inherits="" Debug="False" Description="Template description here." %>   

    4. 

  4.  2   
    5. 

  5. 
 3 <%@ Import NameSpace="System" %>   

    6. 

  6. 
 4 <%@ Import NameSpace="System.Xml" %>   

    7. 

  7. 
 5 <%@ Import NameSpace="System.Text" %>   

    8. 

  8. 
 6 <%@ Import NameSpace="System.Collections.Generic" %>   

    9. 

  9. 
 7 <%@ Assembly Name="DbmlToTable" %>   

    10. 

  10.  8   
    11. 

  11. 
 9 --Code By Wolf   

    12. 

  12. 
10 <script runat="template">   

    13. 

  13. 11 private List<DbmlToTable.DBTable> _DbTableCollection;   
    14. 

  14. 
12 public List<DbmlToTable.DBTable> DbTableCollection   

    15. 

  15. 13 {   
    16. 

  16. 14     get   
    17. 

  17. 15     {   
    18. 

  18. 
16         return _DbTableCollection;   

    19. 

  19. 17     }   
    20. 

  20. 
18     set      

    21. 

  21. 19     {   
    22. 

  22. 20         _DbTableCollection=value;   
    23. 

  23. 21     }   
    24. 

  24. 22 }   
    25. 

  25. 23   
    26. 

  26. 
24 public  string GeneratorTableSql(List<DbmlToTable.DBTable> collection)   

    27. 

  27. 25 {   
    28. 

  28. 26     StringBuilder sb = new StringBuilder();   
    29. 

  29. 27     StringBuilder sbAssocation = new StringBuilder();   
    30. 

  30. 
28     foreach (DbmlToTable.DBTable item in collection)   

    31. 

  31. 29     {   
    32. 

  32. 30         List<string> cols = new List<string>();   
    33. 

  33. 
31         foreach (DbmlToTable.DBColumn  col in item.Columns)   

    34. 

  34. 32         {   
    35. 

  35. 
33             cols.Add(string.Format("{0} {1} {2} ", col.Name, col.DBType, col.IsPrimaryKey ? "PRIMARY KEY " : ""));   

    36. 

  36. 34         }   
    37. 

  37. 
35         sb.AppendFormat("\r\nCREATE TABLE {0} \r\n(\r\n{1}\r\n)", item.TableName, string.Join(",\r\n", cols.ToArray()));   

    38. 

  38. 36   
    39. 

  39. 
37         foreach (DbmlToTable.DBAssociation ass in item.Associations)   

    40. 

  40. 38         {   
    41. 

  41. 39             if (ass.IsForeignKey)   
    42. 

  42. 40             {   
    43. 

  43. 
41                 DbmlToTable.DBTable tab = DbmlToTable.DBTableHlper.GetAssociationTable(collection,ass.Name);   

    44. 

  44. 
42                 if (tab != null)   

    45. 

  45. 43                 {   
    46. 

  46. 44                     sbAssocation.AppendLine();   
    47. 

  47. 
45                     sbAssocation.AppendFormat(@"ALTER TABLE {0}  WITH NOCHECK ADD  CONSTRAINT {1} FOREIGN KEY({2}) REFERENCES {3} ({4})",   

    48. 

  48. 
46                         item.TableName, "FK_" + ass.Name, ass.ThisKey, tab.TableName, ass.OtherKey);   

    49. 

  49. 47                 }   
    50. 

  50. 48             }   
    51. 

  51. 49         }   
    52. 

  52. 50     }   
    53. 

  53. 51   
    54. 

  54. 
52     return sb.ToString() + "\r\n" + sbAssocation.ToString();   

    55. 

  55. 53 }   
    56. 

  56. 54 </script>   
    57. 

  57. 55 <%= this.GeneratorTableSql(_DbTableCollection) %>  
    58. 

  58. 56   
    59. 

  59. 57   
    60. 

  60. 複製程式碼 
    61. 

在codeSimth中我們建立了一個集合屬性傳遞實體類DBTable和一個轉化TSql輔助方法.


      在控制檯呼叫編譯模板以及輸出：




  1. 程式碼   
    2. 

  2. 
 1 <%@ CodeTemplate Language="C#" TargetLanguage="Text" Src="" Inherits="" Debug="False" Description="Template description here." %>   

    3. 

  3.  2   
    4. 

  4. 
 3 <%@ Import NameSpace="System" %>   

    5. 

  5. 
 4 <%@ Import NameSpace="System.Xml" %>   

    6. 

  6. 
 5 <%@ Import NameSpace="System.Text" %>   

    7. 

  7. 
 6 <%@ Import NameSpace="System.Collections.Generic" %>   

    8. 

  8. 
 7 <%@ Assembly Name="DbmlToTable" %>   

    9. 

  9.  8   
    10. 

  10. 
 9 --Code By Wolf   

    11. 

  11. 
10 <script runat="template">   

    12. 

  12. 11 private List<DbmlToTable.DBTable> _DbTableCollection;   
    13. 

  13. 
12 public List<DbmlToTable.DBTable> DbTableCollection   

    14. 

  14. 13 {   
    15. 

  15. 14     get   
    16. 

  16. 15     {   
    17. 

  17. 
16         return _DbTableCollection;   

    18. 

  18. 17     }   
    19. 

  19. 
18     set      

    20. 

  20. 19     {   
    21. 

  21. 20         _DbTableCollection=value;   
    22. 

  22. 21     }   
    23. 

  23. 22 }   
    24. 

  24. 23   
    25. 

  25. 
24 public  string GeneratorTableSql(List<DbmlToTable.DBTable> collection)   

    26. 

  26. 25 {   
    27. 

  27. 26     StringBuilder sb = new StringBuilder();   
    28. 

  28. 27     StringBuilder sbAssocation = new StringBuilder();   
    29. 

  29. 
28     foreach (DbmlToTable.DBTable item in collection)   

    30. 

  30. 29     {   
    31. 

  31. 30         List<string> cols = new List<string>();   
    32. 

  32. 
31         foreach (DbmlToTable.DBColumn  col in item.Columns)   

    33. 

  33. 32         {   
    34. 

  34. 
33             cols.Add(string.Format("{0} {1} {2} ", col.Name, col.DBType, col.IsPrimaryKey ? "PRIMARY KEY " : ""));   

    35. 

  35. 34         }   
    36. 

  36. 
35         sb.AppendFormat("\r\nCREATE TABLE {0} \r\n(\r\n{1}\r\n)", item.TableName, string.Join(",\r\n", cols.ToArray()));   

    37. 

  37. 36   
    38. 

  38. 
37         foreach (DbmlToTable.DBAssociation ass in item.Associations)   

    39. 

  39. 38         {   
    40. 

  40. 39             if (ass.IsForeignKey)   
    41. 

  41. 40             {   
    42. 

  42. 
41                 DbmlToTable.DBTable tab = DbmlToTable.DBTableHlper.GetAssociationTable(collection,ass.Name);   

    43. 

  43. 
42                 if (tab != null)   

    44. 

  44. 43                 {   
    45. 

  45. 44                     sbAssocation.AppendLine();   
    46. 

  46. 
45                     sbAssocation.AppendFormat(@"ALTER TABLE {0}  WITH NOCHECK ADD  CONSTRAINT {1} FOREIGN KEY({2}) REFERENCES {3} ({4})",   

    47. 

  47. 
46                         item.TableName, "FK_" + ass.Name, ass.ThisKey, tab.TableName, ass.OtherKey);   

    48. 

  48. 47                 }   
    49. 

  49. 48             }   
    50. 

  50. 49         }   
    51. 

  51. 50     }   
    52. 

  52. 51   
    53. 

  53. 
52     return sb.ToString() + "\r\n" + sbAssocation.ToString();   

    54. 

  54. 53 }   
    55. 

  55. 54 </script>   
    56. 

  56. 55 <%= this.GeneratorTableSql(_DbTableCollection) %>  
    57. 

  57. 56   
    58. 

  58. 57   
    59. 

  59. 複製程式碼  
    60. 

   在CodeSimth中就是這麼簡單，生成相應的模板程式碼（個人理解CodeSmith就是把程式碼作為字串輸出）。


在上面到我的CodeSmith模板編譯輔助類，在上一篇通過程式碼生成機制實現強型別程式設計-CodeSmith版也有，在這裡也附帶上：需要引用CodeSmith.Engine.dll.




  1. 程式碼   
    2. 

  2.  
    3. 

  3.  1 using System;   
    4. 

  4.  2   
    5. 

  5.  3 using System.Collections.Generic;   
    6. 

  6.  4   
    7. 

  7.  5 using System.Linq;   
    8. 

  8.  6   
    9. 

  9.  7 using System.Text;   
    10. 

  10.  8   
    11. 

  11.  9 using CodeSmith.Engine;   
    12. 

  12. 10   
    13. 

  13. 11 using Wolf.NameValueDictionary;   
    14. 

  14. 12   
    15. 

  15. 13 namespace DbmlToTable   
    16. 

  16. 14   
    17. 

  17. 15 {   
    18. 

  18. 16   
    19. 

  19. 
17 public class CodeSimthTemplateHelper   

    20. 

  20. 18   
    21. 

  21. 19 {   
    22. 

  22. 20   
    23. 

  23. 
21      public static CodeTemplate CompileTemplate(string templateName, Action errorWriter)   

    24. 

  24. 22   
    25. 

  25. 23      {   
    26. 

  26. 24   
    27. 

  27. 25            CodeTemplateCompiler compiler = new CodeTemplateCompiler(templateName); compiler.Compile();   
    28. 

  28. 26   
    29. 

  29. 
27           if (compiler.Errors.Count == 0)   

    30. 

  30. 28   
    31. 

  31. 29            {   
    32. 

  32. 30   
    33. 

  33. 
31            return compiler.CreateInstance();  

    34. 

  34. 32   
    35. 

  35. 33            }   
    36. 

  36. 34   
    37. 

  37. 
35        else   

    38. 

  38. 36   
    39. 

  39. 37          {   
    40. 

  40. 38   
    41. 

  41. 
39            for (int i = 0; i < compiler.Errors.Count; i++)   

    42. 

  42. 40   
    43. 

  43. 41         {   
    44. 

  44. 42   
    45. 

  45. 43             errorWriter(compiler.Errors[i].ToString());   
    46. 

  46. 44   
    47. 

  47. 45          }   
    48. 

  48. 46   
    49. 

  49. 
47         return null;   

    50. 

  50. 48   
    51. 

  51. 49        }   
    52. 

  52. 50   
    53. 

  53. 51 }   
    54. 

  54. 52   
    55. 

  55. 53    
    56. 

  56. 54   
    57. 

  57. 
55 public static void AddPropertyParams(CodeTemplate template,object param)   

    58. 

  58. 56   
    59. 

  59. 57 {   
    60. 

  60. 58   
    61. 

  61. 59       NameValueDictionary dict = new NameValueDictionary<object>(param);  
    62. 

  62. 60   
    63. 

  63. 61        AddPropertyParams(template, dict);  
    64. 

  64. 62   
    65. 

  65. 63 }  
    66. 

  66. 64   
    67. 

  67. 65    
    68. 

  68. 66   
    69. 

  69. 
67 public static void AddPropertyParams(CodeTemplate template, NameValueDictionary<object> param)  

    70. 

  70. 68   
    71. 

  71. 69 {  
    72. 

  72. 70   
    73. 

  73. 71           NameValueDictionary<object> dict = new NameValueDictionary<object>(param);  
    74. 

  74. 72   
    75. 

  75. 
73           foreach (var item in dict.Keys)  

    76. 

  76. 74   
    77. 

  77. 75           {  
    78. 

  78. 76   
    79. 

  79. 77                 template.SetProperty(item, dict[item]);  
    80. 

  80. 78   
    81. 

  81. 79            }  
    82. 

  82. 80   
    83. 

  83. 81 }  
    84. 

  84. 82   
    85. 

  85. 83 }  
    86. 

  86. 84   
    87. 

  87. 85 }  
    88. 

  88. 86   
    89. 

  89. 複製程式碼 
    90. 

 














 本文轉自 破狼 51CTO部落格，原文連結：http://blog.51cto.com/whitewolfblog/834728，如需轉載請自行聯絡原作者





 




相關文章