Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
milvus
milvus
提交
d6f95f49
M
milvus
项目概览
milvus
/
milvus
10 个月 前同步成功
通知
260
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,体验更适合开发者的 AI 搜索 >>
未验证
提交
d6f95f49
编写于
10月 18, 2021
作者:
Z
zhenshan.cao
提交者:
GitHub
10月 18, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix golint error in querycoord (#10127)
Signed-off-by:
N
zhenshan.cao
<
zhenshan.cao@zilliz.com
>
上级
32a029fb
变更
7
展开全部
隐藏空白更改
内联
并排
Showing
7 changed file
with
269 addition
and
274 deletion
+269
-274
internal/querycoord/impl.go
internal/querycoord/impl.go
+8
-8
internal/querycoord/impl_test.go
internal/querycoord/impl_test.go
+2
-2
internal/querycoord/query_coord.go
internal/querycoord/query_coord.go
+4
-4
internal/querycoord/task.go
internal/querycoord/task.go
+185
-190
internal/querycoord/task_scheduler.go
internal/querycoord/task_scheduler.go
+18
-18
internal/querycoord/task_scheduler_test.go
internal/querycoord/task_scheduler_test.go
+27
-27
internal/querycoord/task_test.go
internal/querycoord/task_test.go
+25
-25
未找到文件。
internal/querycoord/impl.go
浏览文件 @
d6f95f49
...
...
@@ -144,8 +144,8 @@ func (qc *QueryCoord) LoadCollection(ctx context.Context, req *querypb.LoadColle
}
baseTask
:=
newBaseTask
(
qc
.
loopCtx
,
querypb
.
TriggerCondition_grpcRequest
)
loadCollectionTask
:=
&
L
oadCollectionTask
{
B
aseTask
:
baseTask
,
loadCollectionTask
:=
&
l
oadCollectionTask
{
b
aseTask
:
baseTask
,
LoadCollectionRequest
:
req
,
rootCoord
:
qc
.
rootCoordClient
,
dataCoord
:
qc
.
dataCoordClient
,
...
...
@@ -193,8 +193,8 @@ func (qc *QueryCoord) ReleaseCollection(ctx context.Context, req *querypb.Releas
}
baseTask
:=
newBaseTask
(
qc
.
loopCtx
,
querypb
.
TriggerCondition_grpcRequest
)
releaseCollectionTask
:=
&
R
eleaseCollectionTask
{
B
aseTask
:
baseTask
,
releaseCollectionTask
:=
&
r
eleaseCollectionTask
{
b
aseTask
:
baseTask
,
ReleaseCollectionRequest
:
req
,
cluster
:
qc
.
cluster
,
meta
:
qc
.
meta
,
...
...
@@ -336,8 +336,8 @@ func (qc *QueryCoord) LoadPartitions(ctx context.Context, req *querypb.LoadParti
}
baseTask
:=
newBaseTask
(
qc
.
loopCtx
,
querypb
.
TriggerCondition_grpcRequest
)
loadPartitionTask
:=
&
L
oadPartitionTask
{
B
aseTask
:
baseTask
,
loadPartitionTask
:=
&
l
oadPartitionTask
{
b
aseTask
:
baseTask
,
LoadPartitionsRequest
:
req
,
dataCoord
:
qc
.
dataCoordClient
,
cluster
:
qc
.
cluster
,
...
...
@@ -407,8 +407,8 @@ func (qc *QueryCoord) ReleasePartitions(ctx context.Context, req *querypb.Releas
req
.
PartitionIDs
=
toReleasedPartitions
baseTask
:=
newBaseTask
(
qc
.
loopCtx
,
querypb
.
TriggerCondition_grpcRequest
)
releasePartitionTask
:=
&
R
eleasePartitionTask
{
B
aseTask
:
baseTask
,
releasePartitionTask
:=
&
r
eleasePartitionTask
{
b
aseTask
:
baseTask
,
ReleasePartitionsRequest
:
req
,
cluster
:
qc
.
cluster
,
}
...
...
internal/querycoord/impl_test.go
浏览文件 @
d6f95f49
...
...
@@ -452,8 +452,8 @@ func TestLoadBalanceTask(t *testing.T) {
BalanceReason
:
querypb
.
TriggerCondition_nodeDown
,
}
loadBalanceTask
:=
&
L
oadBalanceTask
{
BaseTask
:
&
B
aseTask
{
loadBalanceTask
:=
&
l
oadBalanceTask
{
baseTask
:
&
b
aseTask
{
ctx
:
baseCtx
,
Condition
:
NewTaskCondition
(
baseCtx
),
triggerCondition
:
querypb
.
TriggerCondition_nodeDown
,
...
...
internal/querycoord/query_coord.go
浏览文件 @
d6f95f49
...
...
@@ -238,8 +238,8 @@ func (qc *QueryCoord) watchNodeLoop() {
}
baseTask
:=
newBaseTask
(
qc
.
loopCtx
,
querypb
.
TriggerCondition_nodeDown
)
loadBalanceTask
:=
&
L
oadBalanceTask
{
B
aseTask
:
baseTask
,
loadBalanceTask
:=
&
l
oadBalanceTask
{
b
aseTask
:
baseTask
,
LoadBalanceRequest
:
loadBalanceSegment
,
rootCoord
:
qc
.
rootCoordClient
,
dataCoord
:
qc
.
dataCoordClient
,
...
...
@@ -289,8 +289,8 @@ func (qc *QueryCoord) watchNodeLoop() {
}
baseTask
:=
newBaseTask
(
qc
.
loopCtx
,
querypb
.
TriggerCondition_nodeDown
)
loadBalanceTask
:=
&
L
oadBalanceTask
{
B
aseTask
:
baseTask
,
loadBalanceTask
:=
&
l
oadBalanceTask
{
b
aseTask
:
baseTask
,
LoadBalanceRequest
:
loadBalanceSegment
,
rootCoord
:
qc
.
rootCoordClient
,
dataCoord
:
qc
.
dataCoordClient
,
...
...
internal/querycoord/task.go
浏览文件 @
d6f95f49
此差异已折叠。
点击以展开。
internal/querycoord/task_scheduler.go
浏览文件 @
d6f95f49
...
...
@@ -272,8 +272,8 @@ func (scheduler *TaskScheduler) unmarshalTask(taskID UniqueID, t string) (task,
if
err
!=
nil
{
return
nil
,
err
}
loadCollectionTask
:=
&
L
oadCollectionTask
{
B
aseTask
:
baseTask
,
loadCollectionTask
:=
&
l
oadCollectionTask
{
b
aseTask
:
baseTask
,
LoadCollectionRequest
:
&
loadReq
,
rootCoord
:
scheduler
.
rootCoord
,
dataCoord
:
scheduler
.
dataCoord
,
...
...
@@ -287,8 +287,8 @@ func (scheduler *TaskScheduler) unmarshalTask(taskID UniqueID, t string) (task,
if
err
!=
nil
{
return
nil
,
err
}
loadPartitionTask
:=
&
L
oadPartitionTask
{
B
aseTask
:
baseTask
,
loadPartitionTask
:=
&
l
oadPartitionTask
{
b
aseTask
:
baseTask
,
LoadPartitionsRequest
:
&
loadReq
,
dataCoord
:
scheduler
.
dataCoord
,
cluster
:
scheduler
.
cluster
,
...
...
@@ -301,8 +301,8 @@ func (scheduler *TaskScheduler) unmarshalTask(taskID UniqueID, t string) (task,
if
err
!=
nil
{
return
nil
,
err
}
releaseCollectionTask
:=
&
R
eleaseCollectionTask
{
B
aseTask
:
baseTask
,
releaseCollectionTask
:=
&
r
eleaseCollectionTask
{
b
aseTask
:
baseTask
,
ReleaseCollectionRequest
:
&
loadReq
,
cluster
:
scheduler
.
cluster
,
meta
:
scheduler
.
meta
,
...
...
@@ -315,8 +315,8 @@ func (scheduler *TaskScheduler) unmarshalTask(taskID UniqueID, t string) (task,
if
err
!=
nil
{
return
nil
,
err
}
releasePartitionTask
:=
&
R
eleasePartitionTask
{
B
aseTask
:
baseTask
,
releasePartitionTask
:=
&
r
eleasePartitionTask
{
b
aseTask
:
baseTask
,
ReleasePartitionsRequest
:
&
loadReq
,
cluster
:
scheduler
.
cluster
,
}
...
...
@@ -328,8 +328,8 @@ func (scheduler *TaskScheduler) unmarshalTask(taskID UniqueID, t string) (task,
if
err
!=
nil
{
return
nil
,
err
}
loadSegmentTask
:=
&
L
oadSegmentTask
{
B
aseTask
:
baseTask
,
loadSegmentTask
:=
&
l
oadSegmentTask
{
b
aseTask
:
baseTask
,
LoadSegmentsRequest
:
&
loadReq
,
cluster
:
scheduler
.
cluster
,
meta
:
scheduler
.
meta
,
...
...
@@ -343,8 +343,8 @@ func (scheduler *TaskScheduler) unmarshalTask(taskID UniqueID, t string) (task,
if
err
!=
nil
{
return
nil
,
err
}
releaseSegmentTask
:=
&
R
eleaseSegmentTask
{
B
aseTask
:
baseTask
,
releaseSegmentTask
:=
&
r
eleaseSegmentTask
{
b
aseTask
:
baseTask
,
ReleaseSegmentsRequest
:
&
loadReq
,
cluster
:
scheduler
.
cluster
,
}
...
...
@@ -356,8 +356,8 @@ func (scheduler *TaskScheduler) unmarshalTask(taskID UniqueID, t string) (task,
if
err
!=
nil
{
return
nil
,
err
}
watchDmChannelTask
:=
&
W
atchDmChannelTask
{
B
aseTask
:
baseTask
,
watchDmChannelTask
:=
&
w
atchDmChannelTask
{
b
aseTask
:
baseTask
,
WatchDmChannelsRequest
:
&
loadReq
,
cluster
:
scheduler
.
cluster
,
meta
:
scheduler
.
meta
,
...
...
@@ -371,8 +371,8 @@ func (scheduler *TaskScheduler) unmarshalTask(taskID UniqueID, t string) (task,
if
err
!=
nil
{
return
nil
,
err
}
watchQueryChannelTask
:=
&
W
atchQueryChannelTask
{
B
aseTask
:
baseTask
,
watchQueryChannelTask
:=
&
w
atchQueryChannelTask
{
b
aseTask
:
baseTask
,
AddQueryChannelRequest
:
&
loadReq
,
cluster
:
scheduler
.
cluster
,
}
...
...
@@ -384,8 +384,8 @@ func (scheduler *TaskScheduler) unmarshalTask(taskID UniqueID, t string) (task,
if
err
!=
nil
{
return
nil
,
err
}
loadBalanceTask
:=
&
L
oadBalanceTask
{
B
aseTask
:
baseTask
,
loadBalanceTask
:=
&
l
oadBalanceTask
{
b
aseTask
:
baseTask
,
LoadBalanceRequest
:
&
loadReq
,
rootCoord
:
scheduler
.
rootCoord
,
dataCoord
:
scheduler
.
dataCoord
,
...
...
internal/querycoord/task_scheduler_test.go
浏览文件 @
d6f95f49
...
...
@@ -25,7 +25,7 @@ import (
)
type
testTask
struct
{
B
aseTask
b
aseTask
baseMsg
*
commonpb
.
MsgBase
cluster
Cluster
meta
Meta
...
...
@@ -59,8 +59,8 @@ func (tt *testTask) execute(ctx context.Context) error {
switch
tt
.
baseMsg
.
MsgType
{
case
commonpb
.
MsgType_LoadSegments
:
childTask
:=
&
L
oadSegmentTask
{
BaseTask
:
&
B
aseTask
{
childTask
:=
&
l
oadSegmentTask
{
baseTask
:
&
b
aseTask
{
ctx
:
tt
.
ctx
,
Condition
:
NewTaskCondition
(
tt
.
ctx
),
triggerCondition
:
tt
.
triggerCondition
,
...
...
@@ -77,8 +77,8 @@ func (tt *testTask) execute(ctx context.Context) error {
}
tt
.
addChildTask
(
childTask
)
case
commonpb
.
MsgType_WatchDmChannels
:
childTask
:=
&
W
atchDmChannelTask
{
BaseTask
:
&
B
aseTask
{
childTask
:=
&
w
atchDmChannelTask
{
baseTask
:
&
b
aseTask
{
ctx
:
tt
.
ctx
,
Condition
:
NewTaskCondition
(
tt
.
ctx
),
triggerCondition
:
tt
.
triggerCondition
,
...
...
@@ -95,8 +95,8 @@ func (tt *testTask) execute(ctx context.Context) error {
}
tt
.
addChildTask
(
childTask
)
case
commonpb
.
MsgType_WatchQueryChannels
:
childTask
:=
&
W
atchQueryChannelTask
{
BaseTask
:
&
B
aseTask
{
childTask
:=
&
w
atchQueryChannelTask
{
baseTask
:
&
b
aseTask
{
ctx
:
tt
.
ctx
,
Condition
:
NewTaskCondition
(
tt
.
ctx
),
triggerCondition
:
tt
.
triggerCondition
,
...
...
@@ -134,7 +134,7 @@ func TestWatchQueryChannel_ClearEtcdInfoAfterAssignedNodeDown(t *testing.T) {
nodeID
:=
queryNode
.
queryNodeID
waitQueryNodeOnline
(
queryCoord
.
cluster
,
nodeID
)
testTask
:=
&
testTask
{
BaseTask
:
B
aseTask
{
baseTask
:
b
aseTask
{
ctx
:
baseCtx
,
Condition
:
NewTaskCondition
(
baseCtx
),
triggerCondition
:
querypb
.
TriggerCondition_grpcRequest
,
...
...
@@ -173,8 +173,8 @@ func TestUnMarshalTask(t *testing.T) {
cancel
:
cancel
,
}
t
.
Run
(
"Test
L
oadCollectionTask"
,
func
(
t
*
testing
.
T
)
{
loadTask
:=
&
L
oadCollectionTask
{
t
.
Run
(
"Test
l
oadCollectionTask"
,
func
(
t
*
testing
.
T
)
{
loadTask
:=
&
l
oadCollectionTask
{
LoadCollectionRequest
:
&
querypb
.
LoadCollectionRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_LoadCollection
,
...
...
@@ -195,7 +195,7 @@ func TestUnMarshalTask(t *testing.T) {
})
t
.
Run
(
"Test LoadPartitionsTask"
,
func
(
t
*
testing
.
T
)
{
loadTask
:=
&
L
oadPartitionTask
{
loadTask
:=
&
l
oadPartitionTask
{
LoadPartitionsRequest
:
&
querypb
.
LoadPartitionsRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_LoadPartitions
,
...
...
@@ -215,8 +215,8 @@ func TestUnMarshalTask(t *testing.T) {
assert
.
Equal
(
t
,
task
.
msgType
(),
commonpb
.
MsgType_LoadPartitions
)
})
t
.
Run
(
"Test
R
eleaseCollectionTask"
,
func
(
t
*
testing
.
T
)
{
releaseTask
:=
&
R
eleaseCollectionTask
{
t
.
Run
(
"Test
r
eleaseCollectionTask"
,
func
(
t
*
testing
.
T
)
{
releaseTask
:=
&
r
eleaseCollectionTask
{
ReleaseCollectionRequest
:
&
querypb
.
ReleaseCollectionRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_ReleaseCollection
,
...
...
@@ -236,8 +236,8 @@ func TestUnMarshalTask(t *testing.T) {
assert
.
Equal
(
t
,
task
.
msgType
(),
commonpb
.
MsgType_ReleaseCollection
)
})
t
.
Run
(
"Test
R
eleasePartitionTask"
,
func
(
t
*
testing
.
T
)
{
releaseTask
:=
&
R
eleasePartitionTask
{
t
.
Run
(
"Test
r
eleasePartitionTask"
,
func
(
t
*
testing
.
T
)
{
releaseTask
:=
&
r
eleasePartitionTask
{
ReleasePartitionsRequest
:
&
querypb
.
ReleasePartitionsRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_ReleasePartitions
,
...
...
@@ -257,8 +257,8 @@ func TestUnMarshalTask(t *testing.T) {
assert
.
Equal
(
t
,
task
.
msgType
(),
commonpb
.
MsgType_ReleasePartitions
)
})
t
.
Run
(
"Test
L
oadSegmentTask"
,
func
(
t
*
testing
.
T
)
{
loadTask
:=
&
L
oadSegmentTask
{
t
.
Run
(
"Test
l
oadSegmentTask"
,
func
(
t
*
testing
.
T
)
{
loadTask
:=
&
l
oadSegmentTask
{
LoadSegmentsRequest
:
&
querypb
.
LoadSegmentsRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_LoadSegments
,
...
...
@@ -278,8 +278,8 @@ func TestUnMarshalTask(t *testing.T) {
assert
.
Equal
(
t
,
task
.
msgType
(),
commonpb
.
MsgType_LoadSegments
)
})
t
.
Run
(
"Test
R
eleaseSegmentTask"
,
func
(
t
*
testing
.
T
)
{
releaseTask
:=
&
R
eleaseSegmentTask
{
t
.
Run
(
"Test
r
eleaseSegmentTask"
,
func
(
t
*
testing
.
T
)
{
releaseTask
:=
&
r
eleaseSegmentTask
{
ReleaseSegmentsRequest
:
&
querypb
.
ReleaseSegmentsRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_ReleaseSegments
,
...
...
@@ -299,8 +299,8 @@ func TestUnMarshalTask(t *testing.T) {
assert
.
Equal
(
t
,
task
.
msgType
(),
commonpb
.
MsgType_ReleaseSegments
)
})
t
.
Run
(
"Test
W
atchDmChannelTask"
,
func
(
t
*
testing
.
T
)
{
watchTask
:=
&
W
atchDmChannelTask
{
t
.
Run
(
"Test
w
atchDmChannelTask"
,
func
(
t
*
testing
.
T
)
{
watchTask
:=
&
w
atchDmChannelTask
{
WatchDmChannelsRequest
:
&
querypb
.
WatchDmChannelsRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_WatchDmChannels
,
...
...
@@ -320,8 +320,8 @@ func TestUnMarshalTask(t *testing.T) {
assert
.
Equal
(
t
,
task
.
msgType
(),
commonpb
.
MsgType_WatchDmChannels
)
})
t
.
Run
(
"Test
W
atchQueryChannelTask"
,
func
(
t
*
testing
.
T
)
{
watchTask
:=
&
W
atchQueryChannelTask
{
t
.
Run
(
"Test
w
atchQueryChannelTask"
,
func
(
t
*
testing
.
T
)
{
watchTask
:=
&
w
atchQueryChannelTask
{
AddQueryChannelRequest
:
&
querypb
.
AddQueryChannelRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_WatchQueryChannels
,
...
...
@@ -341,8 +341,8 @@ func TestUnMarshalTask(t *testing.T) {
assert
.
Equal
(
t
,
task
.
msgType
(),
commonpb
.
MsgType_WatchQueryChannels
)
})
t
.
Run
(
"Test
L
oadBalanceTask"
,
func
(
t
*
testing
.
T
)
{
loadBalanceTask
:=
&
L
oadBalanceTask
{
t
.
Run
(
"Test
l
oadBalanceTask"
,
func
(
t
*
testing
.
T
)
{
loadBalanceTask
:=
&
l
oadBalanceTask
{
LoadBalanceRequest
:
&
querypb
.
LoadBalanceRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_LoadBalanceSegments
,
...
...
@@ -379,7 +379,7 @@ func TestReloadTaskFromKV(t *testing.T) {
}
kvs
:=
make
(
map
[
string
]
string
)
triggerTask
:=
&
L
oadCollectionTask
{
triggerTask
:=
&
l
oadCollectionTask
{
LoadCollectionRequest
:
&
querypb
.
LoadCollectionRequest
{
Base
:
&
commonpb
.
MsgBase
{
Timestamp
:
1
,
...
...
@@ -392,7 +392,7 @@ func TestReloadTaskFromKV(t *testing.T) {
triggerTaskKey
:=
fmt
.
Sprintf
(
"%s/%d"
,
triggerTaskPrefix
,
100
)
kvs
[
triggerTaskKey
]
=
string
(
triggerBlobs
)
activeTask
:=
&
L
oadSegmentTask
{
activeTask
:=
&
l
oadSegmentTask
{
LoadSegmentsRequest
:
&
querypb
.
LoadSegmentsRequest
{
Base
:
&
commonpb
.
MsgBase
{
Timestamp
:
2
,
...
...
internal/querycoord/task_test.go
浏览文件 @
d6f95f49
...
...
@@ -21,7 +21,7 @@ import (
"github.com/milvus-io/milvus/internal/proto/querypb"
)
func
genLoadCollectionTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
)
*
L
oadCollectionTask
{
func
genLoadCollectionTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
)
*
l
oadCollectionTask
{
req
:=
&
querypb
.
LoadCollectionRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_LoadCollection
,
...
...
@@ -30,8 +30,8 @@ func genLoadCollectionTask(ctx context.Context, queryCoord *QueryCoord) *LoadCol
Schema
:
genCollectionSchema
(
defaultCollectionID
,
false
),
}
baseTask
:=
newBaseTask
(
ctx
,
querypb
.
TriggerCondition_grpcRequest
)
loadCollectionTask
:=
&
L
oadCollectionTask
{
B
aseTask
:
baseTask
,
loadCollectionTask
:=
&
l
oadCollectionTask
{
b
aseTask
:
baseTask
,
LoadCollectionRequest
:
req
,
rootCoord
:
queryCoord
.
rootCoordClient
,
dataCoord
:
queryCoord
.
dataCoordClient
,
...
...
@@ -41,7 +41,7 @@ func genLoadCollectionTask(ctx context.Context, queryCoord *QueryCoord) *LoadCol
return
loadCollectionTask
}
func
genLoadPartitionTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
)
*
L
oadPartitionTask
{
func
genLoadPartitionTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
)
*
l
oadPartitionTask
{
req
:=
&
querypb
.
LoadPartitionsRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_LoadPartitions
,
...
...
@@ -50,8 +50,8 @@ func genLoadPartitionTask(ctx context.Context, queryCoord *QueryCoord) *LoadPart
PartitionIDs
:
[]
UniqueID
{
defaultPartitionID
},
}
baseTask
:=
newBaseTask
(
ctx
,
querypb
.
TriggerCondition_grpcRequest
)
loadPartitionTask
:=
&
L
oadPartitionTask
{
B
aseTask
:
baseTask
,
loadPartitionTask
:=
&
l
oadPartitionTask
{
b
aseTask
:
baseTask
,
LoadPartitionsRequest
:
req
,
dataCoord
:
queryCoord
.
dataCoordClient
,
cluster
:
queryCoord
.
cluster
,
...
...
@@ -60,7 +60,7 @@ func genLoadPartitionTask(ctx context.Context, queryCoord *QueryCoord) *LoadPart
return
loadPartitionTask
}
func
genReleaseCollectionTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
)
*
R
eleaseCollectionTask
{
func
genReleaseCollectionTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
)
*
r
eleaseCollectionTask
{
req
:=
&
querypb
.
ReleaseCollectionRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_ReleaseCollection
,
...
...
@@ -68,8 +68,8 @@ func genReleaseCollectionTask(ctx context.Context, queryCoord *QueryCoord) *Rele
CollectionID
:
defaultCollectionID
,
}
baseTask
:=
newBaseTask
(
ctx
,
querypb
.
TriggerCondition_grpcRequest
)
releaseCollectionTask
:=
&
R
eleaseCollectionTask
{
B
aseTask
:
baseTask
,
releaseCollectionTask
:=
&
r
eleaseCollectionTask
{
b
aseTask
:
baseTask
,
ReleaseCollectionRequest
:
req
,
rootCoord
:
queryCoord
.
rootCoordClient
,
cluster
:
queryCoord
.
cluster
,
...
...
@@ -79,7 +79,7 @@ func genReleaseCollectionTask(ctx context.Context, queryCoord *QueryCoord) *Rele
return
releaseCollectionTask
}
func
genReleasePartitionTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
)
*
R
eleasePartitionTask
{
func
genReleasePartitionTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
)
*
r
eleasePartitionTask
{
req
:=
&
querypb
.
ReleasePartitionsRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_ReleasePartitions
,
...
...
@@ -88,8 +88,8 @@ func genReleasePartitionTask(ctx context.Context, queryCoord *QueryCoord) *Relea
PartitionIDs
:
[]
UniqueID
{
defaultPartitionID
},
}
baseTask
:=
newBaseTask
(
ctx
,
querypb
.
TriggerCondition_grpcRequest
)
releasePartitionTask
:=
&
R
eleasePartitionTask
{
B
aseTask
:
baseTask
,
releasePartitionTask
:=
&
r
eleasePartitionTask
{
b
aseTask
:
baseTask
,
ReleasePartitionsRequest
:
req
,
cluster
:
queryCoord
.
cluster
,
}
...
...
@@ -97,7 +97,7 @@ func genReleasePartitionTask(ctx context.Context, queryCoord *QueryCoord) *Relea
return
releasePartitionTask
}
func
genReleaseSegmentTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
,
nodeID
int64
)
*
R
eleaseSegmentTask
{
func
genReleaseSegmentTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
,
nodeID
int64
)
*
r
eleaseSegmentTask
{
req
:=
&
querypb
.
ReleaseSegmentsRequest
{
Base
:
&
commonpb
.
MsgBase
{
MsgType
:
commonpb
.
MsgType_ReleaseSegments
,
...
...
@@ -108,15 +108,15 @@ func genReleaseSegmentTask(ctx context.Context, queryCoord *QueryCoord, nodeID i
SegmentIDs
:
[]
UniqueID
{
defaultSegmentID
},
}
baseTask
:=
newBaseTask
(
ctx
,
querypb
.
TriggerCondition_grpcRequest
)
releaseSegmentTask
:=
&
R
eleaseSegmentTask
{
B
aseTask
:
baseTask
,
releaseSegmentTask
:=
&
r
eleaseSegmentTask
{
b
aseTask
:
baseTask
,
ReleaseSegmentsRequest
:
req
,
cluster
:
queryCoord
.
cluster
,
}
return
releaseSegmentTask
}
func
genWatchDmChannelTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
,
nodeID
int64
)
*
W
atchDmChannelTask
{
func
genWatchDmChannelTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
,
nodeID
int64
)
*
w
atchDmChannelTask
{
schema
:=
genCollectionSchema
(
defaultCollectionID
,
false
)
vChannelInfo
:=
&
datapb
.
VchannelInfo
{
CollectionID
:
defaultCollectionID
,
...
...
@@ -134,8 +134,8 @@ func genWatchDmChannelTask(ctx context.Context, queryCoord *QueryCoord, nodeID i
}
baseTask
:=
newBaseTask
(
ctx
,
querypb
.
TriggerCondition_grpcRequest
)
baseTask
.
taskID
=
100
watchDmChannelTask
:=
&
W
atchDmChannelTask
{
B
aseTask
:
baseTask
,
watchDmChannelTask
:=
&
w
atchDmChannelTask
{
b
aseTask
:
baseTask
,
WatchDmChannelsRequest
:
req
,
cluster
:
queryCoord
.
cluster
,
meta
:
queryCoord
.
meta
,
...
...
@@ -151,8 +151,8 @@ func genWatchDmChannelTask(ctx context.Context, queryCoord *QueryCoord, nodeID i
}
baseParentTask
:=
newBaseTask
(
ctx
,
querypb
.
TriggerCondition_grpcRequest
)
baseParentTask
.
taskID
=
10
parentTask
:=
&
L
oadCollectionTask
{
B
aseTask
:
baseParentTask
,
parentTask
:=
&
l
oadCollectionTask
{
b
aseTask
:
baseParentTask
,
LoadCollectionRequest
:
parentReq
,
rootCoord
:
queryCoord
.
rootCoordClient
,
dataCoord
:
queryCoord
.
dataCoordClient
,
...
...
@@ -167,7 +167,7 @@ func genWatchDmChannelTask(ctx context.Context, queryCoord *QueryCoord, nodeID i
queryCoord
.
meta
.
addCollection
(
defaultCollectionID
,
schema
)
return
watchDmChannelTask
}
func
genLoadSegmentTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
,
nodeID
int64
)
*
L
oadSegmentTask
{
func
genLoadSegmentTask
(
ctx
context
.
Context
,
queryCoord
*
QueryCoord
,
nodeID
int64
)
*
l
oadSegmentTask
{
schema
:=
genCollectionSchema
(
defaultCollectionID
,
false
)
segmentInfo
:=
&
querypb
.
SegmentLoadInfo
{
SegmentID
:
defaultSegmentID
,
...
...
@@ -184,8 +184,8 @@ func genLoadSegmentTask(ctx context.Context, queryCoord *QueryCoord, nodeID int6
}
baseTask
:=
newBaseTask
(
ctx
,
querypb
.
TriggerCondition_grpcRequest
)
baseTask
.
taskID
=
100
loadSegmentTask
:=
&
L
oadSegmentTask
{
B
aseTask
:
baseTask
,
loadSegmentTask
:=
&
l
oadSegmentTask
{
b
aseTask
:
baseTask
,
LoadSegmentsRequest
:
req
,
cluster
:
queryCoord
.
cluster
,
meta
:
queryCoord
.
meta
,
...
...
@@ -201,8 +201,8 @@ func genLoadSegmentTask(ctx context.Context, queryCoord *QueryCoord, nodeID int6
}
baseParentTask
:=
newBaseTask
(
ctx
,
querypb
.
TriggerCondition_grpcRequest
)
baseParentTask
.
taskID
=
10
parentTask
:=
&
L
oadCollectionTask
{
B
aseTask
:
baseParentTask
,
parentTask
:=
&
l
oadCollectionTask
{
b
aseTask
:
baseParentTask
,
LoadCollectionRequest
:
parentReq
,
rootCoord
:
queryCoord
.
rootCoordClient
,
dataCoord
:
queryCoord
.
dataCoordClient
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录