public static string GenerateAllTables(this System.Data.Common.DbConnection connection, GeneratorBehavior generatorBehavior = GeneratorBehavior.Default)
{
if (connection.State != ConnectionState.Open)
{
connection.Open();
}
var conneciontName = connection.GetType().Name.ToLower();
var tables = new List <string>();
var sql = generatorBehavior.HasFlag(GeneratorBehavior.View) ? TableSchemaSqls[conneciontName].Split("where")[0] : TableSchemaSqls[conneciontName];
using (var command = connection.CreateCommand(sql))
using (var reader = command.ExecuteReader())
{
while (reader.Read())
{
tables.Add(reader.GetString(0));
}
}
var sb = new StringBuilder();
sb.AppendLine("namespace Models { ");
tables.ForEach(table => sb.Append(connection.GenerateClass(
string.Format(QuerySqls[conneciontName], table), table, generatorBehavior: generatorBehavior
)));
sb.AppendLine("}");
return(sb.ToString());
}
public static string GenerateClass(this IDbConnection connection, string sql, string className = null, GeneratorBehavior generatorBehavior = GeneratorBehavior.Default)
{
if (connection.State != ConnectionState.Open)
{
connection.Open();
}
var builder = new StringBuilder();
//Get Table Name
//Fix : [When View using CommandBehavior.KeyInfo will get duplicate columns �P Issue #8 �P shps951023/PocoClassGenerator](https://github.com/shps951023/PocoClassGenerator/issues/8 )
var isFromMutiTables = false;
using (var command = connection.CreateCommand(sql))
using (var reader = command.ExecuteReader(CommandBehavior.KeyInfo | CommandBehavior.SingleRow))
{
var tables = reader.GetSchemaTable().Select().Select(s => s["BaseTableName"] as string).Distinct();
var tableName = string.IsNullOrWhiteSpace(className) ? tables.First() ?? "Info" : className;
isFromMutiTables = tables.Count() > 1;
if (generatorBehavior.HasFlag(GeneratorBehavior.DapperContrib) && !isFromMutiTables)
{
builder.AppendFormat(" [Dapper.Contrib.Extensions.Table(\"{0}\")]{1}", tableName, Environment.NewLine);
}
builder.AppendFormat(" public class {0}{1}", tableName.Replace(" ", ""), Environment.NewLine);
builder.AppendLine(" {");
}
//Get Columns
var behavior = isFromMutiTables ? (CommandBehavior.SchemaOnly | CommandBehavior.SingleRow) : (CommandBehavior.KeyInfo | CommandBehavior.SingleRow);
using (var command = connection.CreateCommand(sql))
using (var reader = command.ExecuteReader(behavior))
{
do
{
var schema = reader.GetSchemaTable();
foreach (DataRow row in schema.Rows)
{
var type = (Type)row["DataType"];
var name = TypeAliases.ContainsKey(type) ? TypeAliases[type] : type.FullName;
var isNullable = (bool)row["AllowDBNull"] && NullableTypes.Contains(type);
var collumnName = (string)row["ColumnName"];
if (generatorBehavior.HasFlag(GeneratorBehavior.Comment) && !isFromMutiTables)
{
var comments = new[] { "DataTypeName", "IsUnique", "IsKey", "IsAutoIncrement", "IsReadOnly" }
.Select(s =>
{
if (row[s] is bool && ((bool)row[s]))
{
return(s);
}
if (row[s] is string && !string.IsNullOrWhiteSpace((string)row[s]))
{
return(string.Format(" {0} : {1} ", s, row[s]));
}
return(null);
}).Where(w => w != null).ToArray();
var sComment = string.Join(" , ", comments);
builder.AppendFormat(" /// <summary>{0}</summary>{1}", sComment, Environment.NewLine);
}
if (generatorBehavior.HasFlag(GeneratorBehavior.DapperContrib) && !isFromMutiTables)
{
var isKey = (bool)row["IsKey"];
var isAutoIncrement = (bool)row["IsAutoIncrement"];
if (isKey && isAutoIncrement)
{
builder.AppendLine(" [Dapper.Contrib.Extensions.Key]");
}
if (isKey && !isAutoIncrement)
{
builder.AppendLine(" [Dapper.Contrib.Extensions.ExplicitKey]");
}
if (!isKey && isAutoIncrement)
{
builder.AppendLine(" [Dapper.Contrib.Extensions.Computed]");
}
}
builder.AppendLine(string.Format(" public {0}{1} {2} {{ get; set; }}", name, isNullable ? "?" : string.Empty, collumnName));
}
builder.AppendLine(" }");
builder.AppendLine();
} while (reader.NextResult());
return(builder.ToString());
}
}