feat: add func setrange()

wy1433 · Apr 24, 2024 · 7565895 · 7565895
1 parent a4f8a3a
commit 7565895
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 2 deletions.
diff --git a/include/expr/internal_functions.h b/include/expr/internal_functions.h
@@ -83,7 +83,7 @@ ExprValue field(const std::vector<ExprValue>& input);
 ExprValue quote(const std::vector<ExprValue>& input);
 ExprValue func_char(const std::vector<ExprValue>& input);
 ExprValue soundex(const std::vector<ExprValue>& input);
-
+ExprValue setrange(const std::vector<ExprValue>& input);
 
 // datetime functions
 ExprValue unix_timestamp(const std::vector<ExprValue>& input);

diff --git a/src/expr/fn_manager.cpp b/src/expr/fn_manager.cpp
@@ -171,7 +171,7 @@ void FunctionManager::register_operators() {
     register_object_ret("quote", quote, pb::STRING);
     register_object_ret("char", func_char, pb::STRING);
     register_object_ret("soundex", soundex, pb::STRING);
-
+    register_object_ret("setrange", setrange, pb::STRING);
 
     // date funcs
     register_object_ret("unix_timestamp", unix_timestamp, pb::INT64);

diff --git a/src/expr/internal_functions.cpp b/src/expr/internal_functions.cpp
@@ -900,6 +900,32 @@ ExprValue substring_index(const std::vector<ExprValue>& input) {
     return tmp;
 }
 
+ExprValue setrange(const std::vector<ExprValue>& input) {
+    if (input.size() != 3) {
+        return ExprValue::Null();
+    }
+    ExprValue tmp(pb::STRING);
+    std::string str = input[0].get_string();
+    int64_t offset = input[1].get_numberic<int64_t>();
+    std::string value = input[2].get_string();
+    if (offset < 0 || offset > UINT16_MAX) {
+        return ExprValue::Null();
+    }
+    if (offset > str.length()) {
+        tmp.str_val = str;
+        tmp.str_val += std::string(offset - str.length(), '\x00');
+        tmp.str_val += value;
+        return tmp;
+    }
+    tmp.str_val.append(str.begin(), str.begin() + offset);
+    tmp.str_val += value;
+    if (offset + value.length() >= str.length()) {
+        return tmp;
+    }
+    tmp.str_val.append(str.begin() + offset + value.length(), str.end());
+    return tmp;
+}
+
 ExprValue unix_timestamp(const std::vector<ExprValue>& input) {
     ExprValue tmp(pb::UINT32);
     if (input.size() == 0) {