Skip to content

Commit 0f507eb

Browse files
schambersschambers
committed
MySqlGenerator almost complete
1 parent 0bff066 commit 0f507eb

File tree

6 files changed

+289
-1
lines changed

6 files changed

+289
-1
lines changed

‎lib/MySql.Data.dll

212 KB
Binary file not shown.

‎src/FluentMigrator.Runner/Generators/GeneratorBase.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ public virtual string GenerateDDLForColumn(ColumnDefinition column)
9494
if (column.DefaultValue != null)
9595
{
9696
sb.Append(" DEFAULT ");
97-
sb.Append(GetConstantValue(column.DefaultValue));
97+
sb.Append(GetConstantValue(column.DefaultValue));
9898
}
9999

100100
if (column.IsIdentity)

‎src/FluentMigrator.Runner/Generators/MySqlGenerator.cs

Lines changed: 191 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,191 @@
1+
using System;
2+
using System.Collections.Generic;
3+
using System.Data;
4+
using System.Text;
5+
using FluentMigrator.Builders.Insert;
6+
using FluentMigrator.Expressions;
7+
using FluentMigrator.Model;
8+
9+
namespace FluentMigrator.Runner.Generators
10+
{
11+
public class MySqlGenerator : GeneratorBase
12+
{
13+
public const int AnsiTinyStringCapacity = 127;
14+
public const int StringCapacity = 255;
15+
public const int TextCapacity = 65535;
16+
public const int MediumTextCapacity = 16777215;
17+
public const int DecimalCapacity = 19;
18+
19+
protected override void SetupTypeMaps()
20+
{
21+
SetTypeMap(DbType.AnsiStringFixedLength, "CHAR(255)");
22+
SetTypeMap(DbType.AnsiStringFixedLength, "CHAR($size)", StringCapacity);
23+
SetTypeMap(DbType.AnsiStringFixedLength, "TEXT", TextCapacity);
24+
SetTypeMap(DbType.AnsiStringFixedLength, "MEDIUMTEXT", MediumTextCapacity);
25+
SetTypeMap(DbType.AnsiString, "VARCHAR(255)");
26+
SetTypeMap(DbType.AnsiString, "VARCHAR($size)", StringCapacity);
27+
SetTypeMap(DbType.AnsiString, "TEXT", TextCapacity);
28+
SetTypeMap(DbType.AnsiString, "MEDIUMTEXT", MediumTextCapacity);
29+
SetTypeMap(DbType.Binary, "LONGBLOB");
30+
SetTypeMap(DbType.Binary, "TINYBLOB", AnsiTinyStringCapacity);
31+
SetTypeMap(DbType.Binary, "BLOB", TextCapacity);
32+
SetTypeMap(DbType.Binary, "MEDIUMBLOB", MediumTextCapacity);
33+
SetTypeMap(DbType.Boolean, "TINYINT(1)");
34+
SetTypeMap(DbType.Byte, "TINYINT UNSIGNED");
35+
SetTypeMap(DbType.Currency, "MONEY");
36+
SetTypeMap(DbType.Date, "DATE");
37+
SetTypeMap(DbType.DateTime, "DATETIME");
38+
SetTypeMap(DbType.Decimal, "DECIMAL(19,5)");
39+
SetTypeMap(DbType.Decimal, "DECIMAL($size,$precision)", DecimalCapacity);
40+
SetTypeMap(DbType.Double, "DOUBLE");
41+
SetTypeMap(DbType.Guid, "VARCHAR(40)");
42+
SetTypeMap(DbType.Int16, "SMALLINT");
43+
SetTypeMap(DbType.Int32, "INTEGER");
44+
SetTypeMap(DbType.Int64, "BIGINT");
45+
SetTypeMap(DbType.Single, "FLOAT");
46+
SetTypeMap(DbType.StringFixedLength, "CHAR(255)");
47+
SetTypeMap(DbType.StringFixedLength, "CHAR($size)", StringCapacity);
48+
SetTypeMap(DbType.StringFixedLength, "TEXT", TextCapacity);
49+
SetTypeMap(DbType.StringFixedLength, "MEDIUMTEXT", MediumTextCapacity);
50+
SetTypeMap(DbType.String, "VARCHAR(255)");
51+
SetTypeMap(DbType.String, "VARCHAR($size)", StringCapacity);
52+
SetTypeMap(DbType.String, "TEXT", TextCapacity);
53+
SetTypeMap(DbType.String, "MEDIUMTEXT", MediumTextCapacity);
54+
SetTypeMap(DbType.Time, "DATETIME");
55+
}
56+
57+
public override string Generate(CreateTableExpression expression)
58+
{
59+
return FormatExpression("CREATE TABLE {0} ({1}) ENGINE = INNODB", expression.TableName, GetColumnDDL(expression.Columns));
60+
}
61+
62+
public override string Generate(CreateColumnExpression expression)
63+
{
64+
65+
return FormatExpression("ALTER TABLE {0} ADD {1}", expression.TableName, GenerateDDLForColumn(expression.Column));
66+
}
67+
68+
public override string Generate(DeleteTableExpression expression)
69+
{
70+
return FormatExpression("DROP TABLE {0}", expression.TableName);
71+
}
72+
73+
public override string Generate(DeleteColumnExpression expression)
74+
{
75+
return FormatExpression("ALTER TABLE {0} DROP COLUMN {1}", expression.TableName, expression.ColumnName);
76+
}
77+
78+
public override string Generate(CreateForeignKeyExpression expression)
79+
{
80+
string primaryColumns = GetColumnList(expression.ForeignKey.PrimaryColumns);
81+
string foreignColumns = GetColumnList(expression.ForeignKey.ForeignColumns);
82+
83+
string sql = "ALTER TABLE {0} ADD CONSTRAINT {1} FOREIGN KEY ({2}) REFERENCES {3} ({4})";
84+
85+
return String.Format(sql,
86+
expression.ForeignKey.ForeignTable,
87+
expression.ForeignKey.Name,
88+
foreignColumns,
89+
expression.ForeignKey.PrimaryTable,
90+
primaryColumns
91+
);
92+
}
93+
94+
public override string Generate(DeleteForeignKeyExpression expression)
95+
{
96+
string sql = "ALTER TABLE {0} DROP FOREIGN KEY `{1}`";
97+
return String.Format(sql, expression.ForeignKey.PrimaryTable, expression.ForeignKey.Name);
98+
}
99+
100+
public override string Generate(CreateIndexExpression expression)
101+
{
102+
var result = new StringBuilder("CREATE");
103+
if (expression.Index.IsUnique)
104+
result.Append(" UNIQUE");
105+
106+
result.Append(" INDEX {0} ON {1} (");
107+
108+
bool first = true;
109+
foreach (IndexColumnDefinition column in expression.Index.Columns)
110+
{
111+
if (first)
112+
first = false;
113+
else
114+
result.Append(",");
115+
116+
result.Append(column.Name);
117+
if (column.Direction == Direction.Ascending)
118+
{
119+
result.Append(" ASC");
120+
}
121+
else
122+
{
123+
result.Append(" DESC");
124+
}
125+
}
126+
result.Append(")");
127+
128+
return FormatExpression(result.ToString(), expression.Index.Name, expression.Index.TableName);
129+
}
130+
131+
public override string Generate(DeleteIndexExpression expression)
132+
{
133+
return FormatExpression("DROP INDEX {0}", expression.Index.Name, expression.Index.TableName);
134+
}
135+
136+
public override string Generate(RenameTableExpression expression)
137+
{
138+
return FormatExpression("RENAME TABLE {0} TO {1}", expression.OldName, expression.NewName);
139+
}
140+
141+
public override string Generate(RenameColumnExpression expression)
142+
{
143+
return FormatExpression("ALTER TABLE {0} CHANGE COLUMN {1} {2} definition", expression.TableName, expression.OldName, expression.NewName);
144+
}
145+
146+
public override string Generate(InsertDataExpression expression)
147+
{
148+
var result = new StringBuilder();
149+
foreach (InsertionData row in expression.Rows)
150+
{
151+
List<string> columnNames = new List<string>();
152+
List<object> columnData = new List<object>();
153+
foreach (KeyValuePair<string, object> item in row)
154+
{
155+
columnNames.Add(item.Key);
156+
columnData.Add(item.Value);
157+
}
158+
159+
string columns = GetColumnList(columnNames);
160+
string data = GetDataList(columnData);
161+
result.Append(FormatExpression("INSERT INTO [{0}] ({1}) VALUES ({2});", expression.TableName, columns, data));
162+
}
163+
return result.ToString();
164+
}
165+
166+
public string FormatExpression(string template, params object[] args)
167+
{
168+
return String.Format(template, args);
169+
}
170+
171+
private string GetColumnList(IEnumerable<string> columns)
172+
{
173+
string result = "";
174+
foreach (string column in columns)
175+
{
176+
result += column + ",";
177+
}
178+
return result.TrimEnd(',');
179+
}
180+
181+
private string GetDataList(List<object> data)
182+
{
183+
string result = "";
184+
foreach (object column in data)
185+
{
186+
result += GetConstantValue(column) + ",";
187+
}
188+
return result.TrimEnd(',');
189+
}
190+
}
191+
}

