_ _ _ _ _ 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 IDTSOutput100 = 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 IDTSOutputColumn100 = 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 IDTSOutput100 = 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 IDTSOutputColumn100 = 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 = String.Empty 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 = Nothing Dim errMessage As String = String.Empty 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 = String.Empty Dim table As String = String.Empty 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 ]]> _ Public Class OutputBuffer Inherits ScriptBufferPlus Public Sub New(ByVal Component As ScriptComponent, ByVal ObjectID As Integer, ByVal IsInput As Boolean, ByVal Buffer As PipelineBuffer, ByVal OutputMap As OutputNameMap) MyBase.New(Component, ObjectID, IsInput, Buffer, OutputMap) End Sub Public Overrides ReadOnly Property StaticInputColumns() As String() Get Return New String() {} End Get End Property Public Overrides ReadOnly Property StaticOutputColumns() As String() Get Return New String() {} End Get End Property Public Overloads Sub AddRow() MyBase.AddRow() End Sub Public Overloads Sub SetEndOfRowset() MyBase.SetEndOfRowset() End Sub Public Overloads Function EndOfRowset() As Boolean EndOfRowset = MyBase.EndOfRowset End Function End Class ]]> {30D016F9-3734-4E33-A861-5E7D899E18F3};{F184B08F-C81C-45F6-A57F-5ABD9991F28F} Debug AnyCPU 8.0.30703 2.0 {FCE7DEB0-BD42-4853-BB71-E80A52A8A950} Library My Project ScriptComponent_bd09eb48a8dc4dc18010cf2a314b1aad ScriptComponent_bd09eb48a8dc4dc18010cf2a314b1aad v4.0 512 Binary On On true full false .\bin\Debug\ false true true prompt 4 false true .\bin\Release\ false false true prompt 4 False C:\Program Files (x86)\Microsoft SQL Server\120\SDK\Assemblies\CozyRoc.SSISPlus.2014.dll False C:\Program Files (x86)\Microsoft SQL Server\120\SDK\Assemblies\Microsoft.SqlServer.ConnectionInfoExtended.dll False C:\Program Files (x86)\Microsoft SQL Server\120\SDK\Assemblies\Microsoft.SqlServer.Management.Sdk.Sfc.dll False False False False False False SSIS_SC120 ]]> _ Public Class UserComponent Inherits ScriptComponentPlus Public Connections As New Connections(Me) Public Variables As New Variables(Me) Public OutputBuffer As OutputBuffer Public Overridable Sub FinishOutputs() End Sub Private Sub MarkOutputsFinished() If OutputBuffer IsNot Nothing Then OutputBuffer.SetEndOfRowset OutputBuffer = Nothing End If End Sub Public Overrides Sub PrimeOutput(ByVal Outputs As Integer, ByVal OutputIDs() As Integer, ByVal Buffers() As PipelineBuffer, ByVal OutputMap As OutputNameMap) For I As Integer = 0 To Outputs - 1 If OutputIDs(I) = GetOutputID(OutputMap, "Output") Then OutputBuffer = New OutputBuffer(Me, OutputIDs(I), False, Buffers(I), OutputMap) End If Next CreateNewOutputRows() FinishOutputs() MarkOutputsFinished() End Sub Public Overridable Sub CreateNewOutputRows() End Sub End Class Public Class Connections Dim ParentComponent As ScriptComponent _ Public Sub New(ByVal Component As ScriptComponent) ParentComponent = Component End Sub Public ReadOnly Property TraceSource() As IDTSConnectionManager100 Get Return ParentComponent.ComponentMetaData.RuntimeConnectionCollection("TraceSource").ConnectionManager End Get End Property Public ReadOnly Property TraceProfile() As IDTSConnectionManager100 Get Return ParentComponent.ComponentMetaData.RuntimeConnectionCollection("TraceProfile").ConnectionManager End Get End Property End Class Public Class Variables Dim ParentComponent As ScriptComponent _ Public Sub New(ByVal Component As ScriptComponent) ParentComponent = Component End Sub End Class ]]> ScriptComponent_bd09eb48a8dc4dc18010cf2a314b1aad msBuild ScriptComponent_bd09eb48a8dc4dc18010cf2a314b1aad {184DDC77-4B94-464A-8728-57DB155D2239} ]]>CozyRoc.SqlServer.SSIS.ScriptComponentHostPlus, CozyRoc.SSISPlus.2014, Version=1.0.0.0, Culture=neutral, PublicKeyToken=16cf490bb80c34eaScriptComponent_bd09eb48a8dc4dc18010cf2a314b1aadVisualBasic