ustore

ustore

模块化多模态事务数据库,AI和语义搜索的高性能解决方案

UStore是一款面向AI和语义搜索的高性能数据库系统。它支持多种存储引擎,可处理二进制对象、文档、图形和向量数据。UStore提供多语言驱动,集成Pandas和NetworkX API,并支持向量搜索。其模块化、多模态和事务性设计使其成为构建灵活数据存储解决方案的有力工具。

UStore数据库多模态事务性人工智能Github开源项目
<h1 align="center">UStore</h1> <h3 align="center"> 模块化<sup>1</sup> 多模态<sup>2</sup> 事务性<sup>3</sup> 数据库<br/> 适用于人工智能<sup>4</sup>和语义搜索<sup>5</sup><br/> </h3> <br/> <p align="center"> <a href="https://www.youtube.com/watch?v=ybWeUf_hC7o"><img height="25" width="25" src="https://yellow-cdn.veclightyear.com/835a84d5/c7d7e233-9837-47d6-af6e-0b9040f7395c.svg" alt="Youtube"></a> &nbsp;&nbsp;&nbsp; <a href="https://discord.gg/4mxGrenbNt"><img height="25" width="25" src="https://yellow-cdn.veclightyear.com/835a84d5/4a5dd702-8aa9-47dc-8d07-ce5317f4ff8d.svg" alt="Discord"></a> &nbsp;&nbsp;&nbsp; <a href="https://www.linkedin.com/company/unum-cloud/"><img height="25" width="25" src="https://yellow-cdn.veclightyear.com/835a84d5/ecfcca7c-fc41-49ea-bb85-9ec1f0ae560a.svg" alt="LinkedIn"></a> &nbsp;&nbsp;&nbsp; <a href="https://twitter.com/unum_cloud"><img height="25" width="25" src="https://yellow-cdn.veclightyear.com/835a84d5/054f916c-ed5c-4beb-aed6-3824498ea1d8.svg" alt="Twitter"></a> &nbsp;&nbsp;&nbsp; <a href="https://unum.cloud/post"><img height="25" width="25" src="https://yellow-cdn.veclightyear.com/835a84d5/5056887a-cb71-4faf-8377-2fee4b6895e1.svg" alt="博客"></a> &nbsp;&nbsp;&nbsp; <a href="https://github.com/unum-cloud/ukv"><img height="25" width="25" src="https://yellow-cdn.veclightyear.com/835a84d5/cbd1a2d6-e687-4b18-a205-234b39d34679.svg" alt="GitHub"></a> </p> <div align="center"> 1. 支持: <a href="https://github.com/facebook/rocksdb">RocksDB</a> • <a href="https://github.com/google/leveldb">LevelDB</a> • <a href="https://github.com/unum-cloud/udisk">UDisk</a> • <a href="https://github.com/unum-cloud/ucset">UCSet</a> <a href="backends">后端</a> <br/> 2. 可存储: <a href="#Blobs">二进制大对象</a> • <a href="#Documents">文档</a> • <a href="#Graphs">图</a> • 🔜 特征 • 🔜 文本 <br/> 3: 保证 <a href="#Atomicity">原子性</a> • <a href="#Consistency">一致性</a> • <a href="#Isolation">隔离性</a> • <a href="#Durability">持久性</a> <br/> 4: 附带 Pandas 和 NetworkX API 以及 🔜 PyTorch 数据加载器 <br/> 5: 提供 向量搜索 集成 <a href="https://github.com/unum-cloud/usearch">USearch</a> 和 <a href="https://github.com/unum-cloud/uform">UForm</a> </div>
<div align="center"> <b>驱动</b>: Python • C • C++ • GoLang • Java <br/> <b>包</b>: <a href="https://pypi.org/project/ukv/">PyPI</a> • <a href="#cmake">CMake</a> • <a href="https://hub.docker.com/repository/docker/unum/ustore">Docker Hub</a>

<a href="https://www.youtube.com/watch?v=ybWeUf_hC7o">Youtube</a> 介绍 • <a href="https://discord.gg/4mxGrenbNt">Discord</a> 聊天 • 完整 <a href="https://unum-cloud.github.io/ustore">文档</a>

