fix create table error : Sqlite 'table TelemetryData_202209" already exists'. #801

IoTSharp.Data/Shardings/GuessTableEnsureManager.cs

+using System;
+using Microsoft.EntityFrameworkCore;
+using ShardingCore.Sharding.Abstractions;
+using ShardingCore.TableExists.Abstractions;
+using System.Collections.Generic;
+using System.Data.Common;
+using ShardingCore.Core.VirtualRoutes.TableRoutes.RouteTails.Abstractions;
+using Microsoft.Extensions.FileSystemGlobbing.Internal.PathSegments;
+
+namespace ShardingCore.TableExists
+{
+    public class GuessTableEnsureManager : AbstractTableEnsureManager
+    {
+        private readonly EmptyTableEnsureManager _tem_empyt;
+        private readonly MySqlTableEnsureManager _tem_mysql;
+        private readonly SqliteTableEnsureManager _tem_sqlite;
+        private readonly PostgreSqlTableEnsureManager _tem_pgsql;
+        private readonly SqlServerTableEnsureManager _tem_mssql;
+
+        public GuessTableEnsureManager(IRouteTailFactory routeTailFactory) : base(routeTailFactory)
+        {
+            _tem_empyt = new EmptyTableEnsureManager();
+            _tem_mysql = new MySqlTableEnsureManager(routeTailFactory);
+            _tem_sqlite = new SqliteTableEnsureManager(routeTailFactory);
+            _tem_pgsql = new PostgreSqlTableEnsureManager(routeTailFactory);
+            _tem_mssql = new SqlServerTableEnsureManager(routeTailFactory);
+        }
+
+        public override ISet<string> DoGetExistTables(DbConnection connection, string dataSourceName)
+        {
+            ISet<string> result = new HashSet<string>(StringComparer.OrdinalIgnoreCase);
+            var dbctypename = connection.GetType().FullName;
+            switch (dbctypename)
+            {
+                case "Npgsql.NpgsqlConnection":
+                    result = _tem_pgsql.DoGetExistTables(connection, dataSourceName);
+                    break;
+                case "System.Data.Sqlite.SqliteConnection":
+                case "Microsoft.Data.Sqlite.SqliteConnection":
+                    result = _tem_sqlite.DoGetExistTables(connection, dataSourceName);
+                    break;
+                case "MySqlConnector.MySqlConnection":
+                    result = _tem_mysql.DoGetExistTables(connection, dataSourceName);
+                    break;
+                case "Microsoft.Data.SqlClient.SqlConnection":
+                case "System.Data.SqlClient.SqlConnection":
+                    result = _tem_mssql.DoGetExistTables(connection, dataSourceName);
+                    break;
+                default:
+                    break;
+            }
+            return result;
+        }
+
+    }
+}

IoTSharp.Data/Shardings/PostgreSqlTableEnsureManager.cs

+using System;
+using ShardingCore.TableExists.Abstractions;
+using System.Collections.Generic;
+using System.Data.Common;
+using ShardingCore.Core.VirtualRoutes.TableRoutes.RouteTails.Abstractions;
+
+namespace ShardingCore.TableExists
+{
+    public class PostgreSqlTableEnsureManager : AbstractTableEnsureManager
+    {
+        public PostgreSqlTableEnsureManager(IRouteTailFactory routeTailFactory) : base(routeTailFactory)
+        {
+        }
+        private const string Tables = "Tables";
+        private const string TABLE_NAME = "TABLE_NAME";
+
+        public override ISet<string> DoGetExistTables(DbConnection connection, string dataSourceName)
+        {
+            ISet<string> result = new HashSet<string>(StringComparer.OrdinalIgnoreCase);
+            using (var dataTable = connection.GetSchema(Tables))
+            {
+                for (int i = 0; i < dataTable.Rows.Count; i++)
+                {
+                    result.Add(dataTable.Rows[i][TABLE_NAME].ToString());
+                }
+            }
+            return result;
+        }
+
+    }
+}

IoTSharp.Data/Shardings/SqliteTableEnsureManager.cs

