| 1 | #include "duckdb/function/scalar/string_functions.hpp" |
| 2 | |
| 3 | #include "duckdb/common/exception.hpp" |
| 4 | #include "duckdb/common/vector_operations/vector_operations.hpp" |
| 5 | #include "duckdb/common/vector_operations/unary_executor.hpp" |
| 6 | #include "duckdb/planner/expression/bound_function_expression.hpp" |
| 7 | #include "duckdb/common/vector_operations/unary_executor.hpp" |
| 8 | |
| 9 | #include <ctype.h> |
| 10 | #include <cstring> |
| 11 | |
| 12 | using namespace std; |
| 13 | |
| 14 | namespace duckdb { |
| 15 | |
| 16 | static bool contains_strstr(const string_t &str, const string_t &pattern); |
| 17 | |
| 18 | struct ContainsOperator { |
| 19 | template <class TA, class TB, class TR> static inline TR Operation(TA left, TB right) { |
| 20 | return contains_strstr(left, right); |
| 21 | } |
| 22 | }; |
| 23 | |
| 24 | static bool contains_strstr(const string_t &str, const string_t &pattern) { |
| 25 | auto str_data = str.GetData(); |
| 26 | auto patt_data = pattern.GetData(); |
| 27 | return (strstr(str_data, patt_data) != nullptr); |
| 28 | } |
| 29 | |
| 30 | ScalarFunction ContainsFun::GetFunction() { |
| 31 | return ScalarFunction("contains" , // name of the function |
| 32 | {SQLType::VARCHAR, SQLType::VARCHAR}, // argument list |
| 33 | SQLType::BOOLEAN, // return type |
| 34 | ScalarFunction::BinaryFunction<string_t, string_t, bool, ContainsOperator, true>); |
| 35 | } |
| 36 | |
| 37 | void ContainsFun::RegisterFunction(BuiltinFunctions &set) { |
| 38 | set.AddFunction(GetFunction()); |
| 39 | } |
| 40 | |
| 41 | } // namespace duckdb |
| 42 | |