优化

2026-08-20 18:02:38 +08:00
parent b55b049e77
commit 1cd8b6a473
210 changed files with 6974 additions and 825 deletions
@@ -0,0 +1,82 @@
+package com.baiyue.lastleg.repository;
+
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
+import com.baiyue.common.event.EventTypes;
+import com.baiyue.lastleg.entity.LastlegEventOutbox;
+import com.baiyue.lastleg.mapper.LastlegEventOutboxMapper;
+import jakarta.annotation.Resource;
+import org.springframework.stereotype.Repository;
+
+import java.time.LocalDateTime;
+import java.util.List;
+
+@Repository
+public class UnloadCompletedOutboxRepository {
+    @Resource private LastlegEventOutboxMapper mapper;
+
+    public List<LastlegEventOutbox> pending(int limit, LocalDateTime now, LocalDateTime expiredLeaseBefore) {
+        return mapper.selectList(new LambdaQueryWrapper<LastlegEventOutbox>()
+            .eq(LastlegEventOutbox::getEventType, EventTypes.OVERSEAS_CONTAINER_UNLOADED)
+            .and(status -> status.eq(LastlegEventOutbox::getStatus, "PENDING")
+                .and(retry -> retry.isNull(LastlegEventOutbox::getNextRetryAt)
+                    .or().le(LastlegEventOutbox::getNextRetryAt, now))
+                .or(stale -> stale.eq(LastlegEventOutbox::getStatus, "PROCESSING")
+                    .le(LastlegEventOutbox::getLockedAt, expiredLeaseBefore)))
+            .orderByAsc(LastlegEventOutbox::getCreatedAt).orderByAsc(LastlegEventOutbox::getId)
+            .last("LIMIT " + Math.min(Math.max(limit, 1), 100)));
+    }
+
+    public boolean claim(long id, String worker, LocalDateTime now, LocalDateTime expiredLeaseBefore) {
+        return mapper.update(null, new LambdaUpdateWrapper<LastlegEventOutbox>()
+            .eq(LastlegEventOutbox::getId, id)
+            .eq(LastlegEventOutbox::getEventType, EventTypes.OVERSEAS_CONTAINER_UNLOADED)
+            .and(status -> status.eq(LastlegEventOutbox::getStatus, "PENDING")
+                .or(stale -> stale.eq(LastlegEventOutbox::getStatus, "PROCESSING")
+                    .le(LastlegEventOutbox::getLockedAt, expiredLeaseBefore)))
+            .set(LastlegEventOutbox::getStatus, "PROCESSING")
+            .set(LastlegEventOutbox::getLockedBy, worker)
+            .set(LastlegEventOutbox::getLockedAt, now)) == 1;
+    }
+
+    public LastlegEventOutbox claimed(long id, String worker) {
+        return mapper.selectOne(new LambdaQueryWrapper<LastlegEventOutbox>()
+            .eq(LastlegEventOutbox::getId, id)
+            .eq(LastlegEventOutbox::getEventType, EventTypes.OVERSEAS_CONTAINER_UNLOADED)
+            .eq(LastlegEventOutbox::getStatus, "PROCESSING")
+            .eq(LastlegEventOutbox::getLockedBy, worker));
+    }
+
+    public void published(long id, String worker, LocalDateTime now) {
+        mapper.update(null, claimedBase(id, worker)
+            .set(LastlegEventOutbox::getStatus, "PUBLISHED")
+            .set(LastlegEventOutbox::getPublishedAt, now)
+            .set(LastlegEventOutbox::getNextRetryAt, null)
+            .set(LastlegEventOutbox::getLockedBy, null)
+            .set(LastlegEventOutbox::getLockedAt, null)
+            .set(LastlegEventOutbox::getLastError, null));
+    }
+
+    public void retry(long id, String worker, LocalDateTime retryAt, String error) {
+        mapper.update(null, claimedBase(id, worker)
+            .setSql("retry_count = retry_count + 1")
+            .set(LastlegEventOutbox::getStatus, "PENDING")
+            .set(LastlegEventOutbox::getNextRetryAt, retryAt)
+            .set(LastlegEventOutbox::getLockedBy, null)
+            .set(LastlegEventOutbox::getLockedAt, null)
+            .set(LastlegEventOutbox::getLastError, truncate(error)));
+    }
+
+    private LambdaUpdateWrapper<LastlegEventOutbox> claimedBase(long id, String worker) {
+        return new LambdaUpdateWrapper<LastlegEventOutbox>()
+            .eq(LastlegEventOutbox::getId, id)
+            .eq(LastlegEventOutbox::getEventType, EventTypes.OVERSEAS_CONTAINER_UNLOADED)
+            .eq(LastlegEventOutbox::getStatus, "PROCESSING")
+            .eq(LastlegEventOutbox::getLockedBy, worker);
+    }
+
+    private static String truncate(String value) {
+        String safe = value == null || value.isBlank() ? "unknown" : value.replaceAll("[\\r\\n\\t]+", " ");
+        return safe.length() > 500 ? safe.substring(0, 500) : safe;
+    }
+}