+using System;
+using ShardingCore.TableExists.Abstractions;
+using System.Collections.Generic;
+using System.Data.Common;
+using ShardingCore.Core.VirtualRoutes.TableRoutes.RouteTails.Abstractions;
+
+namespace ShardingCore.TableExists
+{
+    public class SqliteTableEnsureManager : AbstractTableEnsureManager
+    {
+        public SqliteTableEnsureManager(IRouteTailFactory routeTailFactory) : base(routeTailFactory)
+        {
+        }
+
+        public override ISet<string> DoGetExistTables(DbConnection connection, string dataSourceName)
+        {
+            ISet<string> result = new HashSet<string>(StringComparer.OrdinalIgnoreCase);
+            using (var query = connection.CreateCommand())
+            {
+                query.CommandText = "SELECT tbl_name FROM  sqlite_master;";
+                using (var reader = query.ExecuteReader())
+                {
+                    while (reader.Read())
+                    {
+                         var   str = (string)reader["tbl_name"];
+                        result.Add(str);
+                    }
+                }
+            }
+            return result;
+        }
+
+    }
+}

IoTSharp/Startup.cs

@@ -40,6 +40,8 @@ using IoTSharp.EventBus.Shashlik;
 using Microsoft.EntityFrameworkCore;
 using ShardingCore;
 using Storage.Net;
+using ShardingCore.TableExists.Abstractions;
+using ShardingCore.TableExists;
 
 namespace IoTSharp
 {
@@ -223,7 +225,8 @@ namespace IoTSharp
                 case TelemetryStorage.Sharding:
                     ShardingByDateMode settingsShardingByDateMode = settings.ShardingByDateMode;
                      var   sbt =  DateTime.Now.Subtract( settings.ShardingBeginTime);
-                    services.AddShardingDbContext<ShardingDbContext>().UseRouteConfig(o =>
+                    var _sharding = services.AddShardingDbContext<ShardingDbContext>();
+                    _sharding.UseRouteConfig(o =>
                     {
                         switch (settingsShardingByDateMode)
                         {
@@ -237,7 +240,7 @@ namespace IoTSharp
                                 break;
                             case ShardingByDateMode.PerDay:
                                 if (sbt.TotalDays < 1) throw new ArgumentException($"按日分表时间至少大于当前时间一天。");
-                                o.AddShardingTableRoute<TelemetryDataDayRoute>(); 
+                                o.AddShardingTableRoute<TelemetryDataDayRoute>();
                                 break;
                             case ShardingByDateMode.PerMonth:
                                 if (sbt.TotalDays < DateTime.Now.Subtract(DateTime.Now.Date.AddDays(-DateTime.Now.Day)).TotalDays) throw new ArgumentException($"按月分表时间至少大于当前时间一个月。");
@@ -249,11 +252,12 @@ namespace IoTSharp
                                 break;
                             default: throw new InvalidOperationException($"unknown sharding mode:{settingsShardingByDateMode}");
                         }
-                    }).UseConfig(o =>
+                    });
+                    _sharding.UseConfig(o =>
                     {
                         o.ThrowIfQueryRouteNotMatch = false;
-                        o.UseShellDbContextConfigure(builder=>builder.UseQueryTrackingBehavior(QueryTrackingBehavior.NoTracking));
-                        o.AddDefaultDataSource("ds0",Configuration.GetConnectionString("TelemetryStorage"));
+                        o.UseShellDbContextConfigure(builder => builder.UseQueryTrackingBehavior(QueryTrackingBehavior.NoTracking));
+                        o.AddDefaultDataSource("ds0", Configuration.GetConnectionString("TelemetryStorage"));
                         switch (settings.DataBase)
                         {
                             case DataBaseType.MySql:
@@ -277,7 +281,29 @@ namespace IoTSharp
                                 break;
 
                         }
-                    }).AddShardingCore();
+                    });
+                    _sharding.AddShardingCore();
+                    switch (settings.DataBase)
+                    {
+                        case DataBaseType.MySql:
+                            _sharding.ReplaceService<ITableEnsureManager, MySqlTableEnsureManager>();
+                            break;
+
+                        case DataBaseType.SqlServer:
+                            _sharding.ReplaceService<ITableEnsureManager, SqlServerTableEnsureManager>();
+                            break;
+                        case DataBaseType.Oracle:
+                            break;
+                        case DataBaseType.Sqlite:
+                            _sharding.ReplaceService<ITableEnsureManager, SqliteTableEnsureManager>();
+                            break;
+                        case DataBaseType.PostgreSql:
+                            _sharding.ReplaceService<ITableEnsureManager, GuessTableEnsureManager>();
+                            break;
+                        default:
+                            break;
+
+                    }
                     services.AddSingleton<IStorage, ShardingStorage>();
                     break;