diff --git a/routers/api/v1/robot/robot.go b/routers/api/v1/robot/robot.go index 78f7130..f26eb9f 100644 --- a/routers/api/v1/robot/robot.go +++ b/routers/api/v1/robot/robot.go @@ -120,14 +120,28 @@ func FirstRequest(c *fiber.Ctx) error { func addRobotSSEMessage(newRobot *structs.Robot) { totalPages := cache.GetAllRobots(rspagination.PageQuery{Page: 1}).TotalPages + apiRobot := structs.APIRobot{ + Id: newRobot.Id, + Type: newRobot.Type, + Name: newRobot.Name, + Status: newRobot.Status, + Address: newRobot.Address, + CurrentJobName: newRobot.CurrentJobName, + JobsWaitingCount: newRobot.JobsWaitingCount, + JobsWaitingNameList: []string{}, + FirmwareVersion: newRobot.FirmwareVersion, + ConnectedAt: newRobot.ConnectedAt, + CreatedAt: newRobot.CreatedAt, + } + robot.BroadcastSSEMessage(structs.SSEMessage{ Cmd: utils.SSESentCmdAddRobot, Body: struct { - Robot *structs.Robot + Robot *structs.APIRobot TotalPages int UnauthorizedRobotsTotalPages int }{ - Robot: newRobot, + Robot: &apiRobot, TotalPages: totalPages, UnauthorizedRobotsTotalPages: cache.GetAllUnauthorizedRobots(rspagination.PageQuery{Page: 1}).TotalPages, },