為什么使用zset實現延遲任務
ZSET（有序集合）適合實現延遲任務的原因主要有以下幾點：
- 排序特性：
ZSET根據分數（score）自動排序，這使得我們可以將任務的執行時間作為分數，從而能夠輕松地獲取到即將執行的任務。 - 范圍查詢：
ZSET支持范圍查詢，我們可以查詢分數在某個區間內的所有元素，這對于獲取所有已到期的任務非常方便。 - 高效性：
ZSET的操作通常都是高效的，比如添加、刪除和范圍查詢等操作的時間復雜度通常為O(log(N))，這使得它適合用于需要頻繁操作的任務隊列。 - 原子操作：
ZSET提供了原子操作，比如添加元素時如果元素已存在則會更新其分數，這有助于避免在并發環境中出現數據一致性問題。 - 靈活性：
ZSET允許同一個集合中存在多個相同分數的元素，這使得我們可以存儲多個同一時間到期的任務。
實現步驟
1、添加redis依賴
<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>
2、配置redis
spring.redis.host=localhost
spring.redis.port=6379
3、創建ZSET操作服務
創建一個服務類來封裝對ZSET的操作：
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.stereotype.Service;
import java.util.Set;
@Service
public class DelayTaskService {// 定義一個常量作為延遲任務隊列的鍵private static final String DELAY_TASK_KEY = "delayTaskQueue";// 注入RedisTemplate用于操作Redisprivate final RedisTemplate<String, String> redisTemplate;// 通過構造函數注入RedisTemplate@Autowiredpublic DelayTaskService(RedisTemplate<String, String> redisTemplate) {this.redisTemplate = redisTemplate;}/*** 添加一個延遲任務* @param taskId 任務ID* @param delayMillis 延遲執行的毫秒數*/public void addTask(String taskId, long delayMillis) {// 計算任務的執行時間（當前時間 + 延遲時間）long executeTime = System.currentTimeMillis() + delayMillis;// 將任務添加到ZSET中，分數為執行時間redisTemplate.opsForZSet().add(DELAY_TASK_KEY, taskId, executeTime);}/*** 處理到期的任務*/public void processTasks() {// 獲取當前時間long now = System.currentTimeMillis();// 查詢ZSET中小于或等于當前時間的所有任務Set<String> tasks = redisTemplate.opsForZSet().rangeByScore(DELAY_TASK_KEY, 0, now);// 如果有任務，則進行處理if (tasks != null && !tasks.isEmpty()) {for (String taskId : tasks) {// 從ZSET中移除任務，如果移除成功（任務存在），則處理任務if (redisTemplate.opsForZSet().remove(DELAY_TASK_KEY, taskId) > 0) {handleTask(taskId);}}}}/*** 處理具體的任務* @param taskId 任務ID*/private void handleTask(String taskId) {// 這里實現任務處理的邏輯System.out.println("處理任務：" + taskId);}
}
4、定時輪詢ZSET
創建一個定時任務來定期檢查ZSET中是否有到期的任務：
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
@Component
@EnableScheduling
public class TaskScheduler {// 注入DelayTaskService用于處理任務private final DelayTaskService delayTaskService;// 通過構造函數注入DelayTaskService@Autowiredpublic TaskScheduler(DelayTaskService delayTaskService) {this.delayTaskService = delayTaskService;}/*** 定時檢查并處理任務*/@Scheduled(fixedRate = 1000) // 每1000毫秒執行一次public void checkAndProcessTasks() {// 調用DelayTaskService的processTasks方法delayTaskService.processTasks();}
}
DelayTaskService負責添加和處理延遲任務，而TaskScheduler負責定期調用processTasks方法來檢查和處理到期的任務。
)
