Guwan
/
yl-backend
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(视频): 

视频初步
This commit is contained in:
Guwan 2024-12-08 19:25:20 +08:00
parent 92ce67245d
commit 12cfe0a184
12 changed files with 423 additions and 384 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pom.xml
View File

@ -229,13 +229,6 @@
<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<!-- SRS Java SDK -->
<dependency>
<groupId>com.github.ossrs</groupId>
<artifactId>srs-sdk</artifactId>
<version>1.0.0</version>
</dependency>
<!-- Netty -->
<dependency>
<groupId>io.netty</groupId>
@ -245,7 +238,7 @@
<!-- Easy-Es -->
<dependency>
<groupId>cn.easyes</groupId>
<groupId>cn.easy-es</groupId>
<artifactId>easy-es-boot-starter</artifactId>
<version>1.1.1</version>
</dependency>

166
src/main/java/com/guwan/backend/client/SrsClient.java
View File

@ -1,83 +1,83 @@
package com.guwan.backend.client;
import lombok.Data;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;
@Slf4j
@Component
@RequiredArgsConstructor
public class SrsClient {
private final RestTemplate restTemplate;
@Value("${srs.server.url}")
private String srsServerUrl;
/**
* 开始录制
*/
public void startRecord(String streamKey, RecordConfig config) {
String url = String.format("%s/api/v1/streams/%s/recording/start", srsServerUrl, streamKey);
try {
ResponseEntity<String> response = restTemplate.postForEntity(url, config, String.class);
if (!response.getStatusCode().is2xxSuccessful()) {
throw new RuntimeException("开始录制失败: " + response.getBody());
}
} catch (Exception e) {
log.error("开始录制失败", e);
throw new RuntimeException("开始录制失败", e);
}
}
/**
* 停止录制
*/
public void stopRecord(String streamKey) {
String url = String.format("%s/api/v1/streams/%s/recording/stop", srsServerUrl, streamKey);
try {
ResponseEntity<String> response = restTemplate.postForEntity(url, null, String.class);
if (!response.getStatusCode().is2xxSuccessful()) {
throw new RuntimeException("停止录制失败: " + response.getBody());
}
} catch (Exception e) {
log.error("停止录制失败", e);
throw new RuntimeException("停止录制失败", e);
}
}
/**
* 获取流信息
*/
public StreamInfo getStreamInfo(String streamKey) {
String url = String.format("%s/api/v1/streams/%s", srsServerUrl, streamKey);
try {
ResponseEntity<StreamInfo> response = restTemplate.getForEntity(url, StreamInfo.class);
return response.getBody();
} catch (Exception e) {
log.error("获取流信息失败", e);
throw new RuntimeException("获取流信息失败", e);
}
}
@Data
public static class RecordConfig {
private String format = "flv"; // 录制格式
private String filePath; // 文件路径
}
@Data
public static class StreamInfo {
private String streamId; // 流ID
private String clientId; // 客户端ID
private String ip; // 客户端IP
private Long startTime; // 开始时间
private String status; // 状态
private Long bytesIn; // 输入字节数
private Long bytesOut; // 输出字节数
}
}
//package com.guwan.backend.client;
//
//import lombok.Data;
//import lombok.RequiredArgsConstructor;
//import lombok.extern.slf4j.Slf4j;
//import org.springframework.beans.factory.annotation.Value;
//import org.springframework.http.ResponseEntity;
//import org.springframework.stereotype.Component;
//import org.springframework.web.client.RestTemplate;
//
//@Slf4j
//@Component
//@RequiredArgsConstructor
//public class SrsClient {
//
// private final RestTemplate restTemplate;
//
// @Value("${srs.server.url}")
// private String srsServerUrl;
//
// /**
// * 开始录制
// */
// public void startRecord(String streamKey, RecordConfig config) {
// String url = String.format("%s/api/v1/streams/%s/recording/start", srsServerUrl, streamKey);
// try {
// ResponseEntity<String> response = restTemplate.postForEntity(url, config, String.class);
// if (!response.getStatusCode().is2xxSuccessful()) {
// throw new RuntimeException("开始录制失败: " + response.getBody());
// }
// } catch (Exception e) {
// log.error("开始录制失败", e);
// throw new RuntimeException("开始录制失败", e);
// }
// }
//
// /**
// * 停止录制
// */
// public void stopRecord(String streamKey) {
// String url = String.format("%s/api/v1/streams/%s/recording/stop", srsServerUrl, streamKey);
// try {
// ResponseEntity<String> response = restTemplate.postForEntity(url, null, String.class);
// if (!response.getStatusCode().is2xxSuccessful()) {
// throw new RuntimeException("停止录制失败: " + response.getBody());
// }
// } catch (Exception e) {
// log.error("停止录制失败", e);
// throw new RuntimeException("停止录制失败", e);
// }
// }
//
// /**
// * 获取流信息
// */
// public StreamInfo getStreamInfo(String streamKey) {
// String url = String.format("%s/api/v1/streams/%s", srsServerUrl, streamKey);
// try {
// ResponseEntity<StreamInfo> response = restTemplate.getForEntity(url, StreamInfo.class);
// return response.getBody();
// } catch (Exception e) {
// log.error("获取流信息失败", e);
// throw new RuntimeException("获取流信息失败", e);
// }
// }
//
// @Data
// public static class RecordConfig {
// private String format = "flv"; // 录制格式
// private String filePath; // 文件路径
// }
//
// @Data
// public static class StreamInfo {
// private String streamId; // 流ID
// private String clientId; // 客户端ID
// private String ip; // 客户端IP
// private Long startTime; // 开始时间
// private String status; // 状态
// private Long bytesIn; // 输入字节数
// private Long bytesOut; // 输出字节数
// }
//}

