package com.ruoyi.web.controller.webSocket;
|
|
import com.ruoyi.common.core.controller.BaseController;
|
import com.ruoyi.common.core.domain.R;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Set;
|
|
/**
|
* WebSocket控制器
|
* 提供REST接口用于触发服务端主动发送消息
|
* 支持基于AppUser的消息发送
|
*/
|
@RestController
|
@RequestMapping("/open/websocket")
|
@Slf4j
|
public class WebSocketController extends BaseController {
|
|
|
@Autowired
|
private WebSocketUserService webSocketUserService;
|
|
// ==================== 基于AppUser的WebSocket接口 ====================
|
|
/**
|
* 向指定用户发送消息
|
* @param appUserId 用户ID
|
* @param request 请求参数
|
* @return 操作结果
|
*/
|
@PostMapping("/user/send/{appUserId}")
|
public R<String> sendMessageToUser(@PathVariable String appUserId, @RequestBody Map<String, String> request) {
|
try {
|
String message = request.get("message");
|
if (message == null || message.trim().isEmpty()) {
|
return R.fail("消息内容不能为空");
|
}
|
|
int deviceCount = webSocketUserService.sendMessageToUser(appUserId, message);
|
if (deviceCount > 0) {
|
return R.ok("消息发送成功，发送到 " + deviceCount + " 个设备");
|
} else {
|
return R.fail("用户不在线或发送失败");
|
}
|
} catch (Exception e) {
|
log.error("向用户发送消息失败", e);
|
return R.fail("发送消息失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 向指定用户发送JSON格式消息
|
* @param appUserId 用户ID
|
* @param request 请求参数
|
* @return 操作结果
|
*/
|
@PostMapping("/user/send-json/{appUserId}")
|
public R<String> sendJsonMessageToUser(@PathVariable String appUserId, @RequestBody Map<String, Object> request) {
|
try {
|
String messageType = (String) request.get("messageType");
|
Object data = request.get("data");
|
|
if (messageType == null || messageType.trim().isEmpty()) {
|
return R.fail("消息类型不能为空");
|
}
|
|
int deviceCount = webSocketUserService.sendJsonMessageToUser(appUserId, messageType, data);
|
if (deviceCount > 0) {
|
return R.ok("JSON消息发送成功，发送到 " + deviceCount + " 个设备");
|
} else {
|
return R.fail("用户不在线或发送失败");
|
}
|
} catch (Exception e) {
|
log.error("向用户发送JSON消息失败", e);
|
return R.fail("发送消息失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 向多个用户发送消息
|
* @param request 请求参数
|
* @return 操作结果
|
*/
|
@PostMapping("/user/send-multiple")
|
public R<String> sendMessageToUsers(@RequestBody Map<String, Object> request) {
|
try {
|
@SuppressWarnings("unchecked")
|
List<String> appUserIds = (List<String>) request.get("appUserIds");
|
String message = (String) request.get("message");
|
|
if (appUserIds == null || appUserIds.isEmpty()) {
|
return R.fail("用户ID列表不能为空");
|
}
|
if (message == null || message.trim().isEmpty()) {
|
return R.fail("消息内容不能为空");
|
}
|
|
int successUserCount = webSocketUserService.sendMessageToUsers(appUserIds, message);
|
return R.ok("消息发送完成，成功发送到 " + successUserCount + " 个用户");
|
} catch (Exception e) {
|
log.error("向多个用户发送消息失败", e);
|
return R.fail("发送消息失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 发送任务分配消息
|
* @param request 请求参数
|
* @return 操作结果
|
*/
|
@PostMapping("/user/task-assignment")
|
public R<String> sendTaskAssignment(@RequestBody Map<String, String> request) {
|
try {
|
String taskId = request.get("taskId");
|
String taskName = request.get("taskName");
|
String assignee = request.get("assignee");
|
String assigneeUserId = request.get("assigneeUserId");
|
|
if (taskId == null || taskName == null || assignee == null) {
|
return R.fail("任务ID、任务名称和分配人不能为空");
|
}
|
|
int successCount = webSocketUserService.sendTaskAssignment(taskId, taskName, assignee, assigneeUserId);
|
return R.ok("任务分配消息发送完成，成功发送到 " + successCount + " 个设备");
|
} catch (Exception e) {
|
log.error("发送任务分配消息失败", e);
|
return R.fail("发送任务分配消息失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 发送状态更新消息
|
* @param request 请求参数
|
* @return 操作结果
|
*/
|
@PostMapping("/user/status-update")
|
public R<String> sendStatusUpdate(@RequestBody Map<String, String> request) {
|
try {
|
String status = request.get("status");
|
String details = request.get("details");
|
String targetUserId = request.get("targetUserId");
|
|
if (status == null || status.trim().isEmpty()) {
|
return R.fail("状态不能为空");
|
}
|
|
int successCount = webSocketUserService.sendStatusUpdate(status, details, targetUserId);
|
return R.ok("状态更新消息发送完成，成功发送到 " + successCount + " 个设备");
|
} catch (Exception e) {
|
log.error("发送状态更新消息失败", e);
|
return R.fail("发送状态更新消息失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 发送任务状态更新消息
|
* @param request 请求参数
|
* @return 操作结果
|
*/
|
@PostMapping("/user/task-status-update")
|
public R<String> sendTaskStatusUpdate(@RequestBody Map<String, String> request) {
|
try {
|
String taskId = request.get("taskId");
|
String taskStatus = request.get("taskStatus");
|
String message = request.get("message");
|
String targetUserId = request.get("targetUserId");
|
|
if (taskId == null || taskStatus == null) {
|
return R.fail("任务ID和任务状态不能为空");
|
}
|
|
int successCount = webSocketUserService.sendTaskStatusUpdate(taskId, taskStatus, message, targetUserId);
|
return R.ok("任务状态更新消息发送完成，成功发送到 " + successCount + " 个设备");
|
} catch (Exception e) {
|
log.error("发送任务状态更新消息失败", e);
|
return R.fail("发送任务状态更新消息失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 发送位置更新消息
|
* @param request 请求参数
|
* @return 操作结果
|
*/
|
@PostMapping("/user/location-update")
|
public R<String> sendLocationUpdate(@RequestBody Map<String, String> request) {
|
try {
|
String userId = request.get("userId");
|
String longitude = request.get("longitude");
|
String latitude = request.get("latitude");
|
String address = request.get("address");
|
|
if (userId == null || longitude == null || latitude == null) {
|
return R.fail("用户ID、经度和纬度不能为空");
|
}
|
|
int successCount = webSocketUserService.sendLocationUpdate(userId, longitude, latitude, address);
|
return R.ok("位置更新消息发送完成，成功发送到 " + successCount + " 个设备");
|
} catch (Exception e) {
|
log.error("发送位置更新消息失败", e);
|
return R.fail("发送位置更新消息失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 获取用户连接信息
|
* @param appUserId 用户ID
|
* @return 连接信息
|
*/
|
@GetMapping("/user/info/{appUserId}")
|
public R getUserConnectionInfo(@PathVariable String appUserId) {
|
try {
|
Map<String, Object> info = webSocketUserService.getUserConnectionInfo(appUserId);
|
return R.ok(info, "获取用户连接信息成功");
|
} catch (Exception e) {
|
log.error("获取用户连接信息失败", e);
|
return R.fail("获取用户连接信息失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 获取所有连接统计信息
|
* @return 统计信息
|
*/
|
@GetMapping("/user/stats")
|
public R getConnectionStats() {
|
try {
|
Map<String, Object> stats = webSocketUserService.getConnectionStats();
|
return R.ok(stats, "获取连接统计信息成功");
|
} catch (Exception e) {
|
log.error("获取连接统计信息失败", e);
|
return R.fail("获取连接统计信息失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 检查用户是否在线
|
* @param appUserId 用户ID
|
* @return 检查结果
|
*/
|
@GetMapping("/user/online/{appUserId}")
|
public R checkUserOnline(@PathVariable String appUserId) {
|
try {
|
boolean isOnline = webSocketUserService.isUserOnline(appUserId);
|
return R.ok(isOnline, "用户在线状态检查完成");
|
} catch (Exception e) {
|
log.error("检查用户在线状态失败", e);
|
return R.fail("检查用户在线状态失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 获取在线用户列表
|
* @return 在线用户列表
|
*/
|
@GetMapping("/user/online-list")
|
public R getOnlineUserIds() {
|
try {
|
Set<String> onlineUserIds = webSocketUserService.getOnlineUserIds();
|
return R.ok(onlineUserIds, "获取在线用户列表成功");
|
} catch (Exception e) {
|
log.error("获取在线用户列表失败", e);
|
return R.fail("获取在线用户列表失败: " + e.getMessage());
|
}
|
}
|
}
|
