タスクキュー
データベース永続化によるバックグラウンド処理でタスクをキューイングするREST APIを構築します。
概要
このチュートリアルでは以下を実演するタスク管理APIを作成します:
- RESTエンドポイント - タスクのPOST、結果のGET
- キューパブリッシュ - 非同期ジョブディスパッチ
- キューコンシューマー - バックグラウンドワーカー
- データベース永続化 - SQLiteストレージ
- マイグレーション - 終了するワンショットプロセス
flowchart LR
subgraph api["HTTPサーバー"]
POST["/tasks POST"]
GET["/tasks GET"]
end
subgraph queue["キュー"]
Q[("tasksキュー")]
end
subgraph workers["ワーカー"]
W1["コンシューマー1"]
W2["コンシューマー2"]
end
subgraph storage["ストレージ"]
DB[(SQLite)]
end
POST -->|publish| Q
Q --> W1
Q --> W2
W1 -->|INSERT| DB
W2 -->|INSERT| DB
GET -->|SELECT| DB
プロジェクト構造
task-queue/
├── wippy.lock
└── src/
├── _index.yaml
├── migrate.lua
├── create_task.lua
├── list_tasks.lua
└── process_task.lua
エントリ定義
src/_index.yamlを作成:
version: "1.0"
namespace: app
entries:
# SQLiteデータベース
- name: db
kind: db.sql.sqlite
file: "./data/tasks.db"
lifecycle:
auto_start: true
# メモリキュードライバ
- name: queue_driver
kind: queue.driver.memory
lifecycle:
auto_start: true
# タスクキュー
- name: tasks_queue
kind: queue.queue
driver: app:queue_driver
# HTTPサーバー
- name: gateway
kind: http.service
addr: ":8080"
lifecycle:
auto_start: true
# ルーター
- name: router
kind: http.router
meta:
server: app:gateway
# マイグレーションプロセス(一度実行して終了)
- name: migrate
kind: process.lua
source: file://migrate.lua
method: main
modules:
- sql
- logger
# マイグレーションサービス(自動起動、成功時に終了)
- name: migrate-service
kind: process.service
process: app:migrate
host: app:processes
lifecycle:
auto_start: true
# プロセスホスト
- name: processes
kind: process.host
lifecycle:
auto_start: true
# APIハンドラ
- name: create_task
kind: function.lua
source: file://create_task.lua
method: handler
modules:
- http
- queue
- uuid
- name: list_tasks
kind: function.lua
source: file://list_tasks.lua
method: handler
modules:
- http
- sql
# キューワーカー
- name: process_task
kind: function.lua
source: file://process_task.lua
method: main
modules:
- sql
- logger
- json
# エンドポイント
- name: create_task.endpoint
kind: http.endpoint
meta:
router: app:router
method: POST
path: /tasks
func: app:create_task
- name: list_tasks.endpoint
kind: http.endpoint
meta:
router: app:router
method: GET
path: /tasks
func: app:list_tasks
# キューコンシューマー
- name: task_consumer
kind: queue.consumer
queue: app:tasks_queue
func: app:process_task
concurrency: 2
prefetch: 5
lifecycle:
auto_start: true
マイグレーションプロセス
src/migrate.luaを作成:
local sql = require("sql")
local logger = require("logger")
local function main()
local db, err = sql.get("app:db")
if err then
logger:error("failed to connect", {error = tostring(err)})
return 1
end
local _, exec_err = db:execute([[
CREATE TABLE IF NOT EXISTS tasks (
id TEXT PRIMARY KEY,
payload TEXT NOT NULL,
status TEXT NOT NULL DEFAULT 'pending',
result TEXT,
created_at INTEGER NOT NULL,
processed_at INTEGER
)
]])
db:release()
if exec_err then
logger:error("migration failed", {error = tostring(exec_err)})
return 1
end
logger:info("migration complete")
return 0
end
return { main = main }
タスク作成エンドポイント
src/create_task.luaを作成:
local http = require("http")
local queue = require("queue")
local uuid = require("uuid")
local function handler()
local req = http.request()
local res = http.response()
local body, parse_err = req:body_json()
if parse_err then
res:set_status(http.STATUS.BAD_REQUEST)
res:write_json({error = "invalid JSON"})
return
end
if not body.action then
res:set_status(http.STATUS.BAD_REQUEST)
res:write_json({error = "action required"})
return
end
local task_id = uuid.v4()
local task = {
id = task_id,
action = body.action,
data = body.data or {},
created_at = os.time()
}
local ok, err = queue.publish("app:tasks_queue", task)
if err then
res:set_status(http.STATUS.INTERNAL_ERROR)
res:write_json({error = "failed to queue task"})
return
end
res:set_status(http.STATUS.ACCEPTED)
res:write_json({
id = task_id,
status = "queued"
})
end
return { handler = handler }
タスク一覧エンドポイント
src/list_tasks.luaを作成:
local http = require("http")
local sql = require("sql")
local function handler()
local req = http.request()
local res = http.response()
local db, db_err = sql.get("app:db")
if db_err then
res:set_status(http.STATUS.INTERNAL_ERROR)
res:write_json({error = "database unavailable"})
return
end
local status_filter = req:query("status")
local query = sql.builder.select("id", "payload", "status", "result", "created_at", "processed_at")
:from("tasks")
:order_by("created_at DESC")
:limit(100)
if status_filter then
query = query:where({status = status_filter})
end
local rows, query_err = query:run_with(db):query()
db:release()
if query_err then
res:set_status(http.STATUS.INTERNAL_ERROR)
res:write_json({error = "query failed"})
return
end
res:set_status(http.STATUS.OK)
res:write_json({
tasks = rows,
count = #rows
})
end
return { handler = handler }
キューワーカー
src/process_task.luaを作成:
local sql = require("sql")
local logger = require("logger")
local json = require("json")
local function main(task)
logger:info("processing task", {
id = task.id,
action = task.action
})
local result
if task.action == "uppercase" then
result = {output = string.upper(task.data.text or "")}
elseif task.action == "sum" then
local nums = task.data.numbers or {}
local total = 0
for _, n in ipairs(nums) do
total = total + n
end
result = {output = total}
else
result = {output = "processed"}
end
local db, db_err = sql.get("app:db")
if db_err then
error("database unavailable: " .. tostring(db_err))
end
local _, exec_err = db:execute(
"INSERT OR REPLACE INTO tasks (id, payload, status, result, created_at, processed_at) VALUES (?, ?, ?, ?, ?, ?)",
{ task.id, json.encode(task), "completed", json.encode(result), task.created_at, os.time() }
)
db:release()
if exec_err then
error("failed to store result: " .. tostring(exec_err))
end
logger:info("task completed", {id = task.id})
end
return { main = main }
queue.message()経由でmsg:ack()またはmsg:nack()を呼び出してください。
サービスの実行
初期化と実行:
mkdir -p data
wippy init
wippy run
APIをテスト:
# タスクを作成
curl -X POST http://localhost:8080/tasks \
-H "Content-Type: application/json" \
-d '{"action": "uppercase", "data": {"text": "hello world"}}'
# レスポンス: {"id": "550e8400-...", "status": "queued"}
# 処理を待ってからタスクを一覧表示
curl http://localhost:8080/tasks
# レスポンス: {"tasks": [...], "count": 1}
# ステータスでフィルタ
curl "http://localhost:8080/tasks?status=completed"
メッセージフロー
- POST /tasksがリクエストを受信し、UUIDを生成し、キューにパブリッシュ
- キューコンシューマーがメッセージを取得(2つの並行ワーカー)
- ワーカーがタスクを処理し、結果をSQLiteに書き込み
- GET /tasksがデータベースから完了したタスクを読み取り
次のステップ
- HTTPモジュール - リクエスト/レスポンス処理
- Queueモジュール - メッセージキュー操作
- SQLモジュール - データベースアクセス
- キューコンシューマー - キュー設定