Experiments/Attic/Generator/Castle.ActiveRecord.Generator.Components/Database/DatabaseDefinitionBuilder.cs

   1 // Copyright 2004-2008 Castle Project - http://www.castleproject.org/
   2 //
   3 // Licensed under the Apache License, Version 2.0 (the "License");
   4 // you may not use this file except in compliance with the License.
   5 // You may obtain a copy of the License at
   6 //
   7 //     http://www.apache.org/licenses/LICENSE-2.0
   8 //
   9 // Unless required by applicable law or agreed to in writing, software
  10 // distributed under the License is distributed on an "AS IS" BASIS,
  11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 // See the License for the specific language governing permissions and
  13 // limitations under the License.
  14
  15 namespace Castle.ActiveRecord.Generator.Components.Database
  16 {
  17         using System;
  18         using System.Data;
  19         using System.Data.OleDb;
  20
  21
  22         public class DatabaseDefinitionBuilder : IDatabaseDefinitionBuilder
  23         {
  24                 private IConnectionFactory _connectionFactory;
  25
  26                 public DatabaseDefinitionBuilder(IConnectionFactory connectionFactory)
  27                 {
  28                         _connectionFactory = connectionFactory;
  29                 }
  30
  31                 public DatabaseDefinition Build(String alias, String connectionString)
  32                 {
  33                         DatabaseDefinition def = new DatabaseDefinition(alias);
  34
  35                         using(OleDbConnection connection = _connectionFactory.CreateConnection(connectionString))
  36                         {
  37                                 BuildTables(connection, def);
  38                                 BuildPks(connection, def);
  39                                 BuildFks(connection, def);
  40                         }
  41
  42                         return def;
  43                 }
  44
  45                 private void BuildPks(OleDbConnection connection, DatabaseDefinition def)
  46                 {
  47                         object[] objArrRestrict = new object[] {null, null, null, null};
  48                         DataTable indexes = connection.GetOleDbSchemaTable(OleDbSchemaGuid.Indexes, objArrRestrict);
  49
  50                         foreach(DataRow index in indexes.Rows)
  51                         {
  52                                 TableDefinition tableDef = def.Tables[ (String) index["TABLE_NAME"] ];
  53
  54                                 if (tableDef == null) continue;
  55
  56                                 ColumnDefinition colDef = tableDef.Columns[ (String) index["COLUMN_NAME"] ];
  57
  58                                 colDef.PrimaryKey = (bool) index["PRIMARY_KEY"];
  59                                 colDef.Unique = (bool) index["UNIQUE"];
  60                         }
  61                 }
  62
  63                 private void BuildFks(OleDbConnection connection, DatabaseDefinition def)
  64                 {
  65                         object[] objArrRestrict = new object[] {null, null, null, null};
  66                         DataTable indexes = connection.GetOleDbSchemaTable(OleDbSchemaGuid.Foreign_Keys, objArrRestrict);
  67
  68                         foreach(DataRow index in indexes.Rows)
  69                         {
  70                                 TableDefinition fkTableDef = def.Tables[ (String) index["FK_TABLE_NAME"] ];
  71                                 TableDefinition pkTableDef = def.Tables[ (String) index["PK_TABLE_NAME"] ];
  72
  73                                 if (fkTableDef == null || pkTableDef == null) continue;
  74
  75                                 ColumnDefinition colDef = fkTableDef.Columns[ (String) index["FK_COLUMN_NAME"] ];
  76
  77                                 colDef.ForeignKey = true;
  78                                 colDef.RelatedTable = pkTableDef;
  79
  80                                 pkTableDef.AddManyRelation(fkTableDef);
  81                         }
  82                 }
  83
  84                 private void BuildTables(OleDbConnection connection, DatabaseDefinition def)
  85                 {
  86                         object[] objArrRestrict = new object[] {null, null, null, "TABLE"};
  87                         DataTable tables = connection.GetOleDbSchemaTable(OleDbSchemaGuid.Tables, objArrRestrict);
  88
  89                         foreach(DataRow table in tables.Rows)
  90                         {
  91                                 TableDefinition tableDef =
  92                                         def.AddTable( new TableDefinition( (String) table["TABLE_NAME"], def ) );
  93
  94                                 objArrRestrict = new object[] {null, null, tableDef.Name, null};
  95
  96                                 DataTable columns = connection.GetOleDbSchemaTable(OleDbSchemaGuid.Columns, objArrRestrict);
  97
  98                                 foreach(DataRow column in columns.Rows)
  99                                 {
 100                                         ColumnDefinition colDef = tableDef.AddColumn( new ColumnDefinition( (String) column["COLUMN_NAME"] ) );
 101                                         colDef.Nullable = (bool) column["IS_NULLABLE"];
 102                                         colDef.Type = (OleDbType) Enum.Parse( typeof(OleDbType), column["DATA_TYPE"].ToString() );
 103                                 }
 104                         }
 105                 }
 106         }
 107 }