package com.ruoyi.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
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.common.utils.MapUtils;
|
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.ArrayList;
|
import java.util.Arrays;
|
import java.util.HashMap;
|
import java.util.List;
|
|
@Service
|
public class ZfShareDataServiceImpl extends ServiceImpl<ZfShareDataMapper, ZfShareData> implements ZfShareDataService {
|
|
@Resource
|
private ZInfoUserService zInfoUserService;
|
|
@Resource
|
private RestHighLevelClient restHighLevelClient;
|
|
@Resource
|
private ZfClanService zfClanService;
|
|
|
@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() {
|
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();
|
|
}
|
|
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 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();
|
}
|
}
|
|
@Override
|
public AjaxResult setAdmin(Long userId,Long id) {
|
List<ZfShareData> list = list(buildCondition(getSource(userId)));
|
list.addAll(getShareId(userId));
|
for (ZfShareData zfShareData1:list){
|
zfShareData1.setMasterAccount(id.intValue());
|
updateById(zfShareData1);
|
}
|
return AjaxResult.success();
|
}
|
|
@Override
|
public AjaxResult allUser(Long id) {
|
List<ZfShareData> list = list(buildCondition(getSource(id)));
|
list.addAll(getShareId(id));
|
for (ZfShareData zfShareData1:list){
|
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);
|
}
|
}
|
