fix: handle input array for extract_domain

Author: hatamiarash7

src/functions/extract_domain.cpp

@@ -11,18 +11,23 @@ namespace duckdb
     {
         // Extract the input from the arguments
         auto &input_vector = args.data[0];
-        auto input         = input_vector.GetValue (0).ToString ();
+        auto result_data   = FlatVector::GetData<string_t> (result);
 
-        if (input.empty ())
+        for (idx_t i = 0; i < args.size (); i++)
         {
-            result.SetValue (0, Value (""));
-            return;
-        }
-
-        // Extract the domain using the utility function
-        auto domain = netquack::ExtractDomain (state, input);
+            auto input = input_vector.GetValue (i).ToString ();
 
-        result.SetValue (0, Value (domain));
+            try
+            {
+                // Extract the domain using the utility function
+                auto domain    = netquack::ExtractDomain (state, input);
+                result_data[i] = StringVector::AddString (result, domain);
+            }
+            catch (const std::exception &e)
+            {
+                result_data[i] = "Error extracting domain: " + std::string (e.what ());
+            }
+        }
     }
 
     namespace netquack

test/sql/extract_domain.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_domain('example.com');
 ----
@@ -172,4 +175,17 @@ SELECT extract_domain('http:/example.com.ac/path');
 query I
 SELECT extract_domain('http:/example.com.ac:443/path');
 ----
-(empty)
+(empty)
+
+query I
+SELECT extract_domain(uri) from uri_list;
+----
+example.com
+example.com.ac
+example.com.co
+example.com
+example.com
+example.com.ac
+example.com
+example.com
+example.com.ac