mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-04 18:02:08 +08:00
a602171d06
#28165 Signed-off-by: luzhang <luzhang@zilliz.com> Co-authored-by: luzhang <luzhang@zilliz.com>
146 lines
4.3 KiB
C++
146 lines
4.3 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 <boost/dynamic_bitset.hpp>
|
|
#include <memory>
|
|
|
|
#include "Plan.h"
|
|
#include "PlanNode.h"
|
|
#include "common/Schema.h"
|
|
#include "pb/plan.pb.h"
|
|
#include "plan/PlanNode.h"
|
|
|
|
namespace milvus::query {
|
|
|
|
class ProtoParser {
|
|
public:
|
|
explicit ProtoParser(const Schema& schema) : schema(schema) {
|
|
}
|
|
|
|
// ExprPtr
|
|
// ExprFromProto(const proto::plan::Expr& expr_proto);
|
|
|
|
ExprPtr
|
|
ParseBinaryArithOpEvalRangeExpr(
|
|
const proto::plan::BinaryArithOpEvalRangeExpr& expr_pb);
|
|
|
|
ExprPtr
|
|
ParseUnaryRangeExpr(const proto::plan::UnaryRangeExpr& expr_pb);
|
|
|
|
ExprPtr
|
|
ParseBinaryRangeExpr(const proto::plan::BinaryRangeExpr& expr_pb);
|
|
|
|
ExprPtr
|
|
ParseCompareExpr(const proto::plan::CompareExpr& expr_pb);
|
|
|
|
ExprPtr
|
|
ParseTermExpr(const proto::plan::TermExpr& expr_pb);
|
|
|
|
ExprPtr
|
|
ParseUnaryExpr(const proto::plan::UnaryExpr& expr_pb);
|
|
|
|
ExprPtr
|
|
ParseBinaryExpr(const proto::plan::BinaryExpr& expr_pb);
|
|
|
|
ExprPtr
|
|
ParseExistExpr(const proto::plan::ExistsExpr& expr_pb);
|
|
|
|
ExprPtr
|
|
ParseJsonContainsExpr(const proto::plan::JSONContainsExpr& expr_pb);
|
|
|
|
ExprPtr
|
|
ParseExpr(const proto::plan::Expr& expr_pb);
|
|
|
|
std::unique_ptr<VectorPlanNode>
|
|
PlanNodeFromProto(const proto::plan::PlanNode& plan_node_proto);
|
|
|
|
std::unique_ptr<RetrievePlanNode>
|
|
RetrievePlanNodeFromProto(const proto::plan::PlanNode& plan_node_proto);
|
|
|
|
std::unique_ptr<Plan>
|
|
CreatePlan(const proto::plan::PlanNode& plan_node_proto);
|
|
|
|
std::unique_ptr<RetrievePlan>
|
|
CreateRetrievePlan(const proto::plan::PlanNode& plan_node_proto);
|
|
|
|
expr::TypedExprPtr
|
|
ParseUnaryRangeExprs(const proto::plan::UnaryRangeExpr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
ParseExprs(const proto::plan::Expr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
ParseBinaryArithOpEvalRangeExprs(
|
|
const proto::plan::BinaryArithOpEvalRangeExpr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
ParseBinaryRangeExprs(const proto::plan::BinaryRangeExpr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
ParseCompareExprs(const proto::plan::CompareExpr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
ParseTermExprs(const proto::plan::TermExpr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
ParseUnaryExprs(const proto::plan::UnaryExpr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
ParseBinaryExprs(const proto::plan::BinaryExpr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
ParseExistExprs(const proto::plan::ExistsExpr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
ParseJsonContainsExprs(const proto::plan::JSONContainsExpr& expr_pb);
|
|
|
|
expr::TypedExprPtr
|
|
CreateAlwaysTrueExprs();
|
|
|
|
private:
|
|
const Schema& schema;
|
|
};
|
|
|
|
} // namespace milvus::query
|
|
//
|
|
template <>
|
|
struct fmt::formatter<milvus::proto::plan::GenericValue::ValCase>
|
|
: formatter<string_view> {
|
|
auto
|
|
format(milvus::proto::plan::GenericValue::ValCase c,
|
|
format_context& ctx) const {
|
|
string_view name = "unknown";
|
|
switch (c) {
|
|
case milvus::proto::plan::GenericValue::ValCase::kBoolVal:
|
|
name = "kBoolVal";
|
|
break;
|
|
case milvus::proto::plan::GenericValue::ValCase::kInt64Val:
|
|
name = "kInt64Val";
|
|
break;
|
|
case milvus::proto::plan::GenericValue::ValCase::kFloatVal:
|
|
name = "kFloatVal";
|
|
break;
|
|
case milvus::proto::plan::GenericValue::ValCase::kStringVal:
|
|
name = "kStringVal";
|
|
break;
|
|
case milvus::proto::plan::GenericValue::ValCase::kArrayVal:
|
|
name = "kArrayVal";
|
|
break;
|
|
case milvus::proto::plan::GenericValue::ValCase::VAL_NOT_SET:
|
|
name = "VAL_NOT_SET";
|
|
break;
|
|
}
|
|
return formatter<string_view>::format(name, ctx);
|
|
}
|
|
};
|