<a href="https://discord.gg/4mxGrenbNt"><img src="https://img.shields.io/discord/1063947616615923875?label=discord"></a>    <a href="https://www.linkedin.com/company/unum-cloud/"><img src="https://yellow-cdn.veclightyear.com/835a84d5/96e30c27-b31f-4b41-9a4c-1595e56be943.svg?"/></a>    <a href="https://twitter.com/unum_cloud"><img src="https://yellow-cdn.veclightyear.com/835a84d5/cadf9329-5f13-4bbe-b77d-a70ad4effe88.svg?"/></a>    <a href="https://zenodo.org/badge/latestdoi/502647695"><img src="https://yellow-cdn.veclightyear.com/835a84d5/bcc86041-7307-4834-903b-0a09cfb8eec8.svg" alt="DOI"></a>    <a href="https://www.github.com/unum-cloud/"><img src="https://img.shields.io/github/issues-closed-raw/unum-cloud/ustore?"/></a>    <a href="https://www.github.com/unum-cloud/"><img src="https://img.shields.io/github/stars/unum-cloud/ustore?"/></a>    <a href="#"><img src="https://img.shields.io/github/workflow/status/unum-cloud/ustore/Build"/></a>

</div>

快速开始

安装UStore非常简单,使用起来就像Python的dict一样简单。

$ pip install ukv $ python from ukv import umem db = umem.DataBase() db.main[42] = 'Hi'

我们刚刚创建了一个内存嵌入式事务性数据库,并在其main集合中添加了一个条目。 你想要将数据存储在磁盘上吗? 只需更改一行代码。

from ukv import rocksdb db = rocksdb.DataBase('/some-folder/')

你想要连接到远程UStore服务器吗? UStore带有Apache Arrow Flight RPC接口!

from ukv import flight_client db = flight_client.DataBase('grpc://0.0.0.0:38709')

你是否在存储类似[NetworkX][networkx]的MultiDiGraph? 或者类似[Pandas][pandas]的DataFrame

db = rocksdb.DataBase() users_table = db['users'].table users_table.merge(pd.DataFrame([ {'id': 1, 'name': 'Lex', 'lastname': 'Fridman'}, {'id': 2, 'name': 'Joe', 'lastname': 'Rogan'}, ])) friends_graph = db['friends'].graph friends_graph.add_edge(1, 2) assert friends_graph.has_edge(1, 2) and \ friends_graph.has_node(1) and \ friends_graph.number_of_edges(1, 2) == 1

函数调用看起来可能相同,但底层实现可能涉及处理远程机器持久内存中的数百TB数据。


是否有其他人同时更新这些集合? 将操作捆绑在一起以保证一致性!

db = rocksdb.DataBase() with db.transact() as txn: txn['users'].table.merge(...) txn['friends'].graph.add_edge(1, 2)

到目前为止,我们仅涉及了UStore的冰山一角。 您可以使用它来...

  1. 获取RocksDB或LevelDB的C99、Python、GoLang或Java包装器。
  2. 通过Apache Arrow Flight RPC将它们提供给Spark、Kafka或PyTorch。
  3. 在嵌入式数据库中存储文档和图形,避免网络开销。
  4. 在一个API下将DBMS分层为内存和持久化后端。

但UStore能做的更多。 以下是功能地图:


基本用法

UStore不仅是一个数据库,还是一个"构建您自己的数据库"工具包和潜在事务性NoSQL数据库的开放标准,为"创建、读取、更新、删除"操作(简称CRUD)定义零拷贝二进制接口。

几个简单的C99头文件可以将几乎任何底层存储引擎链接到众多高级语言驱动程序,将其对二进制字符串值的支持扩展到图、灵活模式文档和其他模态,旨在用单一的ACID事务系统取代MongoDB、Neo4J、Pinecone和ElasticSearch。

UStore: 小地图

例如,Redis提供了RediSearch、RedisJSON和RedisGraph,具有类似的目标。UStore做得更好,允许您添加您喜欢的键值存储(KVS),无论是嵌入式、独立还是分片的,如FoundationDB,从而倍增其功能。

模态

二进制大对象

二进制大对象可以放置在UStore中。性能会根据所使用的底层技术而有很大差异。内存中的UCSet将是最快的,但最不适合较大的对象。当正确配置时,持久化的UDisk可以完全绕过Linux内核,包括文件系统层,直接访问块设备。

UDisk和RocksDB的二进制处理性能图表

