package jnpf.dmsService.search;
|
|
import com.fasterxml.jackson.core.JsonProcessingException;
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.data.redis.core.StringRedisTemplate;
|
import org.springframework.stereotype.Component;
|
|
import java.time.Duration;
|
|
@Component
|
@RequiredArgsConstructor
|
public class DmsSearchIndexQueue {
|
private final StringRedisTemplate redisTemplate;
|
private final ObjectMapper objectMapper;
|
private final DmsSearchProperties properties;
|
|
public void push(DmsSearchIndexTask task) {
|
try {
|
redisTemplate.opsForList().leftPush(properties.getRedisQueueKey(),
|
objectMapper.writeValueAsString(task));
|
} catch (JsonProcessingException exception) {
|
throw new IllegalArgumentException("全文检索任务序列化失败", exception);
|
}
|
}
|
|
public DmsSearchIndexTask take() throws JsonProcessingException {
|
String value = redisTemplate.opsForList().rightPop(properties.getRedisQueueKey(),
|
Duration.ofSeconds(Math.max(1, properties.getRedisPollTimeoutSeconds())));
|
return value == null ? null : objectMapper.readValue(value, DmsSearchIndexTask.class);
|
}
|
}
|
