mirror of
				https://github.com/Alfresco/alfresco-community-repo.git
				synced 2025-10-29 15:21:53 +00:00 
			
		
		
		
	23430: Merged V3.3 to V3.3-BUG-FIX
      23429: Merged PATCHES/V3.2.0 to V3.3
         23428: ALF-5141: Further refinement of transaction limiting behaviour
            - Once accepted a transaction will be retried until it succeeds or the retry limit is exceeded. No limiting of retries by time as otherwise likelihood of any transaction succeeding under load is small
            - Default web transaction threshold is now 20 seconds to avoid problems on non-loaded systems
            - Record stack traces of tracked transactions to help debug slow transactions
git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@23431 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
		
	
		
			
				
	
	
		
			682 lines
		
	
	
		
			26 KiB
		
	
	
	
		
			Java
		
	
	
	
	
	
			
		
		
	
	
			682 lines
		
	
	
		
			26 KiB
		
	
	
	
		
			Java
		
	
	
	
	
	
| /*
 | |
|  * Copyright (C) 2005-2010 Alfresco Software Limited.
 | |
|  *
 | |
|  * This file is part of Alfresco
 | |
|  *
 | |
|  * Alfresco is free software: you can redistribute it and/or modify
 | |
|  * it under the terms of the GNU Lesser General Public License as published by
 | |
|  * the Free Software Foundation, either version 3 of the License, or
 | |
|  * (at your option) any later version.
 | |
|  *
 | |
|  * Alfresco is distributed in the hope that it will be useful,
 | |
|  * but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
|  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | |
|  * GNU Lesser General Public License for more details.
 | |
|  *
 | |
|  * You should have received a copy of the GNU Lesser General Public License
 | |
|  * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
 | |
|  */
 | |
| package org.alfresco.repo.transaction;
 | |
| 
 | |
| import java.util.Collections;
 | |
| import java.util.ConcurrentModificationException;
 | |
| import java.util.LinkedList;
 | |
| import java.util.List;
 | |
| import java.util.concurrent.ArrayBlockingQueue;
 | |
| import java.util.concurrent.CountDownLatch;
 | |
| import java.util.concurrent.ExecutorService;
 | |
| import java.util.concurrent.ThreadPoolExecutor;
 | |
| import java.util.concurrent.TimeUnit;
 | |
| 
 | |
| import javax.transaction.Status;
 | |
| import javax.transaction.UserTransaction;
 | |
| 
 | |
| import junit.framework.TestCase;
 | |
| 
 | |
| import org.alfresco.error.ExceptionStackUtil;
 | |
| import org.alfresco.model.ContentModel;
 | |
| import org.alfresco.repo.security.authentication.AuthenticationComponent;
 | |
| import org.alfresco.repo.security.authentication.AuthenticationUtil;
 | |
| import org.alfresco.repo.transaction.AlfrescoTransactionSupport.TxnReadState;
 | |
| import org.alfresco.repo.transaction.RetryingTransactionHelper.RetryingTransactionCallback;
 | |
| import org.alfresco.service.ServiceRegistry;
 | |
| import org.alfresco.service.cmr.repository.InvalidNodeRefException;
 | |
| import org.alfresco.service.cmr.repository.NodeRef;
 | |
| import org.alfresco.service.cmr.repository.NodeService;
 | |
| import org.alfresco.service.cmr.repository.StoreRef;
 | |
| import org.alfresco.service.namespace.NamespaceService;
 | |
| import org.alfresco.service.namespace.QName;
 | |
| import org.alfresco.service.transaction.TransactionService;
 | |
| import org.alfresco.util.ApplicationContextHelper;
 | |
| import org.alfresco.util.Pair;
 | |
| import org.apache.commons.lang.mutable.MutableInt;
 | |
| import org.apache.commons.logging.Log;
 | |
| import org.apache.commons.logging.LogFactory;
 | |
