package com.ruoyi.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.common.core.domain.AjaxResult;
|
import com.ruoyi.common.core.domain.entity.EsModel;
|
import com.ruoyi.domain.*;
|
import com.ruoyi.mapper.ZfShareDataMapper;
|
import com.ruoyi.service.EsService;
|
import com.ruoyi.service.ZInfoUserService;
|
import com.ruoyi.service.ZfClanService;
|
import com.ruoyi.service.ZfShareDataService;
|
import org.elasticsearch.action.delete.DeleteRequest;
|
import org.elasticsearch.client.RequestOptions;
|
import org.elasticsearch.client.RestHighLevelClient;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import java.io.IOException;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
@Service
|
public class ZfShareDataServiceImpl extends ServiceImpl<ZfShareDataMapper, ZfShareData> implements ZfShareDataService {
|
|
@Resource
|
private ZInfoUserService zInfoUserService;
|
|
@Resource
|
private RestHighLevelClient restHighLevelClient;
|
|
@Resource
|
private EsService esSer;
|
|
private LambdaQueryWrapper<ZfShareData> buildCondition(Long userId) {
|
LambdaQueryWrapper<ZfShareData> lqw = new LambdaQueryWrapper<>();
|
lqw.orderByDesc(ZfShareData::getId);
|
lqw.eq(ZfShareData::getUserId,userId);
|
return lqw;
|
}
|
|
private LambdaQueryWrapper<ZfShareData> uniqueCondition(ZfShareData zfShareData) {
|
LambdaQueryWrapper<ZfShareData> lqw = new LambdaQueryWrapper<>();
|
lqw.orderByDesc(ZfShareData::getId);
|
lqw.eq(ZfShareData::getSharedId,zfShareData.getSharedId());
|
return lqw;
|
}
|
|
@Override
|
public List<ZfShareData> selectList(ZfShareData zfShareData) {
|
LambdaQueryWrapper<ZfShareData> lqw = buildCondition(getUserId());
|
|
return list(lqw);
|
}
|
|
@Override
|
public List<ZfShareData> selectByIds(Long[] ids) {
|
List<ZfShareData> list = new ArrayList<>();
|
if(ids.length!=0)
|
list = listByIds(Arrays.asList(ids));
|
else
|
list = list();
|
return list;
|
}
|
|
public Long getUserId(){
|
ZInfoUser myself = zInfoUserService.getMyself();
|
return myself.getUserId();
|
|
}
|
@Override
|
public Long getSource(Long userId){
|
LambdaQueryWrapper<ZfShareData> lqw = new LambdaQueryWrapper<>();
|
lqw.orderByDesc(ZfShareData::getId);
|
lqw.eq(ZfShareData::getSharedId,userId);
|
if (list(lqw).size() == 0){
|
return userId;
|
}
|
return getSource(list(lqw).get(0).getUserId().longValue());
|
}
|
@Override
|
public Long getEnd(Long userId){
|
LambdaQueryWrapper<ZfShareData> lqw1 = new LambdaQueryWrapper<>();
|
lqw1.eq(ZfShareData::getUserId,userId);
|
if (list(lqw1).size() == 0){
|
return userId;
|
}
|
return getEnd(list(lqw1).get(list(lqw1).size()-1).getSharedId().longValue());
|
}
|
|
@Override
|
public AjaxResult addData(Long[] userIds) {
|
|
for (Long userId : userIds) {
|
ZfShareData zfShareData = new ZfShareData();
|
zfShareData.setUserId(getUserId().intValue());
|
zfShareData.setSharedId(userId.intValue());
|
zfShareData.setMasterAccount(0);
|
for(ZfShareData zfShareData1:list(buildCondition(getUserId()))){
|
System.out.println(zfShareData1);
|
}
|
if (list(uniqueCondition(zfShareData)).size() == 0) save(zfShareData);
|
else return AjaxResult.error("请勿重复共享");
|
}
|
return AjaxResult.success("数据共享成功");
|
|
|
}
|
|
@Override
|
public AjaxResult deleteData(Long[] ids) {
|
List<ZfShareData> meetings = listByIds(Arrays.asList(ids));
|
|
if (removeByIds(Arrays.asList(ids))) {
|
|
//删除es中的数据
|
meetings.forEach(meeting -> {
|
EsModel esModel = esSer.findByCtId((meeting.getId().intValue()), "数据共享");
|
if (esModel != null){
|
DeleteRequest deleteRequest = new DeleteRequest("allsearchdata", esModel.getId());
|
try {
|
restHighLevelClient.delete(deleteRequest, RequestOptions.DEFAULT);
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
}
|
});
|
|
return AjaxResult.success();
|
}else {
|
return AjaxResult.error();
|
}
|
}
|
|
|
private LambdaQueryWrapper<ZfShareData> selectByUserId(List<Long> userId) {
|
LambdaQueryWrapper<ZfShareData> lqw = new LambdaQueryWrapper<>();
|
lqw.in(ZfShareData::getSharedId,userId);
|
return lqw;
|
}
|
|
private LambdaQueryWrapper<ZfShareData> selectBySharedId(List<Long> userId) {
|
LambdaQueryWrapper<ZfShareData> lqw = new LambdaQueryWrapper<>();
|
lqw.in(ZfShareData::getUserId,userId);
|
return lqw;
|
}
|
|
@Override
|
public List<Long> getAllPerson(List<Long> userIds){
|
List<ZfShareData> lists = new ArrayList<>(list(selectByUserId(userIds)));
|
lists.addAll(list(selectBySharedId(userIds)));
|
|
if (userIds.contains(getEnd(getSource(userIds.get(0))))){
|
return userIds;
|
}
|
|
return getAllPerson(lists.stream().map(ZfShareData::getSharedId).map(Integer::longValue).collect(Collectors.toList()));
|
|
}
|
|
@Override
|
public AjaxResult setAdmin(Long userId,Long id) {
|
Long source = getSource(userId);
|
List<Long> all = getAllPerson(Collections.singletonList(source));
|
List<Long> collect = all.stream().distinct().collect(Collectors.toList());
|
collect.add(source);
|
|
List<ZfShareData> list = list(selectByUserId(collect));
|
list.addAll(list(selectBySharedId(collect)));
|
List<ZfShareData> collect1 = list.stream().distinct().collect(Collectors.toList());
|
for (ZfShareData zfShareData1:collect1){
|
zfShareData1.setMasterAccount(id.intValue());
|
updateById(zfShareData1);
|
}
|
|
return AjaxResult.success();
|
}
|
|
@Override
|
public AjaxResult allUser(Long id) {
|
Long source = getSource(id);
|
List<Long> all = getAllPerson(Collections.singletonList(source));
|
List<Long> collect = all.stream().distinct().collect(Collectors.toList());
|
collect.add(source);
|
|
List<ZfShareData> list = list(selectByUserId(collect));
|
list.addAll(list(selectBySharedId(collect)));
|
List<ZfShareData> collect1 = list.stream().distinct().collect(Collectors.toList());
|
for (ZfShareData zfShareData1:collect1){
|
zfShareData1.setMasterAccount(0);
|
updateById(zfShareData1);
|
}
|
return AjaxResult.success();
|
}
|
|
@Override
|
public List<ZfShareData> getShareId(Long id) {
|
LambdaQueryWrapper<ZfShareData> lqw = new LambdaQueryWrapper<>();
|
lqw.orderByDesc(ZfShareData::getId);
|
lqw.eq(ZfShareData::getSharedId,id);
|
return list(lqw);
|
}
|
|
@Override
|
public List<ZfShareData> getByShareId(List<Long> id) {
|
LambdaQueryWrapper<ZfShareData> lqw = new LambdaQueryWrapper<>();
|
lqw.orderByDesc(ZfShareData::getId);
|
lqw.in(ZfShareData::getSharedId,id);
|
return list(lqw);
|
}
|
}
|
