parent
5dd7ee034e
commit
340b83b4e2
@ -0,0 +1,32 @@
|
|||||||
|
package com.ruoyi.im.domain;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.annotation.*;
|
||||||
|
import com.fasterxml.jackson.annotation.JsonFormat;
|
||||||
|
import lombok.Data;
|
||||||
|
import java.util.Date;
|
||||||
|
|
||||||
|
@Data
|
||||||
|
@TableName("im_user_black")
|
||||||
|
public class ImUserBlack {
|
||||||
|
|
||||||
|
@TableId(type = IdType.AUTO)
|
||||||
|
private Long id;
|
||||||
|
|
||||||
|
/** 拉黑人uid */
|
||||||
|
private Long fromUid;
|
||||||
|
|
||||||
|
/** 被拉黑uid */
|
||||||
|
private Long toUid;
|
||||||
|
|
||||||
|
@TableField(value = "create_time")
|
||||||
|
@JsonFormat(shape = JsonFormat.Shape.STRING, pattern="yyyy-MM-dd HH:mm:ss",timezone = "GMT+8")
|
||||||
|
private Date createTime;
|
||||||
|
|
||||||
|
@TableField(value = "update_time")
|
||||||
|
@JsonFormat(shape = JsonFormat.Shape.STRING, pattern="yyyy-MM-dd HH:mm:ss",timezone = "GMT+8")
|
||||||
|
private Date updateTime;
|
||||||
|
|
||||||
|
/** 若依逻辑删除 0正常 1删除 */
|
||||||
|
@TableField(value = "del_flag")
|
||||||
|
private Integer delFlag;
|
||||||
|
}
|
||||||
@ -0,0 +1,18 @@
|
|||||||
|
package com.ruoyi.im.mapper;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||||
|
import com.ruoyi.im.domain.ImUserBlack;
|
||||||
|
import org.apache.ibatis.annotations.Param;
|
||||||
|
import org.apache.ibatis.annotations.Select;
|
||||||
|
|
||||||
|
public interface ImUserBlackMapper extends BaseMapper<ImUserBlack> {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 判断 receiver 是否拉黑 sender
|
||||||
|
* @param receiver 接收方A
|
||||||
|
* @param sender 发送方B
|
||||||
|
* @return true A拉黑B
|
||||||
|
*/
|
||||||
|
@Select("SELECT COUNT(1) FROM im_user_black WHERE from_uid = #{receiver} AND to_uid = #{sender} AND del_flag = 0")
|
||||||
|
boolean existsBlack(@Param("receiver") Long receiver, @Param("sender") Long sender);
|
||||||
|
}
|
||||||
@ -0,0 +1,68 @@
|
|||||||
|
package com.ruoyi.im.service.impl;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
||||||
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||||||
|
import com.ruoyi.im.domain.ImUserBlack;
|
||||||
|
import com.ruoyi.im.mapper.ImUserBlackMapper;
|
||||||
|
import com.ruoyi.im.service.ImUserBlackService;
|
||||||
|
import org.springframework.stereotype.Service;
|
||||||
|
import org.springframework.transaction.annotation.Transactional;
|
||||||
|
import javax.annotation.Resource;
|
||||||
|
import java.util.Date;
|
||||||
|
|
||||||
|
@Service
|
||||||
|
public class ImUserBlackServiceImpl extends ServiceImpl<ImUserBlackMapper, ImUserBlack> implements ImUserBlackService {
|
||||||
|
|
||||||
|
@Resource
|
||||||
|
private ImUserBlackMapper imUserBlackMapper;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isBlack(Long receiver, Long sender) {
|
||||||
|
if (receiver == null || sender == null) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
// receiver 是否拉黑 sender
|
||||||
|
return imUserBlackMapper.existsBlack(receiver, sender);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
@Transactional(rollbackFor = Exception.class)
|
||||||
|
public boolean blackUser(Long fromUid, Long toUid) {
|
||||||
|
if(fromUid == null || toUid == null || fromUid.equals(toUid)){
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
// 先查是否已有拉黑记录（逻辑删除的旧记录）
|
||||||
|
LambdaQueryWrapper<ImUserBlack> wrapper = new LambdaQueryWrapper<>();
|
||||||
|
wrapper.eq(ImUserBlack::getFromUid, fromUid);
|
||||||
|
wrapper.eq(ImUserBlack::getToUid, toUid);
|
||||||
|
ImUserBlack oldRecord = getOne(wrapper);
|
||||||
|
|
||||||
|
if(oldRecord != null){
|
||||||
|
// 存在记录，恢复（取消逻辑删除）
|
||||||
|
oldRecord.setDelFlag(0);
|
||||||
|
oldRecord.setUpdateTime(new Date());
|
||||||
|
return updateById(oldRecord);
|
||||||
|
}else{
|
||||||
|
ImUserBlack black = new ImUserBlack();
|
||||||
|
black.setFromUid(fromUid);
|
||||||
|
black.setToUid(toUid);
|
||||||
|
black.setDelFlag(0);
|
||||||
|
black.setCreateTime(new Date());
|
||||||
|
return save(black);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
@Transactional(rollbackFor = Exception.class)
|
||||||
|
public boolean unBlackUser(Long fromUid, Long toUid) {
|
||||||
|
LambdaQueryWrapper<ImUserBlack> wrapper = new LambdaQueryWrapper<>();
|
||||||
|
wrapper.eq(ImUserBlack::getFromUid, fromUid);
|
||||||
|
wrapper.eq(ImUserBlack::getToUid, toUid);
|
||||||
|
ImUserBlack record = getOne(wrapper);
|
||||||
|
if(record == null){
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
// 逻辑删除（若依标准）
|
||||||
|
return removeById(record.getId());
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue