完善全网搜（腾讯/优酷）

admin

2021-03-25 9438097857e16f4929924d0d349c346d36b5f947

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
package com.yeshi.buwan.service.imp.juhe;
 
import com.yeshi.buwan.dao.juhe.youku.InternetSearchVideoDao;
import com.yeshi.buwan.domain.video.InternetSearchVideo;
import com.yeshi.buwan.exception.ParamsException;
import com.yeshi.buwan.service.inter.juhe.InternetSearchVideoService;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;
import org.yeshi.utils.StringUtil;
 
import javax.annotation.Resource;
import javax.validation.Valid;
import java.util.*;
 
@Service
public class InternetSearchVideoServiceImpl implements InternetSearchVideoService {
 
    @Resource
    private InternetSearchVideoDao internetSearchVideoDao;
 
 
    @Validated
    @Override
    public void save(@Valid InternetSearchVideo video) throws ParamsException, Exception {
        InternetSearchVideo oldVideo = internetSearchVideoDao.get(video.getId());
        Set<Integer> resourceIds = listResourceId(video);
        resourceIds.add(Integer.parseInt(video.getResourceIds()));
        Integer[] rids = new Integer[resourceIds.size()];
        resourceIds.toArray(rids);
        video.setResourceIds(StringUtil.concat(rids, ","));
        if (oldVideo == null) {
            video.setCreateTime(new Date());
            internetSearchVideoDao.save(video);
        } else {
            internetSearchVideoDao.updateSelective(video);
        }
    }
 
    private Set<Integer> listResourceId(InternetSearchVideo video) {
        if (video == null) {
            return new HashSet<>();
        }
        List<String> list = Arrays.asList(video.getResourceIds().split(","));
        Set<Integer> sets = new HashSet<>();
        for (String st : list) {
            sets.add(Integer.parseInt(st));
        }
        return sets;
    }
 
    @Override
    public Set<Integer> listResourceId(String id) {
        InternetSearchVideo oldVideo = internetSearchVideoDao.get(id);
        return listResourceId(oldVideo);
    }
 
    @Override
    public long countAll() {
 
        return internetSearchVideoDao.count(new Query());
    }
 
    @Override
    public List<InternetSearchVideo> listAll(int page, int pageSize) {
        Query query = new Query();
        query.skip((page - 1) * pageSize);
        query.limit(pageSize);
        return internetSearchVideoDao.findList(query);
    }
 
    @Override
    public InternetSearchVideo selectByPrimaryKey(String id) {
        return internetSearchVideoDao.get(id);
    }
 
 
}