queue api + queue limits

1 files changed, 35 insertions, 0 deletions

diff --git a/include/queue_fn.php b/include/queue_fn.php
index bc47ceffd..2168d42d7 100644
--- a/include/queue_fn.php
+++ b/include/queue_fn.php
@@ -14,3 +14,38 @@ function remove_queue_item($id) {
 		intval($id)
 	);
 }
+
+
+function add_to_queue($cid,$network,$msg,$batch = false) {
+
+	$max_queue = get_config('system','max_contact_queue');
+	if($max_queue < 1)
+		$max_queue = 500;
+
+	$batch_queue = get_config('system','max_batch_queue');
+	if($batch_queue < 1)
+		$batch_queue = 1000;
+
+	$r = q("SELECT COUNT(*) AS `total` FROM `queue` left join `contact` WHERE ``queue`.`cid` = %d AND `contact`.`self` = 0 ",
+		intval($cid)
+	);
+	if($r && count($r)) {
+		if($batch &&  ($r[0]['total'] > $batch_queue)) {
+			logger('add_to_queue: too many queued items for batch server ' . $cid . ' - discarding message');
+			return;
+		}
+		elseif((! $batch) && ($r[0]['total'] > $max_queue)) {
+			logger('add_to_queue: too many queued items for contact ' . $cid . ' - discarding message');
+			return;
+		}
+	}
+
+	q("INSERT INTO `queue` ( `cid`, `network`, `created`, `last`, `content`, `batch`)
+		VALUES ( %d, '%s', '%s', '%s', '%s', %d) ",
+		intval($cid),
+		dbesc(datetime_convert()),
+		dbesc(datetime_convert()),
+		dbesc($msg),
+		intval(($batch) ? 1: 0)
+	);
+}