fix

zhanzhiqin

2022-04-26 af9cb02606e8f8d5d8bbad05ba1647dc7a91ffc3

 ximon-admin/src/main/java/com/sandu/ximon/admin/service/LEDProgramFileService.java

@@ -1,5 +1,7 @@
package com.sandu.ximon.admin.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.sandu.common.execption.BusinessException;
import com.sandu.common.service.impl.BaseServiceImpl;
import com.sandu.ximon.admin.param.LEDProgramFileParam;
@@ -15,13 +17,17 @@
public class LEDProgramFileService extends BaseServiceImpl<LEDProgramFileMapper, LEDProgramFile> {

    private final LEDProgramFileMapper ledProgramFileMapper;
    private final ClientService clientService;

    public boolean addProgramFile(LEDProgramFileParam param) {
        LEDProgramFile file = new LEDProgramFile();
        BeanUtils.copyProperties(param, file);
        file.setUserId(SecurityUtils.getClientId());
        if (SecurityUtils.getClientId() != null) {
            file.setUserId(SecurityUtils.getUserId());
            file.setUserName(SecurityUtils.getUsername());
            if (clientService.findClientId()) {
                file.setClientId(clientService.getClientId());
            }
        }
        return save(file);
    }
@@ -34,5 +40,17 @@
        return removeById(id);
    }

    public LambdaQueryWrapper<LEDProgramFile> listFile() {
        LambdaQueryWrapper<LEDProgramFile> ledProgramFileLambdaQueryWrapper = Wrappers.lambdaQuery(LEDProgramFile.class);
        if (SecurityUtils.getClientId() != null) {
            return ledProgramFileLambdaQueryWrapper.eq(LEDProgramFile::getUserId, SecurityUtils.getUserId()).or(w -> {
                w.eq(LEDProgramFile::getClientId, SecurityUtils.getUserId());
            });
        } else {
            return ledProgramFileLambdaQueryWrapper;
        }

    }


}