/// <summary>
/// Generate an ActiveRecord model usable by the Castle Project's
/// ActiveRecord library
/// </summary>
/// <param name="p_Table">database table or view name</param>
/// <param name="p_OutputDir">the target directory</param>
public void GenerateClass(string sTemplate, DbTableInfo p_Table, string p_OutputDir)
{
Template template = engine.GetTemplate(sTemplate);
// Generate ActiveRecord Classes
string className = p_Table.GetClassName();
DbFieldInfo[] fieldList = p_Table.GetFields();
DbRelatedTableInfo[] related = p_Table.GetDbRelatedTableInfo();
// replicate flag to children
//TODO: add to [new] code generation context
for (int i = 0; i < fieldList.Length; i++)
{
fieldList[i].EnableValidationAttributes = _EnableValidationAttributes;
}
Debug.WriteLine("Table: " + p_Table + " -> " + className);
string fileName;
VelocityContext context = new VelocityContext();
context.Put("namespace", _NameSpace);
context.Put("developer", Environment.UserName);
context.Put("Partial", _MakePartial ? "partial" : "");
context.Put("PropChange", _PropChange);
context.Put("table", p_Table);
context.Put("fields", fieldList);
context.Put("related", related);
context.Put("date", DateTime.Now.ToString("yyyy-MM-dd"));
//get suggested class name
fileName = GetFileName(sTemplate, context);
if (!CanWriteThisFile(p_OutputDir, className, fileName))
{
return;
}
StreamWriter wr = new StreamWriter(p_OutputDir + fileName);
try
{
StringWriter writer = new StringWriter();
template.Merge(context, writer);
wr.WriteLine(writer.GetStringBuilder().ToString());
}
finally
{
wr.Close();
}
}
/// <summary>
/// Generate an ActiveRecord model usable by the Castle Project's
/// ActiveRecord library
/// </summary>
/// <param name="p_Table">database table or view name</param>
/// <param name="p_OutputDir">the target directory</param>
public void GenerateClass(string sTemplate, DbTableInfo p_Table, string p_OutputDir)
{
Template template = engine.GetTemplate(sTemplate);
// Generate ActiveRecord Classes
string className = p_Table.GetClassName();
DbFieldInfo[] fieldList = p_Table.GetFields();
DbRelatedTableInfo[] related = p_Table.GetDbRelatedTableInfo();
// replicate flag to children
//TODO: add to [new] code generation context
for (int i = 0; i < fieldList.Length; i++)
{
fieldList[i].EnableValidationAttributes = _EnableValidationAttributes;
}
Debug.WriteLine("Table: " + p_Table + " -> " + className);
string fileName;
VelocityContext context = new VelocityContext();
context.Put("namespace", _NameSpace);
context.Put("developer", Environment.UserName);
context.Put("Partial", _MakePartial ? "partial" : "");
context.Put("PropChange", _PropChange);
context.Put("table", p_Table);
context.Put("fields", fieldList);
context.Put("related", related);
context.Put("date", DateTime.Now.ToString("yyyy-MM-dd"));
//get suggested class name
fileName = GetFileName(sTemplate, context);
if (!CanWriteThisFile(p_OutputDir, className, fileName)) return;
StreamWriter wr = new StreamWriter(p_OutputDir + fileName);
try
{
StringWriter writer = new StringWriter();
template.Merge(context, writer);
wr.WriteLine(writer.GetStringBuilder().ToString());
}
finally
{
wr.Close();
}
}
public string GetClassName()
{
return(DbTableInfo.GetSingularName(_Table_Name));
}
public string GetNetType()
{
if (this.IsForeignKey())
{
return(DbTableInfo.GetSingularName(_DbForeignKeyInfo.PK_Table));
}
string sqlType = Data_Type;
// the suffix will add "?" at end if .net type is not a class and field is nullable
string suf = (Is_Nullable) ? "?" : "";
if (sqlType.Equals("bigint"))
{
return("long" + suf);
}
if (sqlType.Equals("int"))
{
return("int" + suf);
}
if (sqlType.Equals("smallint"))
{
return("short" + suf);
}
if (sqlType.Equals("tinyint"))
{
return("byte" + suf);
}
if (sqlType.Equals("bit"))
{
return("bool" + suf);
}
if (sqlType.Equals("decimal"))
{
return("System.Decimal" + suf);
}
if (sqlType.Equals("numeric"))
{
return("System.Decimal" + suf);
}
if (sqlType.Equals("money"))
{
return("System.Decimal" + suf);
}
if (sqlType.Equals("smallmoney"))
{
return("System.Decimal" + suf);
}
if (sqlType.Equals("float"))
{
return("float" + suf);
}
if (sqlType.Equals("real"))
{
return("double" + suf);
}
if (sqlType.Equals("datetime"))
{
return("DateTime" + suf);
}
if (sqlType.Equals("smalldatetime"))
{
return("DateTime" + suf);
}
if (sqlType.Equals("char"))
{
return("string");
}
if (sqlType.Equals("varchar"))
{
return("string");
}
if (sqlType.Equals("text"))
{
return("string"); // might be HUGE!
}
if (sqlType.Equals("nchar"))
{
return("string");
}
if (sqlType.Equals("nvarchar"))
{
return("string");
}
if (sqlType.Equals("ntext"))
{
return("string");
}
if (sqlType.Equals("binary"))
{
return("byte[]");
}
if (sqlType.Equals("varbinary"))
{
return("byte[]");
}
if (sqlType.Equals("image"))
{
return("byte[]");
}
if (sqlType.Equals("uniqueidentifier"))
{
return("byte[]"); // this MAY be a byte[16] array
}
throw new Exception("Unexpected data type: " + Data_Type);
}
// get a list of tables
//
// if username is blank, use integrated security
// if bInclude is false, use filter to exclude
public static DbTableInfo[] GetTables(BackgroundWorker bw,
string p_Server, string p_Database,
string p_Username, string p_Password,
bool bInclude, string p_FilterPrefix)
{
System.Data.Common.DbProviderFactory sqlFactory = System.Data.SqlClient.SqlClientFactory.Instance;
IDbConnection conn = null;
// Connect to database, collect list of tables
conn = sqlFactory.CreateConnection();
IDbCommand cmd = null;
IDataReader reader = null;
string table;
List <DbTableInfo> dbList = new List <DbTableInfo>();
DbTableInfo dbTableInfo;
if (bw != null)
{
bw.ReportProgress(0, "Connecting ...");
}
if (p_Username.Length == 0)
{
conn.ConnectionString = "Data Source=" + p_Server + ";Initial Catalog=" + p_Database
+ ";Integrated Security=SSPI";
}
else
{
conn.ConnectionString = "Data Source=" + p_Server + ";Initial Catalog=" + p_Database
+ ";user id=" + p_Username + ";password="******"SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME LIKE 'AG%';";
//#else
if (bInclude)
{
// only include tables starting with our prefix
cmd.CommandText = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES" +
" WHERE TABLE_NAME LIKE '" + p_FilterPrefix
+ "%' ;";
}
else
{
cmd.CommandText = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES;";
}
//#endif
reader = cmd.ExecuteReader();
while (reader.Read())
{
table = reader.GetString(0);
if (table.Equals("dtproperties"))
{
continue;
}
if (table.StartsWith("sys"))
{
continue;
}
if ((p_FilterPrefix.Length > 0) && (!bInclude))
{
// exclude tables starting with our prefix
if (table.StartsWith(p_FilterPrefix))
{
continue;
}
}
dbTableInfo = new DbTableInfo(table);
dbTableInfo.GetFields();
dbList.Add(dbTableInfo);
}
reader.Close();
if (bw != null)
{
bw.ReportProgress(50, "Tables collected");
}
foreach (DbTableInfo tableInfo in dbList)
{
tableInfo.CollectFields(conn);
}
if (bw != null)
{
bw.ReportProgress(100, "Fields collected");
}
}
finally
{
if (reader != null)
{
reader.Close();
}
if (cmd != null)
{
cmd.Dispose();
}
conn.Close();
}
return(dbList.ToArray());
}
// get a list of tables
//
// if username is blank, use integrated security
// if bInclude is false, use filter to exclude
public static DbTableInfo[] GetTables(BackgroundWorker bw,
string p_Server, string p_Database,
string p_Username, string p_Password,
bool bInclude, string p_FilterPrefix)
{
System.Data.Common.DbProviderFactory sqlFactory = System.Data.SqlClient.SqlClientFactory.Instance;
IDbConnection conn = null;
// Connect to database, collect list of tables
conn = sqlFactory.CreateConnection();
IDbCommand cmd = null;
IDataReader reader = null;
string table;
List<DbTableInfo> dbList = new List<DbTableInfo>();
DbTableInfo dbTableInfo;
if (bw != null) bw.ReportProgress(0, "Connecting ...");
if (p_Username.Length == 0)
conn.ConnectionString = "Data Source=" + p_Server + ";Initial Catalog=" + p_Database
+ ";Integrated Security=SSPI" ;
else
conn.ConnectionString = "Data Source=" + p_Server + ";Initial Catalog=" + p_Database
+ ";user id=" + p_Username + ";password="******"SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME LIKE 'AG%';";
//#else
if (bInclude)
{
// only include tables starting with our prefix
cmd.CommandText = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES" +
" WHERE TABLE_NAME LIKE '" + p_FilterPrefix
+"%' ;";
}
else
{
cmd.CommandText = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES;";
}
//#endif
reader = cmd.ExecuteReader();
while (reader.Read())
{
table = reader.GetString(0);
if (table.Equals("dtproperties")) continue;
if (table.StartsWith("sys")) continue;
if ((p_FilterPrefix.Length > 0) && (!bInclude))
{
// exclude tables starting with our prefix
if (table.StartsWith(p_FilterPrefix))
continue;
}
dbTableInfo = new DbTableInfo(table);
dbTableInfo.GetFields();
dbList.Add(dbTableInfo);
}
reader.Close();
if (bw != null) bw.ReportProgress(50, "Tables collected");
foreach(DbTableInfo tableInfo in dbList)
{
tableInfo.CollectFields(conn);
}
if (bw != null) bw.ReportProgress(100, "Fields collected");
}
finally
{
if (reader != null) reader.Close();
if (cmd != null) cmd.Dispose();
conn.Close();
}
return dbList.ToArray();
}
