From 840ccb278258f0add9208f65cfab390ef0aaccf1 Mon Sep 17 00:00:00 2001
From: Mikhail <99481254+DCFApixels@users.noreply.github.com>
Date: Tue, 13 Jun 2023 15:41:31 +0800
Subject: [PATCH] add inject null checks

---
 src/Builtin/InjectSystem.cs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/Builtin/InjectSystem.cs b/src/Builtin/InjectSystem.cs
index c7fdf6a..2663a20 100644
--- a/src/Builtin/InjectSystem.cs
+++ b/src/Builtin/InjectSystem.cs
@@ -108,6 +108,7 @@ namespace DCFApixels.DragonECS
             void IEcsInject<PreInitInjectController>.Inject(PreInitInjectController obj) => _injectController = obj;
             public InjectSystem(T injectedData)
             {
+                if (_injectedData == null) throw new ArgumentNullException();
                 _injectedData = injectedData;
             }
             public void PreInit(EcsPipeline pipeline)
@@ -140,6 +141,7 @@ namespace DCFApixels.DragonECS
     {
         public static EcsPipeline.Builder Inject<T>(this EcsPipeline.Builder self, T data)
         {
+            if (data == null) throw new ArgumentNullException();
             return self.Add(new InjectSystem<T>(data));
         }
         public static EcsPipeline.Builder Inject<A, B>(this EcsPipeline.Builder self, A a, B b)