164
src/main/java/com/guwan/backend/controller/LiveController.java
View File

@ -1,82 +1,82 @@
package com.guwan.backend.controller;
import com.guwan.backend.entity.LiveRoom;
import com.guwan.backend.entity.LiveRoomDTO;
import com.guwan.backend.service.LiveService;
import com.guwan.backend.util.Result;
import com.guwan.backend.util.SecurityUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.security.SecurityRequirement;
import io.swagger.v3.oas.annotations.tags.Tag;
@Slf4j
@Tag(name = "直播管理", description = "直播相关接口")
@RestController
@RequestMapping("/api/live")
@RequiredArgsConstructor
public class LiveController {
private final LiveService liveService;
private final SecurityUtil securityUtil;
@Operation(summary = "创建直播间")
@SecurityRequirement(name = "bearer-jwt")
@PostMapping("/room")
public Result<LiveRoom> createLiveRoom(@RequestBody LiveRoomDTO dto) {
try {
return Result.success(liveService.createLiveRoom(dto));
} catch (Exception e) {
log.error("创建直播间失败", e);
return Result.error(e.getMessage());
}
}
@Operation(summary = "开始直播")
@SecurityRequirement(name = "bearer-jwt")
@PostMapping("/room/{id}/start")
public Result<Void> startLive(@PathVariable Long id) {
try {
// 检查权限
checkPermission(id);
liveService.startLive(id);
return Result.success();
} catch (Exception e) {
log.error("开始直播失败", e);
return Result.error(e.getMessage());
}
}
@Operation(summary = "结束直播")
@SecurityRequirement(name = "bearer-jwt")
@PostMapping("/room/{id}/end")
public Result<Void> endLive(@PathVariable Long id) {
try {
// 检查权限
checkPermission(id);
liveService.endLive(id);
return Result.success();
} catch (Exception e) {
log.error("结束直播失败", e);
return Result.error(e.getMessage());
}
}
/**
* 检查权限
*/
private void checkPermission(Long roomId) {
LiveRoom room = liveService.getLiveRoom(roomId);
if (room == null) {
throw new IllegalArgumentException("直播间不存在");
}
Long currentUserId = securityUtil.getCurrentUserId();
if (!room.getUserId().equals(currentUserId)) {
throw new IllegalStateException("无权操作此直播间");
}
}
}
//package com.guwan.backend.controller;
//
//import com.guwan.backend.entity.LiveRoom;
//import com.guwan.backend.entity.LiveRoomDTO;
//import com.guwan.backend.service.LiveService;
//import com.guwan.backend.util.Result;
//import com.guwan.backend.util.SecurityUtil;
//import lombok.RequiredArgsConstructor;
//import lombok.extern.slf4j.Slf4j;
//import org.springframework.web.bind.annotation.*;
//import io.swagger.v3.oas.annotations.Operation;
//import io.swagger.v3.oas.annotations.media.Schema;
//import io.swagger.v3.oas.annotations.security.SecurityRequirement;
//import io.swagger.v3.oas.annotations.tags.Tag;
//
//@Slf4j
//@Tag(name = "直播管理", description = "直播相关接口")
//@RestController
//@RequestMapping("/api/live")
//@RequiredArgsConstructor
//public class LiveController {
//
// private final LiveService liveService;
// private final SecurityUtil securityUtil;
//
// @Operation(summary = "创建直播间")
// @SecurityRequirement(name = "bearer-jwt")
// @PostMapping("/room")
// public Result<LiveRoom> createLiveRoom(@RequestBody LiveRoomDTO dto) {
// try {
// return Result.success(liveService.createLiveRoom(dto));
// } catch (Exception e) {
// log.error("创建直播间失败", e);
// return Result.error(e.getMessage());
// }
// }
//
// @Operation(summary = "开始直播")
// @SecurityRequirement(name = "bearer-jwt")
// @PostMapping("/room/{id}/start")
// public Result<Void> startLive(@PathVariable Long id) {
// try {
// // 检查权限
// checkPermission(id);
// liveService.startLive(id);
// return Result.success();
// } catch (Exception e) {
// log.error("开始直播失败", e);
// return Result.error(e.getMessage());
// }
// }
//
// @Operation(summary = "结束直播")
// @SecurityRequirement(name = "bearer-jwt")
// @PostMapping("/room/{id}/end")
// public Result<Void> endLive(@PathVariable Long id) {
// try {
// // 检查权限
// checkPermission(id);
// liveService.endLive(id);
// return Result.success();
// } catch (Exception e) {
// log.error("结束直播失败", e);
// return Result.error(e.getMessage());
// }
// }
//
// /**
// * 检查权限
// */
// private void checkPermission(Long roomId) {
// LiveRoom room = liveService.getLiveRoom(roomId);
// if (room == null) {
// throw new IllegalArgumentException("直播间不存在");
// }
//
// Long currentUserId = securityUtil.getCurrentUserId();
// if (!room.getUserId().equals(currentUserId)) {
// throw new IllegalStateException("无权操作此直播间");
// }
// }
//}

