]]> _
_
Public Class ScriptMain
Inherits UserComponent
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Public Overrides Sub Input_ProcessInputRow(ByVal Row As InputBuffer)
Try
Dim inputStream As Stream
Select Case Me.Action
Case CompressionMode.Compress
' Get input.
inputStream = GetInputStream_(Row)
' Compress.
Using compress As DeflateStream = New DeflateStream( _
m_outputStream, _
CompressionMode.Compress, _
True)
Call m_outputStream.SetLength(0)
Call CopyStream_(inputStream, compress)
End Using
Case CompressionMode.Decompress
' Get input.
Dim inputBlob As BlobColumn = CType(Row.Buffer(m_inputIndex), BlobColumn)
inputStream = New MemoryStream(inputBlob.GetBlobData(0, CInt(inputBlob.Length)))
' Decompress.
Using decompress As DeflateStream = New DeflateStream( _
inputStream, _
CompressionMode.Decompress)
Call m_outputStream.SetLength(0)
Call CopyStream_(decompress, m_outputStream)
End Using
End Select
' Store in result column.
Dim resultBlob As BlobColumn = CType(Row.Buffer(m_outputIndex), BlobColumn)
resultBlob.AddBlobData(m_outputStream.ToArray())
Catch ex As Exception
Call FireError_(ex.ToString())
End Try
End Sub ' Input_ProcessInputRow
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Public Overrides Sub PreExecute()
Call MyBase.PreExecute()
Dim input As IDTSInput90 = Me.ComponentMetaData.InputCollection(0)
Dim output As IDTSOutput90 = Me.ComponentMetaData.OutputCollection(0)
' Find input index.
Dim inputCol As IDTSInputColumn90 = input.InputColumnCollection(0)
m_inputIndex = MyBase.HostComponent.BufferManager.FindColumnByLineageID( _
input.Buffer, _
inputCol.LineageID)
m_inputType = inputCol.DataType
' Find output index.
m_outputIndex = MyBase.HostComponent.BufferManager.FindColumnByLineageID( _
input.Buffer, _
output.OutputColumnCollection(0).LineageID)
' Setup output stream used during row processing for efficiency.
m_outputStream = New MemoryStream()
' Setup buffer used for copying streams.
m_buf = New Byte(4096 - 1) {}
End Sub ' PreExecute
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Public Function Validate(ByRef errMessage As String) As Boolean
Dim result As Boolean
Try
If String.IsNullOrEmpty(Me.InputColumn) Then
Throw New ApplicationException("Select input column.")
End If
If String.IsNullOrEmpty(Me.ResultColumn) Then
Throw New ApplicationException("Specify result column name.")
End If
Dim input As IDTSInput90 = Me.ComponentMetaData.InputCollection(0)
If input.InputColumnCollection.Count = 0 Then
' Setup new input column.
Dim virtInput As IDTSVirtualInput90 = input.GetVirtualInput()
Dim virtColumn As IDTSVirtualInputColumn90 = virtInput.VirtualInputColumnCollection( _
Me.InputColumn)
Call virtInput.SetUsageType(virtColumn.LineageID, DTSUsageType.UT_READONLY)
End If
' Setup output column.
Dim output As IDTSOutput90 = Me.ComponentMetaData.OutputCollection(0)
If output.OutputColumnCollection.Count = 0 Then
' Setup new result column.
Dim newColumn As IDTSOutputColumn90 = output.OutputColumnCollection.[New]()
newColumn.Name = Me.ResultColumn
newColumn.SetDataTypeProperties(DataType.DT_IMAGE, 0, 0, 0, 0)
End If
Dim outColumn As IDTSOutputColumn90 = output.OutputColumnCollection(0)
outColumn.Name = Me.ResultColumn
result = True
Catch ex As Exception
result = False
errMessage = ex.Message
End Try
Validate = result
End Function 'Validate
#Region "Properties"
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
Public Property Action() As CompressionMode
Get
Action = m_action
End Get
Set(ByVal value As CompressionMode)
m_action = value
End Set
End Property ' Action
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
Public Property InputColumn() As String
Get
InputColumn = m_inputColumn
End Get
Set(ByVal value As String)
If m_inputColumn <> value Then
If Not String.IsNullOrEmpty(m_inputColumn) Then
' Reset input collection.
Dim input As IDTSInput90 = Me.ComponentMetaData.InputCollection(0)
Call input.InputColumnCollection.RemoveAll()
End If
m_inputColumn = value
End If
End Set
End Property ' InputColumn
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
Public Property ResultColumn() As String
Get
ResultColumn = m_resultColumn
End Get
Set(ByVal value As String)
m_resultColumn = value
End Set
End Property ' ResultColumn
#End Region ' Properties
#Region "Internals"
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Private Sub FireError_(ByVal message As String)
Dim cancel As Boolean
Call MyBase.ComponentMetaData.FireError( _
0, _
"Column Compression", _
message, _
String.Empty, _
0, _
cancel)
End Sub ' FireError_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Private Function GetInputStream_(ByVal row As InputBuffer) As Stream
Dim result As Stream
Select Case m_inputType
Case DataType.DT_BYTES, DataType.DT_IMAGE, DataType.DT_NTEXT, DataType.DT_TEXT
Dim blob As BlobColumn = CType(row.Buffer(m_inputIndex), BlobColumn)
result = New MemoryStream(blob.GetBlobData(0, CInt(blob.Length)))
Case DataType.DT_STR, DataType.DT_WSTR
Dim str As String = row.Buffer.GetString(m_inputIndex)
result = New MemoryStream(Encoding.Default.GetBytes(str))
Case Else
Throw New Exception("Unsupported input data type.")
End Select
GetInputStream_ = result
End Function ' GetInputStream_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Private Sub CopyStream_(ByVal source As Stream, ByVal dest As Stream)
Dim bytesRead As Integer = 0
Do While True
bytesRead = source.Read(m_buf, 0, m_buf.Length)
If bytesRead = 0 Then
' Reached the end.
Exit Do
End If
Call dest.Write(m_buf, 0, bytesRead)
Loop
End Sub ' CopyStream_
#End Region ' Internals
#Region "Attributes"
Private m_action As CompressionMode = CompressionMode.Compress
Private m_inputColumn As String
Private m_resultColumn As String
Private m_inputIndex As Integer
Private m_inputType As DataType
Private m_outputIndex As Integer
Private m_outputStream As MemoryStream
Private m_buf() As Byte
#End Region ' Attributes
End Class ' ScriptMain
]]>ScriptComponent_5b4c54949f324729b497b165ad306078trueCozyRoc.SqlServer.SSIS.ScriptComponentHostPlus, CozyRoc.SSISPlus, Version=1.0.0.0, Culture=neutral, PublicKeyToken=16cf490bb80c34ea