mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-30 23:45:28 +08:00
82f4c289eb
* modify proto Signed-off-by: groot <yihua.mo@zilliz.com> * modify c sdk Signed-off-by: groot <yihua.mo@zilliz.com> * rewrite search by id Signed-off-by: groot <yihua.mo@zilliz.com> * change cases Signed-off-by: zw <zw@zilliz.com> * fix cases Signed-off-by: zw <zw@zilliz.com> * fix wal case Signed-off-by: zw <zw@zilliz.com> * fix search by id Signed-off-by: groot <yihua.mo@zilliz.com> * modify sdk Signed-off-by: groot <yihua.mo@zilliz.com> * refine code Signed-off-by: groot <yihua.mo@zilliz.com> * refine code Signed-off-by: groot <yihua.mo@zilliz.com> * changelog Signed-off-by: groot <yihua.mo@zilliz.com> * add search_by_ids case Signed-off-by: zw <zw@zilliz.com> * disable test_search)by_ids Signed-off-by: zw <zw@zilliz.com> Co-authored-by: zw <zw@zilliz.com> Co-authored-by: del-zhenwu <56623710+del-zhenwu@users.noreply.github.com>
125 lines
3.6 KiB
C++
125 lines
3.6 KiB
C++
// Copyright (C) 2019-2020 Zilliz. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software distributed under the License
|
|
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
|
|
// or implied. See the License for the specific language governing permissions and limitations under the License.
|
|
|
|
#pragma once
|
|
|
|
#include "include/MilvusApi.h"
|
|
#include "grpc-gen/gen-milvus/milvus.grpc.pb.h"
|
|
//#include "grpc/gen-status/status.grpc.pb.h"
|
|
|
|
#include <chrono>
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include <random>
|
|
#include <string>
|
|
#include <thread>
|
|
|
|
#include <grpc/grpc.h>
|
|
#include <grpcpp/channel.h>
|
|
#include <grpcpp/client_context.h>
|
|
#include <grpcpp/create_channel.h>
|
|
#include <grpcpp/security/credentials.h>
|
|
|
|
namespace milvus {
|
|
class GrpcClient {
|
|
public:
|
|
explicit GrpcClient(std::shared_ptr<::grpc::Channel>& channel);
|
|
|
|
virtual ~GrpcClient();
|
|
|
|
Status
|
|
CreateCollection(const grpc::CollectionSchema& collection_schema);
|
|
|
|
bool
|
|
HasCollection(const grpc::CollectionName& collection_name, Status& status);
|
|
|
|
Status
|
|
DropCollection(const grpc::CollectionName& collection_name);
|
|
|
|
Status
|
|
CreateIndex(const grpc::IndexParam& index_param);
|
|
|
|
Status
|
|
Insert(const grpc::InsertParam& insert_param, grpc::VectorIds& vector_ids);
|
|
|
|
Status
|
|
GetVectorsByID(const grpc::VectorsIdentity& vectors_identity, ::milvus::grpc::VectorsData& vectors_data);
|
|
|
|
Status
|
|
GetIDsInSegment(const grpc::GetVectorIDsParam& param, grpc::VectorIds& vector_ids);
|
|
|
|
Status
|
|
Search(const grpc::SearchParam& search_param, ::milvus::grpc::TopKQueryResult& topk_query_result);
|
|
|
|
Status
|
|
SearchByID(const grpc::SearchByIDParam& search_param, ::milvus::grpc::TopKQueryResult& topk_query_result);
|
|
|
|
Status
|
|
DescribeCollection(const std::string& collection_name, grpc::CollectionSchema& grpc_schema);
|
|
|
|
int64_t
|
|
CountCollection(grpc::CollectionName& collection_name, Status& status);
|
|
|
|
Status
|
|
ShowCollections(milvus::grpc::CollectionNameList& collection_name_list);
|
|
|
|
Status
|
|
ShowCollectionInfo(grpc::CollectionName& collection_name, grpc::CollectionInfo& collection_info);
|
|
|
|
Status
|
|
Cmd(const std::string& cmd, std::string& result);
|
|
|
|
Status
|
|
DeleteByID(grpc::DeleteByIDParam& delete_by_id_param);
|
|
|
|
Status
|
|
PreloadCollection(grpc::CollectionName& collection_name);
|
|
|
|
Status
|
|
DescribeIndex(grpc::CollectionName& collection_name, grpc::IndexParam& index_param);
|
|
|
|
Status
|
|
DropIndex(grpc::CollectionName& collection_name);
|
|
|
|
Status
|
|
CreatePartition(const grpc::PartitionParam& partition_param);
|
|
|
|
Status
|
|
ShowPartitions(const grpc::CollectionName& collection_name, grpc::PartitionList& partition_array) const;
|
|
|
|
Status
|
|
DropPartition(const ::milvus::grpc::PartitionParam& partition_param);
|
|
|
|
Status
|
|
Flush(const std::string& collection_name);
|
|
|
|
Status
|
|
Compact(milvus::grpc::CollectionName& collection_name);
|
|
|
|
Status
|
|
Disconnect();
|
|
|
|
/*******************************New Interface**********************************/
|
|
Status
|
|
CreateHybridCollection(milvus::grpc::Mapping& mapping);
|
|
|
|
Status
|
|
InsertEntities(milvus::grpc::HInsertParam& entities, milvus::grpc::HEntityIDs& ids);
|
|
|
|
Status
|
|
HybridSearch(milvus::grpc::HSearchParam& search_param, milvus::grpc::TopKQueryResult& result);
|
|
|
|
private:
|
|
std::unique_ptr<grpc::MilvusService::Stub> stub_;
|
|
};
|
|
|
|
} // namespace milvus
|