DataSyncBackgroundService.cs
9.72 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
using System.Linq;
using System.Text.Json;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using Rcs.Application.Services;
using Rcs.Domain.Repositories;
using StackExchange.Redis;
namespace Rcs.Infrastructure.Services
{
/// <summary>
/// 数据同步后台服务 - 程序启动时将地图和机器人数据同步到Redis
/// @author zzy
/// </summary>
public class DataSyncBackgroundService : IHostedService, IDataSyncService
{
private readonly ILogger<DataSyncBackgroundService> _logger;
private readonly IServiceProvider _serviceProvider;
private readonly IConnectionMultiplexer _redis;
// Redis缓存键前缀
private const string MapKeyPrefix = "rcs:map:";
private const string RobotKeyPrefix = "rcs:robot:";
private const string MapListKey = "rcs:maps";
private const string RobotListKey = "rcs:robots";
public DataSyncBackgroundService(
ILogger<DataSyncBackgroundService> logger,
IServiceProvider serviceProvider,
IConnectionMultiplexer redis)
{
_logger = logger;
_serviceProvider = serviceProvider;
_redis = redis;
}
/// <summary>
/// 服务启动时执行数据同步
/// @author zzy
/// </summary>
public async Task StartAsync(CancellationToken cancellationToken)
{
_logger.LogInformation("[数据同步] 数据同步服务启动中...");
try
{
await SyncAllAsync(cancellationToken);
_logger.LogInformation("[数据同步] 数据同步服务启动完成");
}
catch (Exception ex)
{
_logger.LogError(ex, "[数据同步] 数据同步服务启动失败");
}
}
/// <summary>
/// 服务停止
/// @author zzy
/// </summary>
public Task StopAsync(CancellationToken cancellationToken)
{
_logger.LogInformation("[数据同步] 数据同步服务已停止");
return Task.CompletedTask;
}
/// <summary>
/// 同步所有数据到Redis
/// @author zzy
/// </summary>
public async Task SyncAllAsync(CancellationToken cancellationToken = default)
{
await SyncMapsAsync(cancellationToken);
await SyncRobotsAsync(cancellationToken);
}
/// <summary>
/// 同步所有地图数据到Redis（包含节点、边、资源）
/// @author zzy
/// </summary>
public async Task SyncMapsAsync(CancellationToken cancellationToken = default)
{
using var scope = _serviceProvider.CreateScope();
var mapRepo = scope.ServiceProvider.GetRequiredService<IMapRepository>();
var db = _redis.GetDatabase();
var maps = await mapRepo.GetAllAsync(cancellationToken);
var mapIds = new System.Collections.Generic.List<string>();
foreach (var mapBasic in maps)
{
// 获取包含完整关联数据的地图
var map = await mapRepo.GetWithFullDetailsAsync(mapBasic.MapId, cancellationToken);
if (map == null) continue;
var key = $"{MapKeyPrefix}{map.MapId}";
var json = JsonSerializer.Serialize(new
{
map.MapId,
map.MapCode,
map.MapName,
MapType = (int)map.MapType,
map.Version,
map.Description,
map.Active,
map.ResourceUrl,
map.PointsUrl,
map.CreatedAt,
// 节点数据
Nodes = map.MapNodes.Select(n => new
{
n.NodeId,
n.NodeCode,
n.NodeName,
n.X,
n.Y,
n.Theta,
Type = (int)n.Type,
n.IsReverseParking,
n.AllowRotate,
n.MaxCoordinateOffset,
n.Active
}),
// 边数据
Edges = map.MapEdges.Select(e => new
{
e.EdgeId,
e.EdgeCode,
e.EdgeName,
e.FromNode,
e.ToNode,
e.Length,
e.Cost,
e.IsCurve,
e.Radius,
e.CenterX,
e.CenterY,
e.Regress,
e.MaxSpeed,
e.Active
}),
// 资源数据
Resources = map.MapResources.Select(r => new
{
r.ResourceId,
r.ResourceCode,
r.ResourceName,
Type = (int)r.Type,
r.Capacity,
r.LocationCoordinates,
r.Active
})
});
await db.StringSetAsync(key, json);
mapIds.Add(map.MapId.ToString());
}
// 存储地图ID列表
await db.StringSetAsync(MapListKey, JsonSerializer.Serialize(mapIds));
_logger.LogInformation("[数据同步] 已同步 {Count} 个地图数据到Redis（含节点、边、资源）", mapIds.Count);
}
/// <summary>
/// 同步所有机器人数据到Redis（使用Hash结构分离存储基础/状态/位置数据）
/// @author zzy
/// </summary>
public async Task SyncRobotsAsync(CancellationToken cancellationToken = default)
{
using var scope = _serviceProvider.CreateScope();
var robotRepo = scope.ServiceProvider.GetRequiredService<IRobotRepository>();
var db = _redis.GetDatabase();
var robots = await robotRepo.GetAllAsync(cancellationToken);
foreach (var robot in robots)
{
var robotKey = $"{robot.RobotManufacturer}:{robot.RobotSerialNumber}";
// 基础数据 Hash（使用制造商+序列号作为唯一标识）
var basicKey = $"{RobotKeyPrefix}{robotKey}:basic";
await db.HashSetAsync(basicKey, new HashEntry[]
{
new("RobotId", robot.RobotId.ToString()),
new("RobotCode", robot.RobotCode),
new("RobotName", robot.RobotName),
new("RobotVersion", robot.RobotVersion),
new("ProtocolName", robot.ProtocolName),
new("ProtocolVersion", robot.ProtocolVersion),
new("ProtocolType", (int)robot.ProtocolType),
new("RobotManufacturer", robot.RobotManufacturer ?? ""),
new("RobotSerialNumber", robot.RobotSerialNumber),
new("RobotType", (int)robot.RobotType),
new("IpAddress", robot.IpAddress),
new("CoordinateScale", robot.CoordinateScale),
new("Active", robot.Active)
});
// 状态数据 Hash（使用制造商+序列号作为唯一标识）
var statusKey = $"{RobotKeyPrefix}{robotKey}:status";
await db.HashSetAsync(statusKey, new HashEntry[]
{
new("Status", (int)robot.Status),
new("Online", (int)robot.Online),
new("BatteryLevel", robot.BatteryLevel ?? 0),
new("Driving", robot.Driving),
new("Paused", robot.Paused),
new("Charging", robot.Charging),
new("OperatingMode", (int)robot.OperatingMode),
new("Errors", robot.Errors ?? ""),
new("UpdatedAt", DateTime.Now.ToString("O"))
});
// 位置数据 Hash（使用制造商+序列号作为唯一标识）
var locationKey = $"{RobotKeyPrefix}{robotKey}:location";
await db.HashSetAsync(locationKey, new HashEntry[]
{
new("MapId", robot.CurrentMapCodeId?.ToString() ?? ""),
new("NodeId", robot.CurrentNodeId?.ToString() ?? ""),
new("X", robot.CurrentX?.ToString(System.Globalization.CultureInfo.InvariantCulture) ?? ""),
new("Y", robot.CurrentY?.ToString(System.Globalization.CultureInfo.InvariantCulture) ?? ""),
new("Theta", robot.CurrentTheta?.ToString(System.Globalization.CultureInfo.InvariantCulture) ?? ""),
new("Path", ""),
new("UpdatedAt", DateTime.Now.ToString("O"))
});
// 添加到机器人集合（使用制造商:序列号作为标识）
await db.SetAddAsync(RobotListKey, robotKey);
// 更新在线/空闲集合
if (robot.Online == Domain.Entities.OnlineStatus.Online)
await db.SetAddAsync("rcs:robots:online", robotKey);
if (robot.Status == Domain.Entities.RobotStatus.Idle)
await db.SetAddAsync("rcs:robots:idle", robotKey);
}
_logger.LogInformation("[数据同步] 已同步 {Count} 个机器人数据到Redis（Hash结构）", robots.Count());
}
}
}