Update data source context

dogzz9445 · Jul 25, 2024 · 88d3efd · 88d3efd
1 parent e863141
commit 88d3efd
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/Apps/Corathing.Organizer/Services/DataSourceService.cs b/src/Apps/Corathing.Organizer/Services/DataSourceService.cs
@@ -109,7 +109,11 @@ public void AddDataSourceContext(Guid dataSourceStateId, DataSourceContext dataS
 
     public void AddDataSourceContext(DataSourceContext dataSourceContext)
     {
-        DataSourceContexts.Add(dataSourceContext.DataSourceId, dataSourceContext);
+        if (dataSourceContext.DataSourceId == null)
+        {
+            throw new InvalidOperationException("DataSourceId is null");
+        }
+        DataSourceContexts.Add(dataSourceContext.DataSourceId.Value, dataSourceContext);
     }
 
     public void RemoveAllDataSourceContexts<T>()
@@ -124,7 +128,10 @@ public void RemoveAllDataSourceContexts<T>()
 
     public void RemoveDataSourceContext<T>(T dataSourceContext) where T : DataSourceContext
     {
-        DataSourceContexts.Remove(dataSourceContext.DataSourceId);
+        if (dataSourceContext.DataSourceId != null)
+        {
+            DataSourceContexts.Remove(dataSourceContext.DataSourceId.Value);
+        }
     }
 
     public IEnumerable<DataSourceContext> GetAllDataSourceContexts(Type? dataSourceContext)
@@ -139,7 +146,6 @@ public IEnumerable<DataSourceContext> GetAllDataSourceContexts(Type? dataSourceC
     public IEnumerable<T> GetAllDataSourceContexts<T>()
     {
         return DataSourceContexts.Values.OfType<T>();
-        //return DataSourceContexts.Values.Select(item => item as T).Where(x => x != null);
     }
 
     public IEnumerable<T> GetDataSourceContexts<T>()