在高端服务器上,基于用户空间驱动程序(如SPDK)构建的现代持久化IO每个套接字可以超过100 GB/s。这接近高端RAM的实际吞吐量,并解锁了数据库中不常见的新用例。现在可以将一个GB级的视频文件放入ACID事务数据库中,就在其元数据旁边,而不是使用单独的对象存储,如MinIO。

文档

JSON是当今最常用的文档格式。UStore文档集合支持JSON,以及MessagePack和MongoDB使用的BSON。 UStore和MongoDB的文档处理性能图表

UStore目前还不支持水平扩展,但提供了更高的单节点性能,并且由于使用了开源的simdjsonyyjson库,在多核系统上几乎呈线性的垂直扩展性。 此外,与数据交互时,您不需要像MQL这样的自定义查询语言。 相反,我们优先考虑开放的RFC标准,以真正避免供应商锁定:

现代图数据库,如Neo4J,在处理大型工作负载时面临挑战。 它们需要过多的RAM,并且其算法一次只能观察一个数据条目。 我们在两个方面进行了优化:

  • 使用增量编码压缩倒排索引。
  • 更新经典图算法以适应高延迟存储,以批处理或边缘中心的方式处理图。

向量

特征存储和向量数据库,如Pinecone、Milvus和USearch,为向量搜索提供独立的索引。 UStore将其实现为一种独立的模态,与文档和图形并列。 特性:

  • 8位整数量化。
  • 16位浮点数量化。
  • 余弦、内积和欧几里得度量。

驱动程序

UStore的Python和C++版本看起来很不一样。 我们的Python SDK模仿了其他Python库 - [Pandas][pandas]和[NetworkX][networkx]。 同样,C++库提供了C++开发者所期望的接口。

UStore:前端

众所周知,人们使用不同的语言来满足不同的需求。 某些C级功能并未在某些语言中实现。 这或是因为没有需求,或是我们还没有来得及实现。

名称事务集合批处理文档副本
[C99标准][ustore-c]0
[C++ SDK][ustore-cpp]0
[Python SDK][ustore-python]0-1
[GoLang SDK][ustore-golang]1
[Java SDK][ustore-java]1
Arrow Flight API0-2

这里的一些前端有其自己的完整生态系统! 例如,[Apache Arrow Flight][flight] API拥有自己的C、C++、C#、Go、Java、JavaScript、Julia、MATLAB、Python、R、Ruby和Rust驱动程序。

UStore:前端

常见问题解答

常见问题

  • 事务默认是ACI(D)的。这意味着什么?
  • 为什么不使用LevelDB或RocksDB接口?已回答
  • 为什么不使用SQL、MQL或CYPHER?已回答
  • UStore是否支持生存时间(TTL)?已回答
  • UStore是否支持压缩?已回答
  • UStore是否支持队列?已回答
  • 如何为语言X添加驱动程序?[已回答][ustore-new-drivers]
  • 如何将数据库X添加为引擎?[已回答][ustore-new-engine]

高级用法

引擎

以下引擎几乎可以互换使用。 从历史上看,LevelDB是第一个。 然后RocksDB在功能和性能上进行了改进。 现在它成为了一半DBMS创业公司的基础。

LevelDBRocksDBUDiskUCSet
速度1x2x10x30x
持久化
事务支持
块设备支持
加密
[监视][watch]
快照
随机采样
批量枚举
命名集合
开源
兼容性任何任何Linux任何
维护者GoogleFacebookUnumUnum

UCSet和UDisk都由Unum设计和维护。 两者都功能完备,但我们的替代方案提供的最关键特性是性能。 在内存中实现高速很容易。 UCSet的核心逻辑可以在模板化的仅头文件<code class="docutils literal notranslate"><a href="https://github.com/unum-cloud/ucset" class="pre">ucset</a></code>库中找到。

设计UDisk是一项更具挑战性的7年长期努力。 它包括发明新的树状结构、实现使用io_uring的部分内核旁路、使用SPDK的完全旁路、CUDA GPU加速,甚至自定义内部文件系统。 UDisk是第一个从头开始设计时就考虑并行架构和内核旁路的引擎

事务

原子性

始终保证原子性。 即使在非事务性写入中 - 要么所有更新都通过,要么全部失败。

一致性

一致性以最严格的形式实现 - "严格可串行化",意味着:

然而,默认行为可以在特定操作级别进行调整。 为此,可以将::ustore_option_transaction_dont_watch_k传递给ustore_transaction_init()或任何事务性读/写操作,以控制暂存期间的一致性检查。