‎src/FluentMigrator.Runner/MigrationRunner.cs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ public MigrationRunner(IMigrationConventions conventions, IMigrationProcessor pr
1818
: this(conventions, processor, new Announcer(Console.Out), new StopWatch())
1919
{
2020
}
21+
2122
public MigrationRunner(IMigrationConventions conventions, IMigrationProcessor processor, IAnnouncer announcer, IStopWatch stopWatch)
2223
{
2324
_announcer = announcer;

‎src/FluentMigrator.Runner/Processors/MySql/MySqlProcessor.cs

Lines changed: 78 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,78 @@
1+
using System;
2+
using System.Data;
3+
using MySql.Data.MySqlClient;
4+
5+
namespace FluentMigrator.Runner.Processors.MySql
6+
{
7+
public class MySqlProcessor : ProcessorBase
8+
{
9+
public MySqlConnection Connection { get; set; }
10+
11+
public MySqlProcessor(MySqlConnection connection, IMigrationGenerator generator)
12+
{
13+
this.generator = generator;
14+
Connection = connection;
15+
}
16+
17+
public override bool TableExists(string tableName)
18+
{
19+
return Exists("select count(*) from sqlite_master where name='{0}'", tableName);
20+
}
21+
22+
public override void Execute(string template, params object[] args)
23+
{
24+
if (Connection.State != ConnectionState.Open) Connection.Open();
25+
26+
using (var command = new MySqlCommand(String.Format(template, args), Connection))
27+
{
28+
command.ExecuteNonQuery();
29+
}
30+
}
31+
32+
public override bool Exists(string template, params object[] args)
33+
{
34+
if (Connection.State != ConnectionState.Open) Connection.Open();
35+
36+
using (var command = new MySqlCommand(String.Format(template, args), Connection))
37+
using (var reader = command.ExecuteReader())
38+
{
39+
try
40+
{
41+
if (!reader.Read()) return false;
42+
if (int.Parse(reader[0].ToString()) <= 0) return false;
43+
return true;
44+
}
45+
catch
46+
{
47+
return false;
48+
}
49+
}
50+
}
51+
52+
public override DataSet ReadTableData(string tableName)
53+
{
54+
return Read("select * from {0}", tableName);
55+
}
56+
57+
protected override void Process(string sql)
58+
{
59+
if (Connection.State != ConnectionState.Open) Connection.Open();
60+
61+
using (var command = new MySqlCommand(sql, Connection))
62+
command.ExecuteNonQuery();
63+
}
64+
65+
public override DataSet Read(string template, params object[] args)
66+
{
67+
if (Connection.State != ConnectionState.Open) Connection.Open();
68+
69+
DataSet ds = new DataSet();
70+
using (var command = new MySqlCommand(String.Format(template, args), Connection))
71+
using (MySqlDataAdapter adapter = new MySqlDataAdapter(command))
72+
{
73+
adapter.Fill(ds);
74+
return ds;
75+
}
76+
}
77+
}
78+
}

‎src/FluentMigrator.Runner/Processors/MySql/MySqlProcessorFactory.cs

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
using System;
2+
using System.Collections.Generic;
3+
using System.Linq;
4+
using System.Text;
5+
using FluentMigrator.Runner.Generators;
6+
using MySql.Data.MySqlClient;
7+
8+
namespace FluentMigrator.Runner.Processors.MySql
9+
{
10+
public class MySqlProcessorFactory : IMigrationProcessorFactory
11+
{
12+
public IMigrationProcessor Create(string connectionString)
13+
{
14+
var connection = new MySqlConnection(connectionString);
15+
return new MySqlProcessor(connection, new MySqlGenerator());
16+
}
17+
}
18+
}

0 commit comments

Comments
 (0)