84 lines
2.6 KiB
Plaintext
84 lines
2.6 KiB
Plaintext
package com.bonus.sq.service;
|
||
|
||
|
||
import java.util.List;
|
||
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.stereotype.Service;
|
||
|
||
import java.util.Set;
|
||
import java.util.HashSet;
|
||
import java.util.stream.Collectors;
|
||
|
||
import com.bonus.sq.beans.MachineStatusQueryBean;
|
||
import com.bonus.sq.dao.MachineStatusQueryDao;
|
||
import com.bonus.sys.BaseServiceImp;
|
||
|
||
@Service("query")
|
||
public class MachineStatusQueryServiceImp extends BaseServiceImp<MachineStatusQueryBean> implements MachineStatusQueryService {
|
||
|
||
@Autowired
|
||
MachineStatusQueryDao dao;
|
||
|
||
@Override
|
||
public List<MachineStatusQueryBean> machineStatusQuery(MachineStatusQueryBean o) {
|
||
// TODO Auto-generated method stub
|
||
List<MachineStatusQueryBean> list = dao.machineStatusQuery(o);
|
||
if (list != null && !list.isEmpty()) {
|
||
// 查询收藏表，获取收藏机具类型id
|
||
List<MachineStatusQueryBean> typeIdList = dao.findTypeId(o);
|
||
if (typeIdList != null && !typeIdList.isEmpty()) {
|
||
// 提取 typeIdList 中的 typeId 放入新的 Set<String> 中
|
||
Set<String> typeIdSet = typeIdList.stream()
|
||
.map(MachineStatusQueryBean::getTypeId)
|
||
.collect(Collectors.toSet());
|
||
// 使用 Stream 过滤 list 中的 typeId 在 typeIdSet 中的数据
|
||
if (o.getIsCollect() != null && o.getIsCollect().equals(0)) {
|
||
return list.stream()
|
||
.filter(item -> typeIdSet.contains(item.getTypeId()))
|
||
.collect(Collectors.toList());
|
||
} else {
|
||
// 遍历 list，修改每个 item 的 isCollect 字段
|
||
List<MachineStatusQueryBean> filteredList = list.stream()
|
||
.map(item -> {
|
||
// 检查 typeId 是否存在于 typeIdSet 中，并设置 isCollect 值
|
||
if (typeIdSet.contains(item.getTypeId())) {
|
||
// typeId 存在，设置 isCollect 为 0
|
||
item.setIsCollect(0);
|
||
} else {
|
||
// typeId 不存在，设置 isCollect 为 1
|
||
item.setIsCollect(1);
|
||
}
|
||
return item;
|
||
})
|
||
.collect(Collectors.toList());
|
||
return filteredList;
|
||
}
|
||
}
|
||
}
|
||
return list;
|
||
}
|
||
|
||
/**
|
||
* 添加或移除收藏
|
||
* @param o
|
||
* @return
|
||
*/
|
||
@Override
|
||
public int addOrRemoveCollect(MachineStatusQueryBean o) {
|
||
// 判断收藏状态并调用相应的 DAO 方法
|
||
if (o.getIsCollect() != null && o.getIsCollect().equals(0)) {
|
||
// 状态为0时执行添加收藏操作
|
||
return dao.addCollect(o);
|
||
} else if (o.getIsCollect() != null && o.getIsCollect().equals(1)) {
|
||
// 状态为1时执行移除收藏操作
|
||
return dao.removeCollect(o);
|
||
} else {
|
||
// 如果 o.getIsCollect() 不是 0 或 1，抛出异常
|
||
throw new RuntimeException("Invalid collect state: " + o.getIsCollect());
|
||
}
|
||
}
|
||
|
||
|
||
}
|