功能完善

liuhaonan

2021-12-31 0af78b53f3e2b272ea2ec6ff923abe3b9815b19f

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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package com.sandu.ximon.admin.service;
 
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.ClientPrarm;
import com.sandu.ximon.admin.security.SecurityUtils;
import com.sandu.ximon.dao.domain.Client;
import com.sandu.ximon.dao.mapper.ClientMapper;
import lombok.AllArgsConstructor;
import org.springframework.stereotype.Service;
 
import java.util.List;
 
@Service
@AllArgsConstructor
public class ClientService extends BaseServiceImpl<ClientMapper, Client> {
 
    private final ClientMapper clientMapper;
 
    public boolean addClient(ClientPrarm clientPrarm) {
        Client client = new Client();
        if (clientPrarm.getClientSuperior() != null) {
            Client one = getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getClientName, clientPrarm.getClientSuperior()));
            if (one != null) {
                client.setSuperiorId(one.getId());
            } else {
                throw new BusinessException("上级客户不存在");
            }
        }
        client.setClientName(clientPrarm.getClientName());
        client.setLinkMan(clientPrarm.getLinkMan());
        client.setMobile(clientPrarm.getMobile());
        client.setClientSuperior(clientPrarm.getClientSuperior());
 
 
        return save(client);
    }
 
 
    public boolean updateClient(Long id, ClientPrarm clientPrarm) {
        Client one = getById(id);
        if (one == null) {
            throw new BusinessException("该客户不存在");
        }
        Client client = new Client();
        if (clientPrarm.getClientSuperior() != null) {
            Client superior = getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getClientName, clientPrarm.getClientSuperior()));
            if (one != null) {
                client.setSuperiorId(superior.getId());
            } else {
                throw new BusinessException("上级客户不存在");
            }
        }
        client.setId(id);
        client.setClientName(clientPrarm.getClientName());
        client.setLinkMan(clientPrarm.getLinkMan());
        client.setMobile(clientPrarm.getMobile());
        client.setClientSuperior(clientPrarm.getClientSuperior());
 
        // update(client);
        return updateById(client);
    }
 
    public boolean deleteClient(Long id) {
        Client one = getById(id);
        if (one == null) {
            throw new BusinessException("该客户不存在");
        }
        return removeById(id);
    }
 
    public List<Client>  clientList(Long userId) {
        return clientMapper.clientList(userId);
    }
 
    public Client findByPhone(String phone) {
        return getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getMobile, phone).last("limit 1"));
    }
 
    /**\
     * 其他类用来查找客户id使用  如果没有上级客户 这返回用户ID
     * @param
     * @return
     */
    public Long getClientId(){
        Long userId = SecurityUtils.getUserId();
        Client one = getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getId, userId));
        if(one!=null&&one.getSuperiorId()!=null){
            return one.getSuperiorId();
        }else {
            return userId;
        }
 
    }
}