forked from github/dataease
326 lines
13 KiB
Java
326 lines
13 KiB
Java
package io.dataease.service.panel;
|
||
|
||
import com.google.gson.Gson;
|
||
import io.dataease.auth.api.dto.CurrentRoleDto;
|
||
import io.dataease.auth.api.dto.CurrentUserDto;
|
||
import io.dataease.base.domain.PanelGroup;
|
||
import io.dataease.base.domain.PanelShare;
|
||
import io.dataease.base.domain.PanelShareExample;
|
||
import io.dataease.base.mapper.PanelGroupMapper;
|
||
import io.dataease.base.mapper.PanelShareMapper;
|
||
import io.dataease.base.mapper.ext.ExtPanelShareMapper;
|
||
import io.dataease.base.mapper.ext.query.GridExample;
|
||
import io.dataease.commons.model.AuthURD;
|
||
import io.dataease.commons.utils.AuthUtils;
|
||
import io.dataease.commons.utils.BeanUtils;
|
||
import io.dataease.commons.utils.CommonBeanFactory;
|
||
import io.dataease.controller.request.panel.PanelShareFineDto;
|
||
import io.dataease.controller.request.panel.PanelShareRemoveRequest;
|
||
import io.dataease.controller.request.panel.PanelShareRequest;
|
||
import io.dataease.controller.sys.base.BaseGridRequest;
|
||
import io.dataease.dto.panel.PanelShareDto;
|
||
import io.dataease.dto.panel.PanelShareOutDTO;
|
||
import io.dataease.dto.panel.PanelSharePo;
|
||
import io.dataease.service.message.DeMsgutil;
|
||
import lombok.Data;
|
||
import org.apache.commons.collections.CollectionUtils;
|
||
import org.apache.commons.lang3.StringUtils;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.stereotype.Service;
|
||
import org.springframework.transaction.annotation.Transactional;
|
||
import javax.annotation.Resource;
|
||
import java.util.*;
|
||
import java.util.stream.Collectors;
|
||
|
||
|
||
@Service
|
||
public class ShareService {
|
||
|
||
@Autowired(required = false)
|
||
private PanelShareMapper mapper;
|
||
|
||
@Resource
|
||
private PanelGroupMapper panelGroupMapper;
|
||
|
||
@Resource
|
||
private ExtPanelShareMapper extPanelShareMapper;
|
||
|
||
/**
|
||
* 1.查询当前节点已经分享给了哪些目标
|
||
* 2.过滤出新增的目标
|
||
* 3.过滤出减少的目标
|
||
* 4.批量删除
|
||
* 5.批量新增
|
||
* 6.发送取消分享消息
|
||
* 7.发送新增分享消息
|
||
* @param panelShareFineDto
|
||
*/
|
||
@Transactional
|
||
public void fineSave(PanelShareFineDto panelShareFineDto) {
|
||
|
||
List<PanelShare> addShares = new ArrayList<>();//新增的分享
|
||
List<Long> redShareIdLists = new ArrayList<>();//取消的分享
|
||
|
||
String panelGroupId = panelShareFineDto.getResourceId();
|
||
AuthURD authURD = panelShareFineDto.getAuthURD();
|
||
AuthURD sharedAuthURD = new AuthURD();
|
||
AuthURD addAuthURD = new AuthURD();
|
||
|
||
|
||
|
||
Map<Integer, List<Long>> authURDMap = new HashMap<>();
|
||
authURDMap.put(0, authURD.getUserIds());
|
||
authURDMap.put(1, authURD.getRoleIds());
|
||
authURDMap.put(2, authURD.getDeptIds());
|
||
|
||
PanelShareExample example = new PanelShareExample();
|
||
example.createCriteria().andPanelGroupIdEqualTo(panelGroupId);
|
||
List<PanelShare> panelShares = mapper.selectByExample(example);
|
||
Map<Integer, List<TempShareNode>> typeSharedMap = panelShares.stream().map(this::convertNode).collect(Collectors.groupingBy(TempShareNode::getType));
|
||
|
||
for (Map.Entry<Integer, List<Long>> entry : authURDMap.entrySet()) {
|
||
Integer key = entry.getKey();
|
||
List<TempShareNode> shareNodes;
|
||
if (null == typeSharedMap || null == typeSharedMap.get(key)) {
|
||
shareNodes = new ArrayList<>();
|
||
}else{
|
||
shareNodes = typeSharedMap.get(key);
|
||
}
|
||
|
||
if (null != authURDMap.get(key)) {
|
||
Map<String, Object> dataMap = filterData(authURDMap.get(key), shareNodes);
|
||
List<Long> newIds = (List<Long>)dataMap.get("add");
|
||
for (int i = 0; i < newIds.size(); i++) {
|
||
Long id = newIds.get(i);
|
||
PanelShare share = new PanelShare();
|
||
share.setCreateTime(System.currentTimeMillis());
|
||
share.setPanelGroupId(panelGroupId);
|
||
share.setTargetId(id);
|
||
share.setType(key);
|
||
addShares.add(share);
|
||
}
|
||
List<TempShareNode> redNodes = (List<TempShareNode>)dataMap.get("red");
|
||
List<Long> redIds = redNodes.stream().map(TempShareNode::getShareId).distinct().collect(Collectors.toList());
|
||
|
||
redShareIdLists.addAll(redIds);
|
||
buildRedAuthURD(key, redNodes.stream().map(TempShareNode::getTargetId).distinct().collect(Collectors.toList()) , sharedAuthURD);
|
||
buildRedAuthURD(key, newIds, addAuthURD);
|
||
}
|
||
|
||
}
|
||
|
||
if (CollectionUtils.isNotEmpty(redShareIdLists)){
|
||
extPanelShareMapper.batchDelete(redShareIdLists);
|
||
}
|
||
|
||
if (CollectionUtils.isNotEmpty(addShares)){
|
||
extPanelShareMapper.batchInsert(addShares, AuthUtils.getUser().getUsername());
|
||
}
|
||
|
||
// 以上是业务代码
|
||
// 下面是消息发送
|
||
Set<Long> addUserIdSet = AuthUtils.userIdsByURD(addAuthURD);
|
||
Set<Long> redUserIdSet = AuthUtils.userIdsByURD(sharedAuthURD);
|
||
PanelGroup panelGroup = panelGroupMapper.selectByPrimaryKey(panelGroupId);
|
||
CurrentUserDto user = AuthUtils.getUser();
|
||
Gson gson = new Gson();
|
||
String msg = panelGroup.getName();
|
||
|
||
List<String> msgParam = new ArrayList<>();
|
||
msgParam.add(panelGroupId);
|
||
addUserIdSet.forEach(userId -> {
|
||
if (!redUserIdSet.contains(userId) && !user.getUserId().equals(userId)){
|
||
DeMsgutil.sendMsg(userId, 2L,user.getNickName()+" 分享了仪表板【"+msg+"】，请查收!", gson.toJson(msgParam));
|
||
}
|
||
});
|
||
|
||
redUserIdSet.forEach(userId -> {
|
||
if (!addUserIdSet.contains(userId) && !user.getUserId().equals(userId)){
|
||
DeMsgutil.sendMsg(userId, 3L, user.getNickName()+" 取消分享了仪表板【"+msg+"】，请查收!", gson.toJson(msgParam));
|
||
}
|
||
});
|
||
|
||
}
|
||
|
||
private void buildRedAuthURD(Integer type, List<Long> redIds , AuthURD authURD) {
|
||
if (type == 0) {
|
||
authURD.setUserIds(redIds);
|
||
}
|
||
if (type == 1) {
|
||
authURD.setRoleIds(redIds);
|
||
}
|
||
if (type == 2) {
|
||
authURD.setDeptIds(redIds);
|
||
}
|
||
}
|
||
|
||
/**
|
||
*
|
||
* @param newTargets 新的分享目标
|
||
* @param shareNodes 已景分享目标
|
||
* @return
|
||
*/
|
||
private Map<String, Object> filterData(List<Long> newTargets, List<TempShareNode> shareNodes) {
|
||
Map<String, Object> result = new HashMap<>();
|
||
List<Long> newUserIds = new ArrayList<>();
|
||
for (int i = 0; i < newTargets.size(); i++) {
|
||
Long newTargetId = newTargets.get(i);
|
||
Boolean isNew = true;
|
||
for (int j = 0; j < shareNodes.size(); j++) {
|
||
TempShareNode shareNode = shareNodes.get(j);
|
||
Long sharedId = shareNode.getTargetId();
|
||
if (newTargetId.equals(sharedId)) {
|
||
shareNode.setMatched(true); // 已分享 重新命中
|
||
isNew = false;
|
||
}
|
||
}
|
||
if (isNew) {
|
||
// 获取新增的
|
||
newUserIds.add(newTargetId);
|
||
}
|
||
}
|
||
//获取需要取消分享的
|
||
List<TempShareNode> missNodes = shareNodes.stream().filter(item -> !item.getMatched()).collect(Collectors.toList());
|
||
result.put("add", newUserIds);
|
||
result.put("red", missNodes);
|
||
return result;
|
||
}
|
||
|
||
@Data
|
||
private class TempShareNode {
|
||
private Long shareId;
|
||
private Integer type;
|
||
private Long targetId;
|
||
private Boolean matched = false;
|
||
|
||
public boolean targetMatch(Long tid) {
|
||
return targetId.equals(tid);
|
||
}
|
||
}
|
||
|
||
private TempShareNode convertNode(PanelShare panelShare) {
|
||
return BeanUtils.copyBean(new TempShareNode(), panelShare);
|
||
}
|
||
|
||
|
||
@Transactional
|
||
public void save(PanelShareRequest request){
|
||
List<PanelGroup> panelGroups = queryGroup(request.getPanelIds());
|
||
//1.先根据仪表板删除所有已经分享的
|
||
Integer type = request.getType();
|
||
List<String> panelIds = request.getPanelIds();
|
||
List<Long> targetIds = request.getTargetIds();
|
||
// 使用原生对象会导致事物失效 所以这里需要使用spring代理对象
|
||
if (CollectionUtils.isNotEmpty(panelIds)){
|
||
ShareService proxy = CommonBeanFactory.getBean(ShareService.class);
|
||
panelIds.forEach(panelId -> proxy.delete(panelId, type));
|
||
}
|
||
if (CollectionUtils.isEmpty(targetIds)) return;
|
||
|
||
long now = System.currentTimeMillis();
|
||
List<PanelShare> shares = panelIds.stream().flatMap(panelId ->
|
||
targetIds.stream().map(targetId -> {
|
||
PanelShare share = new PanelShare();
|
||
share.setCreateTime(now);
|
||
share.setPanelGroupId(panelId);
|
||
share.setTargetId(targetId);
|
||
share.setType(type);
|
||
return share;
|
||
})
|
||
).collect(Collectors.toList());
|
||
if (CollectionUtils.isNotEmpty(shares)){
|
||
extPanelShareMapper.batchInsert(shares, AuthUtils.getUser().getUsername());
|
||
}
|
||
|
||
// 下面是发送提醒消息逻辑
|
||
Set<Long> userIdSet;
|
||
AuthURD authURD = new AuthURD();
|
||
if (type == 0) {
|
||
authURD.setUserIds(targetIds);
|
||
}
|
||
if (type == 1) {
|
||
authURD.setRoleIds(targetIds);
|
||
}
|
||
if(type == 2) {
|
||
authURD.setDeptIds(targetIds);
|
||
}
|
||
userIdSet = AuthUtils.userIdsByURD(authURD);
|
||
|
||
CurrentUserDto user = AuthUtils.getUser();
|
||
String msg = StringUtils.joinWith("，", panelGroups.stream().map(PanelGroup::getName).collect(Collectors.toList()));
|
||
Gson gson = new Gson();
|
||
userIdSet.forEach(userId -> DeMsgutil.sendMsg(userId, 2L, user.getNickName()+" 分享了仪表板【"+msg+"】给您，请查收!", gson.toJson(panelIds)));
|
||
|
||
}
|
||
|
||
private List<PanelGroup> queryGroup(List<String> panelIds) {
|
||
return panelIds.stream().map(panelGroupMapper::selectByPrimaryKey).collect(Collectors.toList());
|
||
}
|
||
|
||
/**
|
||
* panel_group_id建了索引 效率不会很差
|
||
* @param panel_group_id
|
||
*/
|
||
@Transactional
|
||
public void delete(String panel_group_id, Integer type){
|
||
PanelShareExample example = new PanelShareExample();
|
||
PanelShareExample.Criteria criteria = example.createCriteria();
|
||
criteria.andPanelGroupIdEqualTo(panel_group_id);
|
||
if(type != null){
|
||
criteria.andTypeEqualTo(type);
|
||
}
|
||
mapper.deleteByExample(example);
|
||
}
|
||
|
||
public List<PanelSharePo> shareOut() {
|
||
return null;
|
||
}
|
||
|
||
public List<PanelSharePo> queryShareOut() {
|
||
String username = AuthUtils.getUser().getUsername();
|
||
return extPanelShareMapper.queryOut(username);
|
||
}
|
||
|
||
public List<PanelShareDto> queryTree(BaseGridRequest request){
|
||
CurrentUserDto user = AuthUtils.getUser();
|
||
Long userId = user.getUserId();
|
||
Long deptId = user.getDeptId();
|
||
List<Long> roleIds = user.getRoles().stream().map(CurrentRoleDto::getId).collect(Collectors.toList());
|
||
|
||
Map<String, Object> param = new HashMap<>();
|
||
param.put("userId", userId);
|
||
param.put("deptId", deptId);
|
||
param.put("roleIds", roleIds);
|
||
|
||
List<PanelSharePo> datas = extPanelShareMapper.query(param);
|
||
List<PanelShareDto> dtoLists = datas.stream().map(po -> BeanUtils.copyBean(new PanelShareDto(), po)).collect(Collectors.toList());
|
||
return convertTree(dtoLists);
|
||
}
|
||
|
||
//List构建Tree
|
||
private List<PanelShareDto> convertTree(List<PanelShareDto> datas){
|
||
String username = AuthUtils.getUser().getUsername();
|
||
Map<String, List<PanelShareDto>> map = datas.stream().filter(panelShareDto -> StringUtils.isNotEmpty(panelShareDto.getCreator()) && !StringUtils.equals(username, panelShareDto.getCreator())).collect(Collectors.groupingBy(PanelShareDto::getCreator));
|
||
return map.entrySet().stream().map(entry -> {
|
||
PanelShareDto panelShareDto = new PanelShareDto();
|
||
panelShareDto.setName(entry.getKey());
|
||
panelShareDto.setChildren(entry.getValue());
|
||
return panelShareDto;
|
||
}).collect(Collectors.toList());
|
||
}
|
||
|
||
public List<PanelShare> queryWithResource(BaseGridRequest request){
|
||
GridExample example = request.convertExample();
|
||
return extPanelShareMapper.queryWithResource(example);
|
||
}
|
||
|
||
public List<PanelShareOutDTO> queryTargets(String panelId) {
|
||
return extPanelShareMapper.queryTargets(panelId);
|
||
}
|
||
|
||
public void removeShares(PanelShareRemoveRequest removeRequest) {
|
||
extPanelShareMapper.removeShares(removeRequest);
|
||
}
|
||
|
||
}
|