mirror of https://github.com/milvus-io/milvus.git
refine test code
Former-commit-id: 25266f1e3574d2ffa42ab8299cc8e5a4ac5b651apull/191/head
parent
dbf9de9a62
commit
ae5e672061
|
@ -111,7 +111,52 @@ namespace {
|
||||||
}
|
}
|
||||||
|
|
||||||
void ClientTest::LoopTest() {
|
void ClientTest::LoopTest() {
|
||||||
|
server::TimeRecorder rc("LoopTest");
|
||||||
|
|
||||||
|
std::string address, protocol;
|
||||||
|
int32_t port = 0;
|
||||||
|
GetServerAddress(address, port, protocol);
|
||||||
|
client::ClientSession session(address, port, protocol);
|
||||||
|
|
||||||
|
rc.Record("connection");
|
||||||
|
|
||||||
|
//add group
|
||||||
|
VecGroup group;
|
||||||
|
group.id = "loop_group";
|
||||||
|
group.dimension = VEC_DIMENSION;
|
||||||
|
group.index_type = 0;
|
||||||
|
session.interface()->add_group(group);
|
||||||
|
rc.Record("add group");
|
||||||
|
|
||||||
|
const int64_t batch = 10000;
|
||||||
|
for(int64_t i = 0; i < 1000; i++) {
|
||||||
|
VecTensorList tensor_list;
|
||||||
|
VecBinaryTensorList bin_tensor_list;
|
||||||
|
BuildVectors(i*batch, (i+1)*batch, tensor_list, bin_tensor_list);
|
||||||
|
rc.Record("build batch no." + std::to_string(i));
|
||||||
|
|
||||||
|
session.interface()->add_binary_vector_batch(group.id, bin_tensor_list);
|
||||||
|
rc.Record("add batch no." + std::to_string(i));
|
||||||
|
|
||||||
|
sleep(1);
|
||||||
|
rc.Record("sleep 1 second");
|
||||||
|
|
||||||
|
VecTensor tensor;
|
||||||
|
for (int32_t k = 0; k < VEC_DIMENSION; k++) {
|
||||||
|
tensor.tensor.push_back((double) (k + i*666));
|
||||||
|
}
|
||||||
|
|
||||||
|
//do search
|
||||||
|
VecSearchResult res;
|
||||||
|
VecSearchFilter filter;
|
||||||
|
session.interface()->search_vector(res, group.id, 10, tensor, filter);
|
||||||
|
rc.Record("search finish");
|
||||||
|
|
||||||
|
std::cout << "Search result: " << std::endl;
|
||||||
|
for(VecSearchResultItem& item : res.result_list) {
|
||||||
|
std::cout << "\t" << item.uid << std::endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST(AddVector, CLIENT_TEST) {
|
TEST(AddVector, CLIENT_TEST) {
|
||||||
|
|
Loading…
Reference in New Issue