Blame view

src/main/java/com/objecteye/service/impl/LocalTestServiceImpl.java 2.22 KB
c83b5b39   Liu Haoyu   项目创建, 集成spring-se...
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);
      }
  }