| | |
| | | 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.SysUser; |
| | | import com.ruoyi.common.utils.SecurityUtils; |
| | | import com.ruoyi.domain.MarrySelf; |
| | | import com.ruoyi.domain.MarryUser; |
| | | import com.ruoyi.mapper.MarryUserMapper; |
| | | import com.ruoyi.service.MarryUserService; |
| | | import org.springframework.stereotype.Service; |
| | | |
| | | import java.util.List; |
| | | |
| | | /** |
| | | * @Author Jinquan_Ou |
| | |
| | | **/ |
| | | @Service |
| | | public class MarryUserServiceImpl extends ServiceImpl<MarryUserMapper, MarryUser> implements MarryUserService { |
| | | @Override |
| | | public AjaxResult getInfo() { |
| | | |
| | | SysUser user = SecurityUtils.getLoginUser().getUser(); |
| | | Long userId = user.getUserId(); |
| | | |
| | | LambdaQueryWrapper<MarryUser> lqw = new LambdaQueryWrapper<>(); |
| | | lqw.eq(MarryUser::getUid,userId); |
| | | List<MarryUser> frontList = list(lqw); |
| | | return AjaxResult.success(frontList); |
| | | } |
| | | |
| | | @Override |
| | | public AjaxResult addData(MarryUser marryUser) { |
| | | SysUser user = SecurityUtils.getLoginUser().getUser(); |
| | | Long userId = user.getUserId(); |
| | | |
| | | marryUser.setUid(userId); |
| | | save(marryUser); |
| | | return AjaxResult.success(); |
| | | } |
| | | |
| | | @Override |
| | | public AjaxResult updateData(MarryUser marryUser) { |
| | | // SysUser user = SecurityUtils.getLoginUser().getUser(); |
| | | // Long userId = user.getUserId(); |
| | | // |
| | | // marryUser.setUid(userId); |
| | | |
| | | LambdaQueryWrapper<MarryUser> lqw = new LambdaQueryWrapper<>(); |
| | | lqw.eq(MarryUser::getUid,marryUser.getUid()); |
| | | List<MarryUser> list = list(lqw); |
| | | if(list.size()>0) { |
| | | marryUser.setId(list.get(0).getId()); |
| | | updateById(marryUser); |
| | | }else save(marryUser); |
| | | |
| | | |
| | | return AjaxResult.success(); |
| | | |
| | | } |
| | | |
| | | |
| | | } |