1   /**
2    * Copyright (c) 2000-2010 Liferay, Inc. All rights reserved.
3    *
4    * The contents of this file are subject to the terms of the Liferay Enterprise
5    * Subscription License ("License"). You may not use this file except in
6    * compliance with the License. You can obtain a copy of the License by
7    * contacting Liferay, Inc. See the License for the specific language governing
8    * permissions and limitations under the License, including but not limited to
9    * distribution rights of the Software.
10   *
11   *
12   *
13   */
14  
15  package com.liferay.portal.kernel.util;
16  
17  import java.util.HashSet;
18  import java.util.Set;
19  
20  /**
21   * <a href="ThreadLocalRegistry.java.html"><b><i>View Source</i></b></a>
22   *
23   * @author Shuyang Zhou
24   */
25  public class ThreadLocalRegistry {
26  
27      public static ThreadLocal<?>[] captureSnapshot() {
28          Set<ThreadLocal<?>> threadLocalSet = _threadLocalSet.get();
29  
30          return threadLocalSet.toArray(
31              new ThreadLocal<?>[threadLocalSet.size()]);
32      }
33  
34      public static void registerThreadLocal(ThreadLocal<?> threadLocal) {
35          Set<ThreadLocal<?>> threadLocalSet = _threadLocalSet.get();
36  
37          threadLocalSet.add(threadLocal);
38      }
39  
40      public static void resetThreadLocals() {
41          Set<ThreadLocal<?>> threadLocalSet = _threadLocalSet.get();
42  
43          if (threadLocalSet == null) {
44              return;
45          }
46  
47          for (ThreadLocal<?> threadLocal : threadLocalSet) {
48              threadLocal.remove();
49          }
50      }
51  
52      private static ThreadLocal<Set<ThreadLocal<?>>> _threadLocalSet =
53          new InitialThreadLocal<Set<ThreadLocal<?>>>(
54              new HashSet<ThreadLocal<?>>());
55  
56  }