1
/*******************************************************************************
2
*Copyright (c) 2009 Eucalyptus Systems, Inc.
4
* This program is free software: you can redistribute it and/or modify
5
* it under the terms of the GNU General Public License as published by
6
* the Free Software Foundation, only version 3 of the License.
9
* This file is distributed in the hope that it will be useful, but WITHOUT
10
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14
* You should have received a copy of the GNU General Public License along
15
* with this program. If not, see <http://www.gnu.org/licenses/>.
17
* Please contact Eucalyptus Systems, Inc., 130 Castilian
18
* Dr., Goleta, CA 93101 USA or visit <http://www.eucalyptus.com/licenses/>
19
* if you need additional information or have any questions.
21
* This file may incorporate work covered under the following copyright and
24
* Software License Agreement (BSD License)
26
* Copyright (c) 2008, Regents of the University of California
27
* All rights reserved.
29
* Redistribution and use of this software in source and binary forms, with
30
* or without modification, are permitted provided that the following
33
* Redistributions of source code must retain the above copyright notice,
34
* this list of conditions and the following disclaimer.
36
* Redistributions in binary form must reproduce the above copyright
37
* notice, this list of conditions and the following disclaimer in the
38
* documentation and/or other materials provided with the distribution.
40
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
41
* IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
42
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
43
* PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
44
* OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
45
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
46
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
47
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
48
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
49
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
50
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. USERS OF
51
* THIS SOFTWARE ACKNOWLEDGE THE POSSIBLE PRESENCE OF OTHER OPEN SOURCE
52
* LICENSED MATERIAL, COPYRIGHTED MATERIAL OR PATENTED MATERIAL IN THIS
53
* SOFTWARE, AND IF ANY SUCH MATERIAL IS DISCOVERED THE PARTY DISCOVERING
54
* IT MAY INFORM DR. RICH WOLSKI AT THE UNIVERSITY OF CALIFORNIA, SANTA
55
* BARBARA WHO WILL THEN ASCERTAIN THE MOST APPROPRIATE REMEDY, WHICH IN
56
* THE REGENTS’ DISCRETION MAY INCLUDE, WITHOUT LIMITATION, REPLACEMENT
57
* OF THE CODE SO IDENTIFIED, LICENSING OF THE CODE SO IDENTIFIED, OR
58
* WITHDRAWAL OF THE CODE CAPABILITY TO THE EXTENT NEEDED TO COMPLY WITH
59
* ANY SUCH LICENSES OR RIGHTS.
60
*******************************************************************************/
65
* Author: chris grzegorczyk <grze@eucalyptus.com>
67
package com.eucalyptus.util;
69
import java.util.List;
71
import javax.persistence.EntityManager;
73
import org.apache.log4j.Logger;
74
import org.hibernate.Session;
75
import org.hibernate.criterion.Example;
76
import org.hibernate.criterion.MatchMode;
77
import org.hibernate.exception.JDBCConnectionException;
79
import com.eucalyptus.bootstrap.Component;
80
import com.google.common.collect.Lists;
81
import com.google.common.collect.Sets;
82
import com.mchange.v2.resourcepool.CannotAcquireResourceException;
83
import com.mchange.v2.resourcepool.TimeoutException;
85
import edu.ucsb.eucalyptus.msgs.EventRecord;
87
public class EntityWrapper<TYPE> {
89
static Logger LOG = Logger.getLogger( EntityWrapper.class );
92
public EntityWrapper( ) {
96
@SuppressWarnings( "unchecked" )
97
public EntityWrapper( String persistenceContext ) {
99
LOG.trace( EventRecord.here( Component.db, DbEvent.CREATE.begin( ) ) );
100
this.tx = new TxHandle( persistenceContext );
101
} catch ( Throwable e ) {
102
LOG.trace( EventRecord.here( Component.db, DbEvent.CREATE.fail( ),e.getMessage( ) ) );
103
this.exceptionCaught( e );
104
throw (RuntimeException) e ;
106
LOG.trace( EventRecord.here( Component.db, DbEvent.CREATE.end( ), Long.toString( tx.splitOperation() ), tx.getTxUuid( ) ) );
109
@SuppressWarnings( "unchecked" )
110
public List<TYPE> query( TYPE example ) {
111
LOG.trace( EventRecord.here( Component.db, DbEvent.QUERY.begin( ), tx.getTxUuid( ) ) );
112
Example qbe = Example.create( example ).enableLike( MatchMode.EXACT );
113
List<TYPE> resultList = ( List<TYPE> ) this.getSession( ).createCriteria( example.getClass( ) ).setCacheable( true ).add( qbe ).list( );
114
LOG.trace( EventRecord.here( Component.db, DbEvent.QUERY.end( ), Long.toString( tx.splitOperation( ) ), tx.getTxUuid( ) ) );
115
return Lists.newArrayList( Sets.newHashSet( resultList ) );
118
public TYPE getUnique( TYPE example ) throws EucalyptusCloudException {
119
LOG.trace( EventRecord.here( Component.db, DbEvent.UNIQUE.begin( ), tx.getTxUuid( ) ) );
120
List<TYPE> res = this.query( example );
121
if ( res.size( ) != 1 ) {
123
msg = example.toString( );
124
if ( msg != null && msg.startsWith( example.getClass( ).getCanonicalName( ) ) ) {
125
msg = LogUtil.dumpObject( example );
127
LOG.trace( EventRecord.here( Component.db, DbEvent.QUERY.fail( ), Long.toString( tx.splitOperation( ) ), tx.getTxUuid( ) ) );
128
throw new EucalyptusCloudException( "Error locating information for " + msg );
130
LOG.trace( EventRecord.here( Component.db, DbEvent.QUERY.end( ), Long.toString( tx.splitOperation( ) ), tx.getTxUuid( ) ) );
134
@SuppressWarnings( "unchecked" )
135
private void exceptionCaught( Throwable e ) {
136
Throwable cause = DebugUtil.checkForCauseOfInterest( e, JDBCConnectionException.class, CannotAcquireResourceException.class, TimeoutException.class, IllegalStateException.class );
137
if ( !( cause instanceof ExceptionNotRelatedException ) ) {
138
LOG.error( cause, cause );
139
DatabaseUtil.handleConnectionError( cause );
143
public void add( TYPE newObject ) {
144
this.getEntityManager( ).persist( newObject );
147
public void merge( TYPE newObject ) {
148
this.getEntityManager( ).merge( newObject );
151
public void mergeAndCommit( TYPE newObject ) {
152
this.getEntityManager( ).merge( newObject );
156
public void delete( TYPE deleteObject ) {
157
this.getEntityManager( ).remove( deleteObject );
160
public void rollback( ) {
161
LOG.trace( EventRecord.here( Component.db, DbEvent.ROLLBACK.begin( ), tx.getTxUuid( ) ) );
164
} catch ( Exception e ) {
165
LOG.trace( EventRecord.here( Component.db, DbEvent.ROLLBACK.fail( ), Long.toString( tx.splitOperation() ), tx.getTxUuid( ) ) );
166
this.exceptionCaught( e );
168
LOG.trace( EventRecord.here( Component.db, DbEvent.ROLLBACK.end( ), Long.toString( tx.splitOperation() ), tx.getTxUuid( ) ) );
171
public void commit( ) {
172
LOG.trace( EventRecord.here( Component.db, DbEvent.COMMIT.begin( ), tx.getTxUuid( ) ) );
175
} catch ( Throwable e ) {
176
LOG.trace( EventRecord.here( Component.db, DbEvent.COMMIT.fail( ), Long.toString( tx.splitOperation( ) ), tx.getTxUuid( ) ) );
177
this.exceptionCaught( e );
178
throw (RuntimeException) e ;
180
LOG.trace( EventRecord.here( Component.db, DbEvent.COMMIT.end( ), Long.toString( tx.splitOperation( ) ), tx.getTxUuid( ) ) );
183
public Session getSession( ) {
184
return tx.getSession( );
187
public EntityManager getEntityManager( ) {
188
return tx.getEntityManager( );
191
@SuppressWarnings( "unchecked" )
192
public <NEWTYPE> EntityWrapper<NEWTYPE> recast( Class<NEWTYPE> c ) {
193
return ( EntityWrapper<NEWTYPE> ) this;
196
public static StackTraceElement getMyStackTraceElement( ) {
198
for ( StackTraceElement ste : Thread.currentThread( ).getStackTrace( ) ) {
199
if ( i++ < 2 || ste.getClassName( ).matches( ".*EntityWrapper.*" )
200
|| ste.getClassName( ).matches( ".*TxHandle.*" )
201
|| ste.getMethodName( ).equals( "getEntityWrapper" ) ) {
207
throw new RuntimeException( "BUG: Reached bottom of stack trace without finding any relevent frames." );
216
public String fail() {
217
return this.name( ) + ":FAIL";
219
public String begin() {
220
return this.name( ) + ":BEGIN";
222
public String end() {
223
return this.name( ) + ":END";
225
public String getMessage( ) {
226
if( DebugUtil.TRACE ) {
227
return EntityWrapper.getMyStackTraceElement( ).toString( );