mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-31 07:55:38 +08:00
71 lines
2.5 KiB
C++
71 lines
2.5 KiB
C++
/*******************************************************************************
|
|
* Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
|
|
* Unauthorized copying of this file, via any medium is strictly prohibited.
|
|
* Proprietary and confidential.
|
|
******************************************************************************/
|
|
#pragma once
|
|
|
|
#include "MilvusApi.h"
|
|
#include "ThriftClient.h"
|
|
|
|
namespace milvus {
|
|
|
|
class ClientProxy : public Connection {
|
|
public:
|
|
// Implementations of the Connection interface
|
|
virtual Status Connect(const ConnectParam ¶m) override;
|
|
|
|
virtual Status Connect(const std::string &uri) override;
|
|
|
|
virtual Status Connected() const override;
|
|
|
|
virtual Status Disconnect() override;
|
|
|
|
virtual Status CreateTable(const TableSchema ¶m) override;
|
|
|
|
virtual bool HasTable(const std::string &table_name) override;
|
|
|
|
virtual Status DeleteTable(const std::string &table_name) override;
|
|
|
|
virtual Status DropTable(const std::string &table_name) override;
|
|
|
|
virtual Status BuildIndex(const std::string &table_name) override;
|
|
|
|
virtual Status AddVector(const std::string &table_name,
|
|
const std::vector<RowRecord> &record_array,
|
|
std::vector<int64_t> &id_array) override;
|
|
|
|
virtual Status InsertVector(const std::string &table_name,
|
|
const std::vector<RowRecord> &record_array,
|
|
std::vector<int64_t> &id_array) override;
|
|
|
|
virtual Status SearchVector(const std::string &table_name,
|
|
const std::vector<RowRecord> &query_record_array,
|
|
const std::vector<Range> &query_range_array,
|
|
int64_t topk,
|
|
std::vector<TopKQueryResult> &topk_query_result_array) override;
|
|
|
|
virtual Status DescribeTable(const std::string &table_name, TableSchema &table_schema) override;
|
|
|
|
virtual Status GetTableRowCount(const std::string &table_name, int64_t &row_count) override;
|
|
|
|
virtual Status ShowTables(std::vector<std::string> &table_array) override;
|
|
|
|
virtual std::string ClientVersion() const override;
|
|
|
|
virtual std::string ServerVersion() const override;
|
|
|
|
virtual std::string ServerStatus() const override;
|
|
|
|
private:
|
|
std::shared_ptr<ThriftClient>& ClientPtr() const;
|
|
|
|
bool IsConnected() const;
|
|
|
|
private:
|
|
mutable std::shared_ptr<ThriftClient> client_ptr;
|
|
bool connected_ = false;
|
|
};
|
|
|
|
}
|