读取写入
头部严格可串行化严格可串行化
基于快照的事务可串行化严格可串行化
不带快照的事务严格可串行化严格可串行化
不带监视的事务严格可串行化顺序性

如果你对这个主题不熟悉,请查看Jepsen.io关于一致性的博客。

隔离性

读取写入
基于快照的事务
不带快照的事务

持久性

持久性在定义上不适用于内存系统。 在混合或持久系统中,我们倾向于默认禁用它。 几乎每个构建在KVS之上的DBMS都倾向于实现自己的持久性机制。 在分布式数据库中更是如此,可能存在三个独立的预写日志:

  • 在KVS中,
  • 在DBMS中,
  • 在分布式共识实现中。

如果你仍然需要持久性,可以在提交时使用可选标志刷新写入。 在[C驱动][ustore-c]中,你可以使用::ustore_option_write_flush_k标志调用ustore_transaction_commit()

容器和云部署

整个DBMS适合放入一个不到100 MB的Docker镜像中。 运行以下脚本来拉取和运行容器,在端口38709上暴露[Apache Arrow Flight][flight]服务器。 客户端SDK默认也将通过该端口通信。

docker run -d --rm --name ustore-test -p 38709:38709 unum/ustore

可以通过以下命令获取默认配置文件:

cat /var/lib/ustore/config.json

最简单的连接和测试方法是以下命令:

python ...

UStore的预打包镜像在多个平台上可用:

  • Docker Hub镜像:v0.7
  • RedHat OpenShift操作符:v0.7
  • Amazon AWS Marketplace镜像:
    • 免费社区版:v0.4
    • 内存版:🔜
    • 性能版:🔜

不要hesitate商业化和重新分发UStore。

配置

调优数据库既是艺术也是科学。 像RocksDB这样的项目提供了数十个旋钮来优化行为。 我们允许将专门的配置文件转发给底层引擎。

{ "version": "1.0", "directory": "./tmp/" }

我们还有一个更简单的过程,对80%的用户来说已经足够。 这可以扩展到利用多个设备或目录,或转发专门的引擎配置。

{ "version": "1.0", "directory": "/var/lib/ustore", "data_directories": [ { "path": "/dev/nvme0p0/", "max_size": "100GB" }, { "path": "/dev/nvme1p0/", "max_size": "100GB" } ], "engine": { "config_file_path": "./engine_rocksdb.ini", } }

数据库集合也可以用JSON文件配置。

键大小

在当前版本中,使用64位有符号整数。 它允许在[0, 2^63)范围内的唯一键。 支持128位构建和UUID的版本即将推出,但强烈不建议使用可变长度键。 为什么呢? 使用可变长度的键会给键值存储的设计带来诸多限制。 首先,这意味着需要进行缓慢的逐字符比较——这在现代超标量CPU上会严重影响性能。 其次,它迫使键和值必须在磁盘上相邻存储,以最小化导航所需的元数据。 最后,它违背了我们对KVS作为"持久内存分配器"的简单逻辑视图,给它增加了更多责任。


处理字符串键的推荐方法是:

  1. 选择一种机制来生成唯一的整数键(UID)。例如:单调递增的值。
  2. 使用"路径"模式构建一个字符串到UID的持久哈希映射。
  3. 使用这些UID在二进制、文档和图形模式中处理其余数据。

这将导致从字符串到整数表示的单一转换点,并使系统大部分保持快速响应,同时使C级接口比原本可能的更简单。

值大小

目前我们只能处理4 GB或更小的值。 为什么? 键值存储通常用于高频操作。 在现代硬件上,频繁(每秒数千次)访问和修改4 GB及更大的文件是不可能的。 因此我们坚持使用较小的长度类型,这使得使用Apache Arrow表示稍微容易一些,并允许KVS更好地压缩索引。

路线图

我们的[开发路线图][ustore-roadmap]是公开的,托管在GitHub仓库中。 即将进行的任务包括:

  • 为Arm、MacOS构建。
  • 持久性快照。
  • 连续复制。
  • 文档模式验证。
  • 更丰富的GoLang、Java、JavaScript驱动程序。
  • 改进的向量搜索。
  • 集合级配置。
  • 拥有所有权和非拥有所有权的C++封装器。
  • 水平扩展。

[在我们的文档中阅读完整路线图][ustore-architecture]。

编辑推荐精选

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 的技术优势。

下拉加载更多