添加针对分片键的ValueGenerated的属性判断只允许Never [#196]

Author: xuejmnet

src/ShardingCore/Core/EntityMetadatas/DefaultEntityMetadataManager.cs

@@ -5,6 +5,9 @@
 using System.Text;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Metadata;
+using Microsoft.EntityFrameworkCore.Metadata.Internal;
+using Microsoft.Extensions.Logging;
+using ShardingCore.Core.ShardingConfigurations;
 using ShardingCore.Exceptions;
 using ShardingCore.Extensions;
 using ShardingCore.Sharding.Abstractions;
@@ -16,9 +19,14 @@ namespace ShardingCore.Core.EntityMetadatas
     /// </summary>
     public class DefaultEntityMetadataManager : IEntityMetadataManager
     {
+        private readonly ShardingConfigOptions _shardingConfigOptions;
         private readonly ConcurrentDictionary<Type, EntityMetadata> _caches = new();
         private readonly ConcurrentDictionary<string/*logic table name*/, List<EntityMetadata>> _logicTableCaches = new();
 
+        public DefaultEntityMetadataManager(ShardingConfigOptions shardingConfigOptions)
+        {
+            _shardingConfigOptions = shardingConfigOptions;
+        }
         public bool AddEntityMetadata(EntityMetadata entityMetadata)
         {
             return _caches.TryAdd(entityMetadata.EntityType, entityMetadata);
@@ -104,6 +112,33 @@ public bool TryInitModel(IEntityType efEntityType)
         {
             if (_caches.TryGetValue(efEntityType.ClrType, out var metadata))
             {
+                if (_shardingConfigOptions.CheckShardingKeyValueGenerated)
+                {
+                    if (metadata.IsMultiDataSourceMapping)
+                    {
+                        foreach (var metadataProperty in metadata.ShardingDataSourceProperties)
+                        {
+                            var propertyName = metadataProperty.Key;
+                            var property = efEntityType.GetProperty(propertyName);
+                            if (property.ValueGenerated!=ValueGenerated.Never)
+                            {
+                                throw new ShardingCoreConfigException($"sharding data source key:{propertyName} is not {nameof(ValueGenerated)}.{nameof(ValueGenerated.Never)}");
+                            }
+                        }
+                    }
+                    if (metadata.IsMultiTableMapping)
+                    {
+                        foreach (var metadataProperty in metadata.ShardingTableProperties)
+                        {
+                            var propertyName = metadataProperty.Key;
+                            var property = efEntityType.GetProperty(propertyName);
+                            if (property.ValueGenerated!=ValueGenerated.Never)
+                            {
+                                throw new ShardingCoreConfigException($"sharding table key:{propertyName} is not {nameof(ValueGenerated)}.{nameof(ValueGenerated.Never)}");
+                            }
+                        }
+                    }
+                }
                 metadata.SetEntityModel(efEntityType);
                 if (string.IsNullOrWhiteSpace(metadata.LogicTableName))
                 {

src/ShardingCore/Core/ShardingConfigurations/ShardingConfigOptions.cs

@@ -55,6 +55,12 @@ public class ShardingConfigOptions
         /// 默认数据源链接字符串
         /// </summary>
         public string DefaultConnectionString { get; private set; }
+
+        /// <summary>
+        /// 检测分片键的自动值是否有疑义
+        /// </summary>
+
+        public bool CheckShardingKeyValueGenerated { get; set; } = true;
         /// <summary>
         /// 添加默认数据源
         /// </summary>