LocalTestServiceImpl.java
2.22 KB
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
package com.objecteye.service.impl;
import com.objecteye.entity.ViolateAlarmMsg;
import com.objecteye.pojo.RabbitMQVehicle;
import com.objecteye.service.ILocalTestService;
import com.objecteye.utils.CompareDistance;
import com.objecteye.utils.GlobalUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import java.util.List;
import java.util.stream.Collectors;
/**
* 本次测试
*
* @author officer
*/
@Component
public class LocalTestServiceImpl implements ILocalTestService {
@Autowired
private MongoTemplate mongoTemplate;
@Autowired
private RedisTemplate redisTemplate;
/**
* 创造数据到redis
*/
@Override
public void mongoToRedisVehicleFeature() {
Query query = new Query();
query.fields().include("id");
query.fields().include("vehicle_fea_feature");
List<RabbitMQVehicle> rabbitMqVehicles = mongoTemplate.find(query, RabbitMQVehicle.class);
for (RabbitMQVehicle rabbitMqVehicle : rabbitMqVehicles) {
redisTemplate.opsForHash().put(GlobalUtil.VEHICLE_FEATURE, rabbitMqVehicle.getId(), rabbitMqVehicle.getVehicle_fea_feature());
double dbFea = new CompareDistance().getDistance(rabbitMqVehicle.getVehicle_fea_feature());
redisTemplate.opsForHash().put(GlobalUtil.VEHICLE_FEATURE_DISTANCE, rabbitMqVehicle.getId(), dbFea);
}
}
/**
* 删除多余数据
*/
@Override
public void deleteOldDataNotInRabbitMqVehicleButViolation() {
Query rabbitMqVehicleQuery = new Query();
rabbitMqVehicleQuery.fields().include("id");
List<RabbitMQVehicle> rabbitMqVehicles = mongoTemplate.find(rabbitMqVehicleQuery, RabbitMQVehicle.class);
List<String> ids = rabbitMqVehicles.stream().map(RabbitMQVehicle::getId).collect(Collectors.toList());
Query violationQuery = new Query(Criteria.where("baseId").nin(ids));
mongoTemplate.remove(violationQuery, ViolateAlarmMsg.class);
}
}