From 09fa4778c002ff5603946d3c0eef8cd247582d41 Mon Sep 17 00:00:00 2001
From: Kijin Sung <kijin@kijinsung.com>
Date: Wed, 9 Oct 2024 23:12:55 +0900
Subject: [PATCH] Add DB queue driver

---
 common/framework/drivers/queue/base.php |   0
 common/framework/drivers/queue/db.php   | 123 ++++++++++++++++++++++++
 modules/module/schemas/task_queue.xml   |   8 ++
 3 files changed, 131 insertions(+)
 delete mode 100644 common/framework/drivers/queue/base.php
 create mode 100644 common/framework/drivers/queue/db.php
 create mode 100644 modules/module/schemas/task_queue.xml

diff --git a/common/framework/drivers/queue/base.php b/common/framework/drivers/queue/base.php
deleted file mode 100644
index e69de29bb..000000000
diff --git a/common/framework/drivers/queue/db.php b/common/framework/drivers/queue/db.php
new file mode 100644
index 000000000..90225f02d
--- /dev/null
+++ b/common/framework/drivers/queue/db.php
@@ -0,0 +1,123 @@
+<?php
+
+namespace Rhymix\Framework\Drivers\Queue;
+
+use Rhymix\Framework\DB as RFDB;
+use Rhymix\Framework\Drivers\QueueInterface;
+
+/**
+ * The DB queue driver.
+ */
+class DB implements QueueInterface
+{
+	/**
+	 * The Redis connection is stored here.
+	 */
+	protected $_conn;
+	protected $_key;
+
+	/**
+	 * Create a new instance of the current Queue driver, using the given settings.
+	 *
+	 * @param array $config
+	 * @return void
+	 */
+	public static function getInstance(array $config): self
+	{
+		return new self($config);
+	}
+
+	/**
+	 * Get the human-readable name of this Queue driver.
+	 *
+	 * @return string
+	 */
+	public static function getName(): string
+	{
+		return 'DB';
+	}
+
+	/**
+	 * Get the list of configuration fields required by this Queue driver.
+	 *
+	 * @return array
+	 */
+	public static function getRequiredConfig(): array
+	{
+		return [];
+	}
+
+	/**
+	 * Get the list of configuration fields optionally used by this Queue driver.
+	 *
+	 * @return array
+	 */
+	public static function getOptionalConfig(): array
+	{
+		return [];
+	}
+
+	/**
+	 * Check if this driver is supported on this server.
+	 *
+	 * @return bool
+	 */
+	public static function isSupported(): bool
+	{
+		return true;
+	}
+
+	/**
+	 * Constructor.
+	 *
+	 * @param array $config
+	 */
+	public function __construct(array $config)
+	{
+
+	}
+
+	/**
+	 * Add a task.
+	 *
+	 * @param string $handler
+	 * @param ?object $args
+	 * @param ?object $options
+	 * @return int
+	 */
+	public function addTask(string $handler, ?object $args = null, ?object $options = null): int
+	{
+		$oDB = RFDB::getInstance();
+		$stmt = $oDB->prepare('INSERT INTO task_queue (handler, args, options) VALUES (?, ?, ?)');
+		$result = $stmt->execute([$handler, serialize($args), serialize($options)]);
+		return $result ? $oDB->getInsertID() : false;
+	}
+
+	/**
+	 * Get the first task.
+	 *
+	 * @param int $blocking
+	 * @return ?object
+	 */
+	public function getTask(int $blocking = 0): ?object
+	{
+		$oDB = RFDB::getInstance();
+		$stmt = $oDB->query('SELECT * FROM task_queue ORDER BY id LIMIT 1');
+		$result = $stmt->fetchObject();
+		$stmt->closeCursor();
+
+		if ($result)
+		{
+			$stmt = $oDB->prepare('DELETE FROM task_queue WHERE id = ?');
+			$stmt->execute([$result->id]);
+
+			$result->args = unserialize($result->args);
+			$result->options = unserialize($result->options);
+			return $result;
+		}
+		else
+		{
+			return null;
+		}
+	}
+}
diff --git a/modules/module/schemas/task_queue.xml b/modules/module/schemas/task_queue.xml
new file mode 100644
index 000000000..6e991a700
--- /dev/null
+++ b/modules/module/schemas/task_queue.xml
@@ -0,0 +1,8 @@
+<table name="task_queue">
+	<column name="id" type="bigint" notnull="notnull" primary_key="primary_key" auto_increment="auto_increment" />
+	<column name="handler" type="varchar" size="191" notnull="notnull" />
+	<column name="args" type="longtext" notnull="notnull" />
+	<column name="options" type="longtext" notnull="notnull" />
+	<column name="lock" type="int" notnull="notnull" default="0" index="idx_lock" />
+	<column name="regdate" type="datetime" notnull="notnull" default="current_timestamp()" index="idx_regdate" />
+</table>