ratus

ratus

高效的RESTful异步任务队列服务器

Ratus是一个开源的异步任务队列服务器,提供RESTful API接口。它支持多种存储引擎,实现了至少一次执行保证、统一的调度模型、任务级超时控制和负载均衡等功能。Ratus易于部署和扩展,适用于需要可靠任务调度的分布式系统。其简洁强大的API可轻松集成到不同语言和框架中。

Ratus任务队列RESTful API异步处理分布式系统Github开源项目

Ratus

Go codecov Go Reference Swagger Validator Go Report Card Status

Ratus 是一个 RESTful 异步任务队列服务器。它将分布式任务队列的概念转化为符合 REST 原则的一组资源,并为各种后端提供了一致的 HTTP API

Ratus 的主要特性包括:

  • 具有快速内存存储的独立二进制文件。
  • 支持多种嵌入式或外部存储引擎。
  • 保证任务至少执行一次。
  • 统一的优先级和基于时间的调度模型。
  • 任务级别的超时控制和自动恢复。
  • 语言无关的 RESTful API,内置 Swagger UI。
  • 在动态数量的消费者之间进行负载均衡。
  • 通过复制和分区实现水平扩展。
  • 原生支持 Prometheus 和 Kubernetes。

终端截图

快速开始

安装

Ratus 提供多种安装选项:

  • Docker 镜像可在 Docker HubGitHub Packages 上获取。
  • Kubernetes 和 Docker Compose 示例可在 deployments 目录中找到。
  • 所有主要平台的预编译二进制文件可在 GitHub releases 页面获取。
  • 通过 go install github.com/hyperonym/ratus/cmd/ratus@latest 从源代码构建。

从命令行运行 Ratus 非常简单,只需输入:

$ ratus

上述命令将使用默认的内存存储引擎 memdb 启动一个临时 Ratus 实例,并在默认的 HTTP 端口 80 上监听。

要使用其他端口并启用磁盘快照以实现持久化,请使用以下命令启动 Ratus:

$ ratus --port 8000 --engine memdb --memdb-snapshot-path ratus.db

根据您选择的存储引擎,您可能还需要部署相应的数据库或代理。以 mongodb 引擎为例,假设数据库已在本地运行,则使用以下命令启动 Ratus:

$ ratus --port 8000 --engine mongodb --mongodb-uri mongodb://127.0.0.1:27017

基本用法

Ratus 引入的概念将在下文中以粗体显示,请参阅概念即速查表)了解更多信息。

cURL

生产者创建一个新的任务并将其推送到 example 主题

$ curl -X POST -d '{"payload": "hello world"}' "http://127.0.0.1:8000/v1/topics/example/tasks/1"
<details> <summary>示例响应</summary>
{ "created": 1, "updated": 0 }
</details>

然后,消费者可以发出一个承诺来认领并执行 example 主题中的下一个任务:

$ curl -X POST "http://127.0.0.1:8000/v1/topics/example/promises?timeout=30s"
<details> <summary>示例响应</summary>
{ "_id": "1", "topic": "example", "state": 1, "nonce": "e4SN6Si1nOnE53ou", "produced": "2022-07-29T20:00:00.0Z", "scheduled": "2022-07-29T20:00:00.0Z", "consumed": "2022-07-29T20:00:10.0Z", "deadline": "2022-07-29T20:00:40.0Z", "payload": "hello world" }
</details>

执行任务后,请记得通过提交来告知 Ratus 任务已 completed

$ curl -X PATCH "http://127.0.0.1:8000/v1/topics/example/tasks/1"
<details> <summary>示例响应</summary>
{ "_id": "1", "topic": "example", "state": 2, "nonce": "", "produced": "2022-07-29T20:00:00.0Z", "scheduled": "2022-07-29T20:00:00.0Z", "consumed": "2022-07-29T20:00:10.0Z", "deadline": "2022-07-29T20:00:40.0Z", "payload": "hello world" }
</details>

如果在承诺的截止时间之前未收到提交,任务的状态将被重置为 pending,从而允许消费者再次尝试执行该任务。

Go 客户端

Ratus 附带了一个 Go 客户端库,它不仅封装了所有 API 调用,还提供了符合习惯的轮询-执行-提交工作流,如 Client.PollClient.Subscribeexamples 目录包含了使用该库的可直接运行的示例:

  • hello world示例演示了客户端库的基本用法。
  • 爬虫前沿示例实现了分布式网络爬虫的简单URL前沿。它利用了并发订阅者和基于时间的任务调度等高级功能。