| import org.hibernate.SessionFactory;
 | |
| import org.hibernate.dialect.Dialect;
 | |
| import org.hibernate.dialect.MySQLInnoDBDialect;
 | |
| import org.springframework.context.ApplicationContext;
 | |
| import org.springframework.dao.ConcurrencyFailureException;
 | |
| import org.springframework.orm.hibernate3.support.HibernateDaoSupport;
 | |
| 
 | |
| /**
 | |
|  * Tests the transaction retrying behaviour with various failure modes.
 | |
|  * 
 | |
|  * @see RetryingTransactionHelper
 | |
|  * @see TransactionService
 | |
|  * 
 | |
|  * @author Derek Hulley
 | |
|  * @since 2.1
 | |
|  */
 | |
| public class RetryingTransactionHelperTest extends TestCase
 | |
| {
 | |
|     private static Log logger = LogFactory.getLog("org.alfresco.repo.transaction.RetryingTransactionHelperTest");
 | |
|     
 | |
|     private static final QName PROP_CHECK_VALUE = QName.createQName(NamespaceService.CONTENT_MODEL_1_0_URI, "check_value");
 | |
|     
 | |
|     private static ApplicationContext ctx = ApplicationContextHelper.getApplicationContext();
 | |
| 
 | |
|     private ServiceRegistry serviceRegistry;
 | |
|     private AuthenticationComponent authenticationComponent;
 | |
|     private TransactionService transactionService;
 | |
|     private NodeService nodeService;
 | |
|     private RetryingTransactionHelper txnHelper;
 | |
|     
 | |
|     private Dialect dialect;
 | |
|     
 | |
|     private NodeRef rootNodeRef;
 | |
|     private NodeRef workingNodeRef;
 | |
|     
 | |
|     @Override
 | |
|     public void setUp() throws Exception
 | |
|     {
 | |
|         dialect = (Dialect) ctx.getBean("dialect");
 | |
|         
 | |
|         serviceRegistry = (ServiceRegistry) ctx.getBean(ServiceRegistry.SERVICE_REGISTRY);
 | |
|         authenticationComponent = (AuthenticationComponent) ctx.getBean("authenticationComponent");
 | |
|         transactionService = serviceRegistry.getTransactionService();
 | |
|         nodeService = serviceRegistry.getNodeService();
 | |
|         txnHelper = transactionService.getRetryingTransactionHelper();
 | |
|         
 | |
|         // authenticate
 | |
|         authenticationComponent.setSystemUserAsCurrentUser();
 | |
|         
 | |
|         StoreRef storeRef = nodeService.createStore(
 | |
|                 StoreRef.PROTOCOL_WORKSPACE,
 | |
|                 "test-" + getName() + "-" + System.currentTimeMillis());
 | |
|         rootNodeRef = nodeService.getRootNode(storeRef);
 | |
|         // Create a node to work on
 | |
|         workingNodeRef = nodeService.createNode(
 | |
|                 rootNodeRef,
 | |
|                 ContentModel.ASSOC_CHILDREN,
 | |
|                 QName.createQName(NamespaceService.CONTENT_MODEL_1_0_URI, getName()),
 | |
|                 ContentModel.TYPE_CMOBJECT).getChildRef();
 | |
|     }
 | |
|     
 | |
|     @Override
 | |
|     public void tearDown() throws Exception
 | |
|     {
 | |
|         try { authenticationComponent.clearCurrentSecurityContext(); } catch (Throwable e) {}
 | |
|     }
 | |
|     
 | |
|     public void testSetUp() throws Exception
 | |
|     {
 | |
|         assertNotNull(rootNodeRef);
 | |
|         assertNotNull(workingNodeRef);
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * Get the count, which is 0 to start each test
 | |
|      * 
 | |
|      * @return          Returns the current count
 | |
|      */
 | |
|     private Long getCheckValue()
 | |
|     {
 | |
|         Long checkValue = (Long) nodeService.getProperty(workingNodeRef, PROP_CHECK_VALUE);
 | |
|         if (checkValue == null)
 | |
|         {
 | |
|             checkValue = new Long(0);
 | |
|             nodeService.setProperty(workingNodeRef, PROP_CHECK_VALUE, checkValue);
 | |
|         }
 | |
|         return checkValue;
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * Increment the test count, which is 0 to start each test
 | |
|      * 
 | |
|      * @return          Returns the current count
 | |
|      */
 | |
|     private Long incrementCheckValue()
 | |
|     {
 | |
|         Long checkValue = getCheckValue();
 | |
|         checkValue = new Long(checkValue.longValue() + 1L);
 | |
|         nodeService.setProperty(workingNodeRef, PROP_CHECK_VALUE, checkValue);
 | |
|         return checkValue;
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * @return                          Never returns anything
 | |
|      * @throws InvalidNodeRefException  <b>ALWAYS</b>
 | |
|      */
 | |
|     private Long blowUp()
 | |
|     {
 | |
|         NodeRef invalidNodeRef = new NodeRef(workingNodeRef.getStoreRef(), "BOGUS");
 | |
|         nodeService.setProperty(invalidNodeRef, PROP_CHECK_VALUE, null);
 | |
|         fail("Expected to generate an InvalidNodeRefException");
 | |
|         return null;
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Check that it works without complications.
 | |
|      */
 | |
|     public void testSuccessNoRetry()
 | |
|     {
 | |
|         long beforeValue = getCheckValue();
 | |
|         RetryingTransactionCallback<Long> callback = new RetryingTransactionCallback<Long>()
 | |
|         {
 | |
|             public Long execute() throws Throwable
 | |
|             {
 | |
|                 return incrementCheckValue();
 | |
|             }
 | |
|         };
 | |
|         long txnValue = txnHelper.doInTransaction(callback);
 | |
|         long afterValue = getCheckValue();
 | |
|         assertEquals("The value must have increased", beforeValue + 1, afterValue);
 | |
|         assertEquals("The txn value must be the same as the value after", afterValue, txnValue);
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * Check that the transaction state can be fetched in and around the transaction.
 | |
|      * This also checks that any mischievous attempts to manipulate the transaction
 | |
|      * (other than setRollback) are detected. 
 | |
|      */
 | |
|     public void testUserTransactionStatus()
 | |
|     {
 | |
|         UserTransaction txnBefore = RetryingTransactionHelper.getActiveUserTransaction();
 | |
|         assertNull("Did not expect to get an active UserTransaction", txnBefore);
 | |
|         
 | |
|         RetryingTransactionCallback<Long> callbackOuter = new RetryingTransactionCallback<Long>()
 | |
|         {
 | |
|             public Long execute() throws Throwable
 | |
|             {
 | |
|                 final UserTransaction txnOuter = RetryingTransactionHelper.getActiveUserTransaction();
 | |
|                 assertNotNull("Expected an active UserTransaction", txnOuter);
 | |
|                 assertEquals(
 | |
|                         "Should be read-write txn",
 | |
|                         TxnReadState.TXN_READ_WRITE, AlfrescoTransactionSupport.getTransactionReadState());
 | |
|                 assertEquals("Expected state is active", Status.STATUS_ACTIVE, txnOuter.getStatus());
 | |
|                 RetryingTransactionCallback<Long> callbackInner = new RetryingTransactionCallback<Long>()
 | |
|                 {
 | |
|                     public Long execute() throws Throwable
 | |
|                     {
 | |
|                         UserTransaction txnInner = RetryingTransactionHelper.getActiveUserTransaction();
 | |
|                         assertNotNull("Expected an active UserTransaction", txnInner);
 | |
|                         assertEquals(
 | |
|                                 "Should be read-only txn",
 | |
|                                 TxnReadState.TXN_READ_ONLY, AlfrescoTransactionSupport.getTransactionReadState());
 | |
|                         assertEquals("Expected state is active", Status.STATUS_ACTIVE, txnInner.getStatus());
 | |
|                         // Check blow up
 | |
|                         try
 | |
|                         {
 | |
|                             txnInner.commit();
 | |
|                             fail("Should not be able to commit the UserTransaction.  It is for info only.");
 | |
|                         }
 | |
|                         catch (Throwable e)
 | |
|                         {
 | |
|                             // Expected
 | |
|                         }
 | |
|                         // Force a rollback
 | |
|                         txnInner.setRollbackOnly();
 | |
|                         // Done
 | |
|                         return null;
 | |
|                     }
 | |
|                 };
 | |
|                 return txnHelper.doInTransaction(callbackInner, true, true);
 | |
|             }
 | |
|         };
 | |
|         txnHelper.doInTransaction(callbackOuter);
 | |
|         
 | |
|         UserTransaction txnAfter = RetryingTransactionHelper.getActiveUserTransaction();
 | |
|         assertNull("Did not expect to get an active UserTransaction", txnAfter);
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * Check that the retries happening for simple concurrency exceptions
 | |
|      */
 | |
|     public void testSuccessWithRetry()
 | |
|     {
 | |
|         RetryingTransactionCallback<Long> callback = new RetryingTransactionCallback<Long>()
 | |
|         {
 | |
|             private int maxCalls = 3;
 | |
|             private int callCount = 0;
 | |
|             public Long execute() throws Throwable
 | |
|             {
 | |
|                 callCount++;
 | |
|                 Long checkValue = incrementCheckValue();
 | |
|                 if (callCount == maxCalls)
 | |
|                 {
 | |
|                     return checkValue;
 | |
|                 }
 | |
|                 else
 | |
|                 {
 | |
|                     throw new ConcurrencyFailureException("Testing");
 | |
|                 }
 | |
|             }
 | |
|         };
 | |
|         long txnValue = txnHelper.doInTransaction(callback);
 | |
|         assertEquals("Only one increment expected", 1, txnValue);
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * Checks that a non-retrying exception is passed out and that the transaction is rolled back.
 | |
|      */
 | |
|     public void testNonRetryingFailure()
 | |
|     {
 | |
|         RetryingTransactionCallback<Long> callback = new RetryingTransactionCallback<Long>()
 | |
|         {
 | |
|             public Long execute() throws Throwable
 | |
|             {
 | |
|                 incrementCheckValue();
 | |
|                 return blowUp();
 | |
|             }
 | |
|         };
 | |
|         try
 | |
|         {
 | |
|             txnHelper.doInTransaction(callback);
 | |
|             fail("Wrapper didn't generate an exception");
 | |
|         }
 | |
|         catch (InvalidNodeRefException e)
 | |
|         {
 | |
|             // Correct
 | |
|         }
 | |
|         catch (Throwable e)
 | |
|         {
 | |
|             fail("Incorrect exception from wrapper: " + e);
 | |
|         }
 | |
|         // Check that the value didn't change
 | |
|         long checkValue = getCheckValue();
 | |
|         assertEquals("Check value should not have changed", 0, checkValue);
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * Sometimes, exceptions or other states may cause the transaction to be marked for
 | |
|      * rollback without an exception being generated.  This tests that the exception stays
 | |
|      * absorbed and that another isn't generated, but that the transaction was rolled back
 | |
|      * properly.
 | |
|      */
 | |
|     public void testNonRetryingSilentRollback()
 | |
|     {
 | |
|         RetryingTransactionCallback<Long> callback = new RetryingTransactionCallback<Long>()
 | |
|         {
 | |
|             public Long execute() throws Throwable
 | |
|             {
 | |
|                 incrementCheckValue();
 | |
|                 try
 | |
|                 {
 | |
|                     return blowUp();
 | |
|                 }
 | |
|                 catch (InvalidNodeRefException e)
 | |
|                 {
 | |
|                     // Expected, but absorbed
 | |
|                 }
 | |
|                 return null;
 | |
|             }
 | |
|         };
 | |
|         txnHelper.doInTransaction(callback);
 | |
|         long checkValue = getCheckValue();
 | |
|         assertEquals("Check value should not have changed", 0, checkValue);
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * Checks nesting of two transactions with <code>requiresNew == false</code>
 | |
|      */
 | |
|     public void testNestedWithPropogation()
 | |
|     {
 | |
|         RetryingTransactionCallback<Long> callback = new RetryingTransactionCallback<Long>()
 | |
|         {
 | |
|             public Long execute() throws Throwable
 | |
|             {
 | |
|                 RetryingTransactionCallback<Long> callbackInner = new RetryingTransactionCallback<Long>()
 | |
|                 {
 | |
|                     public Long execute() throws Throwable
 | |
|                     {
 | |
|                         incrementCheckValue();
 | |
|                         incrementCheckValue();
 | |
|                         return getCheckValue();
 | |
|                     }
 | |
|                 };
 | |
|                 txnHelper.doInTransaction(callbackInner, false, false);
 | |
|                 incrementCheckValue();
 | |
|                 incrementCheckValue();
 | |
|                 return getCheckValue();
 | |
|             }
 | |
|         };
 | |
|         long checkValue = txnHelper.doInTransaction(callback);
 | |
|         assertEquals("Nesting requiresNew==false didn't work", 4, checkValue);
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * Checks nesting of two transactions with <code>requiresNew == true</code>
 | |
|      */
 | |
|     public void testNestedWithoutPropogation()
 | |
|     {
 | |
|         RetryingTransactionCallback<Long> callback = new RetryingTransactionCallback<Long>()
 | |
|         {
 | |
|             public Long execute() throws Throwable
 | |
|             {
 | |
|                 RetryingTransactionCallback<Long> callbackInner = new RetryingTransactionCallback<Long>()
 | |
|                 {
 | |
|                     public Long execute() throws Throwable
 | |
|                     {
 | |
|                         incrementCheckValue();
 | |
|                         incrementCheckValue();
 | |
|                         return getCheckValue();
 | |
|                     }
 | |
|                 };
 | |
|                 txnHelper.doInTransaction(callbackInner, false, true);
 | |
|                 incrementCheckValue();
 | |
|                 incrementCheckValue();
 | |
|                 return getCheckValue();
 | |
|             }
 | |
|         };
 | |
|         long checkValue = txnHelper.doInTransaction(callback);
 | |
|         assertEquals("Nesting requiresNew==true didn't work", 4, checkValue);
 | |
|     }
 | |
|     
 | |
|     /**
 | |
|      * Checks nesting of two transactions with <code>requiresNew == true</code>,
 | |
|      * but where the two transactions get involved in a concurrency struggle.
 | |
|      * 
 | |
|      * Note: skip test for non-MySQL
 | |
|      */
 | |
|     public void testNestedWithoutPropogationConcurrentUntilFailureMySQL() throws InterruptedException
 | |
|     {
 | |
|         final RetryingTransactionHelper txnHelperForTest = transactionService.getRetryingTransactionHelper();
 | |
|         txnHelperForTest.setMaxRetries(1);
 | |
|         
 | |
|         if (! (dialect instanceof MySQLInnoDBDialect))
 | |
|         {
 | |
|             // NOOP - skip test for non-MySQL DB dialects to avoid hang if concurrently "nested" (in terms of Spring) since the initial transaction does not complete
 | |
|             // see testConcurrencyRetryingNoFailure instead
 | |
|             logger.warn("NOTE: Skipping testNestedWithoutPropogationConcurrentUntilFailureMySQLOnly for dialect: "+dialect);
 | |
|         }
 | |
|         else
 | |
|         {
 | |
|             RetryingTransactionCallback<Long> callback = new RetryingTransactionCallback<Long>()
 | |
|             {
 | |
|                 public Long execute() throws Throwable
 | |
|                 {
 | |
|                     RetryingTransactionCallback<Long> callbackInner = new RetryingTransactionCallback<Long>()
 | |
|                     {
 | |
|                         public Long execute() throws Throwable
 | |
|                         {
 | |
|                             incrementCheckValue();
 | |
|                             return getCheckValue();
 | |
|                         }
 | |
|                     };
 | |
|                     incrementCheckValue();
 | |
|                     txnHelperForTest.doInTransaction(callbackInner, false, true);
 | |
|                     return getCheckValue();
 | |
|                 }
 | |
|             };
 | |
|             try
 | |
|             {
 | |
|                 txnHelperForTest.doInTransaction(callback);
 | |
|                 fail("Concurrent nested access not leading to failure");
 | |
|             }
 | |
|             catch (Throwable e)
 | |
|             {
 | |
|                 Throwable validCause = ExceptionStackUtil.getCause(e, RetryingTransactionHelper.RETRY_EXCEPTIONS);
 | |
|                 assertNotNull("Unexpected cause of the failure", validCause);
 | |
|             }
 | |
|         }
 | |
|     }
 | |
|     
 | |
|     public void testConcurrencyRetryingNoFailure() throws InterruptedException
 | |
|     {
 | |
|         Thread t1 = new Thread(new ConcurrentTransaction(5000)); 
 | |
|         t1.start();
 | |
|         
 | |
|         Thread.sleep(1000);
 | |
|         
 | |
|         Thread t2 = new Thread(new ConcurrentTransaction(10));
 | |
|         t2.start();
 | |
|         
 | |
|         t1.join();
 | |
|         t2.join();
 | |
|     }
 | |
|     
 | |
|     private class ConcurrentTransaction implements Runnable
 | |
|     {
 | |
|         private long wait;
 | |
|         
 | |
|         public ConcurrentTransaction(long wait)
 | |
|         {
 | |
|             this.wait = wait;
 | |
|         }
 | |
|         
 | |
|         public void run()
 | |
|         {
 | |
|             AuthenticationUtil.setFullyAuthenticatedUser(AuthenticationUtil.getAdminUserName());
 | |
|             
 | |
|             final RetryingTransactionHelper txnHelperForTest = transactionService.getRetryingTransactionHelper();
 | |
|             
 | |
|             RetryingTransactionCallback<Long> callback = new RetryingTransactionCallback<Long>()
 | |
|             {
 | |
|                 public Long execute() throws Throwable
 | |
|                 {
 | |
|                     incrementCheckValue();
 | |
|                     
 | |
|                     System.out.println("Wait started: "+Thread.currentThread()+" ("+wait+")");
 | |
|                     Thread.sleep(wait);
 | |
|                     System.out.println("Wait finished: "+Thread.currentThread()+" ("+wait+")");
 | |
|                     
 | |
|                     return getCheckValue();
 | |
|                 }
 | |
|             };
 | |
|             try
 | |
|             {
 | |
|                 System.out.println("Txn start: "+Thread.currentThread()+" ("+wait+")");
 | |
|                 txnHelperForTest.doInTransaction(callback);
 | |
|                 System.out.println("Txn finish: "+Thread.currentThread()+" ("+wait+")");
 | |
|             }
 | |
|             catch (Throwable e)
 | |
|             {
 | |
|                 Throwable validCause = ExceptionStackUtil.getCause(e, RetryingTransactionHelper.RETRY_EXCEPTIONS);
 | |
|                 assertNotNull("Unexpected cause of the failure", validCause);
 | |
|             }
 | |
|         }
 | |
|     }
 | |
|     
 | |
|     public void testLostConnectionRecovery()
 | |
|     {
 | |
|         RetryingTransactionCallback<Object> killConnectionCallback = new RetryingTransactionCallback<Object>()
 | |
|         {
 | |
|             private boolean killed = false;
 | |
|             public Object execute() throws Throwable
 | |
|             {
 | |
|                 // Do some work
 | |
|                 nodeService.deleteNode(workingNodeRef);
 | |
|                 // Successful upon retry
 | |
|                 if (killed)
 | |
|                 {
 | |
|                     return null;
 | |
|                 }
 | |
|                 // Kill the connection the first time
 | |
|                 HibernateConnectionKiller killer = new HibernateConnectionKiller();
 | |
|                 killer.setSessionFactory((SessionFactory)ctx.getBean("sessionFactory"));
 | |
|                 killer.killConnection();
 | |
|                 killed = true;
 | |
|                 return null;
 | |
|             }
 | |
|         };
 | |
|         // This should work
 | |
|         txnHelper.doInTransaction(killConnectionCallback);
 | |
|     }
 | |
|     
 | |
|     public void testZeroAndNegativeRetries()
 | |
|     {
 | |
|         final MutableInt callCount = new MutableInt(0);
 | |
|         RetryingTransactionCallback<Long> callback = new RetryingTransactionCallback<Long>()
 | |
|         {
 | |
|             public Long execute() throws Throwable
 | |
|             {
 | |
|                 callCount.setValue(callCount.intValue() + 1);
 | |
|                 throw new ConcurrentModificationException();
 | |
|             }
 | |
|         };
 | |
|         // No retries
 | |
|         callCount.setValue(0);
 | |
|         txnHelper.setMaxRetries(0);
 | |
|         try
 | |
|         {
 | |
|             txnHelper.doInTransaction(callback);
 | |
|         }
 | |
|         catch (ConcurrentModificationException e)
 | |
|         {
 | |
|             // Expected
 | |
|         }
 | |
|         assertEquals("Should have been called exactly once", 1, callCount.intValue());
 | |
|         
 | |
|         // Negative retries
 | |
|         callCount.setValue(0);
 | |
|         txnHelper.setMaxRetries(-1);
 | |
|         try
 | |
|         {
 | |
|             txnHelper.doInTransaction(callback);
 | |
|         }
 | |
|         catch (ConcurrentModificationException e)
 | |
|         {
 | |
|             // Expected
 | |
|         }
 | |
|         assertEquals("Should have been called exactly once", 1, callCount.intValue());
 | |
|     }
 | |
|     
 | |
|     public void testTimeLimit()
 | |
|     {
 | |
|         final RetryingTransactionHelper txnHelper = new RetryingTransactionHelper();
 | |
|         txnHelper.setTransactionService(transactionService);
 | |
|         txnHelper.setMaxExecutionMs(3000);
 | |
|         final List<Throwable> caughtExceptions = Collections.synchronizedList(new LinkedList<Throwable>());
 | |
| 
 | |
|         // Try submitting a request after a timeout
 | |
|         runThreads(txnHelper, caughtExceptions, new Pair(0, 1000), new Pair(0, 5000), new Pair(4000, 1000));
 | |
|         assertEquals("Expected 1 exception", 1, caughtExceptions.size());
 | |
|         assertTrue("Excpected TooBusyException", caughtExceptions.get(0) instanceof TooBusyException);
 | |
| 
 | |
|         // Stay within timeout limits
 | |
|         caughtExceptions.clear();
 | |
|         runThreads(txnHelper, caughtExceptions, new Pair(0, 1000), new Pair(0, 2000), new Pair(0, 1000), new Pair(1000, 1000), new Pair(1000, 2000), new Pair(2000, 1000));
 | |
|         if (caughtExceptions.size() > 0)
 | |
|         {
 | |
|             throw new RuntimeException("Unexpected exception", caughtExceptions.get(0));
 | |
|         }
 | |
|     }
 | |
|     
 | |
|     private void runThreads(final RetryingTransactionHelper txnHelper, final List<Throwable> caughtExceptions,
 | |
|             Pair<Integer, Integer>... startDurationPairs)
 | |
|     {
 | |
|         ExecutorService executorService = new ThreadPoolExecutor(10, 10, 0L, TimeUnit.MILLISECONDS,
 | |
|                 new ArrayBlockingQueue<Runnable>(10));
 | |
| 
 | |
|         class Work implements Runnable
 | |
|         {
 | |
|             private final CountDownLatch startLatch;
 | |
|             private final long endTime;
 | |
| 
 | |
|             public Work(CountDownLatch startLatch, long endTime)
 | |
|             {
 | |
|                 this.startLatch = startLatch;
 | |
|                 this.endTime = endTime;
 | |
|             }
 | |
| 
 | |
|             public void run()
 | |
|             {
 | |
|                 try
 | |
|                 {
 | |
|                     txnHelper.doInTransaction(new RetryingTransactionCallback<Void>()
 | |
|                     {
 | |
| 
 | |
|                         public Void execute() throws Throwable
 | |
|                         {
 | |
|                             // Signal that we've started
 | |
|                             startLatch.countDown();
 | |
| 
 | |
|                             long duration = endTime - System.currentTimeMillis();
 | |
|                             if (duration > 0)
 | |
|                             {
 | |
|                                 Thread.sleep(duration);
 | |
|                             }
 | |
|                             return null;
 | |
|                         }
 | |
|                     });
 | |
|                 }
 | |
|                 catch (Throwable e)
 | |
|                 {
 | |
|                     caughtExceptions.add(e);
 | |
|                     // We never got a chance to signal we had started so do it now
 | |
|                     if (startLatch.getCount() > 0)
 | |
|                     {
 | |
|                         startLatch.countDown();
 | |
|                     }
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
|         ;
 | |
| 
 | |
|         // Schedule the transactions at their required start times
 | |
|         long startTime = System.currentTimeMillis();
 | |
|         long currentStart = 0;
 | |
|         for (Pair<Integer, Integer> pair : startDurationPairs)
 | |
|         {
 | |
|             int start = pair.getFirst();
 | |
|             long now = System.currentTimeMillis();
 | |
|             long then = startTime + start;
 | |
|             if (then > now)
 | |
|             {
 | |
|                 try
 | |
|                 {
 | |
|                     Thread.sleep(then - now);
 | |
|                 }
 | |
|                 catch (InterruptedException e)
 | |
|                 {
 | |
|                 }
 | |
|                 currentStart = start;
 | |
|             }
 | |
|             CountDownLatch startLatch = new CountDownLatch(1);
 | |
|             Runnable work = new Work(startLatch, startTime + currentStart + pair.getSecond());
 | |
|             executorService.execute(work);
 | |
|             try
 | |
|             {
 | |
|                 // Wait for the thread to get up and running. We need them starting in sequence
 | |
|                 startLatch.await(60, TimeUnit.SECONDS);
 | |
|             }
 | |
|             catch (InterruptedException e)
 | |
|             {
 | |
|             }
 | |
|         }
 | |
|         // Wait for the threads to have finished
 | |
|         executorService.shutdown();
 | |
|         try
 | |
|         {
 | |
|             executorService.awaitTermination(60, TimeUnit.SECONDS);
 | |
|         }
 | |
|         catch (InterruptedException e)
 | |
|         {
 | |
|         }
 | |
| 
 | |
|     }
 | |
| 
 | |
|     /**
 | |
|      * Helper class to kill the session's DB connection
 | |
|      */
 | |
|     private class HibernateConnectionKiller extends HibernateDaoSupport
 | |
|     {
 | |
|         @SuppressWarnings("deprecation")
 | |
|         private void killConnection() throws Exception
 | |
|         {
 | |
|             getSession().connection().rollback();
 | |
|         }
 | |
|     }
 | |
| }
 |