refactor: 项目结构优化与代码生成模板修复

This commit is contained in:
Ray.Hao
2026-04-02 09:24:24 +08:00
parent 9cd3ff88f8
commit c7c6799af4
52 changed files with 153 additions and 147 deletions

View File

@@ -0,0 +1,45 @@
package com.youlai.boot.message.controller;
import com.youlai.boot.common.result.Result;
import com.youlai.boot.framework.security.model.SysUserDetails;
import com.youlai.boot.framework.security.util.SecurityUtils;
import com.youlai.boot.message.service.SseService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
/**
* SSE 控制器
*/
@Tag(name = "14. SSE连接")
@Slf4j
@RestController
@RequestMapping("/api/v1/sse")
@RequiredArgsConstructor
public class SseController {
private final SseService sseService;
@Operation(summary = "建立SSE连接")
@GetMapping(value = "/connect", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
public SseEmitter connect() {
SysUserDetails user = SecurityUtils.getUser().orElse(null);
if (user == null) {
log.warn("SSE连接失败未获取到当前用户");
return null;
}
return sseService.createConnection(user.getUsername());
}
@Operation(summary = "获取在线用户数")
@GetMapping("/online-count")
public Result<Integer> getOnlineCount() {
return Result.success(sseService.getOnlineUserCount());
}
}