mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 06:55:27 +08:00
3628593d20
OSPP 2024 project: https://summer-ospp.ac.cn/org/prodetail/247410235?list=org&navpage=org Solutions: - parser (planparserv2) - add CallExpr in planparserv2/Plan.g4 - update parser_visitor and show_visitor - grpc protobuf - add CallExpr in plan.proto - execution (`core/src/exec`) - add `CallExpr` `ValueExpr` and `ColumnExpr` (both logical and physical) for function call and function parameters - function factory (`core/src/exec/expression/function`) - create a global hashmap when starting milvus (see server.go) - the global hashmap stores function signatures and their function pointers, the CallExpr in execution engine can get the function pointer by function signature. - custom functions - empty(string) - starts_with(string, string) - add cpp/go unittests and E2E tests closes: #36559 Signed-off-by: Yinzuo Jiang <jiangyinzuo@foxmail.com>
109 lines
3.4 KiB
C++
109 lines
3.4 KiB
C++
// Licensed to the LF AI & Data foundation under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you 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 <memory>
|
|
#include <mutex>
|
|
|
|
#include "common/init_c.h"
|
|
|
|
#include <string>
|
|
#include "common/Slice.h"
|
|
#include "common/Common.h"
|
|
#include "common/Tracer.h"
|
|
#include "log/Log.h"
|
|
|
|
std::once_flag flag1, flag2, flag3, flag4, flag5, flag6;
|
|
std::once_flag traceFlag;
|
|
|
|
void
|
|
InitIndexSliceSize(const int64_t size) {
|
|
std::call_once(
|
|
flag1, [](int64_t size) { milvus::SetIndexSliceSize(size); }, size);
|
|
}
|
|
|
|
void
|
|
InitHighPriorityThreadCoreCoefficient(const int64_t value) {
|
|
std::call_once(
|
|
flag2,
|
|
[](int64_t value) {
|
|
milvus::SetHighPriorityThreadCoreCoefficient(value);
|
|
},
|
|
value);
|
|
}
|
|
|
|
void
|
|
InitMiddlePriorityThreadCoreCoefficient(const int64_t value) {
|
|
std::call_once(
|
|
flag4,
|
|
[](int64_t value) {
|
|
milvus::SetMiddlePriorityThreadCoreCoefficient(value);
|
|
},
|
|
value);
|
|
}
|
|
|
|
void
|
|
InitLowPriorityThreadCoreCoefficient(const int64_t value) {
|
|
std::call_once(
|
|
flag5,
|
|
[](int64_t value) {
|
|
milvus::SetLowPriorityThreadCoreCoefficient(value);
|
|
},
|
|
value);
|
|
}
|
|
|
|
void
|
|
InitCpuNum(const int value) {
|
|
std::call_once(
|
|
flag3, [](int value) { milvus::SetCpuNum(value); }, value);
|
|
}
|
|
|
|
void
|
|
InitDefaultExprEvalBatchSize(int64_t val) {
|
|
std::call_once(
|
|
flag6,
|
|
[](int val) { milvus::SetDefaultExecEvalExprBatchSize(val); },
|
|
val);
|
|
}
|
|
|
|
void
|
|
InitTrace(CTraceConfig* config) {
|
|
auto traceConfig = milvus::tracer::TraceConfig{config->exporter,
|
|
config->sampleFraction,
|
|
config->jaegerURL,
|
|
config->otlpEndpoint,
|
|
config->otlpMethod,
|
|
config->oltpSecure,
|
|
config->nodeID};
|
|
std::call_once(
|
|
traceFlag,
|
|
[](const milvus::tracer::TraceConfig& c) {
|
|
milvus::tracer::initTelemetry(c);
|
|
},
|
|
traceConfig);
|
|
}
|
|
|
|
void
|
|
SetTrace(CTraceConfig* config) {
|
|
auto traceConfig = milvus::tracer::TraceConfig{config->exporter,
|
|
config->sampleFraction,
|
|
config->jaegerURL,
|
|
config->otlpEndpoint,
|
|
config->otlpMethod,
|
|
config->oltpSecure,
|
|
config->nodeID};
|
|
milvus::tracer::initTelemetry(traceConfig);
|
|
}
|