概念

数据模型

  • **任务**指代应异步执行的幂等工作单元。
  • **主题**指具有相同主题名称属性的有序任务子集。
  • **承诺**表示对活动任务所有权的声明。
  • **提交**包含要应用于任务的一组更新。

工作流程

  • 生产者客户端将任务及其期望执行日期(计划时间)推送到主题
  • 消费者客户端承诺执行从主题轮询的任务,并在完成时通过提交确认。

拓扑结构

  • 生产者消费者客户端都可以同时运行多个实例。
  • 可以动态添加消费者实例以提高吞吐量,任务将在消费者之间自然地进行负载均衡。
  • 消费者实例可以随时移除(或崩溃),而不会冒丢失正在执行的任务的风险:在承诺的期限后未收到提交任务将被其他消费者重新拾取并执行。

任务状态

  • 待处理 (0):任务已准备好执行或正在等待将来执行。
  • 活动 (1):任务正在被消费者处理。超时的活动任务将自动重置为"待处理"状态。消费者代码应处理失败并在必要时将状态设置为"待处理"以便稍后重试。
  • 已完成 (2):任务已完成执行。如果存储引擎实现支持TTL,已完成的任务将在保留期过期后自动删除。
  • 已归档 (3):任务作为存档存储。已归档的任务永远不会因过期而被删除。

行为

  • 所有主题的任务ID共享相同的命名空间ADR)。主题只是基于任务的topic属性生成的子集,因此不需要显式创建主题。
  • 当消费者能够跟上任务生成速度时,Ratus是一个任务调度器;当消费者无法跟上任务生成速度时,Ratus是一个优先队列。
  • 任务在计划时间到达之前不会执行。在计划时间之后,多余的任务将按计划时间的顺序执行。

引擎

Ratus为各种后端提供一致的API,允许用户根据需求选择特定引擎,而无需修改客户端代码。

要使用特定引擎,请将--engine标志或ENGINE环境变量设置为以下名称之一:

名称持久化复制分区过期
memdb○/●
mongodb

MemDB

MemDB

MemDB是Ratus的默认存储引擎。它基于go-memdb实现,后者建立在不可变的基数树之上。MemDB适用于开发和不需要持久性的生产环境

持久化

MemDB存储引擎默认是临时性的,但它也提供基于快照的持久化选项。通过将--memdb-snapshot-path标志或MEMDB_SNAPSHOT_PATH环境变量设置为非空文件路径,Ratus将按MEMDB_SNAPSHOT_INTERVAL指定的间隔写入磁盘快照。

MemDB不写入仅追加文件(AOF),这意味着如果Ratus由于任何原因在没有优雅关闭的情况下停止工作,你应该准备好丢失最近几分钟的数据。如果持久性对你的工作流程至关重要,请切换到外部存储引擎,如mongodb

实现细节

  • 列表操作相对昂贵,因为它们需要扫描整个数据库或索引,直到收集到所需数量的结果。幸运的是,这些操作在大多数情况下不会被使用。
  • 快照在适当时与定期后台作业一起执行。如果数据量较大,写入快照文件可能会延迟后台作业的执行
  • 由于MemDB中计划时间的分辨率为毫秒级,并受实例自身时钟的影响,消费者接收任务的顺序不能严格保证
  • 不能为completed任务禁用TTL,为了永久保存任务,请将其设置为archived状态。

MongoDB

MongoDB

Ratus最适合与MongoDB版本~4.4配合使用。MongoDB 5.0+也受支持,但需要额外考虑,请参阅实现细节了解更多信息。

💭 简而言之 使用MongoDB 5.0+时,请设置MONGODB_DISABLE_ATOMIC_POLL=true

复制

使用MongoDB存储引擎时,Ratus实例本身是无状态的。为了实现高可用性,启动多个Ratus实例并将它们连接到同一个MongoDB复制集

所有Ratus实例应运行在配置了健康检查的负载均衡器后面。生产者和消费者客户端应连接到负载均衡器,而不是直接连接到实例。

分区

通过对任务集合进行分片可以实现水平扩展。然而,在TTL机制的帮助下,大多数情况下不需要分区。只有在不分片的情况下才能获得最佳性能和最强的原子性。

如果数据量超过单个节点或复制集的容量,请从以下分片选项中选择:

  • 如果有大量主题,使用topic字段的哈希索引作为分片键,这也将在分片集群上实现最佳的轮询性能。
  • 如果在少数几个主题中有大量任务,使用_id字段的哈希索引作为分片键,这也将导致更均衡的数据分布。

