mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 15:05:31 +08:00
127 lines
4.1 KiB
C++
127 lines
4.1 KiB
C++
// Licensed to the Apache Software Foundation (ASF) 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 "utils/StringHelpFunctions.h"
|
|
|
|
#include <string>
|
|
|
|
namespace milvus {
|
|
namespace server {
|
|
|
|
void
|
|
StringHelpFunctions::TrimStringBlank(std::string& string) {
|
|
if (!string.empty()) {
|
|
static std::string s_format(" \n\r\t");
|
|
string.erase(0, string.find_first_not_of(s_format));
|
|
string.erase(string.find_last_not_of(s_format) + 1);
|
|
}
|
|
}
|
|
|
|
void
|
|
StringHelpFunctions::TrimStringQuote(std::string& string, const std::string& qoute) {
|
|
if (!string.empty()) {
|
|
string.erase(0, string.find_first_not_of(qoute));
|
|
string.erase(string.find_last_not_of(qoute) + 1);
|
|
}
|
|
}
|
|
|
|
Status
|
|
StringHelpFunctions::SplitStringByDelimeter(const std::string& str, const std::string& delimeter,
|
|
std::vector<std::string>& result) {
|
|
if (str.empty()) {
|
|
return Status::OK();
|
|
}
|
|
|
|
size_t last = 0;
|
|
size_t index = str.find_first_of(delimeter, last);
|
|
while (index != std::string::npos) {
|
|
result.emplace_back(str.substr(last, index - last));
|
|
last = index + 1;
|
|
index = str.find_first_of(delimeter, last);
|
|
}
|
|
if (index - last > 0) {
|
|
std::string temp = str.substr(last);
|
|
result.emplace_back(temp);
|
|
}
|
|
|
|
return Status::OK();
|
|
}
|
|
|
|
Status
|
|
StringHelpFunctions::SplitStringByQuote(const std::string& str, const std::string& delimeter, const std::string& quote,
|
|
std::vector<std::string>& result) {
|
|
if (quote.empty()) {
|
|
return SplitStringByDelimeter(str, delimeter, result);
|
|
}
|
|
|
|
size_t last = 0;
|
|
size_t index = str.find_first_of(quote, last);
|
|
if (index == std::string::npos) {
|
|
return SplitStringByDelimeter(str, delimeter, result);
|
|
}
|
|
|
|
std::string process_str = str;
|
|
while (index != std::string::npos) {
|
|
std::string prefix = process_str.substr(last, index - last);
|
|
std::string append_prefix;
|
|
if (!prefix.empty()) {
|
|
std::vector<std::string> prefix_split;
|
|
SplitStringByDelimeter(prefix, delimeter, prefix_split);
|
|
for (size_t i = 0; i < prefix_split.size() - 1; i++) {
|
|
result.push_back(prefix_split[i]);
|
|
}
|
|
append_prefix = prefix_split[prefix_split.size() - 1];
|
|
}
|
|
last = index + 1;
|
|
std::string postfix = process_str.substr(last);
|
|
index = postfix.find_first_of(quote, 0);
|
|
if (index == std::string::npos) {
|
|
return Status(SERVER_UNEXPECTED_ERROR, "");
|
|
}
|
|
std::string quoted_text = postfix.substr(0, index);
|
|
append_prefix += quoted_text;
|
|
|
|
last = index + 1;
|
|
index = postfix.find_first_of(delimeter, last);
|
|
if (index != std::string::npos) {
|
|
if (index > last) {
|
|
append_prefix += postfix.substr(last, index - last);
|
|
}
|
|
} else {
|
|
append_prefix += postfix.substr(last);
|
|
}
|
|
result.emplace_back(append_prefix);
|
|
|
|
if (last == postfix.length()) {
|
|
return Status::OK();
|
|
}
|
|
|
|
process_str = postfix.substr(index + 1);
|
|
last = 0;
|
|
index = process_str.find_first_of(quote, last);
|
|
}
|
|
|
|
if (!process_str.empty()) {
|
|
return SplitStringByDelimeter(process_str, delimeter, result);
|
|
}
|
|
|
|
return Status::OK();
|
|
}
|
|
|
|
} // namespace server
|
|
} // namespace milvus
|