01: /**
02: *
03: * Licensed to the Apache Software Foundation (ASF) under one or more
04: * contributor license agreements. See the NOTICE file distributed with
05: * this work for additional information regarding copyright ownership.
06: * The ASF licenses this file to You under the Apache License, Version 2.0
07: * (the "License"); you may not use this file except in compliance with
08: * the License. You may obtain a copy of the License at
09: *
10: * http://www.apache.org/licenses/LICENSE-2.0
11: *
12: * Unless required by applicable law or agreed to in writing, software
13: * distributed under the License is distributed on an "AS IS" BASIS,
14: * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15: * See the License for the specific language governing permissions and
16: * limitations under the License.
17: */package org.apache.openejb.test.beans;
18:
19: import javax.ejb.Stateless;
20: import java.util.Map;
21: import java.util.TreeMap;
22: import java.util.concurrent.CountDownLatch;
23: import java.util.concurrent.TimeUnit;
24:
25: @Stateless
26: public class TimerSyncBean implements TimerSync {
27: private static Map<String, CountDownLatch> latches = new TreeMap<String, CountDownLatch>();
28:
29: public boolean waitFor(String name) {
30: CountDownLatch latch = getLatch(name);
31: try {
32: return latch.await(1, TimeUnit.SECONDS);
33: } catch (InterruptedException e) {
34: return false;
35: }
36: }
37:
38: public void countDown(String name) {
39: CountDownLatch latch = getLatch(name);
40: latch.countDown();
41: }
42:
43: private synchronized CountDownLatch getLatch(String name) {
44: CountDownLatch latch = latches.get(name);
45: if (latch == null) {
46: latch = new CountDownLatch(1);
47: latches.put(name, latch);
48: }
49: return latch;
50: }
51:
52: }
|