From: Willy Tarreau <w@1wt.eu>
Date: Mon, 25 Jul 2011 12:30:42 +0000 (+0200)
Subject: [MINOR] task: new function task_schedule() to schedule a wake up
X-Git-Tag: v1.5-dev8~166
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=26e4881a2d56cb191e5259ea74528a50ea2c098b;p=thirdparty%2Fhaproxy.git

[MINOR] task: new function task_schedule() to schedule a wake up

This function is used when a task should be woken up at most at a given
date. This will be used with rate shapers.
---

diff --git a/include/proto/task.h b/include/proto/task.h
index 2c02ca6ef5..adcae4cf00 100644
--- a/include/proto/task.h
+++ b/include/proto/task.h
@@ -224,6 +224,23 @@ static inline void task_queue(struct task *task)
 		__task_queue(task);
 }
 
+/* Ensure <task> will be woken up at most at <when>. If the task is already in
+ * the run queue (but not running), nothing is done. It may be used that way
+ * with a delay :  task_schedule(task, tick_add(now_ms, delay));
+ */
+static inline void task_schedule(struct task *task, int when)
+{
+	if (task_in_rq(task))
+		return;
+
+	if (task_in_wq(task))
+		when = tick_first(when, task->expire);
+
+	task->expire = when;
+	if (!task_in_wq(task) || tick_is_lt(task->expire, task->wq.key))
+		__task_queue(task);
+}
+
 /*
  * This does 4 things :
  *   - wake up all expired tasks