mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
e1027e9e65
* #1240 Update license declaration of each files Signed-off-by: jinhai <hai.jin@zilliz.com> * #1240 Update CHANGELOG Signed-off-by: jinhai <hai.jin@zilliz.com>
189 lines
7.6 KiB
C++
189 lines
7.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.
|
|
|
|
#include "server/delivery/RequestHandler.h"
|
|
#include "server/delivery/RequestScheduler.h"
|
|
#include "server/delivery/request/BaseRequest.h"
|
|
#include "server/delivery/request/CmdRequest.h"
|
|
#include "server/delivery/request/CountTableRequest.h"
|
|
#include "server/delivery/request/CreateIndexRequest.h"
|
|
#include "server/delivery/request/CreatePartitionRequest.h"
|
|
#include "server/delivery/request/CreateTableRequest.h"
|
|
#include "server/delivery/request/DeleteByDateRequest.h"
|
|
#include "server/delivery/request/DescribeIndexRequest.h"
|
|
#include "server/delivery/request/DescribeTableRequest.h"
|
|
#include "server/delivery/request/DropIndexRequest.h"
|
|
#include "server/delivery/request/DropPartitionRequest.h"
|
|
#include "server/delivery/request/DropTableRequest.h"
|
|
#include "server/delivery/request/HasTableRequest.h"
|
|
#include "server/delivery/request/InsertRequest.h"
|
|
#include "server/delivery/request/PreloadTableRequest.h"
|
|
#include "server/delivery/request/SearchRequest.h"
|
|
#include "server/delivery/request/ShowPartitionsRequest.h"
|
|
#include "server/delivery/request/ShowTablesRequest.h"
|
|
|
|
namespace milvus {
|
|
namespace server {
|
|
|
|
Status
|
|
RequestHandler::CreateTable(const std::shared_ptr<Context>& context, const std::string& table_name, int64_t dimension,
|
|
int64_t index_file_size, int64_t metric_type) {
|
|
BaseRequestPtr request_ptr =
|
|
CreateTableRequest::Create(context, table_name, dimension, index_file_size, metric_type);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::HasTable(const std::shared_ptr<Context>& context, const std::string& table_name, bool& has_table) {
|
|
BaseRequestPtr request_ptr = HasTableRequest::Create(context, table_name, has_table);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::DropTable(const std::shared_ptr<Context>& context, const std::string& table_name) {
|
|
BaseRequestPtr request_ptr = DropTableRequest::Create(context, table_name);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::CreateIndex(const std::shared_ptr<Context>& context, const std::string& table_name, int64_t index_type,
|
|
int64_t nlist) {
|
|
BaseRequestPtr request_ptr = CreateIndexRequest::Create(context, table_name, index_type, nlist);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::Insert(const std::shared_ptr<Context>& context, const std::string& table_name,
|
|
engine::VectorsData& vectors, const std::string& partition_tag) {
|
|
BaseRequestPtr request_ptr = InsertRequest::Create(context, table_name, vectors, partition_tag);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::ShowTables(const std::shared_ptr<Context>& context, std::vector<std::string>& tables) {
|
|
BaseRequestPtr request_ptr = ShowTablesRequest::Create(context, tables);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::Search(const std::shared_ptr<Context>& context, const std::string& table_name,
|
|
const engine::VectorsData& vectors,
|
|
const std::vector<std::pair<std::string, std::string>>& range_list, int64_t topk, int64_t nprobe,
|
|
const std::vector<std::string>& partition_list, const std::vector<std::string>& file_id_list,
|
|
TopKQueryResult& result) {
|
|
BaseRequestPtr request_ptr = SearchRequest::Create(context, table_name, vectors, range_list, topk, nprobe,
|
|
partition_list, file_id_list, result);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::DescribeTable(const std::shared_ptr<Context>& context, const std::string& table_name,
|
|
TableSchema& table_schema) {
|
|
BaseRequestPtr request_ptr = DescribeTableRequest::Create(context, table_name, table_schema);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::CountTable(const std::shared_ptr<Context>& context, const std::string& table_name, int64_t& count) {
|
|
BaseRequestPtr request_ptr = CountTableRequest::Create(context, table_name, count);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::Cmd(const std::shared_ptr<Context>& context, const std::string& cmd, std::string& reply) {
|
|
BaseRequestPtr request_ptr = CmdRequest::Create(context, cmd, reply);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::DeleteByRange(const std::shared_ptr<Context>& context, const std::string& table_name,
|
|
const Range& range) {
|
|
BaseRequestPtr request_ptr = DeleteByDateRequest::Create(context, table_name, range);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::PreloadTable(const std::shared_ptr<Context>& context, const std::string& table_name) {
|
|
BaseRequestPtr request_ptr = PreloadTableRequest::Create(context, table_name);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::DescribeIndex(const std::shared_ptr<Context>& context, const std::string& table_name,
|
|
IndexParam& param) {
|
|
BaseRequestPtr request_ptr = DescribeIndexRequest::Create(context, table_name, param);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::DropIndex(const std::shared_ptr<Context>& context, const std::string& table_name) {
|
|
BaseRequestPtr request_ptr = DropIndexRequest::Create(context, table_name);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::CreatePartition(const std::shared_ptr<Context>& context, const std::string& table_name,
|
|
const std::string& partition_name, const std::string& tag) {
|
|
BaseRequestPtr request_ptr = CreatePartitionRequest::Create(context, table_name, partition_name, tag);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::ShowPartitions(const std::shared_ptr<Context>& context, const std::string& table_name,
|
|
std::vector<PartitionParam>& partitions) {
|
|
BaseRequestPtr request_ptr = ShowPartitionsRequest::Create(context, table_name, partitions);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
Status
|
|
RequestHandler::DropPartition(const std::shared_ptr<Context>& context, const std::string& table_name,
|
|
const std::string& partition_name, const std::string& tag) {
|
|
BaseRequestPtr request_ptr = DropPartitionRequest::Create(context, table_name, partition_name, tag);
|
|
RequestScheduler::ExecRequest(request_ptr);
|
|
|
|
return request_ptr->status();
|
|
}
|
|
|
|
} // namespace server
|
|
} // namespace milvus
|