package com.zhujizheng.IHome.remark.service; import com.zhujizheng.IHome.generator.dao.PYRemark; import com.zhujizheng.IHome.generator.mapper.PYRemarkMapper; import com.zhujizheng.IHome.remark.dto.PYRemarkUpdateDTO; import com.zhujizheng.IHome.remark.vo.PYRemarkVO; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import java.util.ArrayList; import java.util.List; @Slf4j @Service @Transactional(rollbackFor = RuntimeException.class) public class PYRemarkServiceImpl implements PYRemarkService { @Autowired private PYRemarkMapper remarkMapper; @Override public List getAllRemark(int userId) { List list = remarkMapper.selectAllRemark(userId); if (list == null) { return null; } List voList = new ArrayList<>(); for (PYRemark remark : list) { PYRemarkVO vo = PYRemarkVO.createVOWithRemark(remark); voList.add(vo); } return voList; } @Override public PYRemarkVO getRemark(int userId, int friendId) { PYRemark remark = remarkMapper.selectRemark(userId, friendId); return PYRemarkVO.createVOWithRemark(remark); } @Override public int changeRemark(PYRemarkUpdateDTO updateDTO) { PYRemark oldRemark = remarkMapper.selectRemark(updateDTO.getUserId(), updateDTO.getFriendId()); if (oldRemark == null) { PYRemark remark = PYRemark.createWithUpdateDTO(updateDTO); return remarkMapper.insertRemark(remark); } else { return remarkMapper.updateRemark(oldRemark.getUserId(), oldRemark.getFriendId(), updateDTO.getRemark()); } } }