' This code was generated by a tool.
'
' Changes to this file may cause incorrect behavior and will be lost if
' the code is regenerated.
'
'------------------------------------------------------------------------------
Option Strict Off
Option Explicit On
Partial Friend NotInheritable Class MySettings
Inherits System.Configuration.ApplicationSettingsBase
Private Shared m_Value As MySettings
Private Shared m_SyncObject As Object = New Object
_
Public Shared ReadOnly Property Value() As MySettings
Get
If (MySettings.m_Value Is Nothing) Then
System.Threading.Monitor.Enter(MySettings.m_SyncObject)
If (MySettings.m_Value Is Nothing) Then
Try
MySettings.m_Value = New MySettings
Finally
System.Threading.Monitor.Exit(MySettings.m_SyncObject)
End Try
End If
End If
Return MySettings.m_Value
End Get
End Property
End Class
]]>
'The following GUID is for the ID of the typelib if this project is exposed to COM
' Version information for an assembly consists of the following four values:
'
' Major Version
' Minor Version
' Build Number
' Revision
'
' You can specify all the values or you can default the Build and Revision Numbers
' by using the '*' as shown below:
'
]]>]]> _
Public Class InputBuffer
Inherits ScriptBufferPlus
Dim ErrorOutputOutputID As Integer
Public Sub New(ByVal Component As ScriptComponent, ByVal ObjectID As Integer, ByVal IsInput As Boolean, ByVal Buffer As PipelineBuffer)
MyBase.New(Component, ObjectID, IsInput, Buffer)
ErrorOutputOutputID = Component.ComponentMetaData.OutputCollection("Error Output").ID
End Sub
Public Overrides ReadOnly Property StaticInputColumns() As String()
Get
Return New String() {}
End Get
End Property
Public WriteOnly Property [ErrorDescription]() As String
Set
Me(0) = Value
End Set
End Property
Public WriteOnly Property [ErrorDescription_IsNull] As Boolean
Set
If (value)
SetNull(0)
Else
Throw new InvalidOperationException("IsNull property cannot be set to False. Assign a value to the column instead.")
End If
End Set
End Property
Public Sub DirectRowToErrorOutput()
MyBase.DirectRow(ErrorOutputOutputID)
End Sub
Public Overrides ReadOnly Property StaticOutputColumns() As String()
Get
Return New String() {"ErrorDescription"}
End Get
End Property
Public Overloads Function NextRow() As Boolean
NextRow = MyBase.NextRow()
End Function
Public Overloads Function EndOfRowset() As Boolean
EndOfRowset = MyBase.EndOfRowset
End Function
End Class
]]>{A860303F-1F3F-4691-B57E-529FC101A107};{F184B08F-C81C-45F6-A57F-5ABD9991F28F}DebugAnyCPULibraryScriptComponent_2e3d4c56d0694f9bbd032f1e0950bec1.vbprojScriptComponent_2e3d4c56d0694f9bbd032f1e0950bec1.vbprojOnBinaryOffOn{0241DA19-8516-41A1-9BA8-F82E3331FF03}truetruetruefalsebin\falsefalse42016,42017,42018,42019,42032falsetruefalsetruebin\falsefalse42016,42017,42018,42019,42032CodeVbMyResourcesResXFileCodeGeneratorResources.Designer.vbMy.ResourcesTrueTrueResources.resxCodeSettingsSingleFileGeneratorSettings.Designer.vbTrueSettings.settingsCodeSSIS_ScriptComponent
]]>
' This code was generated by a tool.
'
' Changes to this file may cause incorrect behavior and will be lost if
' the code is regenerated.
'
'------------------------------------------------------------------------------
Option Strict Off
Option Explicit On
Namespace My.Resources
'''
''' A strongly-typed resource class, for looking up localized strings, etc.
'''
'This class was auto-generated by the Strongly Typed Resource Builder
'class via a tool like ResGen or Visual Studio.NET.
'To add or remove a member, edit your .ResX file then rerun ResGen
'with the /str option, or rebuild your VS project.
Class MyResources
Private Shared _resMgr As System.Resources.ResourceManager
Private Shared _resCulture As System.Globalization.CultureInfo
Friend Sub New()
MyBase.New
End Sub
'''
''' Returns the cached ResourceManager instance used by this class.
''' _
Public Shared ReadOnly Property ResourceManager() As System.Resources.ResourceManager
Get
If (_resMgr Is Nothing) Then
Dim temp As System.Resources.ResourceManager = New System.Resources.ResourceManager("My.Resources.MyResources", GetType(MyResources).Assembly)
_resMgr = temp
End If
Return _resMgr
End Get
End Property
'''
''' Overrides the current thread's CurrentUICulture property for all
''' resource lookups using this strongly typed resource class.
''' _
Public Shared Property Culture() As System.Globalization.CultureInfo
Get
Return _resCulture
End Get
Set
_resCulture = value
End Set
End Property
End Class
End Namespace
]]> _
Public Class UserComponent
Inherits ScriptComponentPlus
Public Connections As New Connections(Me)
Public Variables As New Variables(Me)
Public Overrides Sub ProcessInput(ByVal InputID As Integer, ByVal Buffer As PipelineBuffer)
If InputID = MyBase.ComponentMetaData.InputCollection("Input").ID Then
Input_ProcessInput(New InputBuffer(Me, InputID, True, Buffer))
End If
End Sub
Public Overridable Sub Input_ProcessInput(ByVal Buffer As InputBuffer)
While Buffer.NextRow()
Input_ProcessInputRow(Buffer)
End While
End Sub
Public Overridable Sub Input_ProcessInputRow(ByVal Row As InputBuffer)
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 Connection() As IDTSConnectionManager100
Get
Return ParentComponent.ComponentMetaData.RuntimeConnectionCollection("Connection").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
]]>text/microsoft-resx2.0System.Resources.ResXResourceReader, System.Windows.Forms, Version=2.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089System.Resources.ResXResourceWriter, System.Windows.Forms, Version=2.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089]]> _
_
_
_
_
_
Public Class ScriptMain
Inherits UserComponent
Private Const ColumnSeparator As String = "," + vbLf
Private Const Parameter_1P As String = "p{0}"
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Supported destination table actions.
Public Enum ActionTypes
Insert
InsertUpdate ' try an insert first, if it fails based on PK defined in the destination then update
UpdateInsert ' try an update first based on the PK, if it fails then insert
Update
Delete
End Enum ' ActionTypes
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Public Overrides Sub Input_ProcessInputRow(ByVal Row As InputBuffer)
' Setup command parameters.
Dim colsCount As Integer = m_inputIndexes.Length
For colIndex As Integer = 0 To colsCount - 1
Dim bufIndex As Integer = m_inputIndexes(colIndex)
Debug.Assert(bufIndex <> 0)
Dim parameter As DbParameter = m_command.Parameters(colIndex)
Dim value As Object = Row.Buffer(bufIndex)
If TypeOf value Is BlobColumn Then
If parameter.DbType = DbType.Binary Then
value = Row.Buffer.GetBlobData(bufIndex, 0, parameter.Size)
Else
' This is long string.
value = Row.Buffer.GetString(bufIndex)
End If
End If
parameter.Value = value
Next
Try
Select Case Me.Action
Case ActionTypes.Insert
m_command.CommandText = m_insertSql
Call m_command.ExecuteNonQuery()
m_insertedCount += 1
Case ActionTypes.InsertUpdate
Try
m_command.CommandText = m_insertSql
Call m_command.ExecuteNonQuery()
m_insertedCount += 1
Catch ex As Exception
' Insert failed. Try update.
m_command.CommandText = m_updateSql
Call m_command.ExecuteNonQuery()
m_updatedCount += 1
End Try
Case ActionTypes.UpdateInsert
m_command.CommandText = m_selectSql
Dim hasRow As Object = m_command.ExecuteScalar()
If hasRow Is Nothing Then
' Row doesn't exist. Insert new one.
m_command.CommandText = m_insertSql
Call m_command.ExecuteNonQuery()
m_insertedCount += 1
Else
m_command.CommandText = m_updateSql
Call m_command.ExecuteNonQuery()
m_updatedCount += 1
End If
Case ActionTypes.Update
m_command.CommandText = m_updateSql
Call m_command.ExecuteNonQuery()
m_updatedCount += 1
Case ActionTypes.Delete
m_command.CommandText = m_selectSql
Dim hasRow As Object = m_command.ExecuteScalar()
If Not hasRow Is Nothing Then
' Row exists. Delete it.
m_command.CommandText = m_deleteSql
Call m_command.ExecuteNonQuery()
m_deletedCount += 1
End If
End Select
Catch ex As Exception
Select Case Me.ErrorRowDisposition
Case DTSRowDisposition.RD_IgnoreFailure
' Do nothing.
Case DTSRowDisposition.RD_RedirectRow
Row.ErrorDescription = ex.Message
Call Row.DirectRowToErrorOutput()
Case DTSRowDisposition.RD_FailComponent, DTSRowDisposition.RD_NotUsed
Call FireError_(ex.Message)
End Select
End Try
End Sub ' Input_ProcessInputRow
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Public Overrides Sub PreExecute()
Call MyBase.PreExecute()
Dim input As IDTSInput100 = MyBase.ComponentMetaData.InputCollection(0)
' Initialize command.
Dim conn As DbConnection = GetConnection_(False)
m_command = conn.CreateCommand()
Dim primaryKeys As ArrayList = New ArrayList()
If Not String.IsNullOrEmpty(Me.PrimaryKeys) Then
primaryKeys = New ArrayList(Me.PrimaryKeys.Split( _
New String() {ColumnSeparator}, _
StringSplitOptions.None))
End If
Dim colsList As ArrayList = New ArrayList(primaryKeys)
' Include update columns (other than key).
For Each inputCol As IDTSInputColumn100 In input.InputColumnCollection
If inputCol.ExternalMetadataColumnID = 0 Then
' Column not selected. Skip it.
Continue For
End If
Dim extCol As IDTSExternalMetadataColumn100 = input.ExternalMetadataColumnCollection.GetObjectByID( _
inputCol.ExternalMetadataColumnID)
If Not primaryKeys.Contains(extCol.Name) Then
' Key columns will be at the end.
Call colsList.Insert(0, extCol.Name)
End If
Next
' Setup buffer indexes and command parameters.
Dim colsCount As Integer = colsList.Count
m_inputIndexes = New Integer(colsCount - 1) {}
Dim parameters() As DbParameter = New DbParameter(colsCount - 1) {}
For Each inputCol As IDTSInputColumn100 In input.InputColumnCollection
If inputCol.ExternalMetadataColumnID = 0 Then
' Column not selected. Skip it.
Continue For
End If
Dim extCol As IDTSExternalMetadataColumn100 = input.ExternalMetadataColumnCollection.GetObjectByID( _
inputCol.ExternalMetadataColumnID)
Dim colName As String = extCol.Name
Dim paramIndex As Integer = colsList.IndexOf(colName)
Call Debug.Assert(paramIndex <> -1)
' Setup parameter index.
m_inputIndexes(paramIndex) = Me.HostComponent.BufferManager.FindColumnByLineageID( _
input.Buffer, _
inputCol.LineageID)
' Setup command parameter.
Dim parameter As DbParameter = m_command.CreateParameter()
parameter.ParameterName = String.Format(Parameter_1P, 1 + paramIndex)
Call SetupParameter_(inputCol, parameter)
parameters(paramIndex) = parameter
Next
' Include parameters to command.
For paramIndex As Integer = 0 To colsCount - 1
Call m_command.Parameters.Add(parameters(paramIndex))
Next
' Setup SQL statements.
Dim valueList As ArrayList = New ArrayList()
Dim whereList As ArrayList = New ArrayList()
Dim setList As ArrayList = New ArrayList()
Dim selList As ArrayList = New ArrayList()
For colIndex As Integer = 0 To colsCount - 1
Dim colName As String = CStr(colsList(colIndex))
Dim paramName As String = GetParameterName_( _
conn, _
parameters(colIndex).ParameterName)
Call valueList.Add(paramName)
Dim keyValue As String = String.Format("{0} = {1}", colName, paramName)
If primaryKeys.Contains(colName) Then
' Key column.
Call whereList.Add(keyValue)
Else
Call setList.Add(keyValue)
Call selList.Add(paramName)
End If
Next
Dim whereClause As String = String.Join( _
" AND ", _
CType(whereList.ToArray(GetType(String)), String()))
m_selectSql = String.Format( _
"SELECT {0} FROM {1} WHERE {2}", _
String.Join(",", CType(primaryKeys.ToArray(GetType(String)), String())), _
Me.Table, _
whereClause)
m_insertSql = String.Format( _
"INSERT INTO {0} ({1}) VALUES({2})", _
Me.Table, _
String.Join(",", CType(colsList.ToArray(GetType(String)), String())), _
String.Join(",", CType(valueList.ToArray(GetType(String)), String())))
m_updateSql = String.Format( _
"UPDATE {0} SET {1} WHERE {2}", _
Me.Table, _
String.Join(",", CType(setList.ToArray(GetType(String)), String())), _
whereClause)
m_deleteSql = String.Format( _
"DELETE FROM {0} WHERE {1}", _
Me.Table, _
whereClause)
End Sub ' PreExecute
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Public Overrides Sub PostExecute()
Call MyBase.PostExecute()
If Not m_command Is Nothing Then
Dim conn As DbConnection = m_command.Connection
Call m_command.Dispose()
m_command = Nothing
Call conn.Dispose()
End If
' Store diagnostics.
Call SetVariable_(Me.InsertedVariable, m_insertedCount)
Call SetVariable_(Me.UpdatedVariable, m_updatedCount)
Call SetVariable_(Me.DeletedVariable, m_deletedCount)
End Sub ' PostExecute
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Returns list of current properties based on current state.
Public Function GetProperties() As String()
Dim result As New ArrayList
' Setup base properties.
Call result.Add("ConnectionType")
Call result.Add("Connection")
Call result.Add("Table")
Call result.Add("Action")
Call result.Add("ErrorRowDisposition")
Call result.Add("InsertedVariable")
Call result.Add("UpdatedVariable")
Call result.Add("DeletedVariable")
Select Case Me.Action
Case ActionTypes.InsertUpdate, ActionTypes.UpdateInsert, ActionTypes.Update, ActionTypes.Delete
Call result.Add("PrimaryKeys")
End Select
Return CType(result.ToArray(GetType(String)), String())
End Function ' GetProperties
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Public Function Validate(ByRef errMessage As String) As Boolean
Dim result As Boolean
Try
If String.IsNullOrEmpty(Me.Connection) Then
Throw New ApplicationException("Select connection.")
End If
If String.IsNullOrEmpty(Me.Table) Then
Throw New ApplicationException("Select destination table.")
End If
If Me.Action <> ActionTypes.Insert AndAlso _
String.IsNullOrEmpty(Me.PrimaryKeys) Then
Throw New ApplicationException("Specify table primary keys.")
End If
Dim primaryKeys As String() = Me.PrimaryKeys.Split( _
New String() {ColumnSeparator}, _
StringSplitOptions.None)
Dim input As IDTSInput100 = Me.ComponentMetaData.InputCollection(0)
If input.IsAttached AndAlso _
(input.InputColumnCollection.Count = 0 OrElse _
input.InputColumnCollection.Count < primaryKeys.Length) Then
Throw New Exception("Map input columns.")
End If
' Store connection information in the runtime connection collection, too.
' Cannot directly use RuntimeConnectionCollection in the property get/set
' because of issues with multi-threading.
Me.ComponentMetaData.RuntimeConnectionCollection("Connection").ConnectionManagerID = Me.Connection
result = True
Catch ex As Exception
result = False
errMessage = ex.Message
End Try
Validate = result
End Function 'Validate
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Public Overrides Sub ReinitializeMetaData()
Call MyBase.ReinitializeMetaData()
If MyBase.ComponentMetaData.InputCollection.Count = 0 Then
' At least one input must exist.
Throw New Exception("No inputs.")
End If
If String.IsNullOrEmpty(Me.Connection) Then
Throw New Exception("Connection not selected.")
End If
If String.IsNullOrEmpty(Me.Table) Then
Throw New Exception("Destination table not selected.")
End If
Dim input As IDTSInput100 = MyBase.ComponentMetaData.InputCollection(0)
Dim tblCols As DataTable = GetColumns_()
' Cleanup.
input.ExternalMetadataColumnCollection.IsUsed = True
Call input.InputColumnCollection.RemoveAll()
Call input.ExternalMetadataColumnCollection.RemoveAll()
' Setup column metadata.
Dim colsCount As Integer = tblCols.Rows.Count
For colIndex As Integer = 0 To colsCount - 1
Dim columnRow As DataRow = tblCols.Rows(colIndex)
If Me.Action = ActionTypes.Delete AndAlso _
Not CBool(columnRow("IsKey")) Then
' Include only key columns for Delete action.
Continue For
End If
Dim column As IDTSExternalMetadataColumn100 = input.ExternalMetadataColumnCollection.NewAt( _
colIndex)
column.Name = CStr(columnRow("ColumnName"))
' Get type.
Dim rawType As Type = CType(columnRow("DataType"), Type)
Dim columnType As DataType = DataRecordTypeToBufferType_(rawType)
If CBool(columnRow("IsLong")) Then
' This is extended type. Have to modify the column type a bit.
Select Case columnType
Case DataType.DT_WSTR
columnType = DataType.DT_NTEXT
Case DataType.DT_STR
columnType = DataType.DT_TEXT
Case DataType.DT_BYTES
columnType = DataType.DT_IMAGE
End Select
End If
' Get length.
Dim length As Integer = 0
Select Case columnType
Case DataType.DT_WSTR, DataType.DT_STR, DataType.DT_BYTES, _
DataType.DT_IMAGE, DataType.DT_NTEXT, DataType.DT_TEXT
length = CInt(columnRow("ColumnSize"))
End Select
' Get code page.
Dim codepage As Integer = 0
If columnType = DataType.DT_STR Then
codepage = -1
End If
' Get precision and scale.
Dim precision As Integer = 0
Dim scale As Integer = 0
If columnType = DataType.DT_NUMERIC OrElse columnType = DataType.DT_BYREF_DECIMAL Then
If columnRow.IsNull("NumericPrecision") Then
precision = 29
Else
precision = CInt(columnRow("NumericPrecision"))
End If
If columnRow.IsNull("NumericScale") Then
scale = 0
Else
scale = CInt(columnRow("NumericScale"))
End If
If scale = 255 Then
scale = 0
End If
End If
column.DataType = columnType
column.Length = length
column.Precision = precision
column.Scale = scale
column.CodePage = codepage
Next
End Sub ' ReinitializeMetaData
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Returns list of tables.
Public Function GetTables() As Object()
Dim result As ArrayList = New ArrayList()
Try
If String.IsNullOrEmpty(Me.Connection) Then
Throw New Exception("Select connection.")
End If
' Get tables.
Using conn As DbConnection = GetConnection_(True)
Dim tableType As String = "TABLE"
If TypeOf conn Is SqlConnection Then
' SQL uses different table type identifier.
tableType = "BASE TABLE"
End If
Dim tblTables As DataTable = conn.GetSchema( _
"Tables", _
New String() {Nothing, Nothing, Nothing, tableType})
' Get table schema column name.
Dim schemaCol As String = tblTables.Columns(1).ColumnName
For Each tableRow As DataRow In tblTables.Rows
Dim tableSchema As String = CStr(tableRow(schemaCol))
Dim tableName As String = CStr(tableRow("table_name"))
Call result.Add(String.Format("[{0}].[{1}]", tableSchema, tableName))
Next
Call result.Sort()
End Using
Catch ex As Exception
Call MsgBox(ex.Message, MsgBoxStyle.OkOnly Or MsgBoxStyle.Exclamation)
End Try
GetTables = result.ToArray()
End Function ' GetTables
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Returns selected destination table columns.
Public Function GetColumns() As Object()
Dim result As ArrayList = New ArrayList
Try
Dim tblCols As DataTable = GetColumns_()
For Each columnRow As DataRow In tblCols.Rows
Call result.Add(columnRow("ColumnName"))
Next
Call result.Sort()
Catch ex As Exception
Call MsgBox(ex.Message, MsgBoxStyle.OkOnly Or MsgBoxStyle.Exclamation)
End Try
GetColumns = result.ToArray()
End Function ' GetColumns
#Region "Properties"
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
_
_
_
Public Property ConnectionType() As String
Get
ConnectionType = m_connectionType
End Get
Set(ByVal value As String)
If m_connectionType <> value Then
m_connectionType = value
Me.Connection = String.Empty
End If
End Set
End Property ' ConnectionType
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
_
_
Public Property Connection() As String
Get
Connection = m_connection
End Get
Set(ByVal value As String)
If m_connection <> value Then
m_connection = value
Me.Table = String.Empty
End If
End Set
End Property ' Connection
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
_
_
Public Property Table() As String
Get
Table = m_table
End Get
Set(ByVal value As String)
If m_table <> value Then
m_table = value
Me.PrimaryKeys = String.Empty
If Me.Action <> ActionTypes.Insert Then
' Initialize primary keys.
Me.PrimaryKeys = GetPrimaryKeys_()
End If
End If
End Set
End Property ' Table
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
_
_
Public Property Action() As ActionTypes
Get
Action = m_action
End Get
Set(ByVal value As ActionTypes)
If m_action <> value Then
m_action = value
Me.PrimaryKeys = String.Empty
If Me.Action <> ActionTypes.Insert Then
' Initialize primary keys.
Me.PrimaryKeys = GetPrimaryKeys_()
End If
End If
End Set
End Property ' Action
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
_
Public Property PrimaryKeys() As String
Get
PrimaryKeys = m_primaryKeys
End Get
Set(ByVal value As String)
m_primaryKeys = value
End Set
End Property ' PrimaryKeys
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
_
Public Property ErrorRowDisposition() As DTSRowDisposition
Get
ErrorRowDisposition = m_errorRowDisposition
End Get
Set(ByVal value As DTSRowDisposition)
m_errorRowDisposition = value
End Set
End Property ' ErrorRowDisposition
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
_
Public Property InsertedVariable() As String
Get
InsertedVariable = m_insertedVar
End Get
Set(ByVal value As String)
m_insertedVar = value
End Set
End Property ' InsertedVariable
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
_
Public Property UpdatedVariable() As String
Get
UpdatedVariable = m_updatedVar
End Get
Set(ByVal value As String)
m_updatedVar = value
End Set
End Property ' UpdatedVariable
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
_
_
_
Public Property DeletedVariable() As String
Get
DeletedVariable = m_deletedVar
End Get
Set(ByVal value As String)
m_deletedVar = value
End Set
End Property ' DeletedVariable
#End Region ' Properties
#Region "Internals"
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Returns selected destination table primary keys.
Private Function GetPrimaryKeys_() As String
Dim result As ArrayList = New ArrayList
Try
Dim tblCols As DataTable = GetColumns_()
For Each columnRow As DataRow In tblCols.Rows
If CBool(columnRow("IsKey")) Then
' Key column.
Call result.Add(columnRow("ColumnName"))
End If
Next
Catch ex As Exception
' Silently catch the error.
End Try
GetPrimaryKeys_ = String.Join( _
ColumnSeparator, _
CType(result.ToArray(GetType(String)), String()))
End Function ' GetPrimaryKeys_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Private Function GetColumns_() As DataTable
Dim result As DataTable
Using conn As DbConnection = GetConnection_(True)
Using cmd As DbCommand = conn.CreateCommand()
cmd.CommandText = String.Format("SELECT * FROM {0}", Me.Table)
Dim reader As DbDataReader = cmd.ExecuteReader( _
CommandBehavior.SchemaOnly Or CommandBehavior.KeyInfo)
result = reader.GetSchemaTable()
End Using
End Using
GetColumns_ = result
End Function ' GetColumns_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Returns SSIS data type from .NET type. Uses private PipelineComponent method.
Private Function DataRecordTypeToBufferType_(ByVal type As Type) As DataType
DataRecordTypeToBufferType_ = CType( _
GetType(PipelineComponent).InvokeMember( _
"DataRecordTypeToBufferType", _
BindingFlags.InvokeMethod Or BindingFlags.NonPublic Or BindingFlags.Static, _
Nothing, _
Nothing, _
New Object() {type}), _
DataType)
End Function ' DataRecordTypeToBufferType_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Setup parameter data type.
Private Sub SetupParameter_( _
ByVal inputCol As IDTSInputColumn100, _
ByVal parameter As DbParameter)
parameter.DbType = GetDbType_(inputCol.DataType)
parameter.Size = inputCol.Length
If parameter.Size = 0 AndAlso parameter.DbType = DbType.Binary Then
parameter.Size = 8000
End If
' Get precision.
Dim precision As Byte = CByte(inputCol.Precision)
If parameter.DbType = DbType.Decimal AndAlso precision = 0 Then
' Decimal type has precision 29.
' http://msdn.microsoft.com/en-us/library/ms141036.aspx
precision = 29
End If
Dim scale As Byte = CByte(inputCol.Scale)
If parameter.DbType = DbType.DateTime AndAlso scale = 0 Then
' DateTime has maximum scale of 3.
' http://msdn.microsoft.com/en-us/library/ms141036%28v=sql.100%29.aspx
scale = 3
End If
' Setup precision and scale.
Dim iDbDataParameter As IDbDataParameter = parameter
iDbDataParameter.Precision = precision
iDbDataParameter.Scale = scale
End Sub ' SetupParameter_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Return ADO.NET data type from SSIS data type.
Private Function GetDbType_(ByVal dt As DataType) As DbType
Select Case dt
Case DataType.DT_BOOL
Return DbType.Boolean
Case DataType.DT_STR, DataType.DT_WSTR, DataType.DT_TEXT, DataType.DT_NTEXT
Return DbType.String
Case DataType.DT_I1
Return DbType.SByte
Case DataType.DT_UI1
Return DbType.Byte
Case DataType.DT_I2
Return DbType.Int16
Case DataType.DT_UI2
Return DbType.UInt16
Case DataType.DT_I4
Return DbType.Int32
Case DataType.DT_UI4
Return DbType.UInt32
Case DataType.DT_I8
Return DbType.Int64
Case DataType.DT_UI8
Return DbType.UInt64
Case DataType.DT_R4
Return DbType.Single
Case DataType.DT_R8
Return DbType.Double
Case DataType.DT_NUMERIC, DataType.DT_DECIMAL
Return DbType.Decimal
Case DataType.DT_CY
Return DbType.Currency
Case DataType.DT_DBTIMESTAMP
Return DbType.DateTime
Case DataType.DT_DBDATE
Return DbType.Date
Case DataType.DT_DBTIME
Return DbType.Time
Case DataType.DT_BYTES, DataType.DT_IMAGE
Return DbType.Binary
End Select
Throw New ArgumentException(String.Format("Unhandled type. {0}", dt))
End Function ' GetDbType_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Returns parameter name for specified connection type.
Private Function GetParameterName_(ByVal conn As DbConnection, ByVal parameter As String) As String
Dim result As String = parameter
If TypeOf conn Is SqlConnection Then
result = "@" + parameter
Else
' Most probably ODBC or OLEDB
result = "?"
End If
GetParameterName_ = result
End Function ' GetParameterName_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Private Sub FireError_(ByVal message As String)
Dim cancel As Boolean = False
Call MyBase.ComponentMetaData.FireError( _
0, _
"Table Update Destination", _
message, _
String.Empty, _
0, _
cancel)
End Sub ' FireError_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
' Get connection object.
Private Function GetConnection_(ByVal design As Boolean) As DbConnection
Dim result As DbConnection
If design Then
If Me.ConnectionType = "OLEDB" Then
Dim manager As ConnectionManager = MyBase.DesignConnections(Me.Connection)
Dim managerParams As IDTSConnectionManagerDatabaseParameters100 = CType( _
manager.InnerObject, _
IDTSConnectionManagerDatabaseParameters100)
result = CType(managerParams.GetConnectionForSchema(), DbConnection)
Else
result = CType( _
MyBase.DesignConnections(Me.Connection).AcquireConnection(Nothing), _
DbConnection)
End If
Else
If Me.ConnectionType = "OLEDB" Then
Dim host As IDTSObjectHost100 = CType(MyBase.Connections.Connection, IDTSObjectHost100)
Dim managerParams As IDTSConnectionManagerDatabaseParameters100 = CType( _
host.InnerObject, _
IDTSConnectionManagerDatabaseParameters100)
result = CType(managerParams.GetConnectionForSchema(), DbConnection)
Else
result = CType( _
MyBase.Connections.Connection.AcquireConnection(Nothing), _
DbConnection)
End If
End If
GetConnection_ = result
End Function ' GetConnection_
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
Private Sub SetVariable_(ByVal varName As String, ByVal value As Object)
If String.IsNullOrEmpty(varName) Then
' Variable not set.
Exit Sub
End If
Dim vars As IDTSVariables100 = Nothing
Call MyBase.VariableDispenser.LockOneForWrite(varName, vars)
Try
vars(varName).Value = value
Finally
Call vars.Unlock()
End Try
End Sub ' SetVariable_
#End Region ' Internals
#Region "Attributes"
Private m_connectionType As String
Private m_connection As String
Private m_table As String
Private m_primaryKeys As String
Private m_action As ActionTypes
Private m_errorRowDisposition As DTSRowDisposition
Private m_insertedVar As String
Private m_updatedVar As String
Private m_deletedVar As String
Private m_command As DbCommand
Private m_inputIndexes() As Integer
Private m_selectSql As String
Private m_insertSql As String
Private m_updateSql As String
Private m_deleteSql As String
Private m_insertedCount As Integer
Private m_updatedCount As Integer
Private m_deletedCount As Integer
#End Region ' Attributes
End Class ' ScriptMain
]]>CozyRoc.SqlServer.SSIS.ScriptComponentHostPlus, CozyRoc.SSISPlus.2008, Version=1.0.0.0, Culture=neutral, PublicKeyToken=16cf490bb80c34eaScriptComponent_2e3d4c56d0694f9bbd032f1e0950bec1VisualBasic