zqy
5 天以前 b02beccf4567068cb47a3f1181a00039456c872d
zhang-content/src/main/java/com/ruoyi/service/impl/MarryUserServiceImpl.java
@@ -1,10 +1,17 @@
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
@@ -14,4 +21,47 @@
 **/
@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();
    }
}