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
 * @Description
 * @Date 2023-04-23 13:31
 * @Version 1.0.0
 **/
@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();

    }


}