123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173 |
- package com.ym.service.Impl;
- import com.ym.mec.common.entity.ImRoomMessage;
- import com.alibaba.fastjson.JSON;
- import com.alibaba.fastjson.JSONObject;
- import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
- import com.ym.http.HttpHelper;
- import com.ym.mec.biz.dal.dao.ImLiveBroadcastRoomDao;
- import com.ym.mec.biz.dal.entity.ImLiveBroadcastRoom;
- import com.ym.mec.biz.dal.entity.TenantAssetsInfo;
- import com.ym.mec.common.exception.BizException;
- import com.ym.mec.im.IMHelper;
- import com.ym.pojo.IMApiResultInfo;
- import com.ym.pojo.RecordConfig;
- import com.ym.pojo.RecordNotify;
- import com.ym.service.LiveRoomService;
- import lombok.extern.slf4j.Slf4j;
- import org.apache.commons.lang3.StringUtils;
- import org.redisson.api.RBucket;
- import org.redisson.api.RedissonClient;
- import org.springframework.beans.factory.annotation.Autowired;
- import org.springframework.stereotype.Service;
- import java.net.HttpURLConnection;
- import java.util.ArrayList;
- import java.util.List;
- /**
- * @author hgw
- * Created by 2022-02-21
- */
- @Slf4j
- @Service
- public class LiveRoomServiceImpl implements LiveRoomService {
- @Autowired
- private IMHelper imHelper;
- @Autowired
- private HttpHelper httpHelper;
- @Autowired
- private RedissonClient redissonClient;
- @Autowired
- private ImLiveBroadcastRoomDao imLiveBroadcastRoomDao;
- /**
- * 创建房间-聊天室
- *
- * @param roomId 房间Uid
- * @param roomName 房间名称
- */
- public IMApiResultInfo createLiveRoom(String roomId, String roomName) {
- IMApiResultInfo resultInfo;
- try {
- resultInfo = imHelper.createChatRoom(roomId, roomName);
- } catch (Exception e) {
- log.error("create chatRoom error >>>", e.getCause());
- throw new BizException("创建聊天室失败!");
- }
- if (!resultInfo.isSuccess()) {
- log.error("create chatRoom error: {}", resultInfo.getErrorMessage());
- throw new BizException("创建聊天室失败!");
- }
- log.info("create chatRoom success: {}", roomId);
- return resultInfo;
- }
- /**
- * 销毁房间-聊天室
- *
- * @param roomId 房间Uid
- */
- public IMApiResultInfo destroyLiveRoom(String roomId) {
- //删除服务器房间
- List<String> deleteRoomIds = new ArrayList<String>() {{
- add(roomId);
- }};
- IMApiResultInfo resultInfo;
- try {
- resultInfo = imHelper.deleteChrm(deleteRoomIds);
- } catch (Exception e) {
- throw new BizException("关闭聊天室失败!");
- }
- if (!resultInfo.isSuccess()) {
- log.error("destroy chatRoom error: {}", resultInfo.getErrorMessage());
- throw new BizException("关闭聊天室失败!");
- }
- return resultInfo;
- }
- /**
- * 发送消息
- *
- * @param fromUserId 发送者id
- * @param toChatroomId 房间id
- * @param message
- */
- public IMApiResultInfo publishRoomMessage(ImRoomMessage message) {
- IMApiResultInfo resultInfo;
- try {
- resultInfo = imHelper.publishRoomMessage(message.getFromUserId(), message.getToChatroomId(), message);
- } catch (Exception e) {
- throw new BizException("消息发送失败" + e.getMessage());
- }
- if (!resultInfo.isSuccess()) {
- log.error("publishRoomMessage chatRoom error: {}", resultInfo.getErrorMessage());
- throw new BizException("关闭聊天室失败!");
- }
- return resultInfo;
- }
- @Override
- public void startRecord(String roomId) throws Exception {
- log.error("开始录制直播:roomId : {} ",roomId);
- JSONObject paramJson = new JSONObject();
- paramJson.put("sessionId",getRoomSessionId(roomId));
- paramJson.put("config",new RecordConfig());
- String body = paramJson.toJSONString();
- HttpURLConnection conn = httpHelper.createIMRtcPostHttpConnection("/rtc/record/start.json", "application/json",roomId);
- httpHelper.setBodyParameter(body, conn);
- IMApiResultInfo resultInfo = JSON.parseObject(httpHelper.returnResult(conn, body), IMApiResultInfo.class);
- if(resultInfo.getCode() != 200){
- log.error("直播视频录制失败:resultInfo : {} ",resultInfo);
- }
- }
- @Override
- public void stopRecord(String roomId) throws Exception {
- JSONObject paramJson = new JSONObject();
- paramJson.put("sessionId",getRoomSessionId(roomId));
- String body = paramJson.toJSONString();
- HttpURLConnection conn = httpHelper.createIMRtcPostHttpConnection("/rtc/record/stop.json", "application/json",roomId);
- httpHelper.setBodyParameter(body, conn);
- httpHelper.returnResult(conn, body);
- RBucket<String> bucket = redissonClient.getBucket("sessionId:" + roomId);
- bucket.delete();
- log.info("结束录制直播 roomId :{},{}",roomId,httpHelper.returnResult(conn, body));
- }
- @Override
- public void recordSync(RecordNotify recordNotify) {
- log.info("recordSync recordNotify:{}",recordNotify);
- }
- public String getRoomSessionId(String roomId){
- RBucket<String> bucket = redissonClient.getBucket("sessionId:" + roomId);
- String sessionId = bucket.get();
- if(StringUtils.isNotEmpty(sessionId)){
- return sessionId;
- }
- JSONObject jsonObject = new JSONObject();
- jsonObject.put("roomId",roomId);
- HttpURLConnection conn = null;
- try {
- conn = httpHelper.createIMRtcPostHttpConnection("/rtc/room/query.json", "application/json",null);
- httpHelper.setBodyParameter(jsonObject.toJSONString(), conn);
- String returnResult = httpHelper.returnResult(conn, jsonObject.toJSONString());
- JSONObject resultObject = JSONObject.parseObject(returnResult);
- String code = resultObject.get("code").toString();
- if("200".equals(code)){
- sessionId = resultObject.get("sessionId").toString();
- bucket.set(sessionId);
- }else {
- log.error("获取sessionId失败 returnResult:{}",returnResult);
- throw new BizException("获取sessionId失败");
- }
- } catch (Exception e) {
- e.printStackTrace();
- }
- return sessionId;
- }
- }
|