Ejemplo n.º 1
0
        /// <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();
            }
        }
Ejemplo n.º 2
0
        /// <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();
            }
        }
Ejemplo n.º 3
0
 public string GetClassName()
 {
     return(DbTableInfo.GetSingularName(_Table_Name));
 }
Ejemplo n.º 4
0
        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);
        }
Ejemplo n.º 5
0
        // 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());
        }
Ejemplo n.º 6
0
        // 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();
        }