Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
淡淡忧伤的程序员
DolphinScheduler
提交
5e570368
DolphinScheduler
项目概览
淡淡忧伤的程序员
/
DolphinScheduler
与 Fork 源项目一致
Fork自
apache / DolphinScheduler
通知
48
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
DolphinScheduler
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
5e570368
编写于
7月 03, 2019
作者:
L
lidongdai
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update TaskQueueImplTest
上级
c8906686
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
38 addition
and
39 deletion
+38
-39
escheduler-common/src/test/java/cn/escheduler/common/queue/TaskQueueImplTest.java
...st/java/cn/escheduler/common/queue/TaskQueueImplTest.java
+38
-39
未找到文件。
escheduler-common/src/test/java/cn/escheduler/common/queue/TaskQueueImplTest.java
浏览文件 @
5e570368
...
...
@@ -17,12 +17,15 @@
package
cn.escheduler.common.queue
;
import
cn.escheduler.common.Constants
;
import
org.junit.After
;
import
org.junit.Assert
;
import
org.junit.Before
;
import
org.junit.Test
;
import
org.slf4j.Logger
;
import
org.slf4j.LoggerFactory
;
import
java.util.Arrays
;
import
java.util.List
;
import
java.util.Random
;
import
static
org
.
junit
.
Assert
.
assertEquals
;
...
...
@@ -34,61 +37,62 @@ public class TaskQueueImplTest {
private
static
final
Logger
logger
=
LoggerFactory
.
getLogger
(
TaskQueueImplTest
.
class
);
ITaskQueue
tasksQueue
=
null
;
@Before
public
void
before
(){
tasksQueue
=
TaskQueueFactory
.
getTaskQueueInstance
();
//clear all data
tasksQueue
.
delete
();
}
@Test
public
void
testTaskQueue
(){
ITaskQueue
tasksQueue
=
TaskQueueFactory
.
getTaskQueueInstance
();
@After
public
void
after
(){
//clear all data
tasksQueue
.
delete
();
}
@Test
public
void
testAdd
(){
//add
tasksQueue
.
add
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
"1"
);
tasksQueue
.
add
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
"
2
"
);
tasksQueue
.
add
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
"
3
"
);
tasksQueue
.
add
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
"
4
"
);
tasksQueue
.
add
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
"1
_1_1_1_2130706433,3232236775
"
);
tasksQueue
.
add
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
"
0_1_1_1_2130706433,3232236775
"
);
tasksQueue
.
add
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
"
1_1_0_1_2130706433,3232236775
"
);
tasksQueue
.
add
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
"
1_2_1_1_2130706433,3232236775
"
);
//pop
String
node1
=
tasksQueue
.
poll
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
1
).
get
(
0
);
List
<
String
>
tasks
=
tasksQueue
.
poll
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
1
);
assertEquals
(
node1
,
"1"
);
if
(
tasks
.
size
()
<
0
){
return
;
}
String
node2
=
tasksQueue
.
poll
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
1
).
get
(
0
);
assertEquals
(
node2
,
"2"
);
//pop
String
node1
=
tasks
.
get
(
0
);
//sadd
String
task1
=
"1.1.1.1-1-mr"
;
String
task2
=
"1.1.1.2-2-mr"
;
String
task3
=
"1.1.1.3-3-mr"
;
String
task4
=
"1.1.1.4-4-mr"
;
String
task5
=
"1.1.1.5-5-mr"
;
assertEquals
(
node1
,
"0_0000000001_1_0000000001"
);
tasksQueue
.
sadd
(
Constants
.
SCHEDULER_TASKS_KILL
,
task1
);
tasksQueue
.
sadd
(
Constants
.
SCHEDULER_TASKS_KILL
,
task2
);
tasksQueue
.
sadd
(
Constants
.
SCHEDULER_TASKS_KILL
,
task3
);
tasksQueue
.
sadd
(
Constants
.
SCHEDULER_TASKS_KILL
,
task4
);
tasksQueue
.
sadd
(
Constants
.
SCHEDULER_TASKS_KILL
,
task5
);
tasksQueue
.
sadd
(
Constants
.
SCHEDULER_TASKS_KILL
,
task5
);
//repeat task
tasks
=
tasksQueue
.
poll
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
1
);
Assert
.
assertEquals
(
tasksQueue
.
smembers
(
Constants
.
SCHEDULER_TASKS_KILL
).
size
(),
5
);
logger
.
info
(
Arrays
.
toString
(
tasksQueue
.
smembers
(
Constants
.
SCHEDULER_TASKS_KILL
).
toArray
()));
//srem
tasksQueue
.
srem
(
Constants
.
SCHEDULER_TASKS_KILL
,
task5
);
//smembers
Assert
.
assertEquals
(
tasksQueue
.
smembers
(
Constants
.
SCHEDULER_TASKS_KILL
).
size
(),
4
);
logger
.
info
(
Arrays
.
toString
(
tasksQueue
.
smembers
(
Constants
.
SCHEDULER_TASKS_KILL
).
toArray
()));
if
(
tasks
.
size
()
<
0
){
return
;
}
String
node2
=
tasks
.
get
(
0
);
assertEquals
(
node2
,
"0_0000000001_1_0000000001"
);
}
/**
* test one million data from zookeeper queue
*/
@Test
public
void
extremeTest
(){
ITaskQueue
tasksQueue
=
TaskQueueFactory
.
getTaskQueueInstance
();
//clear all data
tasksQueue
.
delete
();
int
total
=
30
*
10000
;
for
(
int
i
=
0
;
i
<
total
;
i
++)
...
...
@@ -104,11 +108,6 @@ public class TaskQueueImplTest {
String
node1
=
tasksQueue
.
poll
(
Constants
.
SCHEDULER_TASKS_QUEUE
,
1
).
get
(
0
);
assertEquals
(
node1
,
"0"
);
//clear all data
tasksQueue
.
delete
();
}
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录