add inject null checks

This commit is contained in:
Mikhail 2023-06-13 15:41:31 +08:00
parent 571c1a1385
commit 840ccb2782

View File

@ -108,6 +108,7 @@ namespace DCFApixels.DragonECS
void IEcsInject<PreInitInjectController>.Inject(PreInitInjectController obj) => _injectController = obj; void IEcsInject<PreInitInjectController>.Inject(PreInitInjectController obj) => _injectController = obj;
public InjectSystem(T injectedData) public InjectSystem(T injectedData)
{ {
if (_injectedData == null) throw new ArgumentNullException();
_injectedData = injectedData; _injectedData = injectedData;
} }
public void PreInit(EcsPipeline pipeline) public void PreInit(EcsPipeline pipeline)
@ -140,6 +141,7 @@ namespace DCFApixels.DragonECS
{ {
public static EcsPipeline.Builder Inject<T>(this EcsPipeline.Builder self, T data) 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)); return self.Add(new InjectSystem<T>(data));
} }
public static EcsPipeline.Builder Inject<A, B>(this EcsPipeline.Builder self, A a, B b) public static EcsPipeline.Builder Inject<A, B>(this EcsPipeline.Builder self, A a, B b)