fix: handle input array for extract_query_string

Author: hatamiarash7

src/functions/extract_query.cpp

@@ -7,15 +7,25 @@ namespace duckdb
     // Function to extract the query string from a URL
     void ExtractQueryStringFunction (DataChunk &args, ExpressionState &state, Vector &result)
     {
-        // Extract the URL from the input
-        auto &url_vector = args.data[0];
-        auto url         = url_vector.GetValue (0).ToString ();
+        // Extract the input from the arguments
+        auto &input_vector = args.data[0];
+        auto result_data   = FlatVector::GetData<string_t> (result);
 
-        // Extract the query string
-        auto query_string = netquack::ExtractQueryString (url);
+        for (idx_t i = 0; i < args.size (); i++)
+        {
+            auto input = input_vector.GetValue (i).ToString ();
 
-        // Set the result
-        result.SetValue (0, Value (query_string));
+            try
+            {
+                // Extract the query string using the utility function
+                auto query_string = netquack::ExtractQueryString (input);
+                result_data[i]    = StringVector::AddString (result, query_string);
+            }
+            catch (const std::exception &e)
+            {
+                result_data[i] = "Error extracting query string: " + std::string (e.what ());
+            }
+        };
     }
 
     namespace netquack

test/sql/extract_query.test

@@ -4,6 +4,9 @@
 
 require netquack
 
+statement ok
+CREATE TABLE uri_list AS SELECT * FROM read_csv('test/data/examples.csv', header=false, columns={'uri': 'VARCHAR'});
+
 query I
 SELECT extract_query_string('http://example.com.ac?a=1');
 ----
@@ -28,3 +31,16 @@ query I
 SELECT extract_query_string('http://example.com.ac/path/?a=1');
 ----
 a=1
+
+query I
+SELECT extract_query_string(uri) from uri_list;
+----
+(empty)
+(empty)
+(empty)
+(empty)
+(empty)
+(empty)
+(empty)
+(empty)
+a=1&b=2&