1254

whywhyo

2023-07-21 d2a1957ec40c70c696ab3260d7f7df90c4539589

 zhang-content/src/main/java/com/ruoyi/service/impl/ZHealthNoteServiceImpl.java

@@ -1,6 +1,10 @@
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.ZHealthNote;
import com.ruoyi.mapper.ZHealthNoteMapper;
import com.ruoyi.service.ZHealthNoteService;
@@ -14,4 +18,40 @@
 **/
@Service
public class ZHealthNoteServiceImpl extends ServiceImpl<ZHealthNoteMapper, ZHealthNote> implements ZHealthNoteService {
    @Override
    public AjaxResult listAll() {
        SysUser user = SecurityUtils.getLoginUser().getUser();
        Long userId = user.getUserId();

        LambdaQueryWrapper<ZHealthNote> lqw = new LambdaQueryWrapper<>();
        lqw.eq(ZHealthNote::getUid,userId);
        ZHealthNote result = getOne(lqw);
        if(result == null){
            throw new RuntimeException("请先填写好数据");
        }

        return AjaxResult.success(result);

    }

    @Override
    public AjaxResult saveData(ZHealthNote zHealthNote) {
        SysUser user = SecurityUtils.getLoginUser().getUser();
        Long userId = user.getUserId();

        LambdaQueryWrapper<ZHealthNote> lqw = new LambdaQueryWrapper<>();
        lqw.eq(ZHealthNote::getUid,userId);
        ZHealthNote result = getOne(lqw);

        if(result == null){
            zHealthNote.setUid(userId);
            save(zHealthNote);
        }else {
            zHealthNote.setId(result.getId());
            zHealthNote.setUid(userId);
            updateById(zHealthNote);
        }

        return AjaxResult.success();
    }
}