1   /**
2    * Copyright (c) 2000-2010 Liferay, Inc. All rights reserved.
3    *
4    * This library is free software; you can redistribute it and/or modify it under
5    * the terms of the GNU Lesser General Public License as published by the Free
6    * Software Foundation; either version 2.1 of the License, or (at your option)
7    * any later version.
8    *
9    * This library is distributed in the hope that it will be useful, but WITHOUT
10   * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
11   * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
12   * details.
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          for (ThreadLocal<?> threadLocal : threadLocalSet) {
44              threadLocal.remove();
45          }
46      }
47  
48      private static ThreadLocal<Set<ThreadLocal<?>>> _threadLocalSet =
49          new InitialThreadLocal<Set<ThreadLocal<?>>>(
50              new HashSet<ThreadLocal<?>>());
51  
52  }