View Javadoc

1   /*
2    * $Id: CounterCallback.java 10489 2008-01-23 17:53:38Z dfeist $
3    * --------------------------------------------------------------------------------------
4    * Copyright (c) MuleSource, Inc.  All rights reserved.  http://www.mulesource.com
5    *
6    * The software in this package is published under the terms of the CPAL v1.0
7    * license, a copy of which has been included with this distribution in the
8    * LICENSE.txt file.
9    */
10  package org.mule.tck.functional;
11  
12  import org.mule.api.MuleEventContext;
13  
14  import edu.emory.mathcs.backport.java.util.concurrent.atomic.AtomicInteger;
15  
16  /**
17   * A test callback that counts the number of messages received.
18   */
19  public class CounterCallback implements EventCallback
20  {
21      private AtomicInteger callbackCount;
22  
23      public CounterCallback()
24      {
25          callbackCount = new AtomicInteger(0);
26      }
27  
28      public CounterCallback(AtomicInteger callbackCount)
29      {
30          this.callbackCount = callbackCount;
31      }
32  
33      public void eventReceived(MuleEventContext context, Object Component) throws Exception
34      {
35          incCallbackCount();
36      }
37  
38      protected void incCallbackCount()
39      {
40          callbackCount.incrementAndGet();
41      }
42  
43      public int getCallbackCount()
44      {
45          return callbackCount.intValue();
46      }
47  }