Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
milvus
提交
5b26d194
milvus
项目概览
BaiXuePrincess
/
milvus
与 Fork 源项目一致
从无法访问的项目Fork
通知
7
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
milvus
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
5b26d194
编写于
8月 16, 2019
作者:
Y
Yu Kun
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add unittest
Former-commit-id: 89139a24caab971fa16b27b814ca4731efd0ea20
上级
fabcbe91
变更
7
隐藏空白更改
内联
并排
Showing
7 changed file
with
22 addition
and
10 deletion
+22
-10
cpp/src/db/insert/MemManagerImpl.cpp
cpp/src/db/insert/MemManagerImpl.cpp
+4
-2
cpp/src/db/insert/MemTable.cpp
cpp/src/db/insert/MemTable.cpp
+1
-1
cpp/src/db/insert/MemTable.h
cpp/src/db/insert/MemTable.h
+1
-1
cpp/src/db/insert/VectorSource.cpp
cpp/src/db/insert/VectorSource.cpp
+3
-3
cpp/src/db/insert/VectorSource.h
cpp/src/db/insert/VectorSource.h
+1
-1
cpp/src/server/grpc_impl/GrpcRequestTask.cpp
cpp/src/server/grpc_impl/GrpcRequestTask.cpp
+9
-2
cpp/unittest/metrics/prometheus_test.cpp
cpp/unittest/metrics/prometheus_test.cpp
+3
-0
未找到文件。
cpp/src/db/insert/MemManagerImpl.cpp
浏览文件 @
5b26d194
...
...
@@ -42,9 +42,11 @@ Status MemManagerImpl::InsertVectorsNoLock(const std::string &table_id,
MemTablePtr
mem
=
GetMemByTable
(
table_id
);
VectorSource
::
Ptr
source
=
std
::
make_shared
<
VectorSource
>
(
n
,
vectors
);
auto
status
=
mem
->
Add
(
source
);
auto
status
=
mem
->
Add
(
source
,
vector_ids
);
if
(
status
.
ok
())
{
vector_ids
=
source
->
GetVectorIds
();
if
(
vector_ids
.
empty
())
{
vector_ids
=
source
->
GetVectorIds
();
}
}
return
status
;
}
...
...
cpp/src/db/insert/MemTable.cpp
浏览文件 @
5b26d194
...
...
@@ -15,7 +15,7 @@ MemTable::MemTable(const std::string &table_id,
}
Status
MemTable
::
Add
(
VectorSource
::
Ptr
&
source
)
{
Status
MemTable
::
Add
(
VectorSource
::
Ptr
&
source
,
IDNumbers
&
vector_ids
)
{
while
(
!
source
->
AllAdded
())
{
...
...
cpp/src/db/insert/MemTable.h
浏览文件 @
5b26d194
...
...
@@ -21,7 +21,7 @@ class MemTable {
MemTable
(
const
std
::
string
&
table_id
,
const
std
::
shared_ptr
<
meta
::
Meta
>
&
meta
,
const
Options
&
options
);
Status
Add
(
VectorSource
::
Ptr
&
source
);
Status
Add
(
VectorSource
::
Ptr
&
source
,
IDNumbers
&
vector_ids
);
void
GetCurrentMemTableFile
(
MemTableFile
::
Ptr
&
mem_table_file
);
...
...
cpp/src/db/insert/VectorSource.cpp
浏览文件 @
5b26d194
...
...
@@ -12,9 +12,9 @@ namespace engine {
VectorSource
::
VectorSource
(
const
size_t
&
n
,
const
float
*
vectors
)
:
n_
(
n
),
vectors_
(
vectors
),
id_generator_
(
new
SimpleIDGenerator
())
{
n_
(
n
),
vectors_
(
vectors
),
id_generator_
(
std
::
make_shared
<
SimpleIDGenerator
>
())
{
current_num_vectors_added
=
0
;
}
...
...
cpp/src/db/insert/VectorSource.h
浏览文件 @
5b26d194
...
...
@@ -37,7 +37,7 @@ class VectorSource {
size_t
current_num_vectors_added
;
IDGenerator
*
id_generator_
;
std
::
shared_ptr
<
IDGenerator
>
id_generator_
;
};
//VectorSource
...
...
cpp/src/server/grpc_impl/GrpcRequestTask.cpp
浏览文件 @
5b26d194
...
...
@@ -381,9 +381,9 @@ InsertTask::InsertTask(const ::milvus::grpc::InsertParam &insert_param,
}
BaseTaskPtr
InsertTask
::
Create
(
const
::
milvus
::
grpc
::
InsertParam
&
insert_
infos
,
InsertTask
::
Create
(
const
::
milvus
::
grpc
::
InsertParam
&
insert_
param
,
::
milvus
::
grpc
::
VectorIds
&
record_ids
)
{
return
std
::
shared_ptr
<
GrpcBaseTask
>
(
new
InsertTask
(
insert_
infos
,
record_ids
));
return
std
::
shared_ptr
<
GrpcBaseTask
>
(
new
InsertTask
(
insert_
param
,
record_ids
));
}
ServerError
...
...
@@ -400,6 +400,13 @@ InsertTask::OnExecute() {
return
SetError
(
SERVER_INVALID_ROWRECORD_ARRAY
,
"Row record array is empty"
);
}
if
(
!
record_ids_
.
vector_id_array
().
empty
())
{
if
(
record_ids_
.
vector_id_array
().
size
()
!=
insert_param_
.
row_record_array_size
())
{
return
SetError
(
SERVER_ILLEGAL_VECTOR_ID
,
"Size of vector ids is not equal to row record array size"
);
}
}
//step 2: check table existence
engine
::
meta
::
TableSchema
table_info
;
table_info
.
table_id_
=
insert_param_
.
table_name
();
...
...
cpp/unittest/metrics/prometheus_test.cpp
浏览文件 @
5b26d194
...
...
@@ -54,4 +54,7 @@ TEST(PrometheusTest, PROMETHEUS_TEST){
instance
.
ConnectionGaugeDecrement
();
instance
.
KeepingAliveCounterIncrement
();
instance
.
OctetsSet
();
instance
.
CPUCoreUsagePercentSet
();
instance
.
GPUTemperature
();
instance
.
CPUTemperature
();
}
\ No newline at end of file
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录