| ThreadLocalRegistry.java |
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 }