added null checks

EAVFW · Aug 28, 2024 · 9b4166d · 9b4166d
1 parent 609c2c3
commit 9b4166d
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/Plugins/EntityPlugin1.cs b/src/Plugins/EntityPlugin1.cs
@@ -145,6 +145,19 @@ public class DynamicEntityPlugin<TContext> : EntityPlugin<TContext>
         public override ValueTask<bool> ShouldPluginBeExecued<T>(T context, TrackedPipelineItem entity)
 
         {
+            if (context is null)
+            {
+                throw new ArgumentNullException(nameof(context));
+            }
+            if (entity.Entity is null)
+            {
+                throw new ArgumentNullException("Entry");
+            }
+            if (entity.Entity.Entity is null)
+            {
+                throw new ArgumentNullException("Entity");
+            }
+
             var type = GetPluginType(context as TContext);
             return ValueTask.FromResult(type.IsAssignableFrom(entity.Entity.Entity.GetType()));