Support Memory<float> as a parameter type.

Signed-off-by: Bradley Grainger <[email protected]>

Author: bgrainger

docs/content/troubleshooting/parameter-types.md

@@ -39,5 +39,6 @@ In some cases, this may be as simple as calling `.ToString()` or `.ToString(Cult
 * .NET primitives: `bool`, `byte`, `char`, `double`, `float`, `int`, `long`, `sbyte`, `short`, `uint`, `ulong`, `ushort`
 * Common types: `BigInteger`, `DateOnly`, `DateTime`, `DateTimeOffset`, `decimal`, `enum`, `Guid`, `string`, `TimeOnly`, `TimeSpan`
 * BLOB types: `ArraySegment<byte>`, `byte[]`, `Memory<byte>`, `ReadOnlyMemory<byte>`
+* Vector types: `float[]`, `Memory<float>`, `ReadOnlyMemory<float>`
 * String types: `Memory<char>`, `ReadOnlyMemory<char>`, `StringBuilder`
 * Custom MySQL types: `MySqlDateTime`, `MySqlDecimal`, `MySqlGeometry`

src/MySqlConnector/MySqlParameter.cs

@@ -285,7 +285,7 @@ internal void AppendSqlString(ByteBufferWriter writer, StatementPreparerOptions
 		{
 			writer.WriteString(ulongValue);
 		}
-		else if (Value is byte[] or ReadOnlyMemory<byte> or Memory<byte> or ArraySegment<byte> or MySqlGeometry or MemoryStream or float[])
+		else if (Value is byte[] or ReadOnlyMemory<byte> or Memory<byte> or ArraySegment<byte> or MySqlGeometry or MemoryStream or float[] or ReadOnlyMemory<float> or Memory<float>)
 		{
 			var inputSpan = Value switch
 			{
@@ -295,6 +295,8 @@ internal void AppendSqlString(ByteBufferWriter writer, StatementPreparerOptions
 				MySqlGeometry geometry => geometry.ValueSpan,
 				MemoryStream memoryStream => memoryStream.TryGetBuffer(out var streamBuffer) ? streamBuffer.AsSpan() : memoryStream.ToArray().AsSpan(),
 				float[] floatArray => MemoryMarshal.AsBytes(floatArray.AsSpan()),
+				Memory<float> memory => MemoryMarshal.AsBytes(memory.Span),
+				ReadOnlyMemory<float> memory => MemoryMarshal.AsBytes(memory.Span),
 				_ => ((ReadOnlyMemory<byte>) Value).Span,
 			};
 
@@ -738,6 +740,16 @@ private void AppendBinary(ByteBufferWriter writer, object value, StatementPrepar
 			writer.WriteLengthEncodedInteger(unchecked((ulong) floatArrayValue.Length * 4));
 			writer.Write(MemoryMarshal.AsBytes(floatArrayValue.AsSpan()));
 		}
+		else if (value is Memory<float> floatMemory)
+		{
+			writer.WriteLengthEncodedInteger(unchecked((ulong) floatMemory.Length * 4));
+			writer.Write(MemoryMarshal.AsBytes(floatMemory.Span));
+		}
+		else if (value is ReadOnlyMemory<float> floatReadOnlyMemory)
+		{
+			writer.WriteLengthEncodedInteger(unchecked((ulong) floatReadOnlyMemory.Length * 4));
+			writer.Write(MemoryMarshal.AsBytes(floatReadOnlyMemory.Span));
+		}
 		else if (value is decimal decimalValue)
 		{
 			writer.WriteLengthEncodedAsciiString(decimalValue.ToString(CultureInfo.InvariantCulture));

tests/IntegrationTests/QueryTests.cs

@@ -1691,9 +1691,13 @@ public void GetBytesByName()
 #endif
 
 	[SkippableTheory(ServerFeatures.Vector)]
-	[InlineData(false)]
-	[InlineData(true)]
-	public void QueryVector(bool prepare)
+	[InlineData(false, 0)]
+	[InlineData(false, 1)]
+	[InlineData(false, 2)]
+	[InlineData(true, 0)]
+	[InlineData(true, 1)]
+	[InlineData(true, 2)]
+	public void QueryVector(bool prepare, int dataFormat)
 	{
 		using var connection = new MySqlConnection(AppConfig.ConnectionString);
 		connection.Open();
@@ -1715,8 +1719,15 @@ public void QueryVector(bool prepare)
 #if MYSQL_DATA
 		// Connector/NET requires the float vector to be passed as a byte array
 		cmd.Parameters[0].Value = MemoryMarshal.AsBytes<float>(floatArray).ToArray();
+		Assert.InRange(dataFormat, 0, 2);
 #else
-		cmd.Parameters[0].Value = floatArray;
+		cmd.Parameters[0].Value = dataFormat switch
+		{
+			0 => floatArray,
+			1 => new Memory<float>(floatArray),
+			2 => new ReadOnlyMemory<float>(floatArray),
+			_ => throw new NotSupportedException(),
+		};
 #endif
 
 		if (prepare)