View Javadoc

1   /*
2    * $Id: CounterCallback.java 21939 2011-05-18 13:32:09Z aperepel $
3    * --------------------------------------------------------------------------------------
4    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.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 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      /**
39       * Increment callback count.
40       * @return current count after increment
41       */
42      protected int incCallbackCount()
43      {
44          return callbackCount.incrementAndGet();
45      }
46  
47      public int getCallbackCount()
48      {
49          return callbackCount.intValue();
50      }
51  }