shayu/cai-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

123333

Browse Source
This commit is contained in:
张良(004796)
2024-02-06 16:56:03 +08:00
parent c827dbc0aa
commit 90356b55c8
40 changed files with 622 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
ruoyi-cai/src/main/java/com/ruoyi/cai/dto/commom/im/ImgContentDTO.java Normal file
View File

@@ -0,0 +1,18 @@
package com.ruoyi.cai.dto.commom.im;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
@Data
public class ImgContentDTO {
@Schema(description = "不知道是撒")
private String serverId;
@Schema(description = "不知道是撒")
private String messageId;
@Schema(description = "时间")
private String time;
@Schema(description = "图片地址")
private String imageUrl;
@Schema(description = "OSS图片地址")
private String ossImage;
}

11
ruoyi-cai/src/main/java/com/ruoyi/cai/enums/ChatTypeEnum.java
View File

@@ -1,5 +1,6 @@
package com.ruoyi.cai.enums;
import com.ruoyi.yunxin.enums.YxImTypeEnum;
import lombok.Getter;
/**
@@ -8,11 +9,11 @@ import lombok.Getter;
* @author duet
*/
public enum ChatTypeEnum {
MESSAGE(1,0),
VOICE(2,2),
PICTURE(3,1),
VIDEO(4,3),
CUSTOM(100,100)
MESSAGE(1, YxImTypeEnum.TXT.getCode()),
VOICE(2,YxImTypeEnum.VOICE.getCode()),
IMAGE(3,YxImTypeEnum.IMAGE.getCode()),
VIDEO(4,YxImTypeEnum.VIDEO.getCode()),
CUSTOM(100,YxImTypeEnum.CUSTOM.getCode())
;
@Getter

20
ruoyi-cai/src/main/java/com/ruoyi/cai/enums/im/ImImgStatusEnum.java Normal file
View File

@@ -0,0 +1,20 @@
package com.ruoyi.cai.enums.im;
import lombok.Getter;
// 图片下载状态:0=未更新,1=已更新,2=已下载,3=下载失败
@Getter
public enum ImImgStatusEnum {
NO(0,"未更新"),
INIT(1,"已初始化"),
DOWNLOAD(2,"已下载"),
DOWNLOAD_FAIL(3,"下载失败"),
;
private final Integer code;
private final String text;
ImImgStatusEnum(Integer code, String text) {
this.code = code;
this.text = text;
}
}

122
ruoyi-cai/src/main/java/com/ruoyi/cai/manager/ImService.java
View File

@@ -1,19 +1,37 @@
package com.ruoyi.cai.manager;
import cn.hutool.core.lang.UUID;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.ruoyi.cai.domain.Account;
import com.ruoyi.cai.domain.User;
import com.ruoyi.cai.domain.UserChatRecord;
import com.ruoyi.cai.dto.app.query.im.ImMessageDTO;
import com.ruoyi.cai.dto.app.query.im.SaveSendImImgReq;
import com.ruoyi.cai.dto.app.vo.ImResp;
import com.ruoyi.cai.enums.GenderEnum;
import com.ruoyi.cai.dto.commom.im.ImgContentDTO;
import com.ruoyi.cai.enums.ChatTypeEnum;
import com.ruoyi.cai.enums.GenderEnum;
import com.ruoyi.cai.enums.UserMemberTypeEnum;
import com.ruoyi.cai.enums.im.ImImgStatusEnum;
import com.ruoyi.cai.mq.AmqpProducer;
import com.ruoyi.cai.mq.handleDelay.dto.SaveSendImImgDelayDto;
import com.ruoyi.cai.service.*;
import com.ruoyi.cai.util.FileUtils;
import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.system.domain.vo.SysOssVo;
import com.ruoyi.system.service.ISysOssService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import java.io.File;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
@Component
@Slf4j
public class ImService {
@Autowired
@@ -28,6 +46,12 @@ public class ImService {
private UserChatRecordService userChatRecordService;
@Autowired
private InnerUserFilter innerUserFilter;
@Autowired
private UserBlacklistService userBlacklistService;
@Autowired
private AmqpProducer amqpProducer;
@Autowired
private ISysOssService sysOssService;
public ImResp sendMessage(Long fromUserId, ImMessageDTO message) {
ChatTypeEnum typeEnum = ChatTypeEnum.getByType(message.getType());
@@ -135,9 +159,6 @@ public class ImService {
return resp;
}
@Autowired
private UserBlacklistService userBlacklistService;
private Long getByImPrice(Long userId){
UserMemberTypeEnum userMemberType = userMemberService.checkUserIsMember(userId);
if(userMemberType == null){
@@ -151,4 +172,97 @@ public class ImService {
}
return 10L;
}
public void saveSendImImg(SaveSendImImgReq saveSendImImgReq) {
if(StringUtils.isBlank(saveSendImImgReq.getImageUrl())){
throw new ServiceException("图片url地址不存在");
}
if(saveSendImImgReq.getRecordId() == null){
throw new ServiceException("记录不存在");
}
UserChatRecord userChatRecord = userChatRecordService.getById(saveSendImImgReq.getRecordId());
if(userChatRecord == null){
throw new ServiceException("记录不存在");
}
if(!ChatTypeEnum.IMAGE.getImType().equals(userChatRecord.getType())){
throw new ServiceException("记录不是图片类型");
}
if(!ImImgStatusEnum.NO.getCode().equals(userChatRecord.getImgStatus())){
return;
}
ImgContentDTO content = new ImgContentDTO();
content.setServerId(saveSendImImgReq.getServerId());
content.setMessageId(saveSendImImgReq.getMessageId());
content.setTime(saveSendImImgReq.getTime());
content.setImageUrl(saveSendImImgReq.getImageUrl());
boolean update = userChatRecordService.update(Wrappers.lambdaUpdate(UserChatRecord.class)
.eq(UserChatRecord::getId, userChatRecord.getId())
.eq(UserChatRecord::getImgStatus, ImImgStatusEnum.NO.getCode())
.set(UserChatRecord::getImgStatus, ImImgStatusEnum.INIT.getCode())
.set(UserChatRecord::getContent, JSON.toJSONString(content)));
if(!update){
log.warn("保存图片消息 忽略，重复调用 record={}",JSON.toJSONString(userChatRecord));
return;
}
SaveSendImImgDelayDto dto = new SaveSendImImgDelayDto();
dto.setRecordId(saveSendImImgReq.getRecordId());
amqpProducer.sendCommonDelayMq(dto,5);
}
public void saveSendImImgDeal(Long recordId) {
UserChatRecord userChatRecord = userChatRecordService.getById(recordId);
if(userChatRecord == null){
log.warn("保存图片消息 忽略，记录不存在 recordId={}",recordId);
return;
}
if(!ChatTypeEnum.IMAGE.getImType().equals(userChatRecord.getType())){
log.warn("保存图片消息 忽略，记录不是图片类型 recordId={}",recordId);
return;
}
if(ImImgStatusEnum.DOWNLOAD.getCode().equals(userChatRecord.getImgStatus())){
log.warn("保存图片消息 忽略，状态错误 已经下载了 recordId={}",recordId);
return;
}
String contentStr = userChatRecord.getContent();
if(StringUtils.isBlank(contentStr)){
log.warn("保存图片消息 忽略，状态错误 内容为空 recordId={}",recordId);
return;
}
ImgContentDTO content = JSON.parseObject(contentStr, ImgContentDTO.class);
if(content == null || StringUtils.isBlank(content.getImageUrl())){
throw new ServiceException("图片url地址不存在");
}
ImImgStatusEnum imImgStatusEnum = ImImgStatusEnum.DOWNLOAD_FAIL;
File imgeFile = null;
try {
File tempFile = FileUtils.getTempFile();
imgeFile = getSaveSendImImgTempFile(tempFile, content.getImageUrl());
FileUtils.downloadPath(content.getImageUrl(),imgeFile);
SysOssVo upload = sysOssService.upload(imgeFile);
content.setOssImage(upload.getUrl());
imImgStatusEnum = ImImgStatusEnum.DOWNLOAD;
}catch (Exception e){
log.error("更新聊天图片失败",e);
} finally {
FileUtils.deleteFile(imgeFile);
}
userChatRecordService.update(Wrappers.lambdaUpdate(UserChatRecord.class)
.eq(UserChatRecord::getId, recordId)
.set(UserChatRecord::getImgStatus, imImgStatusEnum.getCode())
.set(UserChatRecord::getContent, JSON.toJSONString(content)));
}
private File getSaveSendImImgTempFile(File parentFile,String imageUrl){
String day = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
File tempFile = new File(parentFile,day);
if (!tempFile.exists()) {
tempFile.mkdirs();
}
String suffix = StringUtils.substring(imageUrl, imageUrl.lastIndexOf("."), imageUrl.length());
String fileName = UUID.fastUUID() + "." + suffix;
return new File(tempFile,fileName);
}
}

6
ruoyi-cai/src/main/java/com/ruoyi/cai/manager/UserForbidManager.java
View File

@@ -12,7 +12,8 @@ import com.ruoyi.cai.enums.forbid.ForbidTimeEnum;
import com.ruoyi.cai.enums.forbid.ForbidTypeEnum;
import com.ruoyi.cai.mq.AmqpProducer;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.dto.CommonDelayDto;
import com.ruoyi.cai.mq.handleDelay.CommonDelayDto;
import com.ruoyi.cai.mq.handleDelay.dto.ForbidDelayDto;
import com.ruoyi.cai.service.*;
import com.ruoyi.cai.util.CaiDateUtil;
import com.ruoyi.common.exception.ServiceException;
@@ -251,9 +252,8 @@ public class UserForbidManager {
// < 12小时
long between = CaiDateUtil.diff(userForbid.getEndTime(),LocalDateTime.now());
if(between > 0 && between < 60*60*12){
CommonDelayDto dto = new CommonDelayDto();
ForbidDelayDto dto = new ForbidDelayDto();
dto.setForbidId(expireId);
dto.setType(CommonDelayTypeEnum.USER_FORBID.getCode());
amqpProducer.sendCommonDelayMq(dto,(int)between+5);
}
}

6
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/AmqpProducer.java
View File

@@ -7,8 +7,8 @@ import com.ruoyi.cai.mq.config.CommonDelayMqConfig;
import com.ruoyi.cai.mq.consumer.CalculateSalesQueueConsumer;
import com.ruoyi.cai.mq.consumer.CommonConsumer;
import com.ruoyi.cai.mq.consumer.WindowConsumer;
import com.ruoyi.cai.mq.dto.CommonDTO;
import com.ruoyi.cai.mq.dto.CommonDelayDto;
import com.ruoyi.cai.mq.handle.CommonDTO;
import com.ruoyi.cai.mq.handleDelay.CommonDelayDto;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@@ -53,7 +53,7 @@ public class AmqpProducer {
});
}
public void sendCommonDelayMq(CommonDelayDto dto, Integer timeout){
public <T extends CommonDelayDto> void sendCommonDelayMq(T dto, Integer timeout){
rabbitTemplate.convertAndSend(CommonDelayMqConfig.EXCHANGE_NAME,
CommonDelayMqConfig.ROUTING_KEY,
JSON.toJSONString(dto),

2
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/dto/CommonConsumerEnum.java → ruoyi-cai/src/main/java/com/ruoyi/cai/mq/CommonConsumerEnum.java
View File

@@ -1,4 +1,4 @@
package com.ruoyi.cai.mq.dto;
package com.ruoyi.cai.mq;
public enum CommonConsumerEnum {
WINDOW_GIFT,WINDOW_RECHARGE,

11
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/CommonDelayTypeEnum.java
View File

@@ -4,12 +4,7 @@ import lombok.Getter;
@Getter
public enum CommonDelayTypeEnum {
USER_FORBID(1),
SYS_PUSH(2),
;
private final Integer code;
CommonDelayTypeEnum(Integer code) {
this.code = code;
}
USER_FORBID,
SYS_PUSH,
SAVE_SEND_IM_IMG,
}

5
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/HandleConfig.java → ruoyi-cai/src/main/java/com/ruoyi/cai/mq/config/HandleConfig.java
View File

@@ -1,5 +1,6 @@
package com.ruoyi.cai.mq.handle;
package com.ruoyi.cai.mq.config;
import com.ruoyi.cai.mq.handle.IHandle;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@@ -11,7 +12,7 @@ import java.util.Map;
@Component
public class HandleConfig {
public static Map<String,IHandle> MAP = new HashMap<>();
public static Map<String, IHandle> MAP = new HashMap<>();
@Autowired
private List<IHandle> handles;

32
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/config/HandleDelayConfig.java Normal file
View File

@@ -0,0 +1,32 @@
package com.ruoyi.cai.mq.config;
import com.ruoyi.cai.mq.handleDelay.IHandleDelay;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@Component
public class HandleDelayConfig {
public static Map<String, IHandleDelay> MAP = new HashMap<>();
@Autowired
private List<IHandleDelay> handles;
@PostConstruct
public void init(){
for (IHandleDelay handle : handles) {
MAP.put(handle.getType().name(),handle);
}
}
public IHandleDelay getHandle(String type){
return MAP.get(type);
}
}

3
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/consumer/CommonConsumer.java
View File

@@ -2,9 +2,8 @@ package com.ruoyi.cai.mq.consumer;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.ruoyi.cai.mq.handle.HandleConfig;
import com.ruoyi.cai.mq.config.HandleConfig;
import com.ruoyi.cai.mq.handle.IHandle;
import com.ruoyi.cai.service.ConsumeLogService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.annotation.Exchange;
import org.springframework.amqp.rabbit.annotation.Queue;

41
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/consumer/CommonDelayMqConsumer.java
View File

@@ -1,15 +1,18 @@
package com.ruoyi.cai.mq.consumer;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.ruoyi.cai.manager.UserForbidManager;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.config.CommonDelayMqConfig;
import com.ruoyi.cai.mq.dto.CommonDelayDto;
import com.ruoyi.cai.mq.config.HandleDelayConfig;
import com.ruoyi.cai.mq.handle.IHandle;
import com.ruoyi.cai.mq.handleDelay.CommonDelayDto;
import com.ruoyi.cai.mq.handleDelay.IHandleDelay;
import com.ruoyi.cai.service.SysPushService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.Cache;
import org.springframework.stereotype.Component;
@Slf4j
@@ -19,33 +22,21 @@ public class CommonDelayMqConsumer {
private UserForbidManager userForbidManager;
@Autowired
private SysPushService sysPushService;
@Autowired
private HandleDelayConfig handleDelayConfig;
@RabbitListener(queues = CommonDelayMqConfig.QUEUE_NAME
,containerFactory = "customContainerFactory")
public void checkTimeOutMq(String message) {
log.info("CommonDelayMqConsumer: " + message);
CommonDelayDto dto = JSON.parseObject(message, CommonDelayDto.class);
CommonDelayTypeEnum typeEnum = dto.getTypeEnum();
if(typeEnum == null){
log.warn("延时任务执行失败，未检测到正确的类型 dto={}",JSON.toJSONString(dto));
return;
}
switch (typeEnum){
case USER_FORBID:
try {
userForbidManager.checkExpire(dto.getForbidId());
}catch (Exception e){
log.error("检查用户封禁状态失败!需要开发确认！",e);
}
break;
case SYS_PUSH:
try {
sysPushService.fastStart(dto.getSysPushId());
}catch (Exception e){
log.error("检查系统推送失败！需要开发确认！", e);
}
default:
break;
log.info("公共延时队列消息处理-开始: message=" + message);
try {
JSONObject object = JSON.parseObject(message);
String type = object.getString("type");
IHandleDelay handle = handleDelayConfig.getHandle(type);
handle.run(message);
}catch (Exception e){
log.error("公共延时队列消息处理-失败: message=" + message,e);
}
log.info("公共延时队列消息处理-结束: message=" + message);
}
}

2
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/consumer/WindowConsumer.java
View File

@@ -2,7 +2,7 @@ package com.ruoyi.cai.mq.consumer;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.ruoyi.cai.mq.handle.HandleConfig;
import com.ruoyi.cai.mq.config.HandleConfig;
import com.ruoyi.cai.mq.handle.IHandle;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.annotation.Exchange;

24
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/dto/CommonDelayDto.java
View File

@@ -1,24 +0,0 @@
package com.ruoyi.cai.mq.dto;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import lombok.Data;
@Data
public class CommonDelayDto {
/**
* @see com.ruoyi.cai.mq.CommonDelayTypeEnum
*/
private Integer type;
private Long forbidId;
private Long sysPushId;
public CommonDelayTypeEnum getTypeEnum(){
CommonDelayTypeEnum[] values = CommonDelayTypeEnum.values();
for (CommonDelayTypeEnum value : values) {
if(value.getCode().equals(this.type)){
return value;
}
}
return null;
}
}

3
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/dto/CommonDTO.java → ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/CommonDTO.java
View File

@@ -1,5 +1,6 @@
package com.ruoyi.cai.mq.dto;
package com.ruoyi.cai.mq.handle;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import lombok.Data;
@Data

2
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/IHandle.java
View File

@@ -1,6 +1,6 @@
package com.ruoyi.cai.mq.handle;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.CommonConsumerEnum;
public interface IHandle {

2
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/LoginNotifyHandle.java
View File

@@ -8,7 +8,7 @@ import com.ruoyi.cai.dto.commom.IdDTO;
import com.ruoyi.cai.enums.GenderEnum;
import com.ruoyi.cai.manager.LockManager;
import com.ruoyi.cai.mapper.UserFollowMapper;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.dto.LoginNotifyDTO;
import com.ruoyi.cai.notice.YunxinHttpService;
import com.ruoyi.cai.service.UserInfoService;

2
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/RankNotifyHandle.java
View File

@@ -1,7 +1,7 @@
package com.ruoyi.cai.mq.handle;
import com.alibaba.fastjson2.JSON;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.dto.RankNotifyDTO;
import com.ruoyi.cai.rank.RankManager;
import lombok.extern.slf4j.Slf4j;

2
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/SysPushNotifyHandle.java
View File

@@ -1,7 +1,7 @@
package com.ruoyi.cai.mq.handle;
import com.alibaba.fastjson2.JSON;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.dto.SysPushNotifyDTO;
import com.ruoyi.cai.service.SysPushService;
import lombok.extern.slf4j.Slf4j;

2
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/WindowGiftNotifyHandle.java
View File

@@ -4,7 +4,7 @@ import com.alibaba.fastjson2.JSON;
import com.google.common.collect.Lists;
import com.ruoyi.cai.domain.Gift;
import com.ruoyi.cai.domain.User;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.dto.WindowGiftNotifyDTO;
import com.ruoyi.cai.notice.YunxinHttpService;
import com.ruoyi.cai.notice.data.child.SendGiftWindowsAmountNoticeData;

2
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/WindowRechargeNotifyHandle.java
View File

@@ -2,7 +2,7 @@ package com.ruoyi.cai.mq.handle;
import com.alibaba.fastjson2.JSON;
import com.google.common.collect.Lists;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.dto.WindowRechargeNotifyDTO;
import com.ruoyi.cai.notice.YunxinHttpService;
import com.ruoyi.framework.OnlineUserTodayCache;

4
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/dto/LoginNotifyDTO.java
View File

@@ -1,7 +1,7 @@
package com.ruoyi.cai.mq.handle.dto;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.dto.CommonDTO;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.CommonDTO;
import lombok.Data;
import java.time.LocalDateTime;

4
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/dto/RankNotifyDTO.java
View File

@@ -1,7 +1,7 @@
package com.ruoyi.cai.mq.handle.dto;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.dto.CommonDTO;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.CommonDTO;
import lombok.Data;
import java.time.LocalDateTime;

4
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/dto/SysPushNotifyDTO.java
View File

@@ -1,7 +1,7 @@
package com.ruoyi.cai.mq.handle.dto;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.dto.CommonDTO;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.CommonDTO;
import lombok.Data;
@Data

4
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/dto/WindowGiftNotifyDTO.java
View File

@@ -1,8 +1,8 @@
package com.ruoyi.cai.mq.handle.dto;
import com.ruoyi.cai.domain.Gift;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.dto.CommonDTO;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.CommonDTO;
import lombok.Data;
import java.time.LocalDateTime;

4
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handle/dto/WindowRechargeNotifyDTO.java
View File

@@ -1,7 +1,7 @@
package com.ruoyi.cai.mq.handle.dto;
import com.ruoyi.cai.mq.dto.CommonConsumerEnum;
import com.ruoyi.cai.mq.dto.CommonDTO;
import com.ruoyi.cai.mq.CommonConsumerEnum;
import com.ruoyi.cai.mq.handle.CommonDTO;
import lombok.Data;
import java.time.LocalDateTime;

9
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handleDelay/CommonDelayDto.java Normal file
View File

@@ -0,0 +1,9 @@
package com.ruoyi.cai.mq.handleDelay;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import lombok.Data;
@Data
public class CommonDelayDto {
private CommonDelayTypeEnum type;
}

31
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handleDelay/ForbidDelayHandle.java Normal file
View File

@@ -0,0 +1,31 @@
package com.ruoyi.cai.mq.handleDelay;
import com.alibaba.fastjson2.JSON;
import com.ruoyi.cai.manager.UserForbidManager;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.handleDelay.dto.ForbidDelayDto;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@Component
@Slf4j
public class ForbidDelayHandle implements IHandleDelay {
@Autowired
private UserForbidManager userForbidManager;
@Override
public void run(String message) {
try {
ForbidDelayDto dto = JSON.parseObject(message, ForbidDelayDto.class);
userForbidManager.checkExpire(dto.getForbidId());
}catch (Exception e){
log.error("处理延时推送消息失败!",e);
}
}
@Override
public CommonDelayTypeEnum getType() {
return CommonDelayTypeEnum.USER_FORBID;
}
}

10
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handleDelay/IHandleDelay.java Normal file
View File

@@ -0,0 +1,10 @@
package com.ruoyi.cai.mq.handleDelay;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
public interface IHandleDelay {
void run(String message);
CommonDelayTypeEnum getType();
}

32
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handleDelay/SaveSendImImgDelayHandle.java Normal file
View File

@@ -0,0 +1,32 @@
package com.ruoyi.cai.mq.handleDelay;
import com.alibaba.fastjson2.JSON;
import com.ruoyi.cai.manager.ImService;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.handleDelay.dto.SaveSendImImgDelayDto;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@Component
@Slf4j
public class SaveSendImImgDelayHandle implements IHandleDelay {
@Autowired
private ImService imService;
@Override
public void run(String message) {
try {
SaveSendImImgDelayDto dto = JSON.parseObject(message, SaveSendImImgDelayDto.class);
imService.saveSendImImgDeal(dto.getRecordId());
}catch (Exception e){
log.error("处理延时推送消息失败!",e);
}
}
@Override
public CommonDelayTypeEnum getType() {
return CommonDelayTypeEnum.SAVE_SEND_IM_IMG;
}
}

32
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handleDelay/SysPushDelayHandle.java Normal file
View File

@@ -0,0 +1,32 @@
package com.ruoyi.cai.mq.handleDelay;
import com.alibaba.fastjson2.JSON;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.handleDelay.dto.SysPushDelayDto;
import com.ruoyi.cai.service.SysPushService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@Component
@Slf4j
public class SysPushDelayHandle implements IHandleDelay {
@Autowired
private SysPushService sysPushService;
@Override
public void run(String message) {
try {
SysPushDelayDto dto = JSON.parseObject(message, SysPushDelayDto.class);
sysPushService.fastStart(dto.getSysPushId());
}catch (Exception e){
log.error("处理延时推送消息失败!",e);
}
}
@Override
public CommonDelayTypeEnum getType() {
return CommonDelayTypeEnum.SYS_PUSH;
}
}

14
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handleDelay/dto/ForbidDelayDto.java Normal file
View File

@@ -0,0 +1,14 @@
package com.ruoyi.cai.mq.handleDelay.dto;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.handleDelay.CommonDelayDto;
import lombok.Data;
@Data
public class ForbidDelayDto extends CommonDelayDto {
private Long forbidId;
public ForbidDelayDto() {
this.setType(CommonDelayTypeEnum.USER_FORBID);
}
}

14
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handleDelay/dto/SaveSendImImgDelayDto.java Normal file
View File

@@ -0,0 +1,14 @@
package com.ruoyi.cai.mq.handleDelay.dto;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.handleDelay.CommonDelayDto;
import lombok.Data;
@Data
public class SaveSendImImgDelayDto extends CommonDelayDto {
private Long recordId;
public SaveSendImImgDelayDto() {
this.setType(CommonDelayTypeEnum.SAVE_SEND_IM_IMG);
}
}

14
ruoyi-cai/src/main/java/com/ruoyi/cai/mq/handleDelay/dto/SysPushDelayDto.java Normal file
View File

@@ -0,0 +1,14 @@
package com.ruoyi.cai.mq.handleDelay.dto;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.handleDelay.CommonDelayDto;
import lombok.Data;
@Data
public class SysPushDelayDto extends CommonDelayDto {
private Long sysPushId;
public SysPushDelayDto() {
this.setType(CommonDelayTypeEnum.SYS_PUSH);
}
}

27
ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/SysPushServiceImpl.java
View File

@@ -20,9 +20,8 @@ import com.ruoyi.cai.enums.systempush.*;
import com.ruoyi.cai.manager.SystemConfigManager;
import com.ruoyi.cai.mapper.SysPushMapper;
import com.ruoyi.cai.mq.AmqpProducer;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.dto.CommonDelayDto;
import com.ruoyi.cai.mq.handle.dto.SysPushNotifyDTO;
import com.ruoyi.cai.mq.handleDelay.dto.SysPushDelayDto;
import com.ruoyi.cai.notice.data.NoticeMsgR;
import com.ruoyi.cai.notice.data.NoticeOnlyImageR;
import com.ruoyi.cai.notice.data.NoticeR;
@@ -38,7 +37,7 @@ import com.ruoyi.cai.util.CaiDateUtil;
import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.common.utils.BeanConvertUtil;
import com.ruoyi.yunxin.Yunxin;
import com.ruoyi.yunxin.enums.ImTypeEnum;
import com.ruoyi.yunxin.enums.YxImTypeEnum;
import com.ruoyi.yunxin.resp.YxCommonR;
import com.ruoyi.yunxin.resp.YxDataR;
import lombok.extern.slf4j.Slf4j;
@@ -179,9 +178,8 @@ public class SysPushServiceImpl extends ServiceImpl<SysPushMapper, SysPush> impl
try {
// 小于 < 12小时
long between = CaiDateUtil.diff(sysPush.getSendTime(), LocalDateTime.now());
CommonDelayDto dto = new CommonDelayDto();
SysPushDelayDto dto = new SysPushDelayDto();
dto.setSysPushId(sysPush.getId());
dto.setType(CommonDelayTypeEnum.SYS_PUSH.getCode());
amqpProducer.sendCommonDelayMq(dto,(int)between+2);
}catch (Exception e){
log.error("系统推送任务发送失败!",e);
@@ -321,21 +319,21 @@ public class SysPushServiceImpl extends ServiceImpl<SysPushMapper, SysPush> impl
}
String content = sysPush.getContent();
Object body;
ImTypeEnum imTypeEnum;
YxImTypeEnum yxImTypeEnum;
switch (systemPushType){
case SIMPLE_TEXT:
body = JSON.parseObject(content, NoticeMsgR.class);
imTypeEnum = ImTypeEnum.TXT;
yxImTypeEnum = YxImTypeEnum.TXT;
break;
case TEXT:
NoticeR<ExtNoticeData> noticeR = JSON.parseObject(content, new TypeReference<NoticeR<ExtNoticeData>>() {});
noticeR.getData().setCurrentDate();
body = noticeR;
imTypeEnum = ImTypeEnum.CUSTOM;
yxImTypeEnum = YxImTypeEnum.CUSTOM;
break;
case SIMPLE_IMAGE_TEXT:
NoticeR<SimpleImageTextData> simpleImageTextR = JSON.parseObject(content, new TypeReference<NoticeR<SimpleImageTextData>>() {});
imTypeEnum = ImTypeEnum.CUSTOM;
yxImTypeEnum = YxImTypeEnum.CUSTOM;
SimpleImageTextData data = simpleImageTextR.getData();
if(data.getImage() != null){
data.setImage(cosUrl+data.getImage());
@@ -352,7 +350,7 @@ public class SysPushServiceImpl extends ServiceImpl<SysPushMapper, SysPush> impl
}
}
body = multipleImageTextR;
imTypeEnum = ImTypeEnum.CUSTOM;
yxImTypeEnum = YxImTypeEnum.CUSTOM;
break;
case ONLY_IMAGE:
NoticeOnlyImageR imageR = JSON.parseObject(content, NoticeOnlyImageR.class);
@@ -360,12 +358,12 @@ public class SysPushServiceImpl extends ServiceImpl<SysPushMapper, SysPush> impl
imageR.setUrl(cosUrl+imageR.getUrl());
}
body = imageR;
imTypeEnum = ImTypeEnum.IMAGE;
yxImTypeEnum = YxImTypeEnum.IMAGE;
break;
default:
throw new ServiceException("消息类型有误");
}
YxDataR<YxCommonR> r = yunxin.batchSendToNotice(userIds, body, null, imTypeEnum);
YxDataR<YxCommonR> r = yunxin.batchSendToNotice(userIds, body, null, yxImTypeEnum);
SendSysPushResp resp = new SendSysPushResp();
resp.setImResp(r);
return resp;
@@ -413,9 +411,8 @@ public class SysPushServiceImpl extends ServiceImpl<SysPushMapper, SysPush> impl
// < 12小时
long between = CaiDateUtil.diff(sysPush.getSendTime(),LocalDateTime.now());
if(between > 0 && between < 60*60*12){
CommonDelayDto dto = new CommonDelayDto();
dto.setForbidId(expireId);
dto.setType(CommonDelayTypeEnum.SYS_PUSH.getCode());
SysPushDelayDto dto = new SysPushDelayDto();
dto.setSysPushId(sysPush.getId());
amqpProducer.sendCommonDelayMq(dto,(int)between+5);
}
}

7
ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserForbidServiceImpl.java
View File

@@ -14,7 +14,9 @@ import com.ruoyi.cai.manager.ForbidCache;
import com.ruoyi.cai.mapper.UserForbidMapper;
import com.ruoyi.cai.mq.AmqpProducer;
import com.ruoyi.cai.mq.CommonDelayTypeEnum;
import com.ruoyi.cai.mq.dto.CommonDelayDto;
import com.ruoyi.cai.mq.handleDelay.CommonDelayDto;
import com.ruoyi.cai.mq.handleDelay.dto.ForbidDelayDto;
import com.ruoyi.cai.mq.handleDelay.dto.SysPushDelayDto;
import com.ruoyi.cai.service.AnchorService;
import com.ruoyi.cai.service.UserForbidService;
import com.ruoyi.cai.service.UserInfoService;
@@ -88,9 +90,8 @@ public class UserForbidServiceImpl extends ServiceImpl<UserForbidMapper, UserFor
try {
// 12小时
long between = CaiDateUtil.diff(one.getEndTime(), LocalDateTime.now());
CommonDelayDto dto = new CommonDelayDto();
ForbidDelayDto dto = new ForbidDelayDto();
dto.setForbidId(one.getId());
dto.setType(CommonDelayTypeEnum.USER_FORBID.getCode());
amqpProducer.sendCommonDelayMq(dto,(int)between+2);
}catch (Exception e){
log.error("用户封禁延时任务发送失败!",e);

200
ruoyi-cai/src/main/java/com/ruoyi/cai/util/FileUtils.java Normal file
View File

@@ -0,0 +1,200 @@
package com.ruoyi.cai.util;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.ResourceUtils;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.file.Files;
import java.util.List;
/**
* 文件工具类
* <p>created on 2022/6/16 11:54</p>
* @author ZL
*/
@Slf4j
public class FileUtils {
private final static String TEMP_PATH = "/temp";
public static void main(String[] args) {
String targetFile = "C:\\ffmpeg\\video\\55585858.mp4";
// String filePath = "http://oss.nohi.vip/nono/20220816164358-F4A25960-F837-49B1-83F2-D41865FD46F8.mp4";
String filePath = "https://download-video.hik-express.com/fc/20220817184523-909E6F0C-1FE3-4D4D-8A4E-296107EF08CD.mp4?Expires=1661168603&OSSAccessKeyId=LTAI5t9GNWcPxHVUMNU3WRqD&Signature=ybHhkkdhAqYt%2F7x07KYNnPDA9OY%3D";
byte[] fileStream = getFileStream(filePath);
byteToFile(fileStream,new File(targetFile));
}
/**
* 得到文件流
* @param url 网络图片URL地址
* @return
*/
public static byte[] getFileStream(String url){
InputStream inStream = null;
try {
URL httpUrl = new URL(url);
HttpURLConnection conn = (HttpURLConnection)httpUrl.openConnection();
conn.setRequestMethod("GET");
conn.setConnectTimeout(10 * 1000);
conn.setReadTimeout(10 * 1000);
inStream = conn.getInputStream();//通过输入流获取图片数据
return readInputStream(inStream);//得到图片的二进制数据
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
if(inStream != null){
try {
inStream.close();
} catch (IOException e) {
log.error("流关闭失败",e);
}
}
}
}
public static void downloadPath(String path,File targetFile){
InputStream inStream = null;
try {
URL httpUrl = new URL(path);
HttpURLConnection conn = (HttpURLConnection)httpUrl.openConnection();
conn.setRequestMethod("GET");
conn.setConnectTimeout(10 * 1000);
conn.setReadTimeout(10 * 1000);
inStream = conn.getInputStream();//通过输入流获取图片数据
byte[] buffer = new byte[1024];
try (OutputStream os = Files.newOutputStream(targetFile.toPath())){
int len ;
while((len = inStream.read(buffer)) != -1) {
os.write(buffer, 0, len);
os.flush();
}
}
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
if(inStream != null){
try {
inStream.close();
} catch (IOException e) {
log.error("流关闭失败",e);
}
}
}
}
/**
* 从输入流中获取数据
* @param inStream 输入流
* @return
* @throws Exception
*/
public static byte[] readInputStream(InputStream inStream) throws Exception{
try (ByteArrayOutputStream outStream = new ByteArrayOutputStream()){
byte[] buffer = new byte[1024];
int len;
while( (len=inStream.read(buffer)) != -1 ){
outStream.write(buffer, 0, len);
}
return outStream.toByteArray();
}
}
/**
* 从输入流中获取数据
* @param inputStream 输入流
* @return
* @throws Exception
*/
public static String readInputStreamByLine(InputStream inputStream) throws Exception{
StringBuilder sb = new StringBuilder();
try (BufferedReader br = new BufferedReader(new InputStreamReader(inputStream))){
String str;
while ((str = br.readLine()) != null) {
sb.append(str);
}
}
return sb.toString();
}
public static byte[] fileToByte(File file){
try (InputStream in = Files.newInputStream(file.toPath())){
byte[] data = new byte[in.available()];
in.read(data);
return data;
} catch (IOException e) {
throw new RuntimeException("文件转换字节失败",e);
}
}
public static void byteToFile(byte[] bytes,File file){
File parentFile = file.getParentFile();
if(!parentFile.exists()){
parentFile.mkdirs();
}
try (OutputStream outputStream = Files.newOutputStream(file.toPath())){
outputStream.write(bytes);
}catch (IOException e){
throw new RuntimeException("字节写入文件失败!",e);
}
}
public static File getTempFile() throws FileNotFoundException {
File path = new File(ResourceUtils.getURL("classpath:").getPath());
if (!path.exists()) {
path = new File("");
}
File tempLocation = new File(path.getAbsolutePath() + TEMP_PATH);
if (!tempLocation.exists()) {
tempLocation.mkdirs();
}
return tempLocation;
}
public static String getRootFIle() throws FileNotFoundException {
File path = new File(ResourceUtils.getURL("classpath:").getPath());
if (!path.exists()) {
path = new File("");
}
return path.getAbsolutePath();
}
public static boolean deleteFile(File deleteFile){
if(deleteFile == null || !deleteFile.isFile()){
return false;
}
return deleteFile.delete();
}
/**
* 删除文件
* @param rootFile
* @param deleteFilePath
*/
public static void deleteFile(File rootFile, List<String> deleteFilePath){
if(!rootFile.exists()){
return;
}
File[] files = rootFile.listFiles();
if(files == null){
return;
}
for (File file : files) {
if(!deleteFilePath.contains(file.getName())){
try {
String path = file.getAbsolutePath();
if(file.isDirectory()){
org.apache.commons.io.FileUtils.deleteDirectory(file);
log.info("del dir success. path=[{}]", path);
}
} catch (IOException e) {
log.error("清理文件的时候删除失败,filePath={}",file.getParentFile(),e);
}
}
}
}
}