4
src/main/java/com/guwan/backend/controller/VideoController.java
View File

@ -27,12 +27,12 @@ public class VideoController {
// @SecurityRequirement(name = "bearer-jwt")
@PostMapping("/upload")
public Result<VideoDTO> uploadVideo(
@Parameter(description = "视频文件") @RequestParam("file") MultipartFile file,
@Parameter(description = "视频文件") @RequestParam("fileUrl") String fileUrl,
@Parameter(description = "视频标题") @RequestParam("title") String title,
@Parameter(description = "视频描述") @RequestParam("description") String description,
@Parameter(description = "视频标签，多个用逗号分隔") @RequestParam(value = "tags", required = false) String tags) {
try {
VideoDTO video = videoService.uploadVideo(file, title, description, tags);
VideoDTO video = videoService.uploadVideo(fileUrl, title, description, tags);
return Result.success(video);
} catch (Exception e) {
log.error("上传视频失败", e);

6
src/main/java/com/guwan/backend/entity/LiveMessage.java
View File

@ -1,3 +1,9 @@
package com.guwan.backend.entity;
import lombok.Data;
import java.time.LocalDateTime;
@Data
public class LiveMessage {
private String type; // 消息类型：CHAT-聊天，GIFT-礼物，LIKE-点赞，ENTER-进入，LEAVE-离开

7
src/main/java/com/guwan/backend/entity/LiveRoom.java
View File

@ -1,3 +1,10 @@
package com.guwan.backend.entity;
import com.baomidou.mybatisplus.annotation.*;
import lombok.Data;
import java.time.LocalDateTime;
@Data
@TableName("live_room")
public class LiveRoom {

252
src/main/java/com/guwan/backend/service/LiveService.java
View File

@ -1,120 +1,132 @@
@Slf4j
@Service
@RequiredArgsConstructor
public class LiveService {
private final LiveRoomMapper liveRoomMapper;
private final SrsClient srsClient;
private final SecurityUtil securityUtil;
/**
* 创建直播间
*/
public LiveRoom createLiveRoom(LiveRoomDTO dto) {
// 获取当前用户
Long userId = securityUtil.getCurrentUserId();
String username = securityUtil.getCurrentUsername();
if (userId == null) {
throw new IllegalStateException("用户未登录");
}
LiveRoom room = new LiveRoom();
BeanUtils.copyProperties(dto, room);
// 设置用户信息
room.setUserId(userId);
room.setUsername(username);
// 生成推流密钥
String streamKey = generateStreamKey();
room.setStreamKey(streamKey);
// 生成推流地址
String rtmpUrl = generateRtmpUrl(streamKey);
room.setRtmpUrl(rtmpUrl);
// 生成播放地址
String hlsUrl = generateHlsUrl(streamKey);
room.setHlsUrl(hlsUrl);
room.setStatus("PREPARING");
room.setOnlineCount(0);
room.setLikeCount(0);
liveRoomMapper.insert(room);
return room;
}
/**
* 开始直播
*/
public void startLive(Long roomId) {
LiveRoom room = liveRoomMapper.selectById(roomId);
room.setStatus("LIVING");
liveRoomMapper.updateById(room);
// 开始录制
startRecording(room.getStreamKey());
}
/**
* 结束直播
*/
public void endLive(Long roomId) {
LiveRoom room = liveRoomMapper.selectById(roomId);
room.setStatus("ENDED");
liveRoomMapper.updateById(room);
// 停止录制
stopRecording(room.getStreamKey());
// 生成回放地址
String replayUrl = generateReplayUrl(room.getStreamKey());
room.setReplayUrl(replayUrl);
liveRoomMapper.updateById(room);
}
/**
* 开始录制
*/
private void startRecording(String streamKey) {
SrsClient.RecordConfig config = new SrsClient.RecordConfig();
config.setFilePath("/recordings/" + streamKey + ".flv");
srsClient.startRecord(streamKey, config);
}
/**
* 停止录制
*/
private void stopRecording(String streamKey) {
srsClient.stopRecord(streamKey);
}
/**
* 生成推流密钥
*/
private String generateStreamKey() {
return UUID.randomUUID().toString();
}
/**
* 生成推流地址
*/
private String generateRtmpUrl(String streamKey) {
return "rtmp://localhost:1935/live/" + streamKey;
}
/**
* 生成播放地址
*/
private String generateHlsUrl(String streamKey) {
return "http://localhost:8088/live/" + streamKey + ".m3u8";
}
/**
* 生成回放地址
*/
private String generateReplayUrl(String streamKey) {
return "http://localhost:8080/recordings/" + streamKey + ".flv";
}
}
//package com.guwan.backend.service;
//
//import com.guwan.backend.client.SrsClient;
//import com.guwan.backend.dto.live.LiveRoomDTO;
//import com.guwan.backend.entity.LiveRoom;
//import com.guwan.backend.mapper.LiveRoomMapper;
//import com.guwan.backend.util.SecurityUtil;
//import lombok.RequiredArgsConstructor;
//import lombok.extern.slf4j.Slf4j;
//import org.springframework.beans.BeanUtils;
//import org.springframework.stereotype.Service;
//
//@Slf4j
//@Service
//@RequiredArgsConstructor
//public class LiveService {
//
// private final LiveRoomMapper liveRoomMapper;
// private final SrsClient srsClient;
// private final SecurityUtil securityUtil;
//
// /**
// * 创建直播间
// */
// public LiveRoom createLiveRoom(LiveRoomDTO dto) {
// // 获取当前用户
// Long userId = securityUtil.getCurrentUserId();
// String username = securityUtil.getCurrentUsername();
// if (userId == null) {
// throw new IllegalStateException("用户未登录");
// }
//
// LiveRoom room = new LiveRoom();
// BeanUtils.copyProperties(dto, room);
//
// // 设置用户信息
// room.setUserId(userId);
// room.setUsername(username);
//
// // 生成推流密钥
// String streamKey = generateStreamKey();
// room.setStreamKey(streamKey);
//
// // 生成推流地址
// String rtmpUrl = generateRtmpUrl(streamKey);
// room.setRtmpUrl(rtmpUrl);
//
// // 生成播放地址
// String hlsUrl = generateHlsUrl(streamKey);
// room.setHlsUrl(hlsUrl);
//
// room.setStatus("PREPARING");
// room.setOnlineCount(0);
// room.setLikeCount(0);
//
// liveRoomMapper.insert(room);
// return room;
// }
//
// /**
// * 开始直播
// */
// public void startLive(Long roomId) {
// LiveRoom room = liveRoomMapper.selectById(roomId);
// room.setStatus("LIVING");
// liveRoomMapper.updateById(room);
//
// // 开始录制
// startRecording(room.getStreamKey());
// }
//
// /**
// * 结束直播
// */
// public void endLive(Long roomId) {
// LiveRoom room = liveRoomMapper.selectById(roomId);
// room.setStatus("ENDED");
// liveRoomMapper.updateById(room);
//
// // 停止录制
// stopRecording(room.getStreamKey());
//
// // 生成回放地址
// String replayUrl = generateReplayUrl(room.getStreamKey());
// room.setReplayUrl(replayUrl);
// liveRoomMapper.updateById(room);
// }
//
// /**
// * 开始录制
// */
// private void startRecording(String streamKey) {
// SrsClient.RecordConfig config = new SrsClient.RecordConfig();
// config.setFilePath("/recordings/" + streamKey + ".flv");
// srsClient.startRecord(streamKey, config);
// }
//
// /**
// * 停止录制
// */
// private void stopRecording(String streamKey) {
// srsClient.stopRecord(streamKey);
// }
//
// /**
// * 生成推流密钥
// */
// private String generateStreamKey() {
// return UUID.randomUUID().toString();
// }
//
// /**
// * 生成推流地址
// */
// private String generateRtmpUrl(String streamKey) {
// return "rtmp://localhost:1935/live/" + streamKey;
// }
//
// /**
// * 生成播放地址
// */
// private String generateHlsUrl(String streamKey) {
// return "http://localhost:8088/live/" + streamKey + ".m3u8";
// }
//
// /**
// * 生成回放地址
// */
// private String generateReplayUrl(String streamKey) {
// return "http://localhost:8080/recordings/" + streamKey + ".flv";
// }
//}

3
src/main/java/com/guwan/backend/service/VideoService.java
View File

@ -2,11 +2,10 @@ package com.guwan.backend.service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.guwan.backend.dto.video.VideoDTO;
import org.springframework.web.multipart.MultipartFile;
public interface VideoService {
// 上传视频
VideoDTO uploadVideo(MultipartFile file, String title, String description, String tags);
VideoDTO uploadVideo(String fileUrl, String title, String description, String tags);
// 更新视频信息
VideoDTO updateVideo(VideoDTO videoDTO);

9
src/main/java/com/guwan/backend/service/impl/VideoServiceImpl.java
View File

@ -18,7 +18,6 @@ import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
import java.util.List;
@ -36,7 +35,7 @@ public class VideoServiceImpl implements VideoService {
@Override
@Transactional
@OperationLog(description = "上传视频", operationType = "上传")
public VideoDTO uploadVideo(MultipartFile file, String title, String description, String tags) {
public VideoDTO uploadVideo(String fileUrl, String title, String description, String tags) {
// 获取当前用户
Long userId = securityUtil.getCurrentUserId();
if (userId == null) {
@ -45,15 +44,15 @@ public class VideoServiceImpl implements VideoService {
try {
// 上传视频文件到MinIO
String fileName = minioUtil.uploadFile("videos", file);
String url = minioUtil.getUrl(minioUtil.getFileUrl("videos", fileName));
String url = fileUrl;
// 创建视频记录
Video video = new Video();
video.setTitle(title);
video.setDescription(description);
video.setUrl(url);
video.setSize(file.getSize());
// video.setSize(file.getSize());
video.setUserId(userId);
video.setStatus("PUBLISHED");
video.setTags(tags);

16
src/main/java/com/guwan/backend/util/MinioUtil.java
View File

@ -172,6 +172,22 @@ public class MinioUtil {
return UUID.randomUUID() + "." + extension;
}
public long getFileSize(String bucketName, String objectName) {
try {
StatObjectResponse objectInfo = minioClient.statObject(
StatObjectArgs.builder()
.bucket(bucketName)
.object(objectName)
.build());
return objectInfo.size();
} catch (Exception e) {
e.printStackTrace();
return 0;
}
}
// 初始化时创建视频桶
@PostConstruct
public void init() {

164
src/main/java/com/guwan/backend/websocket/LiveWebSocketHandler.java
View File

@ -1,82 +1,82 @@
package com.guwan.backend.websocket;
import com.alibaba.fastjson.JSON;
import com.guwan.backend.entity.LiveMessage;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;
import java.io.IOException;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentHashSet;
@Slf4j
@Component
public class LiveWebSocketHandler extends TextWebSocketHandler {
private static final Map<String, Set<WebSocketSession>> roomSessions = new ConcurrentHashMap<>();
private static final Map<String, WebSocketSession> userSessions = new ConcurrentHashMap<>();
@Override
public void afterConnectionEstablished(WebSocketSession session) {
String roomId = getRoomId(session);
String userId = getUserId(session);
// 加入房间
roomSessions.computeIfAbsent(roomId, k -> new ConcurrentHashSet<>()).add(session);
userSessions.put(userId, session);
// 广播进入消息
broadcastMessage(roomId, createEnterMessage(userId));
}
@Override
public void handleTextMessage(WebSocketSession session, TextMessage message) {
String roomId = getRoomId(session);
LiveMessage liveMessage = JSON.parseObject(message.getPayload(), LiveMessage.class);
// 处理不同类型的消息
switch (liveMessage.getType()) {
case "CHAT":
broadcastMessage(roomId, message);
break;
case "GIFT":
handleGiftMessage(roomId, liveMessage);
break;
case "LIKE":
handleLikeMessage(roomId, liveMessage);
break;
}
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) {
String roomId = getRoomId(session);
String userId = getUserId(session);
// 离开房间
roomSessions.get(roomId).remove(session);
userSessions.remove(userId);
// 广播离开消息
broadcastMessage(roomId, createLeaveMessage(userId));
}
private void broadcastMessage(String roomId, TextMessage message) {
Set<WebSocketSession> sessions = roomSessions.get(roomId);
if (sessions != null) {
sessions.forEach(session -> {
try {
session.sendMessage(message);
} catch (IOException e) {
log.error("发送消息失败", e);
}
});
}
}
}
//package com.guwan.backend.websocket;
//
//import com.alibaba.fastjson.JSON;
//import com.guwan.backend.entity.LiveMessage;
//import lombok.extern.slf4j.Slf4j;
//import org.springframework.stereotype.Component;
//import org.springframework.web.socket.CloseStatus;
//import org.springframework.web.socket.TextMessage;
//import org.springframework.web.socket.WebSocketSession;
//import org.springframework.web.socket.handler.TextWebSocketHandler;
//
//import java.io.IOException;
//import java.util.Map;
//import java.util.Set;
//import java.util.concurrent.ConcurrentHashMap;
//import java.util.concurrent.ConcurrentHashSet;
//
//@Slf4j
//@Component
//public class LiveWebSocketHandler extends TextWebSocketHandler {
//
// private static final Map<String, Set<WebSocketSession>> roomSessions = new ConcurrentHashMap<>();
// private static final Map<String, WebSocketSession> userSessions = new ConcurrentHashMap<>();
//
// @Override
// public void afterConnectionEstablished(WebSocketSession session) {
// String roomId = getRoomId(session);
// String userId = getUserId(session);
//
// // 加入房间
// roomSessions.computeIfAbsent(roomId, k -> new ConcurrentHashSet<>()).add(session);
// userSessions.put(userId, session);
//
// // 广播进入消息
// broadcastMessage(roomId, createEnterMessage(userId));
// }
//
// @Override
// public void handleTextMessage(WebSocketSession session, TextMessage message) {
// String roomId = getRoomId(session);
// LiveMessage liveMessage = JSON.parseObject(message.getPayload(), LiveMessage.class);
//
// // 处理不同类型的消息
// switch (liveMessage.getType()) {
// case "CHAT":
// broadcastMessage(roomId, message);
// break;
// case "GIFT":
// handleGiftMessage(roomId, liveMessage);
// break;
// case "LIKE":
// handleLikeMessage(roomId, liveMessage);
// break;
// }
// }
//
// @Override
// public void afterConnectionClosed(WebSocketSession session, CloseStatus status) {
// String roomId = getRoomId(session);
// String userId = getUserId(session);
//
// // 离开房间
// roomSessions.get(roomId).remove(session);
// userSessions.remove(userId);
//
// // 广播离开消息
// broadcastMessage(roomId, createLeaveMessage(userId));
// }
//
// private void broadcastMessage(String roomId, TextMessage message) {
// Set<WebSocketSession> sessions = roomSessions.get(roomId);
// if (sessions != null) {
// sessions.forEach(session -> {
// try {
// session.sendMessage(message);
// } catch (IOException e) {
// log.error("发送消息失败", e);
// }
// });
// }
// }
//}

7
src/test/java/com/guwan/backend/util/MinioUtilTest.java
View File

@ -71,4 +71,11 @@ class MinioUtilTest {
// 删除图片
// minioUtil.deleteFile(minioConfig.getBucket().getImages(), fileName);
}
@Test
void testSize() {
System.out.println(minioUtil.getFileSize("images", "7c9360ba-0cf7-41db-9e11-6c02379e1170.jpg"));
}
}