mcs/class/System.Data.Linq/src/DbLinq.MySql/MySqlSchemaLoader.Constraints.cs

   1 #region MIT license\r
   2 // \r
   3 // MIT license\r
   4 //\r
   5 // Copyright (c) 2007-2008 Jiri Moudry, Pascal Craponne\r
   6 // \r
   7 // Permission is hereby granted, free of charge, to any person obtaining a copy\r
   8 // of this software and associated documentation files (the "Software"), to deal\r
   9 // in the Software without restriction, including without limitation the rights\r
  10 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\r
  11 // copies of the Software, and to permit persons to whom the Software is\r
  12 // furnished to do so, subject to the following conditions:\r
  13 // \r
  14 // The above copyright notice and this permission notice shall be included in\r
  15 // all copies or substantial portions of the Software.\r
  16 // \r
  17 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\r
  18 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\r
  19 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\r
  20 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\r
  21 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\r
  22 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\r
  23 // THE SOFTWARE.\r
  24 // \r
  25 #endregion\r
  26 \r
  27 using System.Collections.Generic;\r
  28 using System.Data;\r
  29 using DbLinq.Util;\r
  30 \r
  31 namespace DbLinq.MySql\r
  32 {\r
  33     partial class MySqlSchemaLoader\r
  34     {\r
  35         /// <summary>\r
  36         /// represents one row from MySQL's information_schema.`Key_Column_Usage` table\r
  37         /// </summary>\r
  38         protected class DataConstraint\r
  39         {\r
  40             public string ConstraintName;\r
  41             public string TableSchema;\r
  42             public string TableName;\r
  43             public string ColumnName;\r
  44             public string ReferencedTableSchema;\r
  45             public string ReferencedTableName;\r
  46             public string ReferencedColumnName;\r
  47 \r
  48             public override string ToString()\r
  49             {\r
  50                 string detail = ConstraintName == "PRIMARY"\r
  51                                     ? TableName + " PK"\r
  52                                     : ConstraintName;\r
  53                 return "KeyColUsage " + detail;\r
  54             }\r
  55         }\r
  56 \r
  57         protected virtual DataConstraint ReadConstraint(IDataReader rdr)\r
  58         {\r
  59             DataConstraint constraint = new DataConstraint();\r
  60             int field = 0;\r
  61             constraint.ConstraintName = rdr.GetAsString(field++);\r
  62             constraint.TableSchema = rdr.GetAsString(field++);\r
  63             constraint.TableName = rdr.GetAsString(field++);\r
  64             constraint.ColumnName = rdr.GetAsString(field++);\r
  65             constraint.ReferencedTableSchema = rdr.GetAsString(field++);\r
  66             constraint.ReferencedTableName = rdr.GetAsString(field++);\r
  67             constraint.ReferencedColumnName = rdr.GetAsString(field++);\r
  68             return constraint;\r
  69         }\r
  70 \r
  71         protected virtual List<DataConstraint> ReadConstraints(IDbConnection conn, string db)\r
  72         {\r
  73             string sql = @"\r
  74 SELECT constraint_name,table_schema,table_name\r
  75     ,GROUP_CONCAT(column_name SEPARATOR ',') AS column_name,referenced_table_schema,referenced_table_name,GROUP_CONCAT(referenced_column_name SEPARATOR ',') AS referenced_column_name\r
  76 FROM information_schema.`KEY_COLUMN_USAGE`\r
  77 WHERE table_schema=?db GROUP BY constraint_name,table_schema,table_name,referenced_table_name";\r
  78 \r
  79             return DataCommand.Find<DataConstraint>(conn, sql, "?db", db, ReadConstraint);\r
  80         }\r
  81     }\r
  82 }\r