]]> _ _ _ Public Class ScriptMain Inherits UserComponent Public Enum TraceTypes File Table Server End Enum ' TraceTypes ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' Public Overrides Sub CreateNewOutputRows() Dim colsCount As Integer = m_bufIdx.Length Do While m_traceReader.Read() Call Me.OutputBuffer.AddRow() For colIndex As Integer = 0 To colsCount - 1 Dim bufIndex As Integer = m_bufIdx(colIndex) Dim value As Object = m_traceReader(m_readerIdx(colIndex)) If value Is Nothing Then Call Me.OutputBuffer.Buffer.SetNull(bufIndex) Else Me.OutputBuffer.Buffer(bufIndex) = value End If Next Loop Call Me.OutputBuffer.SetEndOfRowset() End Sub ' CreateNewOutputRows ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' Public Overrides Sub PreExecute() Call MyBase.PreExecute() m_traceReader = GetTraceReader_() ' Initialize column indexes. Dim schemaTable As DataTable = m_traceReader.GetSchemaTable() Dim output As IDTSOutput90 = Me.ComponentMetaData.OutputCollection(0) Dim colsCount As Integer = output.OutputColumnCollection.Count m_bufIdx = New Integer(colsCount - 1) {} m_readerIdx = New Integer(colsCount - 1) {} For colIndex As Integer = 0 To colsCount - 1 Dim column As IDTSOutputColumn90 = output.OutputColumnCollection(colIndex) ' Setup buffer index. m_bufIdx(colIndex) = Me.HostComponent.BufferManager.FindColumnByLineageID( _ output.Buffer, _ column.LineageID) ' Setup reader index. Dim columnIndex As Integer = FindColumn_(schemaTable.Rows, column.Name) m_readerIdx(colIndex) = columnIndex Next Call schemaTable.Dispose() End Sub ' PreExecute ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' Public Overrides Sub PostExecute() Call MyBase.PostExecute() If Not m_traceReader Is Nothing Then Call m_traceReader.Close() m_traceReader = Nothing End If End Sub ' PostExecute ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' Public Overrides Sub ReinitializeMetaData() Call MyBase.ReinitializeMetaData() Dim output As IDTSOutput90 = Me.ComponentMetaData.OutputCollection(0) Call output.OutputColumnCollection.RemoveAll() ' Load columns from schema. Dim traceReader As IDataReader = GetTraceReader_() Try Dim schemaTable As DataTable = traceReader.GetSchemaTable() Dim colsCount As Integer = schemaTable.Rows.Count For colIndex As Integer = 0 To colsCount - 1 Dim columnRow As DataRow = schemaTable.Rows(colIndex) Dim columnName As String = CStr(columnRow("ColumnName")) ' Get column info. Dim dtsType As Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType = _ Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_EMPTY Dim length As Integer Dim precision As Integer Dim scale As Integer Dim codePage As Integer GetTypeFromDataRow_( _ columnRow, _ dtsType, _ length, _ precision, _ scale, _ codePage) ' Setup output column. Dim column As IDTSOutputColumn90 = output.OutputColumnCollection.New() column.Name = columnName Call column.SetDataTypeProperties(dtsType, length, precision, scale, codePage) Next Call schemaTable.Dispose() Finally If Not traceReader Is Nothing Then Call traceReader.Close() End If End Try End Sub ' ReinitializeMetaData ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' ' Returns list of current properties based on current state. Public Function GetProperties() As String() Dim result As New ArrayList ' Base properties. Call result.Add("Source") Select Case Me.Source Case TraceTypes.File Call result.Add("File") Case TraceTypes.Table Call result.Add("Server") Call result.Add("Database") Call result.Add("Table") Case TraceTypes.Server Call result.Add("Server") Call result.Add("Profile") End Select Return CType(result.ToArray(GetType(String)), String()) End Function ' GetProperties ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' Public Overrides Function Validate() As DTSValidationStatus Dim result As DTSValidationStatus = MyBase.Validate() If result = DTSValidationStatus.VS_ISVALID Then Dim errMessage As String If Not ValidateUI(errMessage) Then Dim cancel As Boolean Call Me.ComponentMetaData.FireError(0, "Profile Trace", errMessage, String.Empty, 0, cancel) ElseIf Me.ComponentMetaData.OutputCollection(0).OutputColumnCollection.Count = 0 Then ' Have to load new metadata. result = DTSValidationStatus.VS_NEEDSNEWMETADATA End If End If Validate = result End Function ' Validate ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' Public Function ValidateUI(ByRef errMessage As String) As Boolean Dim result As Boolean Try ' Store connection information in the runtime connection collection, too. ' Cannot directly use RuntimeConnectionCollection in the property get/set ' because of issues with multi-threading. Select Case Me.Source Case TraceTypes.File If String.IsNullOrEmpty(Me.File) Then Throw New ApplicationException("Specify trace file connection.") End If Me.ComponentMetaData.RuntimeConnectionCollection("TraceSource").ConnectionManagerID = Me.File ' This is not used. Just set it to avoid validation issues. Me.ComponentMetaData.RuntimeConnectionCollection("TraceProfile").ConnectionManagerID = Me.File Case TraceTypes.Table If String.IsNullOrEmpty(Me.Server) Then Throw New ApplicationException("Select SQL Server connection.") ElseIf String.IsNullOrEmpty(Me.Table) Then Throw New ApplicationException("Select trace table.") End If Me.ComponentMetaData.RuntimeConnectionCollection("TraceSource").ConnectionManagerID = Me.Server ' This is not used. Just set it to avoid validation issues. Me.ComponentMetaData.RuntimeConnectionCollection("TraceProfile").ConnectionManagerID = Me.Server Case TraceTypes.Server If String.IsNullOrEmpty(Me.Server) Then Throw New ApplicationException("Select SQL Server connection.") ElseIf String.IsNullOrEmpty(Me.Profile) Then Throw New ApplicationException("Select trace profile file connection.") End If Me.ComponentMetaData.RuntimeConnectionCollection("TraceSource").ConnectionManagerID = Me.Server Me.ComponentMetaData.RuntimeConnectionCollection("TraceProfile").ConnectionManagerID = Me.Profile End Select result = True Catch ex As Exception result = False errMessage = ex.Message End Try ValidateUI = result End Function 'ValidateUI ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' ' Get list of databases for selected SQL server. Public Function GetDatabases() As String() Dim result As ArrayList = New ArrayList() Dim server As Server = CType( _ Me.DesignConnections(Me.Server).AcquireConnection(Nothing), _ Server) Dim dbCount As Integer = server.Databases.Count For dbIndex As Integer = 0 To dbCount - 1 Call result.Add(server.Databases(dbIndex).Name) Next Return CType(result.ToArray(GetType(String)), String()) End Function ' GetDatabases ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' ' Get list of tables for selected database. Public Function GetTables() As String() Dim result As ArrayList = New ArrayList() Dim server As Server = CType( _ Me.DesignConnections(Me.Server).AcquireConnection(Nothing), _ Server) Dim database As Database = server.Databases(Me.Database) Dim tblCount As Integer = database.Tables.Count For tblIndex As Integer = 0 To tblCount - 1 Call result.Add(database.Tables(tblIndex).ToString()) Next Return CType(result.ToArray(GetType(String)), String()) End Function ' GetTables #Region "Properties" ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' _ _ _ Public Property Source() As TraceTypes Get Source = m_traceType End Get Set(ByVal value As TraceTypes) If m_traceType <> value Then m_traceType = value m_traceSource = String.Empty m_traceParam1 = String.Empty m_traceParam2 = String.Empty End If End Set End Property ' Source ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' _ _ _ Public Property File() As String Get File = m_traceSource End Get Set(ByVal value As String) m_traceSource = value End Set End Property ' File ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' _ _ _ Public Property Server() As String Get Server = m_traceSource End Get Set(ByVal value As String) m_traceSource = value End Set End Property ' Server ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' _ _ _ Public Property Database() As String Get Database = m_traceParam1 End Get Set(ByVal value As String) m_traceParam1 = value End Set End Property ' Database ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' _ _ _ Public Property Table() As String Get Table = m_traceParam2 End Get Set(ByVal value As String) m_traceParam2 = value End Set End Property ' Table ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' _ _ _ Public Property Profile() As String Get Profile = m_traceParam1 End Get Set(ByVal value As String) m_traceParam1 = value End Set End Property ' Profile #End Region ' Properties #Region "Internals" ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' Private ReadOnly Property SMOConnectionType() As String() Get SMOConnectionType = New String() {"SMOServer"} End Get End Property ' SMOConnectionType ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' Private ReadOnly Property FileConnectionType() As String() Get FileConnectionType = New String() {"FILE"} End Get End Property ' FileConnectionType ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' ' Get SSIS type from specified DataRow. Private Sub GetTypeFromDataRow_( _ ByVal dataRow As DataRow, _ ByRef dtsType As Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType, _ ByRef length As Integer, _ ByRef precision As Integer, _ ByRef scale As Integer, _ ByRef codePage As Integer) ' Get type. Dim type As Type = CType(dataRow("DataType"), Type) dtsType = CType( _ GetType(PipelineComponent).InvokeMember( _ "DataRecordTypeToBufferType", _ BindingFlags.InvokeMethod Or BindingFlags.NonPublic Or BindingFlags.Static, _ Nothing, _ Nothing, _ New Object() {type}), _ Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType) If Not TypeOf dataRow("IsLong") Is DBNull Then Select Case dtsType Case Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_WSTR dtsType = Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_NTEXT Case Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_STR dtsType = Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_TEXT Case Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_BYTES dtsType = Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_IMAGE End Select End If ' Get length. length = 0 Select Case dtsType Case Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_WSTR, _ Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_STR, _ Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_BYTES, _ Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_IMAGE, _ Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_NTEXT, _ Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_TEXT length = CInt(dataRow.Item("ColumnSize")) If length > 4000 Then ' Length is more than the maximum in SSIS. Limit to 4000. length = 4000 End If End Select ' Get precision and scale. If dtsType = Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_BYREF_DECIMAL OrElse _ dtsType = Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_NUMERIC Then If dataRow.IsNull("NumericPrecision") Then precision = &H1D Else precision = Convert.ToInt16(dataRow("NumericPrecision")) End If If dataRow.IsNull("NumericScale") Then scale = 0 Else scale = Convert.ToInt16(dataRow("NumericScale")) End If If scale = &HFF Then scale = 0 End If End If ' Get code page. codePage = 0 If dtsType = Microsoft.SqlServer.Dts.Runtime.Wrapper.DataType.DT_STR Then codePage = -1 End If End Sub ' GetTypeFromDataRow_ ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' ' Find column index in DataRowCollection Private Function FindColumn_( _ ByVal rows As DataRowCollection, _ ByVal columnName As String) As Integer Dim result As Integer = -1 Dim rowsCount As Integer = rows.Count For rowIndex As Integer = 0 To rowsCount - 1 If columnName = CStr(rows(rowIndex)("ColumnName")) Then ' Found column. result = rowIndex Exit For End If Next Call Debug.Assert(result <> -1) FindColumn_ = result End Function ' FindColumn_ ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' ' Get IDataReader based on selected trace type. Private Function GetTraceReader_() As IDataReader Dim result As IDataReader Dim errMessage As String If Not ValidateUI(errMessage) Then ' Task doesn't validate. Return Nothing End If Select Case Me.Source Case TraceTypes.File Dim traceFile As TraceFile = New TraceFile() Dim traceFilename As String = CStr(Me.Connections.TraceSource.AcquireConnection(Nothing)) traceFile.InitializeAsReader(traceFilename) result = traceFile Case TraceTypes.Table ' Get SQL connection info. Dim server As Server = CType( _ Me.Connections.TraceSource.AcquireConnection(Nothing), _ Server) Dim sci As SqlConnectionInfo = New SqlConnectionInfo( _ server.ConnectionContext, _ ConnectionType.Sql) sci.DatabaseName = Me.Database ' Get trace table and owner separate. Dim owner As String Dim table As String Call GetOwnerTable_(Me.Table, owner, table) ' Setup trace table. Dim traceTable As TraceTable = New TraceTable() traceTable.InitializeAsReader(sci, table, owner) result = traceTable Case TraceTypes.Server ' Get SQL connection info. Dim server As Server = CType( _ Me.Connections.TraceSource.AcquireConnection(Nothing), _ Server) Dim sci As SqlConnectionInfo = New SqlConnectionInfo( _ server.ConnectionContext, _ ConnectionType.Sql) ' Get trace profile name. Dim traceProfile As String = CStr(Me.Connections.TraceProfile.AcquireConnection(Nothing)) ' Setup trace server. Dim traceServer As TraceServer = New TraceServer() Call traceServer.InitializeAsReader(sci, traceProfile) result = traceServer End Select GetTraceReader_ = result End Function ' GetTraceReader_ ''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''' ' Get owner and table in separate fields. Private Sub GetOwnerTable_( _ ByVal tableCombo As String, _ ByRef owner As String, _ ByRef table As String) Dim combo As String() = tableCombo.Split( _ New String() {"].["}, _ 2, _ StringSplitOptions.None) owner = combo(0).Substring(1) table = combo(1).Substring(0, combo(1).Length - 1) End Sub ' GetOwnerTable_ #End Region ' Internals #Region "Attributes" Private m_traceType As TraceTypes = TraceTypes.File Private m_traceSource As String Private m_traceParam1 As String Private m_traceParam2 As String Private m_traceReader As IDataReader Private m_bufIdx As Integer() Private m_readerIdx As Integer() #End Region ' Attributes End Class ' ScriptMain ]]> ScriptComponent_bd09eb48a8dc4dc18010cf2a314b1aad true CozyRoc.SqlServer.SSIS.ScriptComponentHostPlus, CozyRoc.SSISPlus, Version=1.0.0.0, Culture=neutral, PublicKeyToken=16cf490bb80c34ea