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());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|