NHibernate生成实体类、xml映射文件

最近工作电脑装完win10后，之前使用的codeSmith安装不了，索性自己写一个。

界面比较简单，如下图:

第一行为Oracle数据库的连接字符串。连接成功后，填充表到第4行的下拉列表中。

第二行为实体类命名空间。

第三行为保存生成类、xml文件选择文件夹。

private void btnConnect_Click(object sender, RoutedEventArgs e)
        {
            try
            {
                using (OracleConnection conn = new OracleConnection())
                {
                    conn.ConnectionString = txtConnStr.Text;

                    using (OracleCommand cmd = conn.CreateCommand())
                    {
                        cmd.CommandText = "select distinct table_name from user_tables order by table_name";

                        DataTable dt = new DataTable();

                        OracleDataAdapter adapter = new OracleDataAdapter();
                        adapter.SelectCommand = cmd;

                        adapter.Fill(dt);

                        cbTables.DisplayMemberPath = "TABLE_NAME";
                        cbTables.SelectedValuePath = "TABLE_NAME";
                        cbTables.ItemsSource = dt.DefaultView;

                    }

                    MessageBoxShow("连接成功!");

                }
            }
            catch
            {
                MessageBoxShow("连接失败!");
            }
        }

private void btnChoseDirectory_Click(object sender, RoutedEventArgs e)
        {
            FolderBrowserDialog folderDialog = new FolderBrowserDialog();
            DialogResult dialogResult = folderDialog.ShowDialog();

            if (dialogResult == System.Windows.Forms.DialogResult.OK)
            {
                labDirectoryPath.Content = folderDialog.SelectedPath.Trim();
            }
        }

private void btnSave_Click(object sender, RoutedEventArgs e)
        {
            if (string.IsNullOrEmpty(txtEntityClassNamespace.Text.Trim()))
            {
                MessageBoxShow("请输入实体命名空间名称!");
                txtEntityClassNamespace.Focus();
                return;
            }
            if (labDirectoryPath.Content == null)
            {
                MessageBoxShow("请选择文件夹保存生成文件!");
                btnChoseDirectory.Focus();
                return;
            }
            if (cbTables.SelectedValue == null)
            {
                MessageBoxShow("请选择表名!");
                return;
            }

            DataTable dt = GetTableSchemaByTableName();

            GenerateClass(dt);

            GenerateMappingXml(dt);

            MessageBoxShow("生成成功!");
        }

        private string OracleTypeToCSType(string oracleType)
        {
            string csType = string.Empty;

            switch (oracleType.ToUpper())
            {
                case "NVARCHAR2":
                case "VARCHAR2":
                case "CHAR":
                    csType = "string";
                    break;
                case "LONG":
                    csType = "long";
                    break;
                case "NUMBER":
                    csType = "int";
                    break;
                case "DATE":
                    csType = "datetime";
                    break;
            }

            return csType;
        }

        private DataTable GetTableSchemaByTableName()
        {
            string selectSQL = @"select t1.COLUMN_NAME,t1.data_type,t1.data_length,t1.nullable,t2.comments,
replace(initcap(t1.TABLE_NAME),'_','') className
from user_tab_columns t1 left
join user_col_comments t2
on t1.TABLE_NAME = t2.table_name and t1.COLUMN_NAME = t2.column_name
where t1.table_name = :table_name
order by to_number(t1.column_id)";

            DataTable dt = new DataTable();
            using (OracleConnection conn = new OracleConnection())
            {
                conn.ConnectionString = txtConnStr.Text;

                using (OracleCommand cmd = conn.CreateCommand())
                {
                    cmd.CommandText = selectSQL;
                    cmd.Parameters.Add(new OracleParameter(":table_name", cbTables.SelectedValue));

                    OracleDataAdapter adapter = new OracleDataAdapter();
                    adapter.SelectCommand = cmd;

                    adapter.Fill(dt);
                }
            }

            return dt;
        }

        private void GenerateClass(DataTable dt)
        {
            string entityClassNamespace = txtEntityClassNamespace.Text;
            if (dt.Rows.Count > 0)
            {
                string className = dt.Rows[0]["className"].ToString();

                using (FileStream csStream = new FileStream(System.IO.Path.Combine(labDirectoryPath.Content.ToString(), className + ".cs"), FileMode.Create, FileAccess.Write))
                {
                    using (StreamWriter writer = new StreamWriter(csStream, Encoding.Default))
                    {
                        writer.WriteLine(string.Format("namespace {0}", entityClassNamespace));

                        writer.WriteLine("{");
                        writer.WriteLine(string.Format(@"   public class {0} : BaseModel
   {{", className));
                        for (int i = 0; i < dt.Rows.Count; i++)
                        {
                            writer.WriteLine(string.Format(@"      /// <summary>
      /// {0}
      /// </summary>", dt.Rows[i]["comments"].ToString()));
                            writer.WriteLine(string.Format("      public virtual {0} {1} {{ set; get; }}",
                                OracleTypeToCSType(dt.Rows[i]["data_type"].ToString()),
                                dt.Rows[i]["COLUMN_NAME"].ToString()));
                            writer.WriteLine();
                        }
                        writer.WriteLine("   }");
                        writer.WriteLine("}");
                    }
                }

            }
        }

        private void GenerateMappingXml(DataTable dt)
        {
            string entityClassNamespace = txtEntityClassNamespace.Text;
            if (dt.Rows.Count > 0)
            {
                string className = dt.Rows[0]["className"].ToString();

                XNamespace xmlns = "urn:nhibernate-mapping-2.2";

                var xDoc = new XDocument(new XElement(xmlns + "hibernate-mapping",
                                new XElement(xmlns + "class",
                                    new XAttribute("name", string.Format("{0}.{1},{0}", entityClassNamespace, className)),
                                    new XAttribute("table", cbTables.SelectedValue),
                                    new XAttribute("lazy", "true"),
                                        new XElement(xmlns + "id",
                                            new XAttribute("column", "ID"),
                                            new XAttribute("name", "ID"),
                                                new XElement(xmlns + "generator",
                                                    new XAttribute("class", "sequence"),
                                                        new XElement(xmlns + "param",
                                                            new XAttribute("name", "sequence"),
                                                                new XText(string.Format("SEQ_{0}", cbTables.SelectedValue))
                                                        ))))));

                XElement _class = xDoc.Root.Element(xmlns + "class");

                for (int i = 0; i < dt.Rows.Count; i++)
                {
                    string columnName = dt.Rows[i]["COLUMN_NAME"].ToString();

                    XElement property = new XElement(xmlns + "property");
                    property.SetAttributeValue("name", columnName);
                    property.SetAttributeValue("column", columnName);

                    _class.Add(property);
                }

                xDoc.Save(System.IO.Path.Combine(labDirectoryPath.Content.ToString(), className + ".hbm.xml"));
            }
        }

总结：

1. 使用user_tables系统表查询用户所有表
2. 使用user_tab_columns查询表的所有列
3. 使用user_col_comments查询表的备注信息
4. 使用文件流生成实体类，使用Linq to xml生成*.hbm.xml文件