feat: 使用SignalR来控制jtag边界扫描

server/src/Database.cs

@@ -133,7 +133,7 @@ public class Board
     /// </summary>
     public enum BoardStatus
     {
-        /// <summary> 
+        /// <summary>
         /// 未启用状态，无法被使用
         /// </summary>
         Disabled,
@@ -558,6 +558,31 @@ public class AppDataConnection : DataConnection
         return new(boards[0]);
     }
 
+    /// <summary>
+    /// 根据用户名获取实验板信息
+    /// </summary>
+    /// <param name="userName">用户名</param>
+    /// <returns>包含实验板信息的结果，如果未找到则返回空</returns>
+    public Result<Optional<Board>> GetBoardByUserName(string userName)
+    {
+        var boards = this.BoardTable.Where(board => board.OccupiedUserName == userName).ToArray();
+
+        if (boards.Length > 1)
+        {
+            logger.Error($"数据库中存在多个相同用户名的实验板: {userName}");
+            return new(new Exception($"数据库中存在多个相同用户名的实验板: {userName}"));
+        }
+
+        if (boards.Length == 0)
+        {
+            logger.Info($"未找到用户名对应的实验板: {userName}");
+            return new(Optional<Board>.None);
+        }
+
+        logger.Debug($"成功获取实验板信息: {userName}");
+        return new(boards[0]);
+    }
+
     /// <summary>
     /// 获取所有实验板信息
     /// </summary>

server/src/Hubs/JtagHub.cs

@@ -0,0 +1,176 @@
+using Microsoft.AspNetCore.Authorization;
+using System.Security.Claims;
+using Microsoft.AspNetCore.SignalR;
+using DotNext;
+using System.Collections.Concurrent;
+using TypedSignalR.Client;
+using Tapper;
+
+namespace server.Hubs.JtagHub;
+
+[Hub]
+public interface IJtagHub
+{
+    Task<bool> SetBoundaryScanFreq(int freq);
+    Task<bool> StartBoundaryScan(int freq = 100);
+    Task<bool> StopBoundaryScan();
+}
+
+[Receiver]
+public interface IJtagReceiver
+{
+    Task OnReceiveBoundaryScanData(Dictionary<string, bool> msg);
+}
+
+[Authorize]
+public class JtagHub : Hub<IJtagReceiver>, IJtagHub
+{
+    private static NLog.Logger logger = NLog.LogManager.GetCurrentClassLogger();
+    private ConcurrentDictionary<string, int> FreqTable = new();
+    private ConcurrentDictionary<string, CancellationTokenSource> CancellationTokenSourceTable = new();
+
+    private Optional<Peripherals.JtagClient.Jtag> GetJtagClient(string userName)
+    {
+        try
+        {
+            using var db = new Database.AppDataConnection();
+            var board = db.GetBoardByUserName(userName);
+            if (!board.IsSuccessful)
+            {
+                logger.Error($"Find Board {board.Value.Value.ID} failed because {board.Error}");
+                return new(null);
+            }
+            if (!board.Value.HasValue)
+            {
+                logger.Error($"Board {board.Value.Value.ID} not found");
+                return new(null);
+            }
+
+            var jtag = new Peripherals.JtagClient.Jtag(board.Value.Value.IpAddr, board.Value.Value.Port);
+            return new(jtag);
+        }
+        catch (Exception error)
+        {
+            logger.Error(error);
+            return new(null);
+        }
+    }
+
+    public async Task<bool> SetBoundaryScanFreq(int freq)
+    {
+        try
+        {
+            var userName = Context.User?.FindFirstValue(ClaimTypes.Name);
+            if (userName is null)
+            {
+                logger.Error("Can't get user info");
+                return false;
+            }
+
+            FreqTable.AddOrUpdate(userName, freq, (key, value) => freq);
+            return true;
+        }
+        catch (Exception error)
+        {
+            logger.Error(error);
+            return false;
+        }
+    }
+
+    public async Task<bool> StartBoundaryScan(int freq = 100)
+    {
+        try
+        {
+            var userName = Context.User?.FindFirstValue(ClaimTypes.Name);
+            if (userName is null)
+            {
+                logger.Error("No Such User");
+                return false;
+            }
+
+            await SetBoundaryScanFreq(freq);
+            var cts = CancellationTokenSource.CreateLinkedTokenSource();
+            CancellationTokenSourceTable.AddOrUpdate(userName, cts, (key, value) => cts);
+
+            _ = Task
+                .Run(
+                    () => BoundaryScanLogicPorts(
+                        Context.ConnectionId,
+                        userName,
+                        cts.Token),
+                    cts.Token)
+                .ContinueWith((task) =>
+                {
+                    if (!task.IsFaulted)
+                    {
+                        return;
+                    }
+
+                    if (task.Exception.InnerException is OperationCanceledException)
+                    {
+                        logger.Info($"Boundary scan operation cancelled for user {userName}");
+                    }
+                    else
+                    {
+                        logger.Error(task.Exception);
+                    }
+                });
+
+            return true;
+        }
+        catch (Exception error)
+        {
+            logger.Error(error);
+            return false;
+        }
+    }
+
+    public async Task<bool> StopBoundaryScan()
+    {
+        var userName = Context.User?.FindFirstValue(ClaimTypes.Name);
+        if (userName is null)
+        {
+            logger.Error("No Such User");
+            return false;
+        }
+
+        if (!CancellationTokenSourceTable.TryGetValue(userName, out var cts))
+        {
+            return false;
+        }
+
+        cts.Cancel();
+        cts.Token.WaitHandle.WaitOne();
+        return true;
+    }
+
+    private async void BoundaryScanLogicPorts(string connectionID, string userName, CancellationToken cancellationToken)
+    {
+        var jtagCtrl = GetJtagClient(userName).OrThrow(() => new InvalidOperationException("JTAG client not found"));
+        var cntFail = 0;
+
+        while (true && cntFail < 5)
+        {
+            cancellationToken.ThrowIfCancellationRequested();
+
+            var ret = await jtagCtrl.BoundaryScanLogicalPorts();
+            if (!ret.IsSuccessful)
+            {
+                logger.Error($"User {userName} boundary scan failed for device {jtagCtrl.address}: {ret.Error}");
+                cntFail++;
+            }
+
+            await this.Clients.Client(connectionID).OnReceiveBoundaryScanData(ret.Value);
+            // logger.Info($"User {userName} successfully completed boundary scan for device {jtagCtrl.address}");
+
+            await Task.Delay(FreqTable.TryGetValue(userName, out var freq) ? 1000 / freq : 1000 / 100, cancellationToken);
+        }
+
+        if (cntFail >= 5)
+        {
+            logger.Error($"User {userName} boundary scan failed for device {jtagCtrl.address} after 5 attempts");
+            throw new InvalidOperationException("Boundary scan failed");
+        }
+    }
+
+}

server/src/Peripherals/JtagClient.cs

@@ -386,7 +386,10 @@ public class Jtag
     readonly int timeout;
 
     readonly int port;
-    readonly string address;
+    /// <summary>
+    /// Jtag控制器IP地址
+    /// </summary>
+    public readonly string address;
     private IPEndPoint ep;
 
     /// <summary>