sql errors and exceptions handling

Author: tomFlidr

Connection.vb

@@ -1,33 +1,45 @@
 Imports System.Data.Common
 Imports System.Data.SqlClient
+Imports System.Reflection
 
 Public Class Connection
-	Inherits Databasic.Connection
+    Inherits Databasic.Connection
 
-	Public Overrides ReadOnly Property Provider As DbConnection
-		Get
-			Return Me._provider
-		End Get
-	End Property
-	Private _provider As SqlConnection
+    Public Overrides ReadOnly Property Provider As DbConnection
+        Get
+            Return Me._provider
+        End Get
+    End Property
+    Private _provider As SqlConnection
 
-	Public Overrides ReadOnly Property ProviderResource As System.Type = GetType(ProviderResource)
+    Public Overrides ReadOnly Property ProviderResource As System.Type = GetType(ProviderResource)
 
-	Public Overrides ReadOnly Property ClientName As String = "System.Data.SqlClient"
+    Public Overrides ReadOnly Property ClientName As String = "System.Data.SqlClient"
 
-	Public Overrides ReadOnly Property Statement As System.Type = GetType(Statement)
+    Public Overrides ReadOnly Property Statement As System.Type = GetType(Statement)
 
-	Public Overrides Sub Open(dsn As String)
-		Me._provider = New SqlConnection(dsn)
-		Me._provider.Open()
-		AddHandler Me._provider.InfoMessage, AddressOf Connection.errorHandler
-	End Sub
+    Public Overrides Sub Open(dsn As String)
+        Me._provider = New SqlConnection(dsn)
+        Me._provider.Open()
+        AddHandler Me._provider.InfoMessage, AddressOf Connection.errorHandler
+    End Sub
 
-	Public Overrides Function CreateAndBeginTransaction(Optional transactionName As String = "", Optional isolationLevel As IsolationLevel = IsolationLevel.Unspecified) As Databasic.Transaction
-        Return New Transaction() With {
+    Protected Shared Sub errorHandler(sender As Object, args As SqlInfoMessageEventArgs)
+        Dim sqlErrors As Databasic.SqlErrorsCollection = New SqlErrorsCollection()
+        For index = 0 To args.Errors.Count - 1
+            sqlErrors.Add(New Databasic.MsSql.SqlError(args.Errors(index)))
+        Next
+        Databasic.Events.RaiseError(sqlErrors)
+    End Sub
+
+    Protected Overrides Function createAndBeginTransaction(Optional transactionName As String = "", Optional isolationLevel As IsolationLevel = IsolationLevel.Unspecified) As Databasic.Transaction
+        Me.OpenedTransaction = New Transaction() With {
             .ConnectionWrapper = Me,
-            .Instance = Me._provider.BeginTransaction(isolationLevel, transactionName.Substring(0, Math.Min(transactionName.Length - 1, 32)))
+            .Instance = Me._provider.BeginTransaction(
+                isolationLevel, transactionName.Substring(0, Math.Min(transactionName.Length - 1, 32))
+            )
         }
+        Return Me.OpenedTransaction
     End Function
 
 End Class

Databasic.MsSql.vbproj

@@ -59,6 +59,7 @@
   </ItemGroup>
   <ItemGroup>
     <Compile Include="Connection.vb" />
+    <Compile Include="SqlError.vb" />
     <Compile Include="My Project\AssemblyInfo.vb" />
     <Compile Include="My Project\Application.Designer.vb">
       <AutoGen>True</AutoGen>

SqlError.vb

@@ -0,0 +1,22 @@
+Public Class SqlError
+    Inherits Databasic.SqlError
+
+    Public Property Server As String
+    Public Property Procedure As String
+    Public Property Source As String
+    Public Property LineNumber As Integer
+    Public Property [Class] As Byte
+    Public Property State As Byte
+
+    Public Sub New(msSqlError As System.Data.SqlClient.SqlError)
+        Me.Message = msSqlError.Message
+        Me.Code = msSqlError.Number
+
+        Me.Server = msSqlError.Server
+        Me.Procedure = msSqlError.Procedure
+        Me.Source = msSqlError.Source
+        Me.LineNumber = msSqlError.LineNumber
+        Me.[Class] = msSqlError.[Class]
+        Me.State = msSqlError.State
+    End Sub
+End Class

Transaction.vb

@@ -3,13 +3,35 @@ Imports System.Data.SqlClient
 
 Public Class Transaction
     Inherits Databasic.Transaction
+
     Public Overrides Property Instance As DbTransaction
-        Get
-            Return Me._instance
-        End Get
-        Set(value As DbTransaction)
-            Me._instance = value
-        End Set
-    End Property
+		Get
+			Return Me._instance
+		End Get
+		Set(value As DbTransaction)
+			Me._instance = value
+		End Set
+	End Property
     Private _instance As SqlTransaction
+
+    Public Shadows Sub Rollback(transactionName As String)
+        Try
+            Me._instance.Rollback(transactionName)
+            Me.ConnectionWrapper.OpenedTransaction = Nothing
+        Catch ex As Exception
+            Me.ConnectionWrapper.OpenedTransaction = Nothing
+            Throw ex
+        End Try
+    End Sub
+
+    Public Sub Save(savePointName As String)
+        Try
+            Me._instance.Save(savePointName)
+            Me.ConnectionWrapper.OpenedTransaction = Nothing
+        Catch ex As Exception
+            Me.ConnectionWrapper.OpenedTransaction = Nothing
+            Throw ex
+        End Try
+    End Sub
+
 End Class