Handle Nullable<JsonElement>

npgsql · Dec 5, 2023 · 357a033 · 357a033
1 parent 0a795e3
commit 357a033
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/EFCore.PG/Metadata/Conventions/NpgsqlJsonElementHackConvention.cs b/src/EFCore.PG/Metadata/Conventions/NpgsqlJsonElementHackConvention.cs
@@ -20,10 +20,11 @@ public class NpgsqlJsonElementHackConvention : IPropertyAddedConvention
     public void ProcessPropertyAdded(IConventionPropertyBuilder propertyBuilder, IConventionContext<IConventionPropertyBuilder> context)
     {
         var property = propertyBuilder.Metadata;
+        var clrType = property.ClrType;
 
-        if (property.ClrType == typeof(JsonElement) && property.GetColumnType() is null)
+        if ((clrType == typeof(JsonElement) || clrType == typeof(JsonElement?)) && property.GetColumnType() is null)
         {
-            property.SetTypeMapping(_jsonTypeMapping ??= new NpgsqlJsonTypeMapping("jsonb", typeof(JsonElement)));
+            property.SetTypeMapping(_jsonTypeMapping ??= new NpgsqlJsonTypeMapping("jsonb", clrType));
         }
     }
 }