perf: optimize TicketSet#getTargetStatus

RelativityMC · Nov 11, 2024 · e3c80d4 · e3c80d4
1 parent 1bd210e
commit e3c80d4
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/main/java/com/ishland/flowsched/scheduler/TicketSet.java b/src/main/java/com/ishland/flowsched/scheduler/TicketSet.java
@@ -11,11 +11,11 @@ public class TicketSet<K, V, Ctx> {
 
     private final ItemStatus<K, V, Ctx> initialStatus;
     private final Set<ItemTicket<K, V, Ctx>>[] status2Tickets;
-//    private volatile int targetStatus = 0;
+    private volatile int targetStatus = 0;
 
     public TicketSet(ItemStatus<K, V, Ctx> initialStatus, ObjectFactory objectFactory) {
         this.initialStatus = initialStatus;
-//        this.targetStatus = initialStatus.ordinal();
+        this.targetStatus = initialStatus.ordinal();
         ItemStatus<K, V, Ctx>[] allStatuses = initialStatus.getAllStatuses();
         this.status2Tickets = new Set[allStatuses.length];
         for (int i = 0; i < allStatuses.length; i++) {
@@ -29,6 +29,8 @@ public boolean add(ItemTicket<K, V, Ctx> ticket) {
         final boolean added = this.status2Tickets[targetStatus.ordinal()].add(ticket);
         if (!added) return false;
 
+        this.targetStatus = this.computeTargetStatusSlow();
+
 //        if (this.targetStatus < targetStatus.ordinal()) {
 //            this.targetStatus = targetStatus.ordinal();
 //        }
@@ -41,6 +43,8 @@ public boolean remove(ItemTicket<K, V, Ctx> ticket) {
         final boolean removed = this.status2Tickets[targetStatus.ordinal()].remove(ticket);
         if (!removed) return false;
 
+        this.targetStatus = this.computeTargetStatusSlow();
+
 //        decreaseStatusAtomically();
 
         return true;
@@ -65,7 +69,7 @@ public boolean remove(ItemTicket<K, V, Ctx> ticket) {
 //    }
 
     public ItemStatus<K, V, Ctx> getTargetStatus() {
-        return this.initialStatus.getAllStatuses()[this.computeTargetStatusSlow()];
+        return this.initialStatus.getAllStatuses()[this.targetStatus];
     }
 
     public Set<ItemTicket<K, V, Ctx>> getTicketsForStatus(ItemStatus<K, V, Ctx> status) {