wang-hao-jie
2021-10-19 e48043a2df9ca0c73fe18298bab3c4d42ca5c0c7
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
package cn.exrick.xboot.social.serviceimpl;
 
import cn.exrick.xboot.core.common.vo.SearchVo;
import cn.exrick.xboot.social.dao.SocialDao;
import cn.exrick.xboot.social.entity.Social;
import cn.exrick.xboot.social.service.SocialService;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.StrUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.lang.Nullable;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import javax.persistence.criteria.*;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
 
/**
 * Github登录接口实现
 * @author Exrick
 */
@Slf4j
@Service
@Transactional
public class SocialServiceImpl implements SocialService {
 
    @Autowired
    private SocialDao socialDao;
 
    @Override
    public SocialDao getRepository() {
        return socialDao;
    }
 
    @Override
    public Social findByOpenIdAndPlatform(String openId, Integer platform) {
 
        return socialDao.findByOpenIdAndPlatform(openId, platform);
    }
 
    @Override
    public Social findByRelateUsernameAndPlatform(String relateUsername, Integer platform) {
 
        return socialDao.findByRelateUsernameAndPlatform(relateUsername, platform);
    }
 
    @Override
    public List<Social> findByRelateUsername(String relateUsername) {
 
        return socialDao.findByRelateUsername(relateUsername);
    }
 
    @Override
    public Page<Social> findByCondition(Social social, SearchVo searchVo, Pageable pageable) {
 
        return socialDao.findAll(new Specification<Social>() {
            @Nullable
            @Override
            public Predicate toPredicate(Root<Social> root, CriteriaQuery<?> cq, CriteriaBuilder cb) {
 
                Path<String> usernameField = root.get("username");
                Path<String> relateUsernameField = root.get("relateUsername");
                Path<Integer> platformField = root.get("platform");
                Path<Date> createTimeField = root.get("createTime");
 
                List<Predicate> list = new ArrayList<>();
 
                if (StrUtil.isNotBlank(social.getUsername())) {
                    list.add(cb.like(usernameField, '%' + social.getUsername() + '%'));
                }
                if (StrUtil.isNotBlank(social.getRelateUsername())) {
                    list.add(cb.like(relateUsernameField, '%' + social.getRelateUsername() + '%'));
                }
                if (social.getPlatform() != null) {
                    list.add(cb.equal(platformField, social.getPlatform()));
                }
                //创建时间
                if (StrUtil.isNotBlank(searchVo.getStartDate()) && StrUtil.isNotBlank(searchVo.getEndDate())) {
                    Date start = DateUtil.parse(searchVo.getStartDate());
                    Date end = DateUtil.parse(searchVo.getEndDate());
                    list.add(cb.between(createTimeField, start, DateUtil.endOfDay(end)));
                }
 
                Predicate[] arr = new Predicate[list.size()];
                cq.where(list.toArray(arr));
                return null;
            }
        }, pageable);
    }
}