Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
阿啄debugIT
transmittable-thread-local
提交
8c86d80e
T
transmittable-thread-local
项目概览
阿啄debugIT
/
transmittable-thread-local
与 Fork 源项目一致
从无法访问的项目Fork
通知
8
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
transmittable-thread-local
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
8c86d80e
编写于
10月 31, 2018
作者:
oldratlee
🔥
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
WIP
上级
911e0cdc
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
152 addition
and
26 deletion
+152
-26
src/main/java/com/alibaba/ttl/TransmittableOnlyThreadLocal.java
...in/java/com/alibaba/ttl/TransmittableOnlyThreadLocal.java
+109
-0
src/main/java/com/alibaba/ttl/TransmittableThreadLocal.java
src/main/java/com/alibaba/ttl/TransmittableThreadLocal.java
+43
-26
未找到文件。
src/main/java/com/alibaba/ttl/TransmittableOnlyThreadLocal.java
0 → 100644
浏览文件 @
8c86d80e
package
com.alibaba.ttl
;
import
java.util.Map
;
import
java.util.logging.Logger
;
/**
* {@link TransmittableOnlyThreadLocal} can transmit value from the thread of submitting task to the thread of executing task.
* <p>
* Note: {@link TransmittableOnlyThreadLocal} extends {@link InheritableThreadLocal},
* so {@link TransmittableOnlyThreadLocal} first is a {@link InheritableThreadLocal}.
*
* @author Jerry Lee (oldratlee at gmail dot com)
* @see TtlRunnable
* @see TtlCallable
* @since 0.10.0
*/
public
class
TransmittableOnlyThreadLocal
<
T
>
extends
InheritableThreadLocal
<
T
>
{
private
static
final
Logger
logger
=
Logger
.
getLogger
(
TransmittableOnlyThreadLocal
.
class
.
getName
());
/**
* Computes the value for this transmittable thread-local variable
* as a function of the source thread's value at the time the task
* Object is created. This method is called from {@link TtlRunnable} or
* {@link TtlCallable} when it create, before the task is started.
* <p>
* This method merely returns reference of its source thread value, and should be overridden
* if a different behavior is desired.
*
* @since 1.0.0
*/
protected
T
copy
(
T
parentValue
)
{
return
parentValue
;
}
/**
* Callback method before task object({@link TtlRunnable}/{@link TtlCallable}) execute.
* <p>
* Default behavior is do nothing, and should be overridden
* if a different behavior is desired.
* <p>
* Do not throw any exception, just ignored.
*
* @since 1.2.0
*/
protected
void
beforeExecute
()
{
}
/**
* Callback method after task object({@link TtlRunnable}/{@link TtlCallable}) execute.
* <p>
* Default behavior is do nothing, and should be overridden
* if a different behavior is desired.
* <p>
* Do not throw any exception, just ignored.
*
* @since 1.2.0
*/
protected
void
afterExecute
()
{
}
/**
* see {@link InheritableThreadLocal#get()}
*/
@Override
public
final
T
get
()
{
T
value
=
super
.
get
();
if
(
null
!=
value
)
addValue
();
return
value
;
}
/**
* see {@link InheritableThreadLocal#set}
*/
@Override
public
final
void
set
(
T
value
)
{
super
.
set
(
value
);
// may set null to remove value
if
(
null
==
value
)
removeValue
();
else
addValue
();
}
/**
* see {@link InheritableThreadLocal#remove()}
*/
@Override
public
final
void
remove
()
{
removeValue
();
super
.
remove
();
}
void
superRemove
()
{
super
.
remove
();
}
T
copyValue
()
{
return
copy
(
get
());
}
private
void
addValue
()
{
final
InheritableThreadLocal
<
Map
<
Object
,
?>>
holder
=
TransmittableThreadLocal
.
holder
;
if
(!
holder
.
get
().
containsKey
(
this
))
{
holder
.
get
().
put
(
this
,
null
);
// WeakHashMap supports null value.
}
}
private
void
removeValue
()
{
TransmittableThreadLocal
.
holder
.
get
().
remove
(
this
);
}
}
src/main/java/com/alibaba/ttl/TransmittableThreadLocal.java
浏览文件 @
8c86d80e
...
...
@@ -105,16 +105,16 @@ public class TransmittableThreadLocal<T> extends InheritableThreadLocal<T> {
// 1. The value of holder is type Map<TransmittableThreadLocal<?>, ?> (WeakHashMap implementation),
// but it is used as *set*.
// 2. WeakHashMap support null value.
private
static
InheritableThreadLocal
<
Map
<
TransmittableThreadLocal
<?>
,
?>>
holder
=
new
InheritableThreadLocal
<
Map
<
TransmittableThreadLocal
<?>
,
?>>()
{
static
InheritableThreadLocal
<
Map
<
Object
,
?>>
holder
=
new
InheritableThreadLocal
<
Map
<
Object
,
?>>()
{
@Override
protected
Map
<
TransmittableThreadLocal
<?>
,
?>
initialValue
()
{
return
new
WeakHashMap
<
TransmittableThreadLocal
<?>
,
Object
>();
protected
Map
<
Object
,
?>
initialValue
()
{
return
new
WeakHashMap
<
Object
,
Object
>();
}
@Override
protected
Map
<
TransmittableThreadLocal
<?>,
?>
childValue
(
Map
<
TransmittableThreadLocal
<?>
,
?>
parentValue
)
{
return
new
WeakHashMap
<
TransmittableThreadLocal
<?>
,
Object
>(
parentValue
);
protected
Map
<
Object
,
?>
childValue
(
Map
<
Object
,
?>
parentValue
)
{
return
new
WeakHashMap
<
Object
,
Object
>(
parentValue
);
}
};
...
...
@@ -129,12 +129,14 @@ public class TransmittableThreadLocal<T> extends InheritableThreadLocal<T> {
}
private
static
void
doExecuteCallback
(
boolean
isBefore
)
{
for
(
Map
.
Entry
<
TransmittableThreadLocal
<?>
,
?>
entry
:
holder
.
get
().
entrySet
())
{
TransmittableThreadLocal
<?>
threadLocal
=
entry
.
getKey
();
for
(
Map
.
Entry
<
Object
,
?>
entry
:
holder
.
get
().
entrySet
())
{
Object
threadLocal
=
entry
.
getKey
();
try
{
if
(
isBefore
)
threadLocal
.
beforeExecute
();
else
threadLocal
.
afterExecute
();
if
(
isBefore
)
if
(
threadLocal
instanceof
TransmittableThreadLocal
)
((
TransmittableThreadLocal
<
Object
>)
threadLocal
).
beforeExecute
();
else
((
TransmittableOnlyThreadLocal
<
Object
>)
threadLocal
).
afterExecute
();
}
catch
(
Throwable
t
)
{
if
(
logger
.
isLoggable
(
Level
.
WARNING
))
{
logger
.
log
(
Level
.
WARNING
,
"TTL exception when "
+
(
isBefore
?
"beforeExecute"
:
"afterExecute"
)
+
", cause: "
+
t
.
toString
(),
t
);
...
...
@@ -153,9 +155,11 @@ public class TransmittableThreadLocal<T> extends InheritableThreadLocal<T> {
System
.
out
.
println
(
"Start TransmittableThreadLocal Dump..."
);
}
for
(
Map
.
Entry
<
TransmittableThreadLocal
<?>,
?>
entry
:
holder
.
get
().
entrySet
())
{
final
TransmittableThreadLocal
<?>
key
=
entry
.
getKey
();
System
.
out
.
println
(
key
.
get
());
for
(
Map
.
Entry
<
Object
,
?>
entry
:
holder
.
get
().
entrySet
())
{
final
Object
threadLocal
=
entry
.
getKey
();
if
(
threadLocal
instanceof
TransmittableThreadLocal
)
System
.
out
.
println
(((
TransmittableThreadLocal
)
threadLocal
).
get
());
else
System
.
out
.
println
(((
TransmittableOnlyThreadLocal
)
threadLocal
).
get
());
}
System
.
out
.
println
(
"TransmittableThreadLocal Dump end!"
);
}
...
...
@@ -251,9 +255,11 @@ public class TransmittableThreadLocal<T> extends InheritableThreadLocal<T> {
*/
@Nonnull
public
static
Object
capture
()
{
Map
<
TransmittableThreadLocal
<?>,
Object
>
captured
=
new
HashMap
<
TransmittableThreadLocal
<?>,
Object
>();
for
(
TransmittableThreadLocal
<?>
threadLocal
:
holder
.
get
().
keySet
())
{
captured
.
put
(
threadLocal
,
threadLocal
.
copyValue
());
Map
<
Object
,
Object
>
captured
=
new
HashMap
<
Object
,
Object
>();
for
(
Object
threadLocal
:
holder
.
get
().
keySet
())
{
if
(
threadLocal
instanceof
TransmittableThreadLocal
)
captured
.
put
(
threadLocal
,
((
TransmittableThreadLocal
)
threadLocal
).
copyValue
());
else
captured
.
put
(
threadLocal
,
((
TransmittableOnlyThreadLocal
)
threadLocal
).
copyValue
());
}
return
captured
;
}
...
...
@@ -271,21 +277,29 @@ public class TransmittableThreadLocal<T> extends InheritableThreadLocal<T> {
public
static
Object
replay
(
@Nonnull
Object
captured
)
{
@SuppressWarnings
(
"unchecked"
)
Map
<
TransmittableThreadLocal
<?>,
Object
>
capturedMap
=
(
Map
<
TransmittableThreadLocal
<?>,
Object
>)
captured
;
Map
<
TransmittableThreadLocal
<?>,
Object
>
backup
=
new
HashMap
<
TransmittableThreadLocal
<?>
,
Object
>();
Map
<
Object
,
Object
>
backup
=
new
HashMap
<
Object
,
Object
>();
for
(
Iterator
<?
extends
Map
.
Entry
<
TransmittableThreadLocal
<?>
,
?>>
iterator
=
holder
.
get
().
entrySet
().
iterator
();
for
(
Iterator
<?
extends
Map
.
Entry
<
Object
,
?>>
iterator
=
holder
.
get
().
entrySet
().
iterator
();
iterator
.
hasNext
();
)
{
Map
.
Entry
<
TransmittableThreadLocal
<?>
,
?>
next
=
iterator
.
next
();
TransmittableThreadLocal
<?>
threadLocal
=
next
.
getKey
();
Map
.
Entry
<
Object
,
?>
next
=
iterator
.
next
();
Object
threadLocal
=
next
.
getKey
();
// backup
backup
.
put
(
threadLocal
,
threadLocal
.
get
());
Object
value
;
if
(
threadLocal
instanceof
TransmittableThreadLocal
)
{
value
=
((
TransmittableThreadLocal
)
threadLocal
).
get
();
}
else
value
=
((
TransmittableOnlyThreadLocal
)
threadLocal
).
get
();
backup
.
put
(
threadLocal
,
value
);
// clear the TTL values that is not in captured
// avoid the extra TTL values after replay when run task
if
(!
capturedMap
.
containsKey
(
threadLocal
))
{
iterator
.
remove
();
threadLocal
.
superRemove
();
if
(
threadLocal
instanceof
TransmittableThreadLocal
)
((
TransmittableThreadLocal
)
threadLocal
).
superRemove
();
else
((
TransmittableOnlyThreadLocal
)
threadLocal
).
superRemove
();
}
}
...
...
@@ -324,16 +338,19 @@ public class TransmittableThreadLocal<T> extends InheritableThreadLocal<T> {
// call afterExecute callback
doExecuteCallback
(
false
);
for
(
Iterator
<?
extends
Map
.
Entry
<
TransmittableThreadLocal
<?>
,
?>>
iterator
=
holder
.
get
().
entrySet
().
iterator
();
for
(
Iterator
<?
extends
Map
.
Entry
<
Object
,
?>>
iterator
=
holder
.
get
().
entrySet
().
iterator
();
iterator
.
hasNext
();
)
{
Map
.
Entry
<
TransmittableThreadLocal
<?>
,
?>
next
=
iterator
.
next
();
TransmittableThreadLocal
<?>
threadLocal
=
next
.
getKey
();
Map
.
Entry
<
Object
,
?>
next
=
iterator
.
next
();
Object
threadLocal
=
next
.
getKey
();
// clear the TTL values that is not in backup
// avoid the extra TTL values after restore
if
(!
backupMap
.
containsKey
(
threadLocal
))
{
iterator
.
remove
();
threadLocal
.
superRemove
();
if
(
threadLocal
instanceof
TransmittableThreadLocal
)
((
TransmittableThreadLocal
)
threadLocal
).
superRemove
();
else
((
TransmittableOnlyThreadLocal
)
threadLocal
).
superRemove
();
}
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录