使用长型键的spring数据redis zadd不起作用

时间:2019-07-06 23:39:01

标签: java redis spring-data-redis

我试图通过举例说明在这里学习Redis。我有一个名为DriverLocation的实体,其中有一个称为updatedAt的时间戳,这是一个以毫秒表示的纪元时间。事情是我想要一个SortedSet,以便我可以进行zrangebyscore查询以获取按其时间戳排序的最后N条记录。

sortedset将基本上具有timestamp --> hash_id结构。如果我想获取最近10分钟添加的记录,那么我将执行zrangebyscore查询以排序的方式获取所有hash_id。然后将hmget与hash_ids一起使用以获取所有哈希对象。

这是redis-cli上的一个非常简单的工作演示,您可以假设我使用的是3位数字而不是毫秒。

localhost:6379> zadd locations_0 213 hash_id_1
(integer) 1
localhost:6379> zadd locations_0 214 hash_id_2
(integer) 1
localhost:6379> zadd locations_0 215 hash_id_3
(integer) 1
localhost:6379> zrangebyscore locations_0 212 214
1) "hash_id_1"
2) "hash_id_2"

在redis cli上一切正常。但是,在Spring方面,我无法实现自己的目标。

DriverLocation.java

@RedisHash("driverLocation")
@Data
@AllArgsConstructor
@NoArgsConstructor
public class DriverLocationEntity {

    @Id
    private Long id;

    @Indexed
    private Long driverId;

    @GeoIndexed
    private Point point;

    private Date updatedAt;
}

RedisConfiguration.java

@Configuration
@EnableRedisRepositories
public class RedisConfiguration {

    @Value("${spring.redis.host}")
    private String redisHost;

    @Value("${spring.redis.port}")
    private int redisPort;

    @Bean
    public LettuceConnectionFactory redisConnectionFactory() {
        return new LettuceConnectionFactory(redisHost, redisPort);
    }

    @Bean
    public RedisTemplate<Object, Object> redisTemplate() {
        RedisTemplate<Object, Object> template = new RedisTemplate<>();
        template.setConnectionFactory(redisConnectionFactory());
        return template;
    }

    @Bean
    public RedisAtomicLong redisAtomicLong() {
        RedisAtomicLong redisAtomicLong = new RedisAtomicLong("DriverLocationIdCounter", redisConnectionFactory(), 0L);
        return redisAtomicLong;
    }
}

Controller.java

@RestController
@RequestMapping("/drivers")
@Slf4j
public class DriverLocationController {

    @Autowired
    private DriverLocationRepository driverLocationRepository;

    @Autowired
    private RedisTemplate<Object, Object> redisTemplate;

    @Autowired
    private RedisAtomicLong redisAtomicLong;

    @Autowired
    private ObjectMapper objectMapper;

    @RequestMapping("/{id}")
    public ResponseEntity<List<DriverLocationEntity>> getDriver(@PathVariable("id") Long driverId) {
        long now = Instant.now().getEpochSecond();

        Set<Object> ids = redisTemplate.opsForZSet().rangeByScore(
                "locations_" + driverId,
                Instant.ofEpochSecond(now).minusSeconds(300).toEpochMilli(),
                Instant.ofEpochSecond(now).toEpochMilli());
        List<Object> driverLocations = redisTemplate.opsForHash().multiGet("driverLocations", ids.stream().map(id -> (Long) id).collect(Collectors.toList()));

        return ResponseEntity.status(HttpStatus.OK).body(driverLocations.stream().map(dLoc -> (DriverLocationEntity) dLoc).collect(Collectors.toList()));
    }

    @RequestMapping(value = "/{id}", method = RequestMethod.POST)
    public ResponseEntity<String> addDriverLocation(@RequestBody DriverLocationMessageEntity messageEntity,
                                                    @PathVariable("id") Long driverId) throws JsonProcessingException {
        long now = Instant.now().toEpochMilli();
        DriverLocationEntity driverLocationEntity = new DriverLocationEntity();
        driverLocationEntity.setDriverId(driverId);
        driverLocationEntity.setPoint(new Point(messageEntity.getLongitude(), messageEntity.getLatitude()));
        driverLocationEntity.setUpdatedAt(new Date(now));
        driverLocationEntity.setId(redisAtomicLong.getAndIncrement());

        String strVal = objectMapper.writeValueAsString(driverLocationEntity);

        // save driver location entity
        driverLocationRepository.save(driverLocationEntity);

        // save timestamp -> hash_id
        redisTemplate.opsForZSet().add("locations_" + driverLocationEntity.getDriverId(), now, driverLocationEntity.getId());

        return ResponseEntity.status(HttpStatus.OK).body("done");
    }
}

感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

我基本上是通过将Redisson客户端与他们的StringCodec一起使用的

        DriverLocationEntity driverLocationEntity = new DriverLocationEntity();
        driverLocationEntity.setDriverId(messageEntity.getDriverId());
        driverLocationEntity.setPoint(new Point(messageEntity.getLongitude(), messageEntity.getLatitude()));
        driverLocationEntity.setUpdatedAt(new Date(now));
        driverLocationEntity.setId(UUID.randomUUID().toString());

        // add hash_id into geoset for distance calculations
        redissonClient.getGeo(GEO_SET_KEY_NAME, new StringCodec()).add(messageEntity.getLongitude(), messageEntity.getLatitude(), HASH_ID_PREFIX + driverLocationEntity.getId());