实现细节

  • 在使用MongoDB存储引擎时,所有主题的任务都存储在同一个集合中
  • 任务是MongoDB存储引擎中唯一的具体数据模型,而主题和承诺仅仅是为了强制执行RESTful设计原则的概念实体。
  • 由于MongoDB中调度时间的分辨率为毫秒级,并且受实例自身时钟的影响,消费者接收任务的顺序并不能严格保证
  • 无法为completed状态的任务禁用TTL,若要永久保存任务,请将其设置为archived状态。
  • 不建议在使用topic字段作为分片键的分片集合上执行任务的upsert操作。由于MongoDB自身的限制,这种情况下无法使用原子操作,只能使用等同于先删除后插入的回退方案,因此无法保证原子性和性能。可以通过使用简单的插入操作并配合精细调整的TTL设置来规避这个问题。
  • 默认情况下,轮询是通过findAndModify实现的。发生冲突时,MongoDB的原生乐观并发控制 (OCC)会透明地重试操作。但在MongoDB 5.0及以上版本中,重试会在数据库服务器的日志中报告WriteConflict错误(尽管从客户端角度看操作仍然成功)。你可以选择忽略这个错误,或者**在使用MongoDB 5.0+时设置MONGODB_DISABLE_ATOMIC_POLL=true**来规避问题。这个选项会使Ratus不使用findAndModify进行轮询,而是依赖应用层的OCC来确保原子性。

索引模型

除非设置MONGODB_DISABLE_INDEX_CREATIONtrue,否则以下索引将在启动时创建:

键模式部分过滤表达式TTL
{"topic": "hashed"}--
{"topic": 1, "scheduled": 1}{"state": 0}-
{"deadline": 1}{"state": 1}-
{"topic": 1}{"state": 1}-
{"consumed": 1}{"state": 2}MONGODB_RETENTION_PERIOD

可观察性

指标和标签

Ratus在/metrics端点上暴露以下Prometheus指标:

名称类型标签
ratus_request_duration_seconds直方图topic, method, endpoint, status_code
ratus_chore_duration_seconds直方图-
ratus_task_schedule_delay_seconds仪表topic, producer, consumer
ratus_task_execution_duration_seconds仪表topic, producer, consumer
ratus_task_produced_count_total计数器topic, producer
ratus_task_consumed_count_total计数器topic, producer, consumer
ratus_task_committed_count_total计数器topic, producer, consumer

存活性和就绪性

Ratus通过HTTP GET请求支持存活性和就绪性探针

  • 如果实例正在运行,/livez端点返回状态码200
  • 如果实例准备好接受流量,/readyz端点返回状态码200

注意事项

  • 🚨 由于gin-gonic/gin#2633的问题,主题名称和任务ID不得包含加号('+')。
  • 不建议将Ratus作为任务的主要存储。相反,考虑将完整的任务记录存储在数据库中,并使用最小化的描述符作为Ratus的有效载荷。
  • Ratus是像Celery这样的任务队列的简单高效替代品。如果你需要高吞吐量的消息传递而不需要任务管理,可以考虑使用RabbitMQKafka

常见问题

更多详情,请参见架构决策记录

为什么选择HTTP API?

异步任务队列通常用于长时间运行的后台任务,因此与任务本身所花费的时间相比,HTTP的开销并不显著。另一方面,基于HTTP的RESTful API可以被所有语言轻松访问,无需使用专门的客户端库。

如何从多个主题轮询?

如果主题数量有限且你不关心它们之间的优先级,可以选择创建多个线程/协程来同时监听它们。或者,你可以创建一个主题的主题来依次获取主题名称,然后从相应的主题获取下一个任务。

路线图

  • 存储引擎选项
    • MemDB
      • 临时存储
      • 使用快照实现持久化
      • 使用AOF实现持久化
    • MongoDB
      • 单机版
      • 复制集
      • 分片集群
    • Redis
      • 单机版
      • 哨兵模式
      • 集群模式
    • 关系型数据库
      • MySQL
      • PostgreSQL
    • 消息代理
      • RabbitMQ
      • Amazon SQS
  • 多语言文档
    • 英语
    • 中文

查看未解决的问题以获取完整的提议功能列表。

贡献

这是一个开源项目。如果你有任何想法或问题,请随时通过创建issue来联系我们!

我们非常感谢您的贡献,请参阅CONTRIBUTING.md以获取更多信息。

许可证

