Implement scheduler_touch_channel()

2 files changed, 27 insertions, 0 deletions

diff --git a/src/or/scheduler.c b/src/or/scheduler.c
index 4f12696c68..140ff2fef1 100644
--- a/src/or/scheduler.c
+++ b/src/or/scheduler.c
@@ -580,6 +580,30 @@ scheduler_channel_wants_writes(channel_t *chan)
 }
 
 /**
+ * Notify the scheduler that a channel's position in the pqueue may have
+ * changed
+ */
+
+void
+scheduler_touch_channel(channel_t *chan)
+{
+  tor_assert(chan);
+
+  if (chan->scheduler_state == SCHED_CHAN_PENDING) {
+    /* Remove and re-add it */
+    smartlist_pqueue_remove(channels_pending,
+                            scheduler_compare_channels,
+                            STRUCT_OFFSET(channel_t, sched_heap_idx),
+                            chan);
+    smartlist_pqueue_add(channels_pending,
+                         scheduler_compare_channels,
+                         STRUCT_OFFSET(channel_t, sched_heap_idx),
+                         chan);
+  }
+  /* else no-op, since it isn't in the queue */
+}
+
+/**
  * Notify the scheduler of a queue size adjustment, to recalculate the
  * queue heuristic.
  */
diff --git a/src/or/scheduler.h b/src/or/scheduler.h
index 8fe59cb0b3..e2d2eb5192 100644
--- a/src/or/scheduler.h
+++ b/src/or/scheduler.h
@@ -30,5 +30,8 @@ void scheduler_release_channel(channel_t *chan);
 /* Notify scheduler of queue size adjustments */
 void scheduler_adjust_queue_size(channel_t *chan, char dir, uint64_t adj);
 
+/* Notify scheduler that a channel's queue position may have changed */
+void scheduler_touch_channel(channel_t *chan);
+
 #endif /* !defined(TOR_SCHEDULER_H) */