Merge pull request #292 from jeffgabhart/pgsql/func-array-arg

Support PostgreSQL functions with integer and string array arguments.

Author: mythz

src/ServiceStack.OrmLite.PostgreSQL.Tests/OrmLiteExecuteProcedureTests.cs

@@ -0,0 +1,64 @@
+using System.Collections.Generic;
+using NUnit.Framework;
+using ServiceStack.DataAnnotations;
+using ServiceStack.OrmLite.Tests;
+
+namespace ServiceStack.OrmLite.PostgreSQL.Tests
+{
+    [TestFixture]
+    public class OrmLiteExecuteProcedureTests : OrmLiteTestBase
+    {
+        private const string Create = @"
+            CREATE OR REPLACE FUNCTION f_service_stack(
+                v_string_values CHARACTER VARYING[],
+                v_integer_values INTEGER[]
+            ) RETURNS BOOLEAN AS
+            $BODY$
+            BEGIN
+                IF v_string_values[1] <> 'ServiceStack' THEN
+                    RAISE EXCEPTION 'Unexpected value in string array[1] %', v_string_values[1];
+                END IF;
+                IF v_string_values[2] <> 'Thoughtfully Architected' THEN
+                    RAISE EXCEPTION 'Unexpected value in string array[2] %', v_string_values[2];
+                END IF;
+                IF v_integer_values[1] <> 1 THEN
+                    RAISE EXCEPTION 'Unexpected value in integer array[1] %', v_integer_values[1];
+                END IF;
+                IF v_integer_values[2] <> 2 THEN
+                    RAISE EXCEPTION 'Unexpected value in integer array[2] %', v_integer_values[2];
+                END IF;
+                IF v_integer_values[3] <> 3 THEN
+                    RAISE EXCEPTION 'Unexpected value in integer array[3] %', v_integer_values[3];
+                END IF;
+                RETURN TRUE;
+            END;
+            $BODY$
+            LANGUAGE plpgsql VOLATILE COST 100;
+            ";
+
+        private const string Drop = "DROP FUNCTION f_service_stack(CHARACTER VARYING[], INTEGER[]);";
+
+        [Alias("f_service_stack")]
+        public class ServiceStackFunction
+        {
+            public string[] StringValues { get; set; }
+            public int[] IntegerValues { get; set; }
+        }
+
+        [Test]
+        public void Can_execute_stored_procedure_with_array_arguments()
+        {
+            using (var db = ConnectionString.OpenDbConnection())
+            {
+                db.ExecuteSql(Create);
+
+                db.ExecuteProcedure(new ServiceStackFunction
+                                        {
+                                            StringValues = new[] { "ServiceStack", "Thoughtfully Architected" },
+                                            IntegerValues = new[] { 1, 2, 3 }
+                                        });
+                db.ExecuteSql(Drop);
+            }
+        }
+    }
+}

src/ServiceStack.OrmLite.PostgreSQL.Tests/ServiceStack.OrmLite.PostgreSQL.Tests.csproj

@@ -89,6 +89,7 @@
     <Compile Include="NorthwindPerfTests.cs" />
     <Compile Include="OrmLiteBasicPersistenceProviderTests.cs" />
     <Compile Include="OrmLiteCreateTableWithNamigStrategyTests.cs" />
+    <Compile Include="OrmLiteExecuteProcedureTests.cs" />
     <Compile Include="OrmLiteInsertTests.cs" />
     <Compile Include="OrmLiteNorthwindTests.cs" />
     <Compile Include="OrmLiteSelectTests.cs" />

src/ServiceStack.OrmLite.PostgreSQL/PostgreSQLDialectProvider.cs

@@ -127,6 +127,16 @@ public override string GetQuotedValue(object value, Type fieldType)
 			{
 				return "E'" + ToBinary(value) + "'";
 			}
+			if (fieldType.IsArray && typeof(string).IsAssignableFrom(fieldType.GetElementType()))
+			{
+				var stringArray = (string[]) value;
+				return ToArray(stringArray);
+			}
+			if (fieldType.IsArray && typeof(int).IsAssignableFrom(fieldType.GetElementType()))
+			{
+				var integerArray = (int[]) value;
+				return ToArray(integerArray);
+			}
 
 			return base.GetQuotedValue(value, fieldType);
 		}
@@ -233,5 +243,16 @@ internal static String ToBinary(Object NativeData)
 						.Append((char)('0' + (7 & b)));
 			return res.ToString();
 		}
+
+		internal string ToArray<T>(T[] source)
+		{
+			var values = new StringBuilder();
+			foreach (var value in source)
+			{
+				if (values.Length > 0) values.Append(",");
+				values.Append(base.GetQuotedValue(value, typeof(T)));
+			}
+			return "ARRAY[" + values + "]";
+		}
 	}
 }