fix

zhanzhiqin

2022-04-26 af9cb02606e8f8d5d8bbad05ba1647dc7a91ffc3

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
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;
import com.sandu.ximon.admin.security.SecurityUtils;
import com.sandu.ximon.dao.domain.LEDProgramFile;
import com.sandu.ximon.dao.mapper.LEDProgramFileMapper;
import lombok.AllArgsConstructor;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
 
@Service
@AllArgsConstructor
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);
        if (SecurityUtils.getClientId() != null) {
            file.setUserId(SecurityUtils.getUserId());
            file.setUserName(SecurityUtils.getUsername());
            if (clientService.findClientId()) {
                file.setClientId(clientService.getClientId());
            }
        }
        return save(file);
    }
 
    public boolean deleteFile(Long id) {
        LEDProgramFile byId = getById(id);
        if (byId == null) {
            throw new BusinessException("未找到该节目");
        }
        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;
        }
 
    }
 
 
}