/// <summary> /// This class represents a provier class to get right schema reader based on database type. /// </summary> public static SchemaReader GetReader(DbServerType dbServerType) { SchemaReader schemaReader = null; switch (dbServerType) { case DbServerType.MsSql: schemaReader = new SqlServerSchemaReader(); break; case DbServerType.Sqlite: schemaReader = new SqliteSchemaReader(); break; case DbServerType.MySql: schemaReader = new MySqlSchemaReader(); break; case DbServerType.Postgres: schemaReader = new PostgreSqlSchemaReader(); break; } return(schemaReader); }
private Tables LoadTables(GenerateContext context) { var cmd = context.Command; if (string.IsNullOrWhiteSpace(cmd.ProviderName)) { WriteLine(""); WriteLine("// -----------------------------------------------------------------------------------------"); WriteLine("// db provider must be provided"); WriteLine("// -----------------------------------------------------------------------------------------"); WriteLine(""); return(new Tables()); } if (string.IsNullOrWhiteSpace(cmd.ConnectionString)) { WriteLine(""); WriteLine("// -----------------------------------------------------------------------------------------"); WriteLine("// connection string must be provided."); WriteLine("// -----------------------------------------------------------------------------------------"); WriteLine(""); return(new Tables()); } DbProviderFactory _factory; try { _factory = DbProviderFactories.GetFactory(cmd.ProviderName); } catch (Exception x) { var error = x.Message.Replace("\r\n", "\n").Replace("\n", " "); WriteLine(""); WriteLine("// -----------------------------------------------------------------------------------------"); WriteLine("// Failed to load provider `{0}` - {1}", cmd.ProviderName, error); WriteLine("// -----------------------------------------------------------------------------------------"); WriteLine(""); return(new Tables()); } try { Tables result; using (var conn = _factory.CreateConnection()) { conn.ConnectionString = cmd.ConnectionString; conn.Open(); SchemaReader reader = null; if (_factory.GetType().Name == "MySqlClientFactory") { reader = new MySqlSchemaReader(); context.EscapeSqlIdentifier = sqlIdentifier => $"`{sqlIdentifier}`"; } else if (_factory.GetType().Name == "SqlCeProviderFactory") { reader = new SqlServerCeSchemaReader(); } else if (_factory.GetType().Name == "NpgsqlFactory") { reader = new PostGreSqlSchemaReader(); context.EscapeSqlIdentifier = sqlIdentifier => $"\"{sqlIdentifier}\""; } else if (_factory.GetType().Name == "OracleClientFactory") { reader = new OracleSchemaReader(); context.EscapeSqlIdentifier = sqlIdentifier => $"\"{sqlIdentifier.ToUpperInvariant()}\""; } else { reader = new SqlServerSchemaReader(); } reader.outer = _outer; result = reader.ReadSchema(conn, _factory); // Remove unrequired tables/views for (int i = result.Count - 1; i >= 0; i--) { if (cmd.SchemaName != null && string.Compare(result[i].Schema, cmd.SchemaName, true) != 0) { result.RemoveAt(i); continue; } if (!cmd.IncludeViews && result[i].IsView) { result.RemoveAt(i); continue; } if (Helpers.StartsWithAny(result[i].ClassName, cmd.ExcludePrefix)) { result.RemoveAt(i); continue; } } conn.Close(); var rxClean = new Regex("^(Equals|GetHashCode|GetType|ToString|repo|Save|IsNew|Insert|Update|Delete|Exists|SingleOrDefault|Single|First|FirstOrDefault|Fetch|Page|Query)$"); foreach (var t in result) { t.ClassName = cmd.ClassPrefix + t.ClassName + cmd.ClassSuffix; foreach (var c in t.Columns) { c.PropertyName = rxClean.Replace(c.PropertyName, "_$1"); // Make sure property name doesn't clash with class name if (c.PropertyName == t.ClassName) { c.PropertyName = "_" + c.PropertyName; } } } return(result); } } catch (Exception x) { var error = x.Message.Replace("\r\n", "\n").Replace("\n", " "); WriteLine(""); WriteLine("// -----------------------------------------------------------------------------------------"); WriteLine("// Failed to read database schema - {0}", error); WriteLine("// -----------------------------------------------------------------------------------------"); WriteLine(""); return(new Tables()); } }