Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
milvus
milvus
提交
b582cf7b
M
milvus
项目概览
milvus
/
milvus
12 个月 前同步成功
通知
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,发现更多精彩内容 >>
提交
b582cf7b
编写于
12月 21, 2020
作者:
C
cai.zhang
提交者:
yefu.chen
12月 21, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add the interface of index in proto
Signed-off-by:
N
cai.zhang
<
cai.zhang@zilliz.com
>
上级
d023c1af
变更
13
展开全部
显示空白变更内容
内联
并排
Showing
13 changed file
with
2963 addition
and
344 deletion
+2963
-344
internal/core/src/pb/service_msg.pb.cc
internal/core/src/pb/service_msg.pb.cc
+1099
-48
internal/core/src/pb/service_msg.pb.h
internal/core/src/pb/service_msg.pb.h
+788
-1
internal/master/grpc_service.go
internal/master/grpc_service.go
+12
-0
internal/proto/internal_msg.proto
internal/proto/internal_msg.proto
+33
-0
internal/proto/internalpb/internal_msg.pb.go
internal/proto/internalpb/internal_msg.pb.go
+403
-109
internal/proto/master.proto
internal/proto/master.proto
+4
-0
internal/proto/masterpb/master.pb.go
internal/proto/masterpb/master.pb.go
+143
-32
internal/proto/service.proto
internal/proto/service.proto
+27
-0
internal/proto/service_msg.proto
internal/proto/service_msg.proto
+31
-0
internal/proto/servicepb/service.pb.go
internal/proto/servicepb/service.pb.go
+174
-27
internal/proto/servicepb/service_msg.pb.go
internal/proto/servicepb/service_msg.pb.go
+204
-49
internal/proxy/grpc_service.go
internal/proxy/grpc_service.go
+12
-0
tests/python/test_search.py
tests/python/test_search.py
+33
-78
未找到文件。
internal/core/src/pb/service_msg.pb.cc
浏览文件 @
b582cf7b
此差异已折叠。
点击以展开。
internal/core/src/pb/service_msg.pb.h
浏览文件 @
b582cf7b
此差异已折叠。
点击以展开。
internal/master/grpc_service.go
浏览文件 @
b582cf7b
...
...
@@ -451,3 +451,15 @@ func (s *Master) AssignSegmentID(ctx context.Context, request *internalpb.Assign
PerChannelAssignment
:
segInfos
,
},
nil
}
func
(
s
*
Master
)
CreateIndex
(
context
.
Context
,
*
internalpb
.
CreateIndexRequest
)
(
*
commonpb
.
Status
,
error
)
{
return
nil
,
nil
}
func
(
s
*
Master
)
DescribeIndex
(
context
.
Context
,
*
internalpb
.
DescribeIndexRequest
)
(
*
servicepb
.
DescribeIndexResponse
,
error
)
{
return
nil
,
nil
}
func
(
s
*
Master
)
DescribeIndexProgress
(
context
.
Context
,
*
internalpb
.
DescribeIndexProgressRequest
)
(
*
servicepb
.
BoolResponse
,
error
)
{
return
nil
,
nil
}
internal/proto/internal_msg.proto
浏览文件 @
b582cf7b
...
...
@@ -314,3 +314,36 @@ message QueryNodeStats {
repeated
SegmentStats
seg_stats
=
3
;
repeated
FieldStats
field_stats
=
4
;
}
message
CreateIndexRequest
{
MsgType
msg_type
=
1
;
int64
reqID
=
2
;
uint64
timestamp
=
3
;
int64
proxyID
=
4
;
string
collection_name
=
5
;
string
field_name
=
6
;
string
index_name
=
7
;
repeated
common.KeyValuePair
extra_params
=
8
;
}
message
DescribeIndexRequest
{
MsgType
msg_type
=
1
;
int64
reqID
=
2
;
uint64
timestamp
=
3
;
int64
proxyID
=
4
;
string
collection_name
=
5
;
string
field_name
=
6
;
string
index_name
=
7
;
repeated
common.KeyValuePair
extra_params
=
8
;
}
message
DescribeIndexProgressRequest
{
MsgType
msg_type
=
1
;
int64
reqID
=
2
;
uint64
timestamp
=
3
;
int64
proxyID
=
4
;
string
collection_name
=
5
;
string
field_name
=
6
;
string
index_name
=
7
;
repeated
common.KeyValuePair
extra_params
=
8
;
}
internal/proto/internalpb/internal_msg.pb.go
浏览文件 @
b582cf7b
此差异已折叠。
点击以展开。
internal/proto/master.proto
浏览文件 @
b582cf7b
...
...
@@ -102,4 +102,8 @@ service Master {
rpc
AllocID
(
internal.IDRequest
)
returns
(
internal.IDResponse
)
{}
rpc
AssignSegmentID
(
internal.AssignSegIDRequest
)
returns
(
internal.AssignSegIDResponse
)
{}
rpc
CreateIndex
(
internal.CreateIndexRequest
)
returns
(
common.Status
)
{}
rpc
DescribeIndex
(
internal.DescribeIndexRequest
)
returns
(
service.DescribeIndexResponse
)
{}
rpc
DescribeIndexProgress
(
internal.DescribeIndexProgressRequest
)
returns
(
service.BoolResponse
)
{}
}
internal/proto/masterpb/master.pb.go
浏览文件 @
b582cf7b
...
...
@@ -30,38 +30,41 @@ const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package
func
init
()
{
proto
.
RegisterFile
(
"master.proto"
,
fileDescriptor_f9c348dec43a6705
)
}
var
fileDescriptor_f9c348dec43a6705
=
[]
byte
{
// 484 bytes of a gzipped FileDescriptorProto
0x1f
,
0x8b
,
0x08
,
0x00
,
0x00
,
0x00
,
0x00
,
0x00
,
0x02
,
0xff
,
0x9c
,
0x95
,
0xdf
,
0x6e
,
0xd3
,
0x30
,
0x14
,
0x87
,
0x7b
,
0x35
,
0x24
,
0xd3
,
0x3f
,
0xcc
,
0xdc
,
0x95
,
0x1b
,
0xd6
,
0x9b
,
0x41
,
0xcb
,
0x12
,
0x04
,
0x2f
,
0xc0
,
0xda
,
0x48
,
0xac
,
0x12
,
0x48
,
0xd3
,
0xb2
,
0x1b
,
0x40
,
0x68
,
0x24
,
0xd9
,
0x21
,
0x35
,
0x24
,
0x76
,
0xf0
,
0x39
,
0x19
,
0xa2
,
0x2f
,
0xc1
,
0x2b
,
0xa3
,
0x26
,
0x75
,
0x52
,
0xd3
,
0xba
,
0x94
,
0xdd
,
0xd5
,
0xf6
,
0xe7
,
0xdf
,
0x57
,
0x9f
,
0x73
,
0xa4
,
0xb0
,
0x6e
,
0x1e
,
0x21
,
0x81
,
0xf6
,
0x0a
,
0xad
,
0x48
,
0xf1
,
0xc7
,
0xb9
,
0xc8
,
0xee
,
0x4a
,
0xac
,
0x57
,
0x5e
,
0x7d
,
0x34
,
0xec
,
0x26
,
0x2a
,
0xcf
,
0x95
,
0xac
,
0x37
,
0x87
,
0x5c
,
0x48
,
0x02
,
0x2d
,
0xa3
,
0xec
,
0x26
,
0xc7
,
0x74
,
0xbd
,
0x77
,
0x8c
,
0xa0
,
0xef
,
0x44
,
0x02
,
0xed
,
0xd6
,
0xab
,
0xdf
,
0x0f
,
0xd9
,
0xd1
,
0xfb
,
0xea
,
0x3e
,
0x8f
,
0xd8
,
0xa3
,
0x99
,
0x86
,
0x88
,
0x60
,
0xa6
,
0xb2
,
0x0c
,
0x12
,
0x12
,
0x4a
,
0x72
,
0xcf
,
0xb3
,
0x4c
,
0x26
,
0xd3
,
0xfb
,
0x1b
,
0xbc
,
0x82
,
0x1f
,
0x25
,
0x20
,
0x0d
,
0x9f
,
0xd8
,
0xfc
,
0xfa
,
0x1f
,
0x85
,
0x14
,
0x51
,
0x89
,
0xa3
,
0x0e
,
0xff
,
0xcc
,
0xfa
,
0x81
,
0x56
,
0xc5
,
0x86
,
0xe0
,
0x85
,
0x43
,
0x60
,
0x63
,
0x07
,
0xc6
,
0xc7
,
0xac
,
0x77
,
0x11
,
0xe1
,
0x46
,
0xfa
,
0xc4
,
0x91
,
0x6e
,
0x51
,
0x26
,
0x7c
,
0x64
,
0xc3
,
0xeb
,
0x5a
,
0x79
,
0x53
,
0xa5
,
0xb2
,
0x2b
,
0xc0
,
0x42
,
0x49
,
0x84
,
0x51
,
0x87
,
0x97
,
0x8c
,
0x07
,
0x80
,
0x89
,
0x16
,
0xf1
,
0x66
,
0x9d
,
0x5e
,
0xba
,
0x9e
,
0xb1
,
0x85
,
0x1a
,
0xdb
,
0x64
,
0xb7
,
0xad
,
0x05
,
0xeb
,
0xab
,
0xc5
,
0xea
,
0xe7
,
0xa8
,
0xc3
,
0xbf
,
0xb3
,
0x41
,
0xb8
,
0x50
,
0x3f
,
0xdb
,
0x63
,
0x74
,
0x96
,
0xce
,
0xe6
,
0x8c
,
0xef
,
0xd9
,
0x6e
,
0x5f
,
0x48
,
0x5a
,
0xc8
,
0xf4
,
0x9d
,
0x40
,
0xda
,
0x78
,
0xe3
,
0x0d
,
0x1b
,
0xd4
,
0x0d
,
0xbe
,
0x8c
,
0x34
,
0x89
,
0xea
,
0x81
,
0x67
,
0x7b
,
0x07
,
0xa1
,
0xe1
,
0x0e
,
0x6c
,
0xd4
,
0x27
,
0xd6
,
0x5b
,
0x35
,
0xb8
,
0x8d
,
0x9f
,
0xec
,
0x19
,
0x83
,
0xff
,
0x0d
,
0xff
,
0xc2
,
0xba
,
0x17
,
0x11
,
0xb6
,
0xd9
,
0x63
,
0xf7
,
0x10
,
0x6c
,
0x45
,
0x1f
,
0x36
,
0x03
,
0x9a
,
0x1d
,
0x9b
,
0xc6
,
0xb6
,
0x1a
,
0xff
,
0x1f
,
0x23
,
0xb0
,
0xe5
,
0x1a
,
0xef
,
0x76
,
0x35
,
0x9c
,
0x3d
,
0x00
,
0x82
,
0xf5
,
0x57
,
0x8d
,
0x6d
,
0x4e
,
0xd1
,
0x59
,
0x33
,
0x0b
,
0xbb
,
0x4f
,
0xfb
,
0x13
,
0xd6
,
0x7b
,
0x0b
,
0x14
,
0xfe
,
0xc2
,
0x99
,
0x92
,
0x5f
,
0x45
,
0x8a
,
0xfc
,
0xd4
,
0x65
,
0x32
,
0x88
,
0xb1
,
0x9c
,
0x3a
,
0x2c
,
0x2d
,
0xd7
,
0x48
,
0x3e
,
0xb0
,
0xfe
,
0x79
,
0x96
,
0xa9
,
0xe4
,
0x5a
,
0xe4
,
0x80
,
0x14
,
0xe5
,
0x05
,
0x3f
,
0x71
,
0x58
,
0xae
,
0x51
,
0x39
,
0xda
,
0x63
,
0x23
,
0x4d
,
0xf4
,
0x25
,
0x7b
,
0x50
,
0x45
,
0xcf
,
0x03
,
0xfe
,
0xd4
,
0x71
,
0x61
,
0x1e
,
0x98
,
0xc8
,
0x93
,
0x3d
,
0x44
,
0x93
,
0xf8
,
0x8d
,
0x0d
,
0xce
,
0x11
,
0x45
,
0x2a
,
0x43
,
0x48
,
0x73
,
0x90
,
0x34
,
0x0f
,
0xf8
,
0x73
,
0xc7
,
0xbd
,
0x86
,
0x6b
,
0x15
,
0xe3
,
0x43
,
0x50
,
0xe3
,
0x9a
,
0x4e
,
0x3f
,
0xbe
,
0x49
,
0x05
,
0x2d
,
0xca
,
0x78
,
0x35
,
0xd8
,
0xfe
,
0x52
,
0x64
,
0x99
,
0x58
,
0x12
,
0x24
,
0x0b
,
0xbf
,
0x0e
,
0x39
,
0xbb
,
0x15
,
0x48
,
0x5a
,
0xc4
,
0x25
,
0xc1
,
0xad
,
0x6f
,
0xa2
,
0xfc
,
0x2a
,
0xd9
,
0xaf
,
0x3f
,
0x04
,
0x45
,
0x1c
,
0x1f
,
0x55
,
0xeb
,
0xd7
,
0x7f
,
0x02
,
0x00
,
0x00
,
0xff
,
0xff
,
0x8d
,
0x75
,
0x7d
,
0xec
,
0x36
,
0x06
,
0x00
,
0x00
,
// 538 bytes of a gzipped FileDescriptorProto
0x1f
,
0x8b
,
0x08
,
0x00
,
0x00
,
0x00
,
0x00
,
0x00
,
0x02
,
0xff
,
0x9c
,
0x95
,
0x5f
,
0x6f
,
0xd3
,
0x3c
,
0x14
,
0x87
,
0x7b
,
0xb5
,
0x57
,
0xf2
,
0xdb
,
0x3f
,
0xcc
,
0x88
,
0x9b
,
0x72
,
0xc3
,
0x7a
,
0x33
,
0x68
,
0x59
,
0x82
,
0xd8
,
0x17
,
0x60
,
0x6d
,
0x24
,
0x56
,
0x09
,
0xa4
,
0x6a
,
0x99
,
0x84
,
0x00
,
0xa1
,
0x91
,
0xa4
,
0x87
,
0xd4
,
0x2c
,
0xb1
,
0x83
,
0x8f
,
0x33
,
0x60
,
0xdf
,
0x90
,
0x6f
,
0x85
,
0x9a
,
0xd4
,
0x49
,
0xbc
,
0xd6
,
0x5d
,
0xc6
,
0x5d
,
0x13
,
0x3f
,
0xe7
,
0xf7
,
0xd4
,
0x3e
,
0xc7
,
0x2d
,
0xe9
,
0xa6
,
0x01
,
0x2a
,
0x90
,
0x4e
,
0x26
,
0x85
,
0x12
,
0xf4
,
0x71
,
0xca
,
0x92
,
0x9b
,
0x1c
,
0xcb
,
0x27
,
0xa7
,
0x5c
,
0x1a
,
0x76
,
0x23
,
0x91
,
0xa6
,
0x82
,
0x97
,
0x2f
,
0x87
,
0x94
,
0x71
,
0x05
,
0x92
,
0x07
,
0xc9
,
0x55
,
0x8a
,
0xf1
,
0xe6
,
0xdd
,
0x21
,
0x82
,
0xbc
,
0x61
,
0x11
,
0xd4
,
0xaf
,
0x5e
,
0xff
,
0xe9
,
0x91
,
0x83
,
0xf7
,
0x45
,
0x3d
,
0x0d
,
0xc8
,
0xa3
,
0x99
,
0x84
,
0x40
,
0xc1
,
0x4c
,
0x24
,
0x09
,
0x44
,
0x8a
,
0x09
,
0x4e
,
0x1d
,
0xc7
,
0x30
,
0xe9
,
0x4c
,
0xe7
,
0x2e
,
0x78
,
0x01
,
0x3f
,
0x72
,
0x40
,
0x35
,
0x7c
,
0x6a
,
0xf2
,
0x9b
,
0x6f
,
0xe4
,
0xab
,
0x40
,
0xe5
,
0x38
,
0xea
,
0xd0
,
0x2f
,
0xa4
,
0xef
,
0x49
,
0x91
,
0x35
,
0x04
,
0x2f
,
0x2d
,
0x02
,
0x13
,
0x6b
,
0x19
,
0x1f
,
0x92
,
0xde
,
0x79
,
0x80
,
0x8d
,
0xf4
,
0x89
,
0x25
,
0xdd
,
0xa0
,
0x74
,
0xf8
,
0xc8
,
0x84
,
0x37
,
0x67
,
0xe5
,
0x4c
,
0x85
,
0x48
,
0x2e
,
0x00
,
0x33
,
0xc1
,
0x11
,
0x46
,
0x1d
,
0x9a
,
0x13
,
0xea
,
0x01
,
0x46
,
0x92
,
0x85
,
0xcd
,
0x73
,
0x7a
,
0x65
,
0xdb
,
0xc6
,
0x16
,
0xaa
,
0x6d
,
0x93
,
0xdd
,
0xb6
,
0x1a
,
0x2c
,
0x4b
,
0xb3
,
0xf5
,
0xc7
,
0x51
,
0x87
,
0x5e
,
0x93
,
0x81
,
0xbf
,
0x12
,
0x3f
,
0xeb
,
0x65
,
0xb4
,
0x1e
,
0x9d
,
0xc9
,
0x69
,
0xdf
,
0xf3
,
0xdd
,
0x3e
,
0x5f
,
0x49
,
0xc6
,
0xe3
,
0x77
,
0x0c
,
0x55
,
0x63
,
0x8f
,
0x57
,
0x64
,
0x50
,
0x36
,
0x78
,
0x11
,
0x48
,
0xc5
,
0x8a
,
0x0d
,
0x9e
,
0xec
,
0x1d
,
0x84
,
0x8a
,
0x6b
,
0xd9
,
0xa8
,
0xcf
,
0xa4
,
0xb7
,
0x6e
,
0x70
,
0x1d
,
0x3f
,
0xd9
,
0x33
,
0x06
,
0x0f
,
0x0d
,
0xff
,
0x4a
,
0xba
,
0xe7
,
0x01
,
0xd6
,
0xd9
,
0x63
,
0xfb
,
0x10
,
0x6c
,
0x45
,
0xb7
,
0x9b
,
0x01
,
0x49
,
0x0e
,
0x75
,
0x63
,
0x6b
,
0x8d
,
0x7b
,
0xcf
,
0x08
,
0x6c
,
0xb9
,
0xc6
,
0xbb
,
0x5d
,
0x15
,
0x67
,
0x0e
,
0x00
,
0x23
,
0xfd
,
0x75
,
0x63
,
0xab
,
0x55
,
0xb4
,
0x9e
,
0x99
,
0x81
,
0xfd
,
0x4b
,
0xfb
,
0x23
,
0xd2
,
0x7b
,
0x0b
,
0xca
,
0xff
,
0x8d
,
0x33
,
0xc1
,
0xbf
,
0xb1
,
0x18
,
0xe9
,
0xb1
,
0xcd
,
0xa4
,
0x11
,
0x6d
,
0x39
,
0xb6
,
0x58
,
0x6a
,
0xae
,
0x92
,
0x7c
,
0x24
,
0xfd
,
0xb3
,
0x24
,
0x11
,
0xd1
,
0x25
,
0x4b
,
0x01
,
0x55
,
0x90
,
0x66
,
0xf4
,
0xc8
,
0x62
,
0xb9
,
0x44
,
0x61
,
0x69
,
0x8f
,
0x89
,
0x54
,
0xd1
,
0x0b
,
0xf2
,
0x5f
,
0x11
,
0x3d
,
0xf7
,
0xe8
,
0x33
,
0x4b
,
0xc1
,
0xdc
,
0xd3
,
0x91
,
0x47
,
0x7b
,
0x88
,
0x2a
,
0xf1
,
0x3b
,
0x19
,
0x9c
,
0x21
,
0xb2
,
0x98
,
0xfb
,
0x10
,
0xa7
,
0xc0
,
0xd5
,
0xdc
,
0xa3
,
0x2f
,
0x2c
,
0x75
,
0x15
,
0x57
,
0x2b
,
0xc6
,
0x6d
,
0xd0
,
0xca
,
0xf5
,
0x81
,
0xfc
,
0x5f
,
0x5e
,
0xaa
,
0x39
,
0x5f
,
0xc2
,
0x2f
,
0xab
,
0xa7
,
0xc1
,
0xb4
,
0xbc
,
0x17
,
0xd7
,
0xa4
,
0xa7
,
0x67
,
0xb1
,
0x8c
,
0x9e
,
0xdc
,
0x33
,
0xb1
,
0x46
,
0xb8
,
0xe5
,
0xf7
,
0xea
,
0x0e
,
0xdb
,
0xb8
,
0x22
,
0x4f
,
0x8c
,
0xa5
,
0x85
,
0x14
,
0xb1
,
0x04
,
0x44
,
0x7a
,
0xda
,
0x46
,
0xaa
,
0xe9
,
0x07
,
0x5d
,
0xcb
,
0xe9
,
0xf4
,
0xd3
,
0x9b
,
0x98
,
0xa9
,
0x55
,
0x1e
,
0xae
,
0xb7
,
0xee
,
0xde
,
0xb2
,
0x24
,
0x61
,
0xb7
,
0x0a
,
0xa2
,
0x95
,
0x5b
,
0x16
,
0x9f
,
0x2c
,
0x19
,
0x2a
,
0xc9
,
0xc2
,
0x5c
,
0xc1
,
0xd2
,
0xd5
,
0x5e
,
0xb7
,
0x48
,
0x74
,
0xcb
,
0xbf
,
0xd0
,
0x2c
,
0x0c
,
0x0f
,
0x8a
,
0xe7
,
0xd3
,
0xbf
,
0x01
,
0x00
,
0x00
,
0xff
,
0xff
,
0x42
,
0xb7
,
0x2a
,
0xc4
,
0x70
,
0x07
,
0x00
,
0x00
,
}
// Reference imports to suppress errors if they are not otherwise used.
...
...
@@ -146,6 +149,9 @@ type MasterClient interface {
AllocTimestamp
(
ctx
context
.
Context
,
in
*
internalpb
.
TsoRequest
,
opts
...
grpc
.
CallOption
)
(
*
internalpb
.
TsoResponse
,
error
)
AllocID
(
ctx
context
.
Context
,
in
*
internalpb
.
IDRequest
,
opts
...
grpc
.
CallOption
)
(
*
internalpb
.
IDResponse
,
error
)
AssignSegmentID
(
ctx
context
.
Context
,
in
*
internalpb
.
AssignSegIDRequest
,
opts
...
grpc
.
CallOption
)
(
*
internalpb
.
AssignSegIDResponse
,
error
)
CreateIndex
(
ctx
context
.
Context
,
in
*
internalpb
.
CreateIndexRequest
,
opts
...
grpc
.
CallOption
)
(
*
commonpb
.
Status
,
error
)
DescribeIndex
(
ctx
context
.
Context
,
in
*
internalpb
.
DescribeIndexRequest
,
opts
...
grpc
.
CallOption
)
(
*
servicepb
.
DescribeIndexResponse
,
error
)
DescribeIndexProgress
(
ctx
context
.
Context
,
in
*
internalpb
.
DescribeIndexProgressRequest
,
opts
...
grpc
.
CallOption
)
(
*
servicepb
.
BoolResponse
,
error
)
}
type
masterClient
struct
{
...
...
@@ -282,6 +288,33 @@ func (c *masterClient) AssignSegmentID(ctx context.Context, in *internalpb.Assig
return
out
,
nil
}
func
(
c
*
masterClient
)
CreateIndex
(
ctx
context
.
Context
,
in
*
internalpb
.
CreateIndexRequest
,
opts
...
grpc
.
CallOption
)
(
*
commonpb
.
Status
,
error
)
{
out
:=
new
(
commonpb
.
Status
)
err
:=
c
.
cc
.
Invoke
(
ctx
,
"/milvus.proto.master.Master/CreateIndex"
,
in
,
out
,
opts
...
)
if
err
!=
nil
{
return
nil
,
err
}
return
out
,
nil
}
func
(
c
*
masterClient
)
DescribeIndex
(
ctx
context
.
Context
,
in
*
internalpb
.
DescribeIndexRequest
,
opts
...
grpc
.
CallOption
)
(
*
servicepb
.
DescribeIndexResponse
,
error
)
{
out
:=
new
(
servicepb
.
DescribeIndexResponse
)
err
:=
c
.
cc
.
Invoke
(
ctx
,
"/milvus.proto.master.Master/DescribeIndex"
,
in
,
out
,
opts
...
)
if
err
!=
nil
{
return
nil
,
err
}
return
out
,
nil
}
func
(
c
*
masterClient
)
DescribeIndexProgress
(
ctx
context
.
Context
,
in
*
internalpb
.
DescribeIndexProgressRequest
,
opts
...
grpc
.
CallOption
)
(
*
servicepb
.
BoolResponse
,
error
)
{
out
:=
new
(
servicepb
.
BoolResponse
)
err
:=
c
.
cc
.
Invoke
(
ctx
,
"/milvus.proto.master.Master/DescribeIndexProgress"
,
in
,
out
,
opts
...
)
if
err
!=
nil
{
return
nil
,
err
}
return
out
,
nil
}
// MasterServer is the server API for Master service.
type
MasterServer
interface
{
//*
...
...
@@ -354,6 +387,9 @@ type MasterServer interface {
AllocTimestamp
(
context
.
Context
,
*
internalpb
.
TsoRequest
)
(
*
internalpb
.
TsoResponse
,
error
)
AllocID
(
context
.
Context
,
*
internalpb
.
IDRequest
)
(
*
internalpb
.
IDResponse
,
error
)
AssignSegmentID
(
context
.
Context
,
*
internalpb
.
AssignSegIDRequest
)
(
*
internalpb
.
AssignSegIDResponse
,
error
)
CreateIndex
(
context
.
Context
,
*
internalpb
.
CreateIndexRequest
)
(
*
commonpb
.
Status
,
error
)
DescribeIndex
(
context
.
Context
,
*
internalpb
.
DescribeIndexRequest
)
(
*
servicepb
.
DescribeIndexResponse
,
error
)
DescribeIndexProgress
(
context
.
Context
,
*
internalpb
.
DescribeIndexProgressRequest
)
(
*
servicepb
.
BoolResponse
,
error
)
}
// UnimplementedMasterServer can be embedded to have forward compatible implementations.
...
...
@@ -402,6 +438,15 @@ func (*UnimplementedMasterServer) AllocID(ctx context.Context, req *internalpb.I
func
(
*
UnimplementedMasterServer
)
AssignSegmentID
(
ctx
context
.
Context
,
req
*
internalpb
.
AssignSegIDRequest
)
(
*
internalpb
.
AssignSegIDResponse
,
error
)
{
return
nil
,
status
.
Errorf
(
codes
.
Unimplemented
,
"method AssignSegmentID not implemented"
)
}
func
(
*
UnimplementedMasterServer
)
CreateIndex
(
ctx
context
.
Context
,
req
*
internalpb
.
CreateIndexRequest
)
(
*
commonpb
.
Status
,
error
)
{
return
nil
,
status
.
Errorf
(
codes
.
Unimplemented
,
"method CreateIndex not implemented"
)
}
func
(
*
UnimplementedMasterServer
)
DescribeIndex
(
ctx
context
.
Context
,
req
*
internalpb
.
DescribeIndexRequest
)
(
*
servicepb
.
DescribeIndexResponse
,
error
)
{
return
nil
,
status
.
Errorf
(
codes
.
Unimplemented
,
"method DescribeIndex not implemented"
)
}
func
(
*
UnimplementedMasterServer
)
DescribeIndexProgress
(
ctx
context
.
Context
,
req
*
internalpb
.
DescribeIndexProgressRequest
)
(
*
servicepb
.
BoolResponse
,
error
)
{
return
nil
,
status
.
Errorf
(
codes
.
Unimplemented
,
"method DescribeIndexProgress not implemented"
)
}
func
RegisterMasterServer
(
s
*
grpc
.
Server
,
srv
MasterServer
)
{
s
.
RegisterService
(
&
_Master_serviceDesc
,
srv
)
...
...
@@ -659,6 +704,60 @@ func _Master_AssignSegmentID_Handler(srv interface{}, ctx context.Context, dec f
return
interceptor
(
ctx
,
in
,
info
,
handler
)
}
func
_Master_CreateIndex_Handler
(
srv
interface
{},
ctx
context
.
Context
,
dec
func
(
interface
{})
error
,
interceptor
grpc
.
UnaryServerInterceptor
)
(
interface
{},
error
)
{
in
:=
new
(
internalpb
.
CreateIndexRequest
)
if
err
:=
dec
(
in
);
err
!=
nil
{
return
nil
,
err
}
if
interceptor
==
nil
{
return
srv
.
(
MasterServer
)
.
CreateIndex
(
ctx
,
in
)
}
info
:=
&
grpc
.
UnaryServerInfo
{
Server
:
srv
,
FullMethod
:
"/milvus.proto.master.Master/CreateIndex"
,
}
handler
:=
func
(
ctx
context
.
Context
,
req
interface
{})
(
interface
{},
error
)
{
return
srv
.
(
MasterServer
)
.
CreateIndex
(
ctx
,
req
.
(
*
internalpb
.
CreateIndexRequest
))
}
return
interceptor
(
ctx
,
in
,
info
,
handler
)
}
func
_Master_DescribeIndex_Handler
(
srv
interface
{},
ctx
context
.
Context
,
dec
func
(
interface
{})
error
,
interceptor
grpc
.
UnaryServerInterceptor
)
(
interface
{},
error
)
{
in
:=
new
(
internalpb
.
DescribeIndexRequest
)
if
err
:=
dec
(
in
);
err
!=
nil
{
return
nil
,
err
}
if
interceptor
==
nil
{
return
srv
.
(
MasterServer
)
.
DescribeIndex
(
ctx
,
in
)
}
info
:=
&
grpc
.
UnaryServerInfo
{
Server
:
srv
,
FullMethod
:
"/milvus.proto.master.Master/DescribeIndex"
,
}
handler
:=
func
(
ctx
context
.
Context
,
req
interface
{})
(
interface
{},
error
)
{
return
srv
.
(
MasterServer
)
.
DescribeIndex
(
ctx
,
req
.
(
*
internalpb
.
DescribeIndexRequest
))
}
return
interceptor
(
ctx
,
in
,
info
,
handler
)
}
func
_Master_DescribeIndexProgress_Handler
(
srv
interface
{},
ctx
context
.
Context
,
dec
func
(
interface
{})
error
,
interceptor
grpc
.
UnaryServerInterceptor
)
(
interface
{},
error
)
{
in
:=
new
(
internalpb
.
DescribeIndexProgressRequest
)
if
err
:=
dec
(
in
);
err
!=
nil
{
return
nil
,
err
}
if
interceptor
==
nil
{
return
srv
.
(
MasterServer
)
.
DescribeIndexProgress
(
ctx
,
in
)
}
info
:=
&
grpc
.
UnaryServerInfo
{
Server
:
srv
,
FullMethod
:
"/milvus.proto.master.Master/DescribeIndexProgress"
,
}
handler
:=
func
(
ctx
context
.
Context
,
req
interface
{})
(
interface
{},
error
)
{
return
srv
.
(
MasterServer
)
.
DescribeIndexProgress
(
ctx
,
req
.
(
*
internalpb
.
DescribeIndexProgressRequest
))
}
return
interceptor
(
ctx
,
in
,
info
,
handler
)
}
var
_Master_serviceDesc
=
grpc
.
ServiceDesc
{
ServiceName
:
"milvus.proto.master.Master"
,
HandlerType
:
(
*
MasterServer
)(
nil
),
...
...
@@ -719,6 +818,18 @@ var _Master_serviceDesc = grpc.ServiceDesc{
MethodName
:
"AssignSegmentID"
,
Handler
:
_Master_AssignSegmentID_Handler
,
},
{
MethodName
:
"CreateIndex"
,
Handler
:
_Master_CreateIndex_Handler
,
},
{
MethodName
:
"DescribeIndex"
,
Handler
:
_Master_DescribeIndex_Handler
,
},
{
MethodName
:
"DescribeIndexProgress"
,
Handler
:
_Master_DescribeIndexProgress_Handler
,
},
},
Streams
:
[]
grpc
.
StreamDesc
{},
Metadata
:
"master.proto"
,
...
...
internal/proto/service.proto
浏览文件 @
b582cf7b
...
...
@@ -105,4 +105,31 @@ service MilvusService {
* @return QueryResult
*/
rpc
Search
(
Query
)
returns
(
QueryResult
)
{}
/**
* @brief This method is used to build index by collection in sync mode.
*
* @param IndexParam, index paramters.
*
* @return Status
*/
rpc
CreateIndex
(
IndexParam
)
returns
(
common.Status
)
{}
/**
* @brief This method is used to describe index
*
* @param IndexParam, target index.
*
* @return IndexParam
*/
rpc
DescribeIndex
(
IndexParam
)
returns
(
DescribeIndexResponse
)
{}
/**
* @brief This method is used to query index building progress
*
* @param IndexParam, target index.
*
* @return IndexParam
*/
rpc
DescribeIndexProgress
(
IndexParam
)
returns
(
BoolResponse
)
{}
}
\ No newline at end of file
internal/proto/service_msg.proto
浏览文件 @
b582cf7b
...
...
@@ -162,3 +162,34 @@ message QueryResult {
repeated
bytes
hits
=
2
;
}
/**
* @brief Index params
* @collection_name: target collection
* @field_name: target field
* @index_name: a name for index provided by user, unique within this field
* @extra_params: index parameters in json format
* for vector field:
* extra_params["index_type"] = one of the values: FLAT, IVF_LAT, IVF_SQ8, NSGMIX, IVFSQ8H,
* PQ, HNSW, HNSW_SQ8NM, ANNOY
* extra_params["metric_type"] = one of the values: L2, IP, HAMMING, JACCARD, TANIMOTO
* SUBSTRUCTURE, SUPERSTRUCTURE
* extra_params["params"] = extra parameters for index, for example ivflat: {nlist: 2048}
* for structured field:
* extra_params["index_type"] = one of the values: SORTED
*/
message
IndexParam
{
string
collection_name
=
2
;
string
field_name
=
3
;
string
index_name
=
4
;
repeated
common.KeyValuePair
extra_params
=
5
;
}
message
DescribeIndexResponse
{
common.Status
status
=
1
;
string
collection_name
=
2
;
string
field_name
=
3
;
string
index_name
=
4
;
repeated
common.KeyValuePair
extra_params
=
5
;
}
internal/proto/servicepb/service.pb.go
浏览文件 @
b582cf7b
...
...
@@ -29,33 +29,36 @@ const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package
func
init
()
{
proto
.
RegisterFile
(
"service.proto"
,
fileDescriptor_a0b84a42fa06f626
)
}
var
fileDescriptor_a0b84a42fa06f626
=
[]
byte
{
// 412 bytes of a gzipped FileDescriptorProto
0x1f
,
0x8b
,
0x08
,
0x00
,
0x00
,
0x00
,
0x00
,
0x00
,
0x02
,
0xff
,
0xac
,
0x92
,
0x51
,
0x6b
,
0xd4
,
0x40
,
0x10
,
0xc7
,
0xf3
,
0x74
,
0xc8
,
0x72
,
0x69
,
0xed
,
0xe2
,
0x53
,
0x0a
,
0x82
,
0x51
,
0x41
,
0x14
,
0x13
,
0xd0
,
0x4f
,
0xe0
,
0x5d
,
0x85
,
0x56
,
0x54
,
0x34
,
0xa1
,
0x62
,
0xf5
,
0x41
,
0x36
,
0xdb
,
0x31
,
0x59
,
0xd8
,
0xec
,
0x86
,
0x9d
,
0x49
,
0x4b
,
0xfb
,
0x51
,
0xfd
,
0x34
,
0x92
,
0x6c
,
0xec
,
0xe5
,
0x20
,
0x7a
,
0x17
,
0xf0
,
0x2d
,
0xfb
,
0x9f
,
0xff
,
0xfc
,
0x76
,
0x26
,
0xff
,
0x65
,
0x21
,
0x82
,
0xbb
,
0x52
,
0x12
,
0x92
,
0xc6
,
0x59
,
0xb2
,
0xfc
,
0x41
,
0xad
,
0xf4
,
0x55
,
0x8b
,
0xfe
,
0x94
,
0x0c
,
0xb5
,
0x68
,
0x29
,
0x6d
,
0x5d
,
0x5b
,
0xe3
,
0xd5
,
0xe8
,
0x68
,
0x90
,
0x7f
,
0xd4
,
0x58
,
0x0e
,
0xd2
,
0x12
,
0x65
,
0x05
,
0xb5
,
0xf0
,
0xa7
,
0x57
,
0xbf
,
0xee
,
0xb1
,
0xf0
,
0x43
,
0xcf
,
0xc9
,
0xbd
,
0x93
,
0x7f
,
0x65
,
0xf7
,
0xd7
,
0x0e
,
0x04
,
0xc1
,
0xda
,
0x6a
,
0x0d
,
0x92
,
0x94
,
0x35
,
0xfc
,
0x69
,
0xb2
,
0x7d
,
0x97
,
0x27
,
0x6c
,
0x0c
,
0x79
,
0x2f
,
0x44
,
0xc7
,
0xdb
,
0xb6
,
0x61
,
0x92
,
0x9c
,
0x04
,
0xb5
,
0x18
,
0x07
,
0xfc
,
0x9c
,
0x1d
,
0x9c
,
0x38
,
0xdb
,
0x8c
,
0xb8
,
0x4f
,
0x92
,
0xa9
,
0x1d
,
0x46
,
0xe0
,
0x8f
,
0xa2
,
0x86
,
0x5d
,
0xd8
,
0xef
,
0x2c
,
0x3c
,
0x15
,
0x38
,
0x9b
,
0x1a
,
0x4f
,
0xbb
,
0x56
,
0xd6
,
0xea
,
0x0c
,
0xb0
,
0xb1
,
0x06
,
0x21
,
0x0e
,
0xb8
,
0x62
,
0xfc
,
0x04
,
0x50
,
0x3a
,
0x55
,
0xc0
,
0xec
,
0x1b
,
0x5e
,
0xec
,
0x72
,
0x79
,
0x72
,
0xd3
,
0x7d
,
0xc6
,
0x01
,
0xbf
,
0x60
,
0x87
,
0x79
,
0x65
,
0xaf
,
0x37
,
0x65
,
0xe4
,
0xd1
,
0xe4
,
0xe6
,
0x6f
,
0xeb
,
0x86
,
0x6e
,
0xa2
,
0x67
,
0xd3
,
0xf4
,
0x9c
,
0x9c
,
0x32
,
0xe5
,
0x7b
,
0x85
,
0x34
,
0xda
,
0xe2
,
0x9c
,
0x1d
,
0xfa
,
0x4c
,
0x3f
,
0x09
,
0x47
,
0xaa
,
0x5f
,
0xe1
,
0xf1
,
0x74
,
0xfb
,
0x9d
,
0x61
,
0x9f
,
0x3f
,
0x9f
,
0xb3
,
0xb0
,
0x0b
,
0xf4
,
0xff
,
0x42
,
0x2f
,
0xd8
,
0xf2
,
0x54
,
0xe0
,
0x4c
,
0xe6
,
0x7e
,
0x61
,
0xfe
,
0x64
,
0x47
,
0x7f
,
0xc2
,
0x9c
,
0xc9
,
0x7f
,
0xbe
,
0xc3
,
0xb4
,
0x9d
,
0x64
,
0xc1
,
0x0e
,
0xba
,
0x24
,
0xef
,
0xaa
,
0xb8
,
0xe7
,
0x83
,
0x99
,
0x13
,
0xe9
,
0x17
,
0xb6
,
0x38
,
0x33
,
0x08
,
0x8e
,
0xf8
,
0xc3
,
0xe9
,
0xae
,
0xcc
,
0x5e
,
0xaf
,
0x04
,
0xc9
,
0xea
,
0x6f
,
0xb3
,
0x9f
,
0x19
,
0x82
,
0x12
,
0x5c
,
0x26
,
0x4c
,
0x09
,
0x23
,
0xee
,
0x3b
,
0xb6
,
0xc8
,
0x41
,
0x38
,
0x59
,
0xf1
,
0xe3
,
0xe9
,
0xbe
,
0xcf
,
0x2d
,
0xb8
,
0x9b
,
0xe8
,
0xd1
,
0x3f
,
0x8a
,
0x19
,
0x60
,
0xab
,
0x29
,
0x0e
,
0x56
,
0xeb
,
0x6f
,
0x6f
,
0x4a
,
0x45
,
0x55
,
0x5b
,
0x74
,
0x21
,
0xa7
,
0xb7
,
0x4a
,
0x6b
,
0x75
,
0x4b
,
0x20
,
0xab
,
0xd4
,
0xf7
,
0xbe
,
0xbc
,
0x54
,
0x48
,
0x4e
,
0x15
,
0x2d
,
0xc1
,
0x65
,
0xaa
,
0x0c
,
0x81
,
0x33
,
0x42
,
0xa7
,
0x3d
,
0x30
,
0x1d
,
0x80
,
0x4d
,
0x51
,
0x2c
,
0x7a
,
0xe1
,
0xf5
,
0xef
,
0x00
,
0x00
,
0x00
,
0xff
,
0xff
,
0x36
,
0x07
,
0xbc
,
0xa2
,
0xfe
,
0x04
,
0x00
,
0x00
,
// 464 bytes of a gzipped FileDescriptorProto
0x1f
,
0x8b
,
0x08
,
0x00
,
0x00
,
0x00
,
0x00
,
0x00
,
0x02
,
0xff
,
0xac
,
0x93
,
0xcf
,
0x6e
,
0xd3
,
0x40
,
0x10
,
0xc6
,
0x73
,
0xca
,
0x61
,
0x88
,
0x5b
,
0xba
,
0x82
,
0x8b
,
0x2b
,
0x21
,
0x30
,
0x20
,
0x21
,
0x10
,
0xb1
,
0x04
,
0x4f
,
0x40
,
0x52
,
0xa4
,
0x06
,
0x41
,
0x15
,
0x62
,
0x15
,
0x51
,
0x38
,
0x94
,
0xb5
,
0x33
,
0xd8
,
0x2b
,
0xd9
,
0xbb
,
0xd6
,
0xce
,
0xb8
,
0xff
,
0xde
,
0x99
,
0x77
,
0x40
,
0xb1
,
0xdd
,
0xd6
,
0x96
,
0x1c
,
0x62
,
0x4b
,
0xdc
,
0xb2
,
0x33
,
0xdf
,
0xfc
,
0x76
,
0xbf
,
0x7c
,
0x63
,
0x70
,
0x08
,
0xed
,
0x85
,
0x8a
,
0x70
,
0x9a
,
0x5b
,
0xc3
,
0x46
,
0x3c
,
0xca
,
0x54
,
0x7a
,
0x51
,
0x50
,
0x75
,
0x9a
,
0xd6
,
0x3d
,
0x77
,
0x12
,
0x99
,
0x2c
,
0x33
,
0xba
,
0xaa
,
0xba
,
0x07
,
0x75
,
0xf9
,
0x3c
,
0xa3
,
0xb8
,
0x2e
,
0x4d
,
0x28
,
0x4a
,
0x30
,
0x93
,
0xd5
,
0xe9
,
0xdd
,
0x1f
,
0x00
,
0xe7
,
0x4b
,
0xc9
,
0x09
,
0x2a
,
0xa5
,
0xf8
,
0x0e
,
0x0f
,
0xe7
,
0x16
,
0x25
,
0xe3
,
0xdc
,
0xa4
,
0x29
,
0x46
,
0xac
,
0x8c
,
0x16
,
0x2f
,
0xa7
,
0xed
,
0xbb
,
0x2a
,
0xc2
,
0xbd
,
0x20
,
0x28
,
0x0b
,
0xee
,
0x61
,
0x5b
,
0x56
,
0xbf
,
0x24
,
0x60
,
0xc9
,
0x05
,
0x79
,
0x23
,
0x71
,
0x0a
,
0x7b
,
0x47
,
0xd6
,
0xe4
,
0x0d
,
0xee
,
0x8b
,
0x69
,
0x97
,
0x87
,
0x06
,
0xf8
,
0x44
,
0x66
,
0xb8
,
0x0b
,
0xfb
,
0x13
,
0x9c
,
0x63
,
0x49
,
0x83
,
0xa9
,
0x5e
,
0xb7
,
0x6a
,
0x66
,
0x4c
,
0xba
,
0x42
,
0xca
,
0x8d
,
0x26
,
0xf4
,
0x46
,
0x42
,
0x81
,
0x38
,
0x42
,
0x8a
,
0xac
,
0x0a
,
0x71
,
0xf0
,
0x0d
,
0x6f
,
0x76
,
0xa9
,
0x2a
,
0x72
,
0xbe
,
0xf9
,
0xe9
,
0x8d
,
0xc4
,
0x19
,
0xec
,
0x07
,
0x89
,
0xb9
,
0xbc
,
0x6f
,
0x93
,
0x70
,
0x3b
,
0x9d
,
0x7f
,
0xcc
,
0x72
,
0xbe
,
0x76
,
0x5f
,
0x75
,
0xd3
,
0x03
,
0xb6
,
0x4a
,
0xc7
,
0x9f
,
0x15
,
0x71
,
0xc3
,
0xc5
,
0x29
,
0xec
,
0x57
,
0x99
,
0x2e
,
0xa5
,
0x65
,
0x55
,
0x5a
,
0x78
,
0xde
,
0x3d
,
0x7e
,
0x27
,
0xe8
,
0xf3
,
0xcf
,
0x07
,
0xe0
,
0x6c
,
0x02
,
0xfd
,
0xbf
,
0xd0
,
0x33
,
0x98
,
0x1c
,
0x4b
,
0x1a
,
0xc8
,
0xec
,
0x17
,
0xe6
,
0x6f
,
0x38
,
0xb8
,
0x0d
,
0x73
,
0x20
,
0xff
,
0xf5
,
0x0e
,
0x51
,
0x3b
,
0xc9
,
0x10
,
0xf6
,
0x36
,
0x49
,
0xde
,
0x75
,
0xa9
,
0xe7
,
0xc2
,
0x0c
,
0x89
,
0xf4
,
0x1b
,
0x8c
,
0x17
,
0x9a
,
0xd0
,
0xb2
,
0x78
,
0xd2
,
0x3d
,
0xb5
,
0x32
,
0x97
,
0x33
,
0xc9
,
0x51
,
0xb2
,
0xed
,
0xed
,
0x0b
,
0xcd
,
0x18
,
0xa3
,
0x5d
,
0x49
,
0x1d
,
0x63
,
0x83
,
0xfb
,
0x09
,
0xc6
,
0x01
,
0x4a
,
0x1b
,
0x25
,
0xe2
,
0xb0
,
0x7b
,
0xee
,
0x6b
,
0x81
,
0xf6
,
0xda
,
0x7d
,
0xf6
,
0x8f
,
0xe6
,
0x0a
,
0xa9
,
0x48
,
0xd9
,
0x1b
,
0x89
,
0x13
,
0x78
,
0x50
,
0xad
,
0xdd
,
0x42
,
0xaf
,
0xf1
,
0x4a
,
0x3c
,
0xdd
,
0xf6
,
0x90
,
0x35
,
0x5e
,
0x2d
,
0xa5
,
0x95
,
0xd9
,
0xae
,
0xd5
,
0xf8
,
0x05
,
0xce
,
0x6d
,
0x7e
,
0x7d
,
0x89
,
0x5b
,
0xbe
,
0xc1
,
0x16
,
0xa6
,
0xe1
,
0xfe
,
0x1c
,
0x1e
,
0xb7
,
0x5a
,
0x4b
,
0x6b
,
0x62
,
0x8b
,
0x44
,
0x3d
,
0x6e
,
0xea
,
0xb5
,
0x82
,
0xb3
,
0xf9
,
0x8f
,
0x0f
,
0xb1
,
0xe2
,
0xa4
,
0x08
,
0x37
,
0xe6
,
0xfc
,
0x1b
,
0x95
,
0xa6
,
0xea
,
0x86
,
0x31
,
0x4a
,
0xfc
,
0x6a
,
0xf8
,
0xed
,
0x5a
,
0x11
,
0x5b
,
0x15
,
0x16
,
0x8c
,
0x6b
,
0x5f
,
0x69
,
0x46
,
0xab
,
0x65
,
0xea
,
0x97
,
0x44
,
0xbf
,
0x26
,
0xe6
,
0x61
,
0x38
,
0x2e
,
0x0b
,
0xef
,
0xff
,
0x06
,
0x00
,
0x00
,
0xff
,
0xff
,
0x9b
,
0xf8
,
0xbc
,
0x74
,
0x11
,
0x06
,
0x00
,
0x00
,
}
// Reference imports to suppress errors if they are not otherwise used.
...
...
@@ -144,6 +147,27 @@ type MilvusServiceClient interface {
//
// @return QueryResult
Search
(
ctx
context
.
Context
,
in
*
Query
,
opts
...
grpc
.
CallOption
)
(
*
QueryResult
,
error
)
//*
// @brief This method is used to build index by collection in sync mode.
//
// @param IndexParam, index paramters.
//
// @return Status
CreateIndex
(
ctx
context
.
Context
,
in
*
IndexParam
,
opts
...
grpc
.
CallOption
)
(
*
commonpb
.
Status
,
error
)
//*
// @brief This method is used to describe index
//
// @param IndexParam, target index.
//
// @return IndexParam
DescribeIndex
(
ctx
context
.
Context
,
in
*
IndexParam
,
opts
...
grpc
.
CallOption
)
(
*
DescribeIndexResponse
,
error
)
//*
// @brief This method is used to query index building progress
//
// @param IndexParam, target index.
//
// @return IndexParam
DescribeIndexProgress
(
ctx
context
.
Context
,
in
*
IndexParam
,
opts
...
grpc
.
CallOption
)
(
*
BoolResponse
,
error
)
}
type
milvusServiceClient
struct
{
...
...
@@ -262,6 +286,33 @@ func (c *milvusServiceClient) Search(ctx context.Context, in *Query, opts ...grp
return
out
,
nil
}
func
(
c
*
milvusServiceClient
)
CreateIndex
(
ctx
context
.
Context
,
in
*
IndexParam
,
opts
...
grpc
.
CallOption
)
(
*
commonpb
.
Status
,
error
)
{
out
:=
new
(
commonpb
.
Status
)
err
:=
c
.
cc
.
Invoke
(
ctx
,
"/milvus.proto.service.MilvusService/CreateIndex"
,
in
,
out
,
opts
...
)
if
err
!=
nil
{
return
nil
,
err
}
return
out
,
nil
}
func
(
c
*
milvusServiceClient
)
DescribeIndex
(
ctx
context
.
Context
,
in
*
IndexParam
,
opts
...
grpc
.
CallOption
)
(
*
DescribeIndexResponse
,
error
)
{
out
:=
new
(
DescribeIndexResponse
)
err
:=
c
.
cc
.
Invoke
(
ctx
,
"/milvus.proto.service.MilvusService/DescribeIndex"
,
in
,
out
,
opts
...
)
if
err
!=
nil
{
return
nil
,
err
}
return
out
,
nil
}
func
(
c
*
milvusServiceClient
)
DescribeIndexProgress
(
ctx
context
.
Context
,
in
*
IndexParam
,
opts
...
grpc
.
CallOption
)
(
*
BoolResponse
,
error
)
{
out
:=
new
(
BoolResponse
)
err
:=
c
.
cc
.
Invoke
(
ctx
,
"/milvus.proto.service.MilvusService/DescribeIndexProgress"
,
in
,
out
,
opts
...
)
if
err
!=
nil
{
return
nil
,
err
}
return
out
,
nil
}
// MilvusServiceServer is the server API for MilvusService service.
type
MilvusServiceServer
interface
{
//*
...
...
@@ -338,6 +389,27 @@ type MilvusServiceServer interface {
//
// @return QueryResult
Search
(
context
.
Context
,
*
Query
)
(
*
QueryResult
,
error
)
//*
// @brief This method is used to build index by collection in sync mode.
//
// @param IndexParam, index paramters.
//
// @return Status
CreateIndex
(
context
.
Context
,
*
IndexParam
)
(
*
commonpb
.
Status
,
error
)
//*
// @brief This method is used to describe index
//
// @param IndexParam, target index.
//
// @return IndexParam
DescribeIndex
(
context
.
Context
,
*
IndexParam
)
(
*
DescribeIndexResponse
,
error
)
//*
// @brief This method is used to query index building progress
//
// @param IndexParam, target index.
//
// @return IndexParam
DescribeIndexProgress
(
context
.
Context
,
*
IndexParam
)
(
*
BoolResponse
,
error
)
}
// UnimplementedMilvusServiceServer can be embedded to have forward compatible implementations.
...
...
@@ -380,6 +452,15 @@ func (*UnimplementedMilvusServiceServer) Insert(ctx context.Context, req *RowBat
func
(
*
UnimplementedMilvusServiceServer
)
Search
(
ctx
context
.
Context
,
req
*
Query
)
(
*
QueryResult
,
error
)
{
return
nil
,
status
.
Errorf
(
codes
.
Unimplemented
,
"method Search not implemented"
)
}
func
(
*
UnimplementedMilvusServiceServer
)
CreateIndex
(
ctx
context
.
Context
,
req
*
IndexParam
)
(
*
commonpb
.
Status
,
error
)
{
return
nil
,
status
.
Errorf
(
codes
.
Unimplemented
,
"method CreateIndex not implemented"
)
}
func
(
*
UnimplementedMilvusServiceServer
)
DescribeIndex
(
ctx
context
.
Context
,
req
*
IndexParam
)
(
*
DescribeIndexResponse
,
error
)
{
return
nil
,
status
.
Errorf
(
codes
.
Unimplemented
,
"method DescribeIndex not implemented"
)
}
func
(
*
UnimplementedMilvusServiceServer
)
DescribeIndexProgress
(
ctx
context
.
Context
,
req
*
IndexParam
)
(
*
BoolResponse
,
error
)
{
return
nil
,
status
.
Errorf
(
codes
.
Unimplemented
,
"method DescribeIndexProgress not implemented"
)
}
func
RegisterMilvusServiceServer
(
s
*
grpc
.
Server
,
srv
MilvusServiceServer
)
{
s
.
RegisterService
(
&
_MilvusService_serviceDesc
,
srv
)
...
...
@@ -601,6 +682,60 @@ func _MilvusService_Search_Handler(srv interface{}, ctx context.Context, dec fun
return
interceptor
(
ctx
,
in
,
info
,
handler
)
}
func
_MilvusService_CreateIndex_Handler
(
srv
interface
{},
ctx
context
.
Context
,
dec
func
(
interface
{})
error
,
interceptor
grpc
.
UnaryServerInterceptor
)
(
interface
{},
error
)
{
in
:=
new
(
IndexParam
)
if
err
:=
dec
(
in
);
err
!=
nil
{
return
nil
,
err
}
if
interceptor
==
nil
{
return
srv
.
(
MilvusServiceServer
)
.
CreateIndex
(
ctx
,
in
)
}
info
:=
&
grpc
.
UnaryServerInfo
{
Server
:
srv
,
FullMethod
:
"/milvus.proto.service.MilvusService/CreateIndex"
,
}
handler
:=
func
(
ctx
context
.
Context
,
req
interface
{})
(
interface
{},
error
)
{
return
srv
.
(
MilvusServiceServer
)
.
CreateIndex
(
ctx
,
req
.
(
*
IndexParam
))
}
return
interceptor
(
ctx
,
in
,
info
,
handler
)
}
func
_MilvusService_DescribeIndex_Handler
(
srv
interface
{},
ctx
context
.
Context
,
dec
func
(
interface
{})
error
,
interceptor
grpc
.
UnaryServerInterceptor
)
(
interface
{},
error
)
{
in
:=
new
(
IndexParam
)
if
err
:=
dec
(
in
);
err
!=
nil
{
return
nil
,
err
}
if
interceptor
==
nil
{
return
srv
.
(
MilvusServiceServer
)
.
DescribeIndex
(
ctx
,
in
)
}
info
:=
&
grpc
.
UnaryServerInfo
{
Server
:
srv
,
FullMethod
:
"/milvus.proto.service.MilvusService/DescribeIndex"
,
}
handler
:=
func
(
ctx
context
.
Context
,
req
interface
{})
(
interface
{},
error
)
{
return
srv
.
(
MilvusServiceServer
)
.
DescribeIndex
(
ctx
,
req
.
(
*
IndexParam
))
}
return
interceptor
(
ctx
,
in
,
info
,
handler
)
}
func
_MilvusService_DescribeIndexProgress_Handler
(
srv
interface
{},
ctx
context
.
Context
,
dec
func
(
interface
{})
error
,
interceptor
grpc
.
UnaryServerInterceptor
)
(
interface
{},
error
)
{
in
:=
new
(
IndexParam
)
if
err
:=
dec
(
in
);
err
!=
nil
{
return
nil
,
err
}
if
interceptor
==
nil
{
return
srv
.
(
MilvusServiceServer
)
.
DescribeIndexProgress
(
ctx
,
in
)
}
info
:=
&
grpc
.
UnaryServerInfo
{
Server
:
srv
,
FullMethod
:
"/milvus.proto.service.MilvusService/DescribeIndexProgress"
,
}
handler
:=
func
(
ctx
context
.
Context
,
req
interface
{})
(
interface
{},
error
)
{
return
srv
.
(
MilvusServiceServer
)
.
DescribeIndexProgress
(
ctx
,
req
.
(
*
IndexParam
))
}
return
interceptor
(
ctx
,
in
,
info
,
handler
)
}
var
_MilvusService_serviceDesc
=
grpc
.
ServiceDesc
{
ServiceName
:
"milvus.proto.service.MilvusService"
,
HandlerType
:
(
*
MilvusServiceServer
)(
nil
),
...
...
@@ -653,6 +788,18 @@ var _MilvusService_serviceDesc = grpc.ServiceDesc{
MethodName
:
"Search"
,
Handler
:
_MilvusService_Search_Handler
,
},
{
MethodName
:
"CreateIndex"
,
Handler
:
_MilvusService_CreateIndex_Handler
,
},
{
MethodName
:
"DescribeIndex"
,
Handler
:
_MilvusService_DescribeIndex_Handler
,
},
{
MethodName
:
"DescribeIndexProgress"
,
Handler
:
_MilvusService_DescribeIndexProgress_Handler
,
},
},
Streams
:
[]
grpc
.
StreamDesc
{},
Metadata
:
"service.proto"
,
...
...
internal/proto/servicepb/service_msg.pb.go
浏览文件 @
b582cf7b
...
...
@@ -900,6 +900,154 @@ func (m *QueryResult) GetHits() [][]byte {
return
nil
}
//*
// @brief Index params
// @collection_name: target collection
// @field_name: target field
// @index_name: a name for index provided by user, unique within this field
// @extra_params: index parameters in json format
// for vector field:
// extra_params["index_type"] = one of the values: FLAT, IVF_LAT, IVF_SQ8, NSGMIX, IVFSQ8H,
// PQ, HNSW, HNSW_SQ8NM, ANNOY
// extra_params["metric_type"] = one of the values: L2, IP, HAMMING, JACCARD, TANIMOTO
// SUBSTRUCTURE, SUPERSTRUCTURE
// extra_params["params"] = extra parameters for index, for example ivflat: {nlist: 2048}
// for structured field:
// extra_params["index_type"] = one of the values: SORTED
type
IndexParam
struct
{
CollectionName
string
`protobuf:"bytes,2,opt,name=collection_name,json=collectionName,proto3" json:"collection_name,omitempty"`
FieldName
string
`protobuf:"bytes,3,opt,name=field_name,json=fieldName,proto3" json:"field_name,omitempty"`
IndexName
string
`protobuf:"bytes,4,opt,name=index_name,json=indexName,proto3" json:"index_name,omitempty"`
ExtraParams
[]
*
commonpb
.
KeyValuePair
`protobuf:"bytes,5,rep,name=extra_params,json=extraParams,proto3" json:"extra_params,omitempty"`
XXX_NoUnkeyedLiteral
struct
{}
`json:"-"`
XXX_unrecognized
[]
byte
`json:"-"`
XXX_sizecache
int32
`json:"-"`
}
func
(
m
*
IndexParam
)
Reset
()
{
*
m
=
IndexParam
{}
}
func
(
m
*
IndexParam
)
String
()
string
{
return
proto
.
CompactTextString
(
m
)
}
func
(
*
IndexParam
)
ProtoMessage
()
{}
func
(
*
IndexParam
)
Descriptor
()
([]
byte
,
[]
int
)
{
return
fileDescriptor_b4b40b84dd2f74cb
,
[]
int
{
16
}
}
func
(
m
*
IndexParam
)
XXX_Unmarshal
(
b
[]
byte
)
error
{
return
xxx_messageInfo_IndexParam
.
Unmarshal
(
m
,
b
)
}
func
(
m
*
IndexParam
)
XXX_Marshal
(
b
[]
byte
,
deterministic
bool
)
([]
byte
,
error
)
{
return
xxx_messageInfo_IndexParam
.
Marshal
(
b
,
m
,
deterministic
)
}
func
(
m
*
IndexParam
)
XXX_Merge
(
src
proto
.
Message
)
{
xxx_messageInfo_IndexParam
.
Merge
(
m
,
src
)
}
func
(
m
*
IndexParam
)
XXX_Size
()
int
{
return
xxx_messageInfo_IndexParam
.
Size
(
m
)
}
func
(
m
*
IndexParam
)
XXX_DiscardUnknown
()
{
xxx_messageInfo_IndexParam
.
DiscardUnknown
(
m
)
}
var
xxx_messageInfo_IndexParam
proto
.
InternalMessageInfo
func
(
m
*
IndexParam
)
GetCollectionName
()
string
{
if
m
!=
nil
{
return
m
.
CollectionName
}
return
""
}
func
(
m
*
IndexParam
)
GetFieldName
()
string
{
if
m
!=
nil
{
return
m
.
FieldName
}
return
""
}
func
(
m
*
IndexParam
)
GetIndexName
()
string
{
if
m
!=
nil
{
return
m
.
IndexName
}
return
""
}
func
(
m
*
IndexParam
)
GetExtraParams
()
[]
*
commonpb
.
KeyValuePair
{
if
m
!=
nil
{
return
m
.
ExtraParams
}
return
nil
}
type
DescribeIndexResponse
struct
{
Status
*
commonpb
.
Status
`protobuf:"bytes,1,opt,name=status,proto3" json:"status,omitempty"`
CollectionName
string
`protobuf:"bytes,2,opt,name=collection_name,json=collectionName,proto3" json:"collection_name,omitempty"`
FieldName
string
`protobuf:"bytes,3,opt,name=field_name,json=fieldName,proto3" json:"field_name,omitempty"`
IndexName
string
`protobuf:"bytes,4,opt,name=index_name,json=indexName,proto3" json:"index_name,omitempty"`
ExtraParams
[]
*
commonpb
.
KeyValuePair
`protobuf:"bytes,5,rep,name=extra_params,json=extraParams,proto3" json:"extra_params,omitempty"`
XXX_NoUnkeyedLiteral
struct
{}
`json:"-"`
XXX_unrecognized
[]
byte
`json:"-"`
XXX_sizecache
int32
`json:"-"`
}
func
(
m
*
DescribeIndexResponse
)
Reset
()
{
*
m
=
DescribeIndexResponse
{}
}
func
(
m
*
DescribeIndexResponse
)
String
()
string
{
return
proto
.
CompactTextString
(
m
)
}
func
(
*
DescribeIndexResponse
)
ProtoMessage
()
{}
func
(
*
DescribeIndexResponse
)
Descriptor
()
([]
byte
,
[]
int
)
{
return
fileDescriptor_b4b40b84dd2f74cb
,
[]
int
{
17
}
}
func
(
m
*
DescribeIndexResponse
)
XXX_Unmarshal
(
b
[]
byte
)
error
{
return
xxx_messageInfo_DescribeIndexResponse
.
Unmarshal
(
m
,
b
)
}
func
(
m
*
DescribeIndexResponse
)
XXX_Marshal
(
b
[]
byte
,
deterministic
bool
)
([]
byte
,
error
)
{
return
xxx_messageInfo_DescribeIndexResponse
.
Marshal
(
b
,
m
,
deterministic
)
}
func
(
m
*
DescribeIndexResponse
)
XXX_Merge
(
src
proto
.
Message
)
{
xxx_messageInfo_DescribeIndexResponse
.
Merge
(
m
,
src
)
}
func
(
m
*
DescribeIndexResponse
)
XXX_Size
()
int
{
return
xxx_messageInfo_DescribeIndexResponse
.
Size
(
m
)
}
func
(
m
*
DescribeIndexResponse
)
XXX_DiscardUnknown
()
{
xxx_messageInfo_DescribeIndexResponse
.
DiscardUnknown
(
m
)
}
var
xxx_messageInfo_DescribeIndexResponse
proto
.
InternalMessageInfo
func
(
m
*
DescribeIndexResponse
)
GetStatus
()
*
commonpb
.
Status
{
if
m
!=
nil
{
return
m
.
Status
}
return
nil
}
func
(
m
*
DescribeIndexResponse
)
GetCollectionName
()
string
{
if
m
!=
nil
{
return
m
.
CollectionName
}
return
""
}
func
(
m
*
DescribeIndexResponse
)
GetFieldName
()
string
{
if
m
!=
nil
{
return
m
.
FieldName
}
return
""
}
func
(
m
*
DescribeIndexResponse
)
GetIndexName
()
string
{
if
m
!=
nil
{
return
m
.
IndexName
}
return
""
}
func
(
m
*
DescribeIndexResponse
)
GetExtraParams
()
[]
*
commonpb
.
KeyValuePair
{
if
m
!=
nil
{
return
m
.
ExtraParams
}
return
nil
}
func
init
()
{
proto
.
RegisterEnum
(
"milvus.proto.service.PlaceholderType"
,
PlaceholderType_name
,
PlaceholderType_value
)
proto
.
RegisterType
((
*
CollectionName
)(
nil
),
"milvus.proto.service.CollectionName"
)
...
...
@@ -918,58 +1066,65 @@ func init() {
proto
.
RegisterType
((
*
SysConfigResponse
)(
nil
),
"milvus.proto.service.SysConfigResponse"
)
proto
.
RegisterType
((
*
Hits
)(
nil
),
"milvus.proto.service.Hits"
)
proto
.
RegisterType
((
*
QueryResult
)(
nil
),
"milvus.proto.service.QueryResult"
)
proto
.
RegisterType
((
*
IndexParam
)(
nil
),
"milvus.proto.service.IndexParam"
)
proto
.
RegisterType
((
*
DescribeIndexResponse
)(
nil
),
"milvus.proto.service.DescribeIndexResponse"
)
}
func
init
()
{
proto
.
RegisterFile
(
"service_msg.proto"
,
fileDescriptor_b4b40b84dd2f74cb
)
}
var
fileDescriptor_b4b40b84dd2f74cb
=
[]
byte
{
// 762 bytes of a gzipped FileDescriptorProto
0x1f
,
0x8b
,
0x08
,
0x00
,
0x00
,
0x00
,
0x00
,
0x00
,
0x02
,
0xff
,
0xac
,
0x54
,
0xdd
,
0x4e
,
0xe3
,
0x46
,
0x14
,
0xae
,
0xe3
,
0x90
,
0x86
,
0x13
,
0x27
,
0x24
,
0xd3
,
0x14
,
0x19
,
0xb8
,
0x49
,
0x8d
,
0x68
,
0xa3
,
0x56
,
0x4d
,
0x24
,
0xa8
,
0x54
,
0x71
,
0x51
,
0xa9
,
0x49
,
0xa0
,
0x2d
,
0x3f
,
0x0a
,
0x74
,
0x12
,
0x21
,
0xd1
,
0x4a
,
0x8d
,
0x26
,
0xf6
,
0xd4
,
0x1e
,
0xd5
,
0xf1
,
0x58
,
0x9e
,
0x31
,
0x51
,
0x78
,
0x90
,
0xbe
,
0xc4
,
0x3e
,
0xc8
,
0xde
,
0xed
,
0x33
,
0xad
,
0x3c
,
0x36
,
0xf9
,
0x61
,
0x59
,
0x2d
,
0x4b
,
0xb8
,
0x3b
,
0xe7
,
0xcc
,
0x9c
,
0xf3
,
0x9d
,
0xdf
,
0x0f
,
0x6a
,
0x82
,
0x46
,
0x77
,
0xcc
,
0xa6
,
0xa3
,
0x89
,
0x70
,
0x5b
,
0x61
,
0xc4
,
0x25
,
0x47
,
0xf5
,
0x09
,
0xf3
,
0xef
,
0x62
,
0x91
,
0x6a
,
0xad
,
0xec
,
0x7d
,
0xd7
,
0xb0
,
0xf9
,
0x64
,
0xc2
,
0x83
,
0xd4
,
0xba
,
0x6b
,
0x08
,
0xdb
,
0xa3
,
0x13
,
0x92
,
0x6a
,
0xd6
,
0x31
,
0x54
,
0x7a
,
0xdc
,
0xf7
,
0xa9
,
0x2d
,
0x19
,
0x0f
,
0xfa
,
0x64
,
0x42
,
0xd1
,
0x77
,
0xb0
,
0x65
,
0xcf
,
0x2d
,
0xa3
,
0x80
,
0x4c
,
0xa8
,
0xa9
,
0x35
,
0xb4
,
0xe6
,
0x26
,
0xae
,
0xd8
,
0x2b
,
0x1f
,
0xad
,
0x73
,
0x28
,
0x5f
,
0x93
,
0x48
,
0xb2
,
0xcf
,
0xf6
,
0x44
,
0x55
,
0xd0
,
0x25
,
0x71
,
0xcd
,
0x9c
,
0x7a
,
0x4c
,
0x44
,
0xeb
,
0x8d
,
0x06
,
0x45
,
0xcc
,
0xa7
,
0x5d
,
0x22
,
0x6d
,
0xef
,
0xf9
,
0x71
,
0xf6
,
0xa1
,
0x1c
,
0x3e
,
0x64
,
0x30
,
0x5a
,
0x44
,
0x34
,
0xe6
,
0xc6
,
0x21
,
0x71
,
0xd1
,
0x4f
,
0x50
,
0x8c
,
0xf8
,
0x74
,
0xe4
,
0x10
,
0x49
,
0x4c
,
0xbd
,
0xa1
,
0x37
,
0x4b
,
0x87
,
0x3b
,
0xad
,
0x95
,
0x36
,
0x65
,
0xdd
,
0xe9
,
0xfa
,
0x7c
,
0x8c
,
0xbf
,
0x8c
,
0xf8
,
0xf4
,
0x84
,
0x48
,
0x82
,
0xf6
,
0x60
,
0xd3
,
0x23
,
0xc2
,
0x1b
,
0xfd
,
0x47
,
0x67
,
0xc2
,
0xcc
,
0x37
,
0xf4
,
0x66
,
0x19
,
0x17
,
0x13
,
0xc3
,
0x05
,
0x9d
,
0x09
,
0x6b
,
0x0a
,
0xd5
,
0x6b
,
0x9f
,
0xd8
,
0xd4
,
0xe3
,
0xbe
,
0x43
,
0xa3
,
0x1b
,
0xe2
,
0xc7
,
0xf3
,
0x9a
,
0xb4
,
0x79
,
0x4d
,
0xe8
,
0x18
,
0xf2
,
0x72
,
0x16
,
0x52
,
0x95
,
0x54
,
0xe5
,
0xf0
,
0xa0
,
0xf5
,
0xd4
,
0x6c
,
0x5a
,
0x4b
,
0x71
,
0x86
,
0xb3
,
0x90
,
0x62
,
0xe5
,
0x82
,
0xb6
,
0xa1
,
0x70
,
0x97
,
0x44
,
0x15
,
0x2a
,
0x63
,
0x03
,
0x67
,
0x9a
,
0xf5
,
0xcf
,
0x0a
,
0xf0
,
0xef
,
0x11
,
0x8f
,
0x43
,
0x74
,
0x0e
,
0x46
,
0xb8
,
0xb0
,
0x09
,
0x53
,
0x53
,
0x35
,
0x7e
,
0xfb
,
0x49
,
0x38
,
0x95
,
0x36
,
0x5e
,
0xf1
,
0xb5
,
0xfe
,
0xd7
,
0x60
,
0xe3
,
0xcf
,
0x98
,
0x46
,
0xb3
,
0xe7
,
0xcf
,
0xe0
,
0x00
,
0x2a
,
0x2b
,
0x33
,
0x10
,
0x66
,
0xae
,
0xa1
,
0x37
,
0x37
,
0x71
,
0x79
,
0x79
,
0x08
,
0x22
,
0x69
,
0x8f
,
0x23
,
0x7c
,
0x53
,
0x4f
,
0xdb
,
0xe3
,
0x08
,
0x1f
,
0xfd
,
0x00
,
0xb5
,
0x25
,
0xec
,
0x91
,
0x9b
,
0x14
,
0x63
,
0xe6
,
0x1b
,
0x5a
,
0xd3
,
0xc0
,
0xd5
,
0xf0
,
0x51
,
0x91
,
0xd6
,
0xdf
,
0x50
,
0x19
,
0xc8
,
0x88
,
0x05
,
0x2e
,
0xa6
,
0x22
,
0xe4
,
0x81
,
0xa0
,
0xe8
,
0x08
,
0x0a
,
0x42
,
0x12
,
0x19
,
0x0b
,
0x95
,
0x57
,
0xe9
,
0x70
,
0xef
,
0xc9
,
0xa1
,
0x0e
,
0xd4
,
0x17
,
0x9c
,
0x7d
,
0x45
,
0x75
,
0xd8
,
0x50
,
0x9d
,
0xcc
,
0x16
,
0x25
,
0x55
,
0xac
,
0x5b
,
0x30
,
0xba
,
0x9c
,
0xfb
,
0xaf
,
0x18
,
0xba
,
0xf8
,
0x10
,
0x9a
,
0x00
,
0x4a
,
0xf3
,
0xbe
,
0x64
,
0x42
,
0xae
,
0x07
,
0xb0
,
0xd8
,
0x89
,
0xb4
,
0xc1
,
0x0f
,
0x3b
,
0x31
,
0x86
,
0xaf
,
0xce
,
0x02
,
0x49
,
0x5d
,
0x1a
,
0xbd
,
0x36
,
0x86
,
0x3e
,
0xc7
,
0x10
,
0x50
,
0xcf
,
0x30
,
0x30
,
0x09
,
0x5c
,
0xba
,
0x76
,
0xa7
,
0xc6
,
0xd4
,
0x65
,
0x81
,
0xea
,
0x94
,
0x8e
,
0x53
,
0x25
,
0x59
,
0x10
,
0x1a
,
0x38
,
0x6a
,
0x41
,
0x74
,
0x9c
,
0x88
,
0xd6
,
0x3b
,
0x0d
,
0xbe
,
0x5e
,
0x70
,
0xd3
,
0x09
,
0x15
,
0x76
,
0xc4
,
0xc2
,
0x44
,
0x7c
,
0x19
,
0xec
,
0x2f
,
0x50
,
0x48
,
0x99
,
0x4f
,
0xe1
,
0x96
,
0x3e
,
0x38
,
0xc8
,
0x94
,
0x15
,
0x17
,
0x80
,
0x03
,
0x65
,
0xc0
,
0x99
,
0x13
,
0xea
,
0x00
,
0x24
,
0x81
,
0x98
,
0x90
,
0xcc
,
0x16
,
0x19
,
0x91
,
0x7c
,
0xf3
,
0x24
,
0xee
,
0x05
,
0x9d
,
0xa9
,
0xdb
,
0xba
,
0x26
,
0x2c
,
0xc2
,
0x4b
,
0x4e
,
0xd6
,
0x5b
,
0x0d
,
0xea
,
0x73
,
0xc6
,
0x5c
,
0xbb
,
0x9e
,
0x9f
,
0x21
,
0xaf
,
0xce
,
0x32
,
0xad
,
0x66
,
0xff
,
0x23
,
0xf7
,
0xbe
,
0x4c
,
0xd0
,
0x58
,
0x39
,
0xbc
,
0x46
,
0x25
,
0x12
,
0x6a
,
0x83
,
0x99
,
0xe8
,
0xf1
,
0xe0
,
0x5f
,
0xb6
,
0xe6
,
0x45
,
0x22
,
0xc8
,
0x2b
,
0x8a
,
0x4d
,
0x77
,
0x5a
,
0xc9
,
0x8f
,
0xd8
,
0x6f
,
0xb1
,
0xe9
,
0x17
,
0x90
,
0xff
,
0x83
,
0x49
,
0xc5
,
0x25
,
0x67
,
0x27
,
0x29
,
0xd1
,
0xe9
,
0x38
,
0x11
,
0xd1
,
0xce
,
0x12
,
0xc7
,
0xe7
,
0x14
,
0x63
,
0xce
,
0x89
,
0x7c
,
0x3b
,
0x19
,
0x3b
,
0x8f
,
0xb2
,
0x60
,
0x39
,
0x9c
,
0x69
,
0xd6
,
0x0d
,
0x94
,
0x14
,
0xd3
,
0x61
,
0x2a
,
0x62
,
0x5f
,
0xbe
,
0x38
,
0x79
,
0x8f
,
0x49
,
0x91
,
0x41
,
0x2a
,
0xf9
,
0xfb
,
0x5f
,
0x61
,
0xeb
,
0x11
,
0xa7
,
0xa3
,
0x22
,
0xe4
,
0xfb
,
0x57
,
0xfd
,
0xd3
,
0xea
,
0x17
,
0xa8
,
0x06
,
0xe5
,
0x9b
,
0xd3
,
0xde
,
0xf0
,
0x0a
,
0x8f
,
0xba
,
0x67
,
0xfd
,
0x0e
,
0xbe
,
0xad
,
0x3a
,
0xa8
,
0x0a
,
0x46
,
0x66
,
0xfa
,
0xed
,
0xf2
,
0xaa
,
0x33
,
0xac
,
0xd2
,
0x6e
,
0xef
,
0xaf
,
0x8e
,
0xcb
,
0xa4
,
0x17
,
0x8f
,
0x13
,
0xd4
,
0xf6
,
0x3d
,
0xf3
,
0x7d
,
0x76
,
0x2f
,
0xa9
,
0xed
,
0xb5
,
0xd3
,
0x8c
,
0x7e
,
0x74
,
0x98
,
0x90
,
0x11
,
0x1b
,
0xc7
,
0x92
,
0x3a
,
0x6d
,
0x16
,
0x48
,
0x1a
,
0x05
,
0xc4
,
0x6f
,
0xab
,
0x34
,
0xdb
,
0xd9
,
0xd8
,
0xc3
,
0xf1
,
0xb8
,
0xa0
,
0x0c
,
0x47
,
0xef
,
0x03
,
0x00
,
0x00
,
0xff
,
0xff
,
0x8d
,
0x5a
,
0x44
,
0x98
,
0x25
,
0x08
,
0x00
,
0x00
,
// 847 bytes of a gzipped FileDescriptorProto
0x1f
,
0x8b
,
0x08
,
0x00
,
0x00
,
0x00
,
0x00
,
0x00
,
0x02
,
0xff
,
0xd4
,
0x55
,
0xdd
,
0x6e
,
0xdc
,
0x44
,
0x14
,
0xc6
,
0xeb
,
0x4d
,
0xd8
,
0x9c
,
0xf5
,
0x6e
,
0x37
,
0x43
,
0x5a
,
0xb9
,
0xad
,
0x90
,
0x16
,
0x57
,
0x85
,
0x15
,
0x88
,
0x5d
,
0x29
,
0x45
,
0x42
,
0xbd
,
0x40
,
0x22
,
0x9b
,
0x14
,
0x48
,
0x53
,
0x25
,
0x61
,
0x12
,
0x45
,
0x2a
,
0x48
,
0x58
,
0x63
,
0x7b
,
0x6a
,
0x8f
,
0xb0
,
0x3d
,
0xd6
,
0xcc
,
0x38
,
0xe9
,
0xf6
,
0x41
,
0x78
,
0x09
,
0x9e
,
0x80
,
0x27
,
0xe0
,
0x8e
,
0xf7
,
0xe1
,
0x0e
,
0x79
,
0xc6
,
0xfb
,
0x17
,
0x16
,
0x11
,
0xb2
,
0xb9
,
0xe1
,
0x6e
,
0xce
,
0x39
,
0x3e
,
0xe7
,
0x3b
,
0xbf
,
0x9f
,
0x61
,
0x5b
,
0x52
,
0x71
,
0xc9
,
0x42
,
0xea
,
0x67
,
0x32
,
0x1e
,
0x16
,
0x82
,
0x2b
,
0x8e
,
0x76
,
0x32
,
0x96
,
0x5e
,
0x96
,
0xd2
,
0x48
,
0xc3
,
0xda
,
0xfe
,
0xc8
,
0x09
,
0x79
,
0x96
,
0xf1
,
0xdc
,
0x68
,
0x1f
,
0x39
,
0x32
,
0x4c
,
0x68
,
0x46
,
0x8c
,
0xe4
,
0x3d
,
0x87
,
0xee
,
0x3e
,
0x4f
,
0x53
,
0x1a
,
0x2a
,
0xc6
,
0xf3
,
0x63
,
0x92
,
0x51
,
0xf4
,
0x09
,
0xdc
,
0x0b
,
0x67
,
0x1a
,
0x3f
,
0x27
,
0x19
,
0x75
,
0xad
,
0xbe
,
0x35
,
0xd8
,
0xc2
,
0xdd
,
0x70
,
0xe9
,
0x43
,
0xef
,
0x25
,
0x74
,
0x4e
,
0x89
,
0x50
,
0xec
,
0x3f
,
0x7b
,
0xa2
,
0x1e
,
0xd8
,
0x8a
,
0xc4
,
0x6e
,
0x43
,
0x1b
,
0xab
,
0xa7
,
0xf7
,
0xab
,
0x05
,
0x2d
,
0xcc
,
0xaf
,
0xc6
,
0x44
,
0x85
,
0xc9
,
0xcd
,
0xe3
,
0x3c
,
0x81
,
0x4e
,
0x31
,
0xcd
,
0xc0
,
0x9f
,
0x47
,
0x74
,
0x66
,
0xca
,
0x73
,
0x12
,
0xa3
,
0x2f
,
0xa0
,
0x25
,
0xf8
,
0x95
,
0x1f
,
0x11
,
0x45
,
0x5c
,
0xbb
,
0x6f
,
0x0f
,
0xda
,
0xbb
,
0x0f
,
0x87
,
0x4b
,
0x6d
,
0xaa
,
0xbb
,
0x33
,
0x4e
,
0x79
,
0x80
,
0xdf
,
0x17
,
0xfc
,
0xea
,
0x80
,
0x28
,
0x82
,
0x1e
,
0xc3
,
0x56
,
0x42
,
0x64
,
0xe2
,
0xff
,
0x4c
,
0x27
,
0xd2
,
0x6d
,
0xf6
,
0xed
,
0x41
,
0x07
,
0xb7
,
0x2a
,
0xc5
,
0x11
,
0x9d
,
0x48
,
0xef
,
0x0a
,
0x7a
,
0xa7
,
0x29
,
0x09
,
0x69
,
0xc2
,
0xd3
,
0x88
,
0x8a
,
0x0b
,
0x92
,
0x96
,
0xb3
,
0x9a
,
0xac
,
0x59
,
0x4d
,
0xe8
,
0x39
,
0x34
,
0xd5
,
0xa4
,
0xa0
,
0x3a
,
0xa9
,
0xee
,
0xee
,
0xd3
,
0xe1
,
0xaa
,
0xd9
,
0x0c
,
0x17
,
0xe2
,
0x9c
,
0x4f
,
0x0a
,
0x8a
,
0xb5
,
0x0b
,
0x7a
,
0x00
,
0x9b
,
0x97
,
0x55
,
0x54
,
0xa9
,
0x33
,
0x76
,
0x70
,
0x2d
,
0x79
,
0x3f
,
0x2d
,
0x01
,
0x7f
,
0x2b
,
0x78
,
0x59
,
0xa0
,
0x97
,
0xe0
,
0x14
,
0x73
,
0x9d
,
0x74
,
0x2d
,
0x5d
,
0xe3
,
0xc7
,
0xff
,
0x0a
,
0xa7
,
0xd3
,
0xc6
,
0x4b
,
0xbe
,
0xde
,
0x2f
,
0x16
,
0x6c
,
0x7c
,
0x5f
,
0x52
,
0x31
,
0xb9
,
0xf9
,
0x0c
,
0x9e
,
0x42
,
0x77
,
0x69
,
0x06
,
0xd2
,
0x6d
,
0xf4
,
0xed
,
0xc1
,
0x16
,
0xee
,
0x2c
,
0x0e
,
0x41
,
0x56
,
0xed
,
0x89
,
0x64
,
0xea
,
0xda
,
0xa6
,
0x3d
,
0x91
,
0x4c
,
0xd1
,
0x67
,
0xb0
,
0xbd
,
0x80
,
0xed
,
0xc7
,
0x55
,
0x31
,
0x6e
,
0xb3
,
0x6f
,
0x0d
,
0x1c
,
0xdc
,
0x2b
,
0xae
,
0x15
,
0xe9
,
0xfd
,
0x08
,
0xdd
,
0x33
,
0x25
,
0x58
,
0x1e
,
0x63
,
0x2a
,
0x0b
,
0x9e
,
0x4b
,
0x8a
,
0x9e
,
0xc1
,
0xa6
,
0x54
,
0x44
,
0x95
,
0x52
,
0xe7
,
0xd5
,
0xde
,
0x7d
,
0xbc
,
0x72
,
0xa8
,
0x67
,
0xfa
,
0x13
,
0x5c
,
0x7f
,
0x8a
,
0x76
,
0x60
,
0x43
,
0x77
,
0xb2
,
0x5e
,
0x14
,
0x23
,
0x78
,
0xaf
,
0xc1
,
0x19
,
0x73
,
0x9e
,
0xde
,
0x61
,
0xe8
,
0xd6
,
0x34
,
0x34
,
0x01
,
0x64
,
0xf2
,
0x7e
,
0xc5
,
0xa4
,
0x5a
,
0x0f
,
0x60
,
0xbe
,
0x13
,
0xa6
,
0xc1
,
0xd3
,
0x9d
,
0x08
,
0xe0
,
0x83
,
0xc3
,
0x5c
,
0xd1
,
0x98
,
0x8a
,
0xbb
,
0xc6
,
0xb0
,
0x67
,
0x18
,
0x12
,
0x76
,
0x6a
,
0x0c
,
0x4c
,
0xf2
,
0x98
,
0xae
,
0xdd
,
0xa9
,
0x80
,
0xc6
,
0x2c
,
0xd7
,
0x9d
,
0xb2
,
0xb1
,
0x11
,
0xaa
,
0x05
,
0xa1
,
0x79
,
0xa4
,
0x17
,
0xc4
,
0xc6
,
0xd5
,
0xd3
,
0xfb
,
0xc3
,
0x82
,
0xfb
,
0x73
,
0x6e
,
0x3a
,
0xa0
,
0x32
,
0x14
,
0xac
,
0xa8
,
0x9e
,
0xb7
,
0x83
,
0xfd
,
0x0a
,
0x36
,
0x0d
,
0xf3
,
0x69
,
0xdc
,
0xf6
,
0xdf
,
0x0e
,
0xd2
,
0xb0
,
0xe2
,
0x1c
,
0xf0
,
0x4c
,
0x2b
,
0x70
,
0xed
,
0x84
,
0xf6
,
0x00
,
0xaa
,
0x40
,
0x4c
,
0x2a
,
0x16
,
0xca
,
0x9a
,
0x48
,
0x3e
,
0x5a
,
0x89
,
0x7b
,
0x44
,
0x27
,
0xfa
,
0xb6
,
0x4e
,
0x09
,
0x13
,
0x78
,
0xc1
,
0xc9
,
0xfb
,
0xdd
,
0x82
,
0x9d
,
0x19
,
0x63
,
0xae
,
0x5d
,
0xcf
,
0x97
,
0xd0
,
0xd4
,
0x67
,
0x69
,
0xaa
,
0x79
,
0xf2
,
0x0f
,
0xf7
,
0xbe
,
0x48
,
0xd0
,
0x58
,
0x3b
,
0xdc
,
0x45
,
0x25
,
0x0a
,
0xb6
,
0xcf
,
0x26
,
0x72
,
0x9f
,
0xe7
,
0x6f
,
0xd8
,
0x9a
,
0x17
,
0x89
,
0xa0
,
0xa9
,
0x29
,
0xd6
,
0xec
,
0xb4
,
0x7e
,
0x5f
,
0x63
,
0xbf
,
0xf9
,
0xa6
,
0x1f
,
0x41
,
0xf3
,
0x3b
,
0xa6
,
0x34
,
0x97
,
0x1c
,
0x1e
,
0x18
,
0xa2
,
0xb3
,
0x71
,
0xf5
,
0x44
,
0x0f
,
0x17
,
0x38
,
0xbe
,
0xa1
,
0x19
,
0x73
,
0x46
,
0xe4
,
0x0f
,
0xaa
,
0xb1
,
0x73
,
0x51
,
0x07
,
0x6b
,
0xe0
,
0x5a
,
0xf2
,
0x2e
,
0xa0
,
0xad
,
0x99
,
0x0e
,
0x53
,
0x59
,
0xa6
,
0xea
,
0xd6
,
0xc9
,
0x27
,
0x4c
,
0xc9
,
0x1a
,
0x52
,
0xbf
,
0xbd
,
0xdf
,
0x2c
,
0x80
,
0xc3
,
0x3c
,
0xa2
,
0x6f
,
0x4f
,
0x89
,
0x20
,
0xd9
,
0x2a
,
0x1e
,
0x6d
,
0xac
,
0xe4
,
0xd1
,
0x0f
,
0x01
,
0xde
,
0x30
,
0x9a
,
0x46
,
0xe6
,
0x1b
,
0xc3
,
0x93
,
0x5b
,
0x5a
,
0x33
,
0x35
,
0xb3
,
0x2a
,
0xaa
,
0x31
,
0x37
,
0x8d
,
0x59
,
0x6b
,
0xb4
,
0xf9
,
0x00
,
0x1c
,
0xfa
,
0x56
,
0x09
,
0xe2
,
0x17
,
0x15
,
0xaa
,
0x74
,
0x37
,
0x6e
,
0x3a
,
0xd5
,
0xb6
,
0x76
,
0xd3
,
0xb9
,
0x4a
,
0xef
,
0x4f
,
0x0b
,
0xee
,
0x9b
,
0xbd
,
0x0c
,
0xa8
,
0xae
,
0x61
,
0xbd
,
0xd9
,
0xfe
,
0x9f
,
0x6a
,
0xff
,
0xf4
,
0x6b
,
0xb8
,
0x77
,
0xed
,
0x5f
,
0x8c
,
0x5a
,
0xd0
,
0x3c
,
0x3e
,
0x39
,
0x7e
,
0xd1
,
0x7b
,
0x0f
,
0x6d
,
0x43
,
0xe7
,
0xe2
,
0xc5
,
0xfe
,
0xf9
,
0x09
,
0xf6
,
0xc7
,
0x87
,
0xc7
,
0x7b
,
0xf8
,
0x75
,
0x2f
,
0x42
,
0x3d
,
0x70
,
0x6a
,
0xd5
,
0x37
,
0xaf
,
0x4e
,
0xf6
,
0xce
,
0x7b
,
0x74
,
0xbc
,
0xff
,
0xc3
,
0x5e
,
0xcc
,
0x54
,
0x52
,
0x06
,
0x15
,
0xd8
,
0xe8
,
0x1d
,
0x4b
,
0x53
,
0xf6
,
0x4e
,
0xd1
,
0x30
,
0x19
,
0x99
,
0x44
,
0x3e
,
0x8f
,
0x98
,
0x54
,
0x82
,
0x05
,
0xa5
,
0xa2
,
0xd1
,
0x88
,
0xe5
,
0x8a
,
0x8a
,
0x9c
,
0xa4
,
0x23
,
0x9d
,
0xdd
,
0xa8
,
0x3e
,
0xd7
,
0x22
,
0x08
,
0x36
,
0xb5
,
0xe2
,
0xd9
,
0x5f
,
0x01
,
0x00
,
0x00
,
0xff
,
0xff
,
0x26
,
0x6f
,
0x59
,
0x0a
,
0xdd
,
0x09
,
0x00
,
0x00
,
}
internal/proxy/grpc_service.go
浏览文件 @
b582cf7b
...
...
@@ -593,3 +593,15 @@ func (p *Proxy) ShowPartitions(ctx context.Context, req *servicepb.CollectionNam
}
return
spt
.
result
,
nil
}
func
(
p
*
Proxy
)
CreateIndex
(
ctx
context
.
Context
,
indexParam
*
servicepb
.
IndexParam
)
(
*
commonpb
.
Status
,
error
)
{
return
nil
,
nil
}
func
(
p
*
Proxy
)
DescribeIndex
(
context
.
Context
,
*
servicepb
.
IndexParam
)
(
*
servicepb
.
DescribeIndexResponse
,
error
)
{
return
nil
,
nil
}
func
(
p
*
Proxy
)
DescribeIndexProgress
(
context
.
Context
,
*
servicepb
.
IndexParam
)
(
*
servicepb
.
BoolResponse
,
error
)
{
return
nil
,
nil
}
tests/python/test_search.py
浏览文件 @
b582cf7b
...
...
@@ -705,8 +705,7 @@ class TestSearchBase:
# TODO:
# assert abs(res[0]._distances[0] - max_distance) <= tmp_epsilon
# DOG: TODO BINARY
@
pytest
.
mark
.
skip
(
"search_distance_jaccard_flat_index"
)
# PASS
def
test_search_distance_jaccard_flat_index
(
self
,
connect
,
binary_collection
):
'''
target: search binary_collection, and check the result: distance
...
...
@@ -740,8 +739,7 @@ class TestSearchBase:
with
pytest
.
raises
(
Exception
)
as
e
:
res
=
connect
.
search
(
binary_collection
,
query
)
# DOG: TODO BINARY
@
pytest
.
mark
.
skip
(
"search_distance_hamming_flat_index"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_search_distance_hamming_flat_index
(
self
,
connect
,
binary_collection
):
'''
...
...
@@ -758,8 +756,7 @@ class TestSearchBase:
res
=
connect
.
search
(
binary_collection
,
query
)
assert
abs
(
res
[
0
][
0
].
distance
-
min
(
distance_0
,
distance_1
).
astype
(
float
))
<=
epsilon
# DOG: TODO BINARY
@
pytest
.
mark
.
skip
(
"search_distance_substructure_flat_index"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_search_distance_substructure_flat_index
(
self
,
connect
,
binary_collection
):
'''
...
...
@@ -777,8 +774,7 @@ class TestSearchBase:
res
=
connect
.
search
(
binary_collection
,
query
)
assert
len
(
res
[
0
])
==
0
# DOG: TODO BINARY
@
pytest
.
mark
.
skip
(
"search_distance_substructure_flat_index_B"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_search_distance_substructure_flat_index_B
(
self
,
connect
,
binary_collection
):
'''
...
...
@@ -797,8 +793,7 @@ class TestSearchBase:
assert
res
[
1
][
0
].
distance
<=
epsilon
assert
res
[
1
][
0
].
id
==
ids
[
1
]
# DOG: TODO BINARY
@
pytest
.
mark
.
skip
(
"search_distance_superstructure_flat_index"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_search_distance_superstructure_flat_index
(
self
,
connect
,
binary_collection
):
'''
...
...
@@ -816,8 +811,7 @@ class TestSearchBase:
res
=
connect
.
search
(
binary_collection
,
query
)
assert
len
(
res
[
0
])
==
0
# DOG: TODO BINARY
@
pytest
.
mark
.
skip
(
"search_distance_superstructure_flat_index_B"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_search_distance_superstructure_flat_index_B
(
self
,
connect
,
binary_collection
):
'''
...
...
@@ -838,8 +832,7 @@ class TestSearchBase:
assert
res
[
1
][
0
].
id
in
ids
assert
res
[
1
][
0
].
distance
<=
epsilon
# DOG: TODO BINARY
@
pytest
.
mark
.
skip
(
"search_distance_tanimoto_flat_index"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_search_distance_tanimoto_flat_index
(
self
,
connect
,
binary_collection
):
'''
...
...
@@ -977,8 +970,7 @@ class TestSearchDSL(object):
******************************************************************
"""
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_no_must"
)
# PASS
def
test_query_no_must
(
self
,
connect
,
collection
):
'''
method: build query without must expr
...
...
@@ -989,8 +981,7 @@ class TestSearchDSL(object):
with
pytest
.
raises
(
Exception
)
as
e
:
res
=
connect
.
search
(
collection
,
query
)
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_no_vector_term_only"
)
# PASS
def
test_query_no_vector_term_only
(
self
,
connect
,
collection
):
'''
method: build query without vector only term
...
...
@@ -1025,8 +1016,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
default_top_k
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_wrong_format"
)
# PASS
def
test_query_wrong_format
(
self
,
connect
,
collection
):
'''
method: build query without must expr, with wrong expr name
...
...
@@ -1168,8 +1158,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
0
# DOG: TODO TRC
@
pytest
.
mark
.
skip
(
"query_complex_dsl"
)
# PASS
def
test_query_complex_dsl
(
self
,
connect
,
collection
):
'''
method: query with complicated dsl
...
...
@@ -1191,9 +1180,7 @@ class TestSearchDSL(object):
******************************************************************
"""
# DOG: TODO INVALID DSL
# TODO
@
pytest
.
mark
.
skip
(
"query_term_key_error"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_term_key_error
(
self
,
connect
,
collection
):
'''
...
...
@@ -1213,8 +1200,7 @@ class TestSearchDSL(object):
def
get_invalid_term
(
self
,
request
):
return
request
.
param
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_term_wrong_format"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_term_wrong_format
(
self
,
connect
,
collection
,
get_invalid_term
):
'''
...
...
@@ -1228,7 +1214,7 @@ class TestSearchDSL(object):
with
pytest
.
raises
(
Exception
)
as
e
:
res
=
connect
.
search
(
collection
,
query
)
# DOG:
TODO UNKNOWN
# DOG:
PLEASE IMPLEMENT connect.count_entities
# TODO
@
pytest
.
mark
.
skip
(
"query_term_field_named_term"
)
@
pytest
.
mark
.
level
(
2
)
...
...
@@ -1244,8 +1230,8 @@ class TestSearchDSL(object):
ids
=
connect
.
bulk_insert
(
collection_term
,
term_entities
)
assert
len
(
ids
)
==
default_nb
connect
.
flush
([
collection_term
])
count
=
connect
.
count_entities
(
collection_term
)
assert
count
==
default_nb
count
=
connect
.
count_entities
(
collection_term
)
# count_entities is not impelmented
assert
count
==
default_nb
# removing these two lines, this test passed
term_param
=
{
"term"
:
{
"term"
:
{
"values"
:
[
i
for
i
in
range
(
default_nb
//
2
)]}}}
expr
=
{
"must"
:
[
gen_default_vector_expr
(
default_query
),
term_param
]}
...
...
@@ -1255,8 +1241,7 @@ class TestSearchDSL(object):
assert
len
(
res
[
0
])
==
default_top_k
connect
.
drop_collection
(
collection_term
)
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_term_one_field_not_existed"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_term_one_field_not_existed
(
self
,
connect
,
collection
):
'''
...
...
@@ -1278,7 +1263,6 @@ class TestSearchDSL(object):
"""
# PASS
# TODO
def
test_query_range_key_error
(
self
,
connect
,
collection
):
'''
method: build query with range key error
...
...
@@ -1298,7 +1282,6 @@ class TestSearchDSL(object):
return
request
.
param
# PASS
# TODO
@
pytest
.
mark
.
level
(
2
)
def
test_query_range_wrong_format
(
self
,
connect
,
collection
,
get_invalid_range
):
'''
...
...
@@ -1366,8 +1349,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
default_top_k
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_range_one_field_not_existed"
)
# PASS
def
test_query_range_one_field_not_existed
(
self
,
connect
,
collection
):
'''
method: build query with two fields ranges, one of fields not existed
...
...
@@ -1387,10 +1369,7 @@ class TestSearchDSL(object):
************************************************************************
"""
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_multi_term_has_common"
)
@
pytest
.
mark
.
level
(
2
)
# PASS
def
test_query_multi_term_has_common
(
self
,
connect
,
collection
):
'''
method: build query with multi term with same field, and values has common
...
...
@@ -1405,9 +1384,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
default_top_k
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_multi_term_no_common"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_multi_term_no_common
(
self
,
connect
,
collection
):
'''
...
...
@@ -1423,9 +1400,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
0
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_multi_term_different_fields"
)
# PASS
def
test_query_multi_term_different_fields
(
self
,
connect
,
collection
):
'''
method: build query with multi range with same field, and ranges no common
...
...
@@ -1441,9 +1416,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
0
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_single_term_multi_fields"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_single_term_multi_fields
(
self
,
connect
,
collection
):
'''
...
...
@@ -1459,9 +1432,7 @@ class TestSearchDSL(object):
with
pytest
.
raises
(
Exception
)
as
e
:
res
=
connect
.
search
(
collection
,
query
)
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_multi_range_has_common"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_multi_range_has_common
(
self
,
connect
,
collection
):
'''
...
...
@@ -1477,9 +1448,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
default_top_k
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_multi_range_no_common"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_multi_range_no_common
(
self
,
connect
,
collection
):
'''
...
...
@@ -1495,9 +1464,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
0
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_multi_range_different_fields"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_multi_range_different_fields
(
self
,
connect
,
collection
):
'''
...
...
@@ -1513,9 +1480,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
0
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_single_range_multi_fields"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_single_range_multi_fields
(
self
,
connect
,
collection
):
'''
...
...
@@ -1537,9 +1502,7 @@ class TestSearchDSL(object):
******************************************************************
"""
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_single_term_range_has_common"
)
# PASS
@
pytest
.
mark
.
level
(
2
)
def
test_query_single_term_range_has_common
(
self
,
connect
,
collection
):
'''
...
...
@@ -1555,9 +1518,7 @@ class TestSearchDSL(object):
assert
len
(
res
)
==
nq
assert
len
(
res
[
0
])
==
default_top_k
# DOG: TODO TRC
# TODO
@
pytest
.
mark
.
skip
(
"query_single_term_range_no_common"
)
# PASS
def
test_query_single_term_range_no_common
(
self
,
connect
,
collection
):
'''
method: build query with single term single range
...
...
@@ -1579,7 +1540,6 @@ class TestSearchDSL(object):
"""
# PASS
# TODO
def
test_query_multi_vectors_same_field
(
self
,
connect
,
collection
):
'''
method: build query with two vectors same field
...
...
@@ -1616,8 +1576,7 @@ class TestSearchDSLBools(object):
with
pytest
.
raises
(
Exception
)
as
e
:
res
=
connect
.
search
(
collection
,
query
)
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_should_only_term"
)
# PASS
def
test_query_should_only_term
(
self
,
connect
,
collection
):
'''
method: build query without must, with should.term instead
...
...
@@ -1628,8 +1587,7 @@ class TestSearchDSLBools(object):
with
pytest
.
raises
(
Exception
)
as
e
:
res
=
connect
.
search
(
collection
,
query
)
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_should_only_vector"
)
# PASS
def
test_query_should_only_vector
(
self
,
connect
,
collection
):
'''
method: build query without must, with should.vector instead
...
...
@@ -1640,8 +1598,7 @@ class TestSearchDSLBools(object):
with
pytest
.
raises
(
Exception
)
as
e
:
res
=
connect
.
search
(
collection
,
query
)
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_must_not_only_term"
)
# PASS
def
test_query_must_not_only_term
(
self
,
connect
,
collection
):
'''
method: build query without must, with must_not.term instead
...
...
@@ -1652,8 +1609,7 @@ class TestSearchDSLBools(object):
with
pytest
.
raises
(
Exception
)
as
e
:
res
=
connect
.
search
(
collection
,
query
)
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_must_not_vector"
)
# PASS
def
test_query_must_not_vector
(
self
,
connect
,
collection
):
'''
method: build query without must, with must_not.vector instead
...
...
@@ -1664,8 +1620,7 @@ class TestSearchDSLBools(object):
with
pytest
.
raises
(
Exception
)
as
e
:
res
=
connect
.
search
(
collection
,
query
)
# DOG: TODO INVALID DSL
@
pytest
.
mark
.
skip
(
"query_must_should"
)
# PASS
def
test_query_must_should
(
self
,
connect
,
collection
):
'''
method: build query must, and with should.term
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录