Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
milvus
milvus
提交
9258f705
M
milvus
项目概览
milvus
/
milvus
大约 1 年 前同步成功
通知
261
Star
22476
Fork
2472
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
milvus
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
9258f705
编写于
1月 19, 2023
作者:
X
Xiaofan
提交者:
GitHub
1月 19, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix Bulkload ut print too much result (#21821)
Signed-off-by:
N
xiaofan-luan
<
xiaofan.luan@zilliz.com
>
上级
54ec22e8
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
22 addition
and
8 deletion
+22
-8
internal/rootcoord/import_manager.go
internal/rootcoord/import_manager.go
+2
-3
internal/rootcoord/import_manager_test.go
internal/rootcoord/import_manager_test.go
+5
-2
internal/rootcoord/scheduler_test.go
internal/rootcoord/scheduler_test.go
+7
-3
internal/util/paramtable/component_param.go
internal/util/paramtable/component_param.go
+8
-0
未找到文件。
internal/rootcoord/import_manager.go
浏览文件 @
9258f705
...
...
@@ -40,8 +40,7 @@ import (
)
const
(
MaxPendingCount
=
65536
// TODO: Make this configurable.
delimiter
=
"/"
delimiter
=
"/"
)
// checkPendingTasksInterval is the default interval to check and send out pending tasks,
...
...
@@ -94,7 +93,7 @@ func newImportManager(ctx context.Context, client kv.TxnKV,
mgr
:=
&
importManager
{
ctx
:
ctx
,
taskStore
:
client
,
pendingTasks
:
make
([]
*
datapb
.
ImportTaskInfo
,
0
,
MaxPendingCount
),
// currently task queue max size is 32
pendingTasks
:
make
([]
*
datapb
.
ImportTaskInfo
,
0
,
Params
.
RootCoordCfg
.
ImportMaxPendingTaskCount
.
GetAsInt
()
),
// currently task queue max size is 32
workingTasks
:
make
(
map
[
int64
]
*
datapb
.
ImportTaskInfo
),
busyNodes
:
make
(
map
[
int64
]
int64
),
pendingLock
:
sync
.
RWMutex
{},
...
...
internal/rootcoord/import_manager_test.go
浏览文件 @
9258f705
...
...
@@ -542,7 +542,10 @@ func TestImportManager_ImportJob(t *testing.T) {
globalCount
++
return
globalCount
,
0
,
nil
}
paramtable
.
Get
()
.
Save
(
Params
.
RootCoordCfg
.
ImportTaskSubPath
.
Key
,
"test_import_task"
)
paramtable
.
Get
()
.
Save
(
Params
.
RootCoordCfg
.
ImportMaxPendingTaskCount
.
Key
,
"16"
)
defer
paramtable
.
Get
()
.
Remove
(
Params
.
RootCoordCfg
.
ImportMaxPendingTaskCount
.
Key
)
colID
:=
int64
(
100
)
mockKv
:=
memkv
.
NewMemoryKV
()
callMarkSegmentsDropped
:=
func
(
ctx
context
.
Context
,
segIDs
[]
typeutil
.
UniqueID
)
(
*
commonpb
.
Status
,
error
)
{
...
...
@@ -654,9 +657,9 @@ func TestImportManager_ImportJob(t *testing.T) {
// the pending list already has one task
// once task count exceeds MaxPendingCount, return error
for
i
:=
0
;
i
<=
MaxPendingCount
;
i
++
{
for
i
:=
0
;
i
<=
Params
.
RootCoordCfg
.
ImportMaxPendingTaskCount
.
GetAsInt
()
;
i
++
{
resp
=
mgr
.
importJob
(
context
.
TODO
(),
rowReq
,
colID
,
0
)
if
i
<
MaxPendingCount
-
1
{
if
i
<
Params
.
RootCoordCfg
.
ImportMaxPendingTaskCount
.
GetAsInt
()
-
1
{
assert
.
Equal
(
t
,
commonpb
.
ErrorCode_Success
,
resp
.
Status
.
ErrorCode
)
}
else
{
assert
.
NotEqual
(
t
,
commonpb
.
ErrorCode_Success
,
resp
.
Status
.
ErrorCode
)
...
...
internal/rootcoord/scheduler_test.go
浏览文件 @
9258f705
...
...
@@ -190,9 +190,13 @@ func Test_scheduler_updateDdlMinTsLoop(t *testing.T) {
paramtable
.
Get
()
.
Save
(
Params
.
ProxyCfg
.
TimeTickInterval
.
Key
,
"1"
)
s
.
Start
()
time
.
Sleep
(
time
.
Millisecond
*
4
)
assert
.
Greater
(
t
,
s
.
GetMinDdlTs
(),
Timestamp
(
100
))
for
i
:=
0
;
i
<
100
;
i
++
{
if
s
.
GetMinDdlTs
()
>
Timestamp
(
100
)
{
break
}
assert
.
True
(
t
,
i
<
100
)
time
.
Sleep
(
time
.
Millisecond
)
}
// add task to queue.
n
:=
10
...
...
internal/util/paramtable/component_param.go
浏览文件 @
9258f705
...
...
@@ -579,6 +579,7 @@ type rootCoordConfig struct {
MinSegmentSizeToEnableIndex
ParamItem
`refreshable:"true"`
ImportTaskExpiration
ParamItem
`refreshable:"true"`
ImportTaskRetention
ParamItem
`refreshable:"true"`
ImportMaxPendingTaskCount
ParamItem
`refreshable:"true"`
ImportTaskSubPath
ParamItem
`refreshable:"true"`
EnableActiveStandby
ParamItem
`refreshable:"false"`
}
...
...
@@ -626,6 +627,13 @@ func (p *rootCoordConfig) init(base *BaseTable) {
}
p
.
ImportTaskSubPath
.
Init
(
base
.
mgr
)
p
.
ImportMaxPendingTaskCount
=
ParamItem
{
Key
:
"rootCoord.importMaxPendingTaskCount"
,
Version
:
"2.2.2"
,
DefaultValue
:
strconv
.
Itoa
(
65535
),
}
p
.
ImportMaxPendingTaskCount
.
Init
(
base
.
mgr
)
p
.
EnableActiveStandby
=
ParamItem
{
Key
:
"rootCoord.enableActiveStandby"
,
Version
:
"2.2.0"
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录