提交 d56bdc53 编写于 作者: 麦壳饼's avatar 麦壳饼

重构一部分遥测数据存储代码

上级 6ba42ec0
...@@ -33,6 +33,8 @@ using System.Text; ...@@ -33,6 +33,8 @@ using System.Text;
using System.Text.RegularExpressions; using System.Text.RegularExpressions;
using IoTSharp.Extensions.X509; using IoTSharp.Extensions.X509;
using IoTSharp.Storage; using IoTSharp.Storage;
using ShardingCore;
using Microsoft.Extensions.Options;
namespace IoTSharp namespace IoTSharp
{ {
...@@ -356,10 +358,16 @@ namespace IoTSharp ...@@ -356,10 +358,16 @@ namespace IoTSharp
internal static void UseTelemetryStorage(this IApplicationBuilder app) internal static void UseTelemetryStorage(this IApplicationBuilder app)
{ {
var options = app.ApplicationServices.GetRequiredService<IOptions<AppSettings>>();
var settings = options.Value;
if (settings.TelemetryStorage == TelemetryStorage.Sharding)
{
app.ApplicationServices.UseAutoTryCompensateTable();
}
using (var scope = app.ApplicationServices.CreateScope()) using (var scope = app.ApplicationServices.CreateScope())
{ {
var _ts_storage = scope.ServiceProvider.GetService<IStorage>(); var _ts_storage = scope.ServiceProvider.GetService<IStorage>();
_ts_storage.CheckTelemetryStorage(); _ts_storage.CheckTelemetryStorage();
} }
} }
} }
......
...@@ -300,11 +300,7 @@ namespace IoTSharp ...@@ -300,11 +300,7 @@ namespace IoTSharp
} }
app.CheckApplicationDBMigrations(); app.CheckApplicationDBMigrations();
//添加定时任务创建表 //添加定时任务创建表
var settings = Configuration.Get<AppSettings>();
if (settings.TelemetryStorage == TelemetryStorage.Sharding)
{
app.ApplicationServices.UseAutoTryCompensateTable();
}
app.UseRouting(); app.UseRouting();
app.UseCors(option => option app.UseCors(option => option
.AllowAnyOrigin() .AllowAnyOrigin()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册