Ratus 采用 Apache License 2.0 许可证。


© 2022-2024 Hyperonym

编辑推荐精选

Trae

Trae

字节跳动发布的AI编程神器IDE

Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。

热门AI工具生产力协作转型TraeAI IDE
问小白

问小白

全能AI智能助手,随时解答生活与工作的多样问题

问小白,由元石科技研发的AI智能助手,快速准确地解答各种生活和工作问题,包括但不限于搜索、规划和社交互动,帮助用户在日常生活中提高效率,轻松管理个人事务。

聊天机器人AI助手热门AI工具AI对话
Transly

Transly

实时语音翻译/同声传译工具

Transly是一个多场景的AI大语言模型驱动的同声传译、专业翻译助手,它拥有超精准的音频识别翻译能力,几乎零延迟的使用体验和支持多国语言可以让你带它走遍全球,无论你是留学生、商务人士、韩剧美剧爱好者,还是出国游玩、多国会议、跨国追星等等,都可以满足你所有需要同传的场景需求,线上线下通用,扫除语言障碍,让全世界的语言交流不再有国界。

讯飞智文

讯飞智文

一键生成PPT和Word,让学习生活更轻松

讯飞智文是一个利用 AI 技术的项目,能够帮助用户生成 PPT 以及各类文档。无论是商业领域的市场分析报告、年度目标制定,还是学生群体的职业生涯规划、实习避坑指南,亦或是活动策划、旅游攻略等内容,它都能提供支持,帮助用户精准表达,轻松呈现各种信息。

热门AI工具AI办公办公工具讯飞智文AI在线生成PPTAI撰写助手多语种文档生成AI自动配图
讯飞星火

讯飞星火

深度推理能力全新升级,全面对标OpenAI o1

科大讯飞的星火大模型,支持语言理解、知识问答和文本创作等多功能,适用于多种文件和业务场景,提升办公和日常生活的效率。讯飞星火是一个提供丰富智能服务的平台,涵盖科技资讯、图像创作、写作辅助、编程解答、科研文献解读等功能,能为不同需求的用户提供便捷高效的帮助,助力用户轻松获取信息、解决问题,满足多样化使用场景。

模型训练热门AI工具内容创作智能问答AI开发讯飞星火大模型多语种支持智慧生活
Spark-TTS

Spark-TTS

一种基于大语言模型的高效单流解耦语音令牌文本到语音合成模型

Spark-TTS 是一个基于 PyTorch 的开源文本到语音合成项目,由多个知名机构联合参与。该项目提供了高效的 LLM(大语言模型)驱动的语音合成方案,支持语音克隆和语音创建功能,可通过命令行界面(CLI)和 Web UI 两种方式使用。用户可以根据需求调整语音的性别、音高、速度等参数,生成高质量的语音。该项目适用于多种场景,如有声读物制作、智能语音助手开发等。

咔片PPT

咔片PPT

AI助力,做PPT更简单!

咔片是一款轻量化在线演示设计工具,借助 AI 技术,实现从内容生成到智能设计的一站式 PPT 制作服务。支持多种文档格式导入生成 PPT,提供海量模板、智能美化、素材替换等功能,适用于销售、教师、学生等各类人群,能高效制作出高品质 PPT,满足不同场景演示需求。

讯飞绘文

讯飞绘文

选题、配图、成文,一站式创作,让内容运营更高效

讯飞绘文,一个AI集成平台,支持写作、选题、配图、排版和发布。高效生成适用于各类媒体的定制内容,加速品牌传播,提升内容营销效果。

AI助手热门AI工具AI创作AI辅助写作讯飞绘文内容运营个性化文章多平台分发
材料星

材料星

专业的AI公文写作平台,公文写作神器

AI 材料星,专业的 AI 公文写作辅助平台,为体制内工作人员提供高效的公文写作解决方案。拥有海量公文文库、9 大核心 AI 功能,支持 30 + 文稿类型生成,助力快速完成领导讲话、工作总结、述职报告等材料,提升办公效率,是体制打工人的得力写作神器。

openai-agents-python

openai-agents-python

OpenAI Agents SDK,助力开发者便捷使用 OpenAI 相关功能。

openai-agents-python 是 OpenAI 推出的一款强大 Python SDK,它为开发者提供了与 OpenAI 模型交互的高效工具,支持工具调用、结果处理、追踪等功能,涵盖多种应用场景,如研究助手、财务研究等,能显著提升开发效率,让开发者更轻松地利用 OpenAI 的技术优势。

下拉加载更多