Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
阿啄debugIT
transmittable-thread-local
提交
4caa49c4
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,体验更适合开发者的 AI 搜索 >>
提交
4caa49c4
编写于
12月 04, 2015
作者:
oldratlee
🔥
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add DistributedTracerUseDemo
上级
68e1d3b5
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
179 addition
and
0 deletion
+179
-0
src/test/java/com/alibaba/mtc/demo/distributed_tracer/DistributedTracerUseDemo.java
...mtc/demo/distributed_tracer/DistributedTracerUseDemo.java
+179
-0
未找到文件。
src/test/java/com/alibaba/mtc/demo/distributed_tracer/DistributedTracerUseDemo.java
0 → 100644
浏览文件 @
4caa49c4
package
com.alibaba.mtc.demo.distributed_tracer
;
import
com.alibaba.mtc.MtContextThreadLocal
;
import
com.alibaba.mtc.Utils
;
import
com.alibaba.mtc.threadpool.MtContextExecutors
;
import
java.util.Map
;
import
java.util.concurrent.*
;
import
java.util.concurrent.atomic.AtomicInteger
;
/**
* DistributedTracer(DT) use demo.
*
* @author Jerry Lee (oldratlee at gmail dot com)
*/
public
class
DistributedTracerUseDemo
{
static
ThreadFactory
threadFactory
=
new
ThreadFactory
()
{
@Override
public
Thread
newThread
(
Runnable
r
)
{
Thread
thread
=
new
Thread
(
r
,
"Executors"
);
thread
.
setDaemon
(
true
);
return
thread
;
}
};
static
ExecutorService
executorService
=
MtContextExecutors
.
getMtcExecutorService
(
Executors
.
newFixedThreadPool
(
1
,
threadFactory
));
static
{
// 挤满线程, 保证线程不是用的时候new的, 确保验证MTC的传递功能
Utils
.
expandThreadPool
(
executorService
);
}
static
class
DtTransferInfo
{
public
String
traceId
;
public
String
baseSpanId
;
public
DtTransferInfo
(
String
traceId
,
String
baseSpanId
)
{
this
.
traceId
=
traceId
;
this
.
baseSpanId
=
baseSpanId
;
}
}
private
static
MtContextThreadLocal
<
DtTransferInfo
>
transferInfo
=
new
MtContextThreadLocal
<
DtTransferInfo
>()
{
@Override
protected
DtTransferInfo
copy
(
DtTransferInfo
parentValue
)
{
increaseSpanIdRefCounter
();
return
super
.
childValue
(
parentValue
);
}
@Override
protected
void
afterExecute
()
{
decreaseSpanIdRefCounter
();
}
};
static
class
LeafSpanIdInfo
{
AtomicInteger
current
=
new
AtomicInteger
(
1
);
AtomicInteger
refCounter
=
new
AtomicInteger
(
0
);
}
private
static
Map
<
String
,
LeafSpanIdInfo
>
traceId2LeafSpanIdInfo
=
new
ConcurrentHashMap
<
String
,
LeafSpanIdInfo
>();
static
void
increaseSpanIdRefCounter
()
{
DtTransferInfo
dtTransferInfo
=
transferInfo
.
get
();
String
traceId
=
dtTransferInfo
.
traceId
;
int
refCounter
=
traceId2LeafSpanIdInfo
.
get
(
traceId
).
refCounter
.
incrementAndGet
();
System
.
out
.
printf
(
"DEBUG: Increase reference counter(%s) for traceId %s in thread %s\n"
,
refCounter
,
traceId
,
Thread
.
currentThread
().
getName
());
}
static
void
decreaseSpanIdRefCounter
()
{
DtTransferInfo
dtTransferInfo
=
transferInfo
.
get
();
String
traceId
=
dtTransferInfo
.
traceId
;
LeafSpanIdInfo
leafSpanIdInfo
=
traceId2LeafSpanIdInfo
.
get
(
traceId
);
int
refCounter
=
leafSpanIdInfo
.
refCounter
.
decrementAndGet
();
System
.
out
.
printf
(
"DEBUG: Decrease reference counter(%s) for traceId %s in thread %s\n"
,
refCounter
,
traceId
,
Thread
.
currentThread
().
getName
());
if
(
refCounter
==
0
)
{
traceId2LeafSpanIdInfo
.
remove
(
traceId
);
System
.
out
.
printf
(
"DEBUG: Clear traceId2LeafSpanIdInfo for traceId %s in thread %s\n"
,
traceId
,
Thread
.
currentThread
().
getName
());
}
else
if
(
refCounter
<
0
)
{
throw
new
IllegalStateException
(
"Leaf Span Id Info Reference counter has Bug!!"
);
}
}
static
int
increaseLeafSpanCurrentAndReturn
()
{
DtTransferInfo
dtTransferInfo
=
transferInfo
.
get
();
String
traceId
=
dtTransferInfo
.
traceId
;
LeafSpanIdInfo
leafSpanIdInfo
=
traceId2LeafSpanIdInfo
.
get
(
traceId
);
return
leafSpanIdInfo
.
current
.
getAndIncrement
();
}
public
static
void
main
(
String
[]
args
)
throws
Exception
{
rpcInvokeIn
();
Thread
.
sleep
(
1000
);
}
static
void
rpcInvokeIn
()
{
////////////////////////////////////////////////
// DistributedTracer Framework Code
////////////////////////////////////////////////
// Get Trace Id and Span Id from RPC Context
String
traceId
=
"traceId_XXXYYY"
;
String
baseSpanId
=
"1.1"
;
transferInfo
.
set
(
new
DtTransferInfo
(
traceId
,
baseSpanId
));
traceId2LeafSpanIdInfo
.
put
(
traceId
,
new
LeafSpanIdInfo
());
increaseSpanIdRefCounter
();
////////////////////////////////////////////////
// Biz Code
////////////////////////////////////////////////
syncMethod
();
////////////////////////////////////////////////
// DistributedTracer Framework Code
////////////////////////////////////////////////
decreaseSpanIdRefCounter
();
}
static
void
syncMethod
()
{
// async call by MTC Executor, Test OK!
executorService
.
submit
(
new
Runnable
()
{
@Override
public
void
run
()
{
asyncMethod
();
}
});
// async call by new Thread
// FIXME Bug!! 没有 Increase/Decrease reference counter操作!
new
Thread
(
new
Runnable
()
{
@Override
public
void
run
()
{
syncMethod_ByNewThread
();
}
},
"Thread-by-new"
).
start
();
invokeServerWithRpc
(
"server 1"
);
}
static
void
asyncMethod
()
{
invokeServerWithRpc
(
"server 2"
);
}
static
void
syncMethod_ByNewThread
()
{
invokeServerWithRpc
(
"server 3"
);
}
// RPC invoke
static
void
invokeServerWithRpc
(
String
server
)
{
////////////////////////////////////////////////
// DistributedTracer Framework Code
////////////////////////////////////////////////
int
leafSpanCurrent
=
increaseLeafSpanCurrentAndReturn
();
// Set RpcContext
// Mocked, should use RPC util to get Rpc Context instead
Map
<
String
,
String
>
rpcContext
=
new
ConcurrentHashMap
<
String
,
String
>();
rpcContext
.
put
(
"traceId"
,
transferInfo
.
get
().
traceId
);
rpcContext
.
put
(
"spanId"
,
transferInfo
.
get
().
baseSpanId
+
"."
+
leafSpanCurrent
);
// Do Rpc
// ...
System
.
out
.
printf
(
"Do Rpc invocation to server %s with %s\n"
,
server
,
rpcContext
);
}
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录