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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
|
// **********************************************************************
//
// Copyright (c) 2003-2008 ZeroC, Inc. All rights reserved.
//
// This copy of Ice is licensed to you under the terms described in the
// ICE_LICENSE file included in this distribution.
//
// **********************************************************************
//
// A SQL request context encapsulates SQL resources allocated in the
// process of executing a request, such as the database connection,
// and associated SQL statements.
//
// The request context is automatically destroyed at the end of a
// request, or if obtain is called it must be destroyed manually by
// calling destroy.
//
// When the request context is destroyed, the transaction is rolled
// back, if not already committed, and all allocated resources are
// released,
//
class SQLRequestContext
{
public static SQLRequestContext
getCurrentContext()
{
synchronized(_contextMap)
{
SQLRequestContext context = _contextMap.get(Thread.currentThread());
if(context != null)
{
//context._logger.trace("SQLRequestContext", "getCurrentContext: thread " + Thread.currentThread() +
//": connection: " + context._conn);
}
return context;
}
}
public java.sql.PreparedStatement
prepareStatement(String sql)
throws java.sql.SQLException
{
java.sql.PreparedStatement stmt = _conn.prepareStatement(sql);
_statements.add(stmt);
return stmt;
}
public java.sql.PreparedStatement
prepareStatement(String sql, int autoGeneratedKeys)
throws java.sql.SQLException
{
java.sql.PreparedStatement stmt = _conn.prepareStatement(sql, autoGeneratedKeys);
_statements.add(stmt);
return stmt;
}
// Called to obtain ownership of the context. The context is no
// longer destroyed automatically when the current request has
// completed.
public void
obtain()
{
// Remove the current context from the map.
synchronized(_contextMap)
{
//_logger.trace("SQLRequestContext", "obtain: connection: " + _conn);
_contextMap.remove(Thread.currentThread());
}
}
public void
destroy()
{
synchronized(_contextMap)
{
SQLRequestContext context = _contextMap.remove(Thread.currentThread());
assert context == null;
}
//_logger.trace("SQLRequestContext", "destroy: connection: " + _conn);
// Release all resources.
try
{
// Rollback the transaction if it was not committed.
if(!_commit)
{
//_logger.trace("SQLRequestContext", "rollback: thread " + Thread.currentThread() + ": connection: " + _conn);
_conn.rollback();
}
java.util.Iterator<java.sql.Statement> p = _statements.iterator();
while(p.hasNext())
{
p.next().close();
}
}
catch(java.sql.SQLException e)
{
error(e);
}
_pool.release(_conn);
_statements.clear();
_conn = null;
_pool = null;
}
public void
commit()
throws java.sql.SQLException
{
_conn.commit();
//_logger.trace("SQLRequestContext", "commit: thread " + Thread.currentThread() + ": connection: " + _conn);
_commit = true;
}
SQLRequestContext(Ice.Logger logger, ConnectionPool pool)
{
_logger = logger;
_pool = pool;
_conn = pool.acquire();
synchronized(_contextMap)
{
//_logger.trace("SQLRequestContext", "associate: thread " + Thread.currentThread() + ": connection: " + _conn);
_contextMap.put(Thread.currentThread(), this);
}
}
// Called only by the servant locator.
void
destroyFromLocator()
{
synchronized(_contextMap)
{
//_logger.trace("SQLRequestContext", "release: connection: " + _conn);
// Remove the current context from the map.
SQLRequestContext context = _contextMap.remove(Thread.currentThread());
assert context == this;
}
destroy();
}
private void
error(Exception ex)
{
java.io.StringWriter sw = new java.io.StringWriter();
java.io.PrintWriter pw = new java.io.PrintWriter(sw);
ex.printStackTrace(pw);
pw.flush();
_logger.error("SQLRequestContext: error:\n" + sw.toString());
}
// A map of threads to request contexts.
private static java.util.Map<Thread, SQLRequestContext> _contextMap =
new java.util.HashMap<Thread, SQLRequestContext>();
private Ice.Logger _logger;
private ConnectionPool _pool;
private java.util.List<java.sql.Statement> _statements = new java.util.LinkedList<java.sql.Statement>();
private java.sql.Connection _conn;
private boolean _commit = false;
}
|
