forked from danbarua/Npgsql
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathNpgsqlTransactionCallbacks.cs
More file actions
142 lines (125 loc) · 4.58 KB
/
NpgsqlTransactionCallbacks.cs
File metadata and controls
142 lines (125 loc) · 4.58 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
// NpgsqlTransactionCallbacks.cs
//
// Author:
// Josh Cooley <jbnpgsql@tuxinthebox.net>
//
// Copyright (C) 2007, The Npgsql Development Team
//
// Permission to use, copy, modify, and distribute this software and its
// documentation for any purpose, without fee, and without a written
// agreement is hereby granted, provided that the above copyright notice
// and this paragraph and the following two paragraphs appear in all copies.
//
// IN NO EVENT SHALL THE NPGSQL DEVELOPMENT TEAM BE LIABLE TO ANY PARTY
// FOR DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES,
// INCLUDING LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS
// DOCUMENTATION, EVEN IF THE NPGSQL DEVELOPMENT TEAM HAS BEEN ADVISED OF
// THE POSSIBILITY OF SUCH DAMAGE.
//
// THE NPGSQL DEVELOPMENT TEAM SPECIFICALLY DISCLAIMS ANY WARRANTIES,
// INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE PROVIDED HEREUNDER IS
// ON AN "AS IS" BASIS, AND THE NPGSQL DEVELOPMENT TEAM HAS NO OBLIGATIONS
// TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
using System;
using System.Data;
using System.Reflection;
using Common.Logging;
namespace Npgsql
{
internal interface INpgsqlTransactionCallbacks : IDisposable
{
string GetName();
void PrepareTransaction();
void CommitTransaction();
void RollbackTransaction();
}
internal class NpgsqlTransactionCallbacks : MarshalByRefObject, INpgsqlTransactionCallbacks
{
private NpgsqlConnection _connection;
private readonly string _connectionString;
private bool _closeConnectionRequired;
private bool _prepared;
private readonly string _txName = Guid.NewGuid().ToString();
private static readonly ILog _log = LogManager.GetCurrentClassLogger();
public NpgsqlTransactionCallbacks(NpgsqlConnection connection)
{
_connection = connection;
_connectionString = _connection.ConnectionString;
_connection.Disposed += new EventHandler(_connection_Disposed);
}
private void _connection_Disposed(object sender, EventArgs e)
{
// TODO: what happens if this is called from another thread?
// connections should not be shared across threads while in a transaction
_connection.Disposed -= new EventHandler(_connection_Disposed);
_connection = null;
}
private NpgsqlConnection GetConnection()
{
if (_connection == null || (_connection.FullState & ConnectionState.Open) != ConnectionState.Open)
{
_connection = new NpgsqlConnection(_connectionString);
_connection.Open();
_closeConnectionRequired = true;
return _connection;
}
else
{
return _connection;
}
}
#region INpgsqlTransactionCallbacks Members
public string GetName()
{
return _txName;
}
public void CommitTransaction()
{
_log.Debug("Commit transaction");
NpgsqlConnection connection = GetConnection();
if (_prepared)
{
connection.Connector.ExecuteBlind(string.Format("COMMIT PREPARED '{0}'", _txName));
}
else
{
connection.Connector.ExecuteBlind(QueryManager.CommitTransaction);
}
}
public void PrepareTransaction()
{
if (!_prepared)
{
_log.Debug("Prepare transaction");
NpgsqlConnection connection = GetConnection();
connection.Connector.ExecuteBlind(string.Format("PREPARE TRANSACTION '{0}'", _txName));
_prepared = true;
}
}
public void RollbackTransaction()
{
_log.Debug("Rollback transaction");
NpgsqlConnection connection = GetConnection();
if (_prepared)
{
connection.Connector.ExecuteBlind(string.Format("ROLLBACK PREPARED '{0}'", _txName));
}
else
{
connection.Connector.ExecuteBlind(QueryManager.RollbackTransaction);
}
}
#endregion
#region IDisposable Members
public void Dispose()
{
if (_closeConnectionRequired)
{
_connection.Close();
}
_closeConnectionRequired = false;
}
#endregion
}
}