用Codesmith写的一个的从SQL Server向Oracle导数据的模板

CodeSmith从4.1版直接支持Oracle Schema Provider了，最近在做一个SQL Server向Oracle迁移的项目，于是写了一个模板来导数据，感觉比较方便，放出来让大家评评，有不足的地方，请高手之处以便改进。

 1<%@ CodeTemplate Language="C#" TargetLanguage="T-SQL" Description="Export sql" Debug="true" %>
 2<%@ Assembly Name="SchemaExplorer" %>
 3<%@ Import Namespace="SchemaExplorer" %>
 4<%@ Property Name="SourceTable" Type="SchemaExplorer.TableSchema" Category="Context" Description="" %>
 5<%@ Property Name="SortBy" Type="String" Default="" Optional="True" Category="SQL" Description="" OnChanged="" Editor="" EditorBase="" Serializer="" %>
 6<%@ Property Name="Filter" Type="String" Default="" Optional="True" Category="" Description="" OnChanged="" Editor="" EditorBase="" Serializer="" %>
 7<%@ Property Name="IsOracle" Type="Boolean" Default="False" Optional="False" Category="" Description="Script target database is oracle" OnChanged="" Editor="" EditorBase="" Serializer="" %>
 8
 9--Created <%=DateTime.Now %>
-- written by wilson.fu
<%for(int intRow=0;intRow<SourceTableData.Rows.Count;intRow++){ %>
insert into <%=GetTableOwner() %><%=SourceTable.Name %>(<%=AllColumns %>)
values (<%=AllDate(intRow) %>)
<%if(!IsOracle){%>
go
<%}else{%>
;
<%}%>
<%}%>

  1<script runat="template">
  2private DataTable _sourceTableData;
  3private string _allColumns;
  4private string _allData;
  5
  6SourceTableData#region SourceTableData
  7private DataTable SourceTableData
  8{
  9    get
 10     {
 11         if (_sourceTableData == null)
 12         {
 13             _sourceTableData = SourceTable.GetTableData();
 14         }
 15         
 16         return _sourceTableData;
 17     }
 18 }
 19 #endregion
 20 
 21AllColumns#region AllColumns
 22 private string AllColumns
 23 {
 24     get
 25     {
 26         if(_allColumns ==null)
 27         {
 28             string strCols = "";
 29             for(int i=0;i<SourceTable.Columns.Count-1;i++)
 30             {
 31                if(IsOracle)
 32                {
 33                    strCols += SourceTable.Columns[i].Name + ",";
 34                }
 35                else if(SourceTable.Columns[i].Name != "fid")
 36                {
 37                    strCols += SourceTable.Columns[i].Name + ",";
 38                }
 39             }
 40             strCols+=SourceTable.Columns[SourceTable.Columns.Count-1].Name;
 41             _allColumns = strCols;
 42         }
 43         return _allColumns;
 44         
 45     }
 46 }
 47 #endregion
 48 
 49AllData#region AllData
 50 private string AllDate(int intRow)
 51 {
 52    string strVal = "";
 53     int intCol=0;
 54    DataTable dt = SourceTable.GetTableData();
 55    DataView dv = new DataView(dt);
 56    if(SortBy!=string.Empty)
 57    {
 58        dv.Sort = SortBy;
 59    }
 60    
 61    if(Filter!=string.Empty)
 62    {
 63        
 64        dv.RowFilter = Filter;
 65    }
 66    dt = dv.ToTable();
 67     for(;intCol<dt.Columns.Count-1;intCol++)
 68     {
 69        if(IsOracle)
 70        {
 71            strVal += Formated(dt.Rows[intRow][intCol],intCol,intRow) + ",";
 72        }
 73        else if(dt.Columns[intCol].ColumnName != "fid")
 74        {
 75            strVal += Formated(dt.Rows[intRow][intCol],intCol,intRow) + ",";
 76        }
 77     }
 78     strVal += Formated(dt.Rows[intRow][intCol],intCol,intRow);
 79     return  strVal;
 80 }
 81 #endregion
 82 
 83FormatData#region FormatData
 84 private string Formated(object obj,int intCol,int intRow)
 85 {
 86     string strVal = "";
 87     switch(SourceTable.Columns[intCol].DataType)
 88     {
 89        case DbType.String:
 90        case DbType.Guid:
 91             strVal = "'" + obj + "'";
 92             break;
 93         case DbType.Int16:
 94         case DbType.Int32:
 95         case DbType.Int64:
 96         case DbType.Byte:
 97         
 98             strVal = obj.ToString();
 99             break;
        case DbType.DateTime:
        case DbType.Date:
        if(!IsOracle)
        {
            strVal = "'" + obj + "'";
        }
        else
        {
            strVal = string.Format("to_date('{0}','yyyy-mm-dd hh24:mi:ss')"
            ,Convert.ToDateTime(obj).ToString("yyyy-MM-dd HH:mm:ss"));
        }
        break;
        
        case DbType.Boolean:
            if(obj == Convert.DBNull)
            {
                strVal = "0";
            }
            else
            {
                if(Convert.ToBoolean(obj))
                {
                    strVal = "1";
                }
                else
                {
                    strVal = "0";
                }
            }
            break;
        /**//* case DbType.DateTime:
             strVal = "('" + obj + "','yyyy-mm-dd HH24:MI:SS')";
             break;*/
         default:
             strVal = "'" + obj + "'";
             break;
     }
     if (SourceTableData.Rows[intRow][intCol]==DBNull.Value)
         strVal = "null";
     return strVal;
 }
 #endregion
 
 public string GetTableOwner()
 {
     return GetTableOwner(true);
 }
 
 public string GetTableOwner(bool includeDot)
 {
     if (SourceTable.Owner.Length > 0&&!IsOracle)
     {
         return  SourceTable.Owner + ".";
     }
     else
     {
         return "";
     }
 }
 
</script>

漂泊雪狼,转载请注明出处，谢谢