[用Apache Cassandra存储聊天记录：从入门到精通]

llzwxh888

于 2024-09-15 20:51:09 发布

阅读量94

点赞数 1

文章标签： apache python

本文链接：https://blog.csdn.net/ppoojjj/article/details/142288713

版权

# 用Apache Cassandra存储聊天记录：从入门到精通

## 引言
Apache Cassandra是一种NoSQL数据库，以其高可扩展性和高可用性著称，非常适合存储大量数据。本文将重点介绍如何使用Cassandra存储聊天消息历史记录，提供实用指导和代码示例，让你快速上手。

## 主要内容

### 为什么选择Cassandra存储聊天记录
- **可扩展性**：Cassandra能够轻松扩展以处理大量写入操作，这对于聊天记录这种高频数据写入场景尤为重要。
- **高可用性**：通过多节点和复制机制，确保数据的持久性和可用性。
- **行导向存储**：适合存储结构化和半结构化数据。

### 环境准备
要运行本示例，你需要一个运行中的Cassandra集群或一个DataStax Astra DB实例（可以在[datastax.com](https://www.datastax.com/)免费获取）。访问[cassio.org](http://cassio.org)获取更多信息。

```bash
%pip install --upgrade --quiet "cassio>=0.1.0 langchain-community"

设置数据库连接参数

首先，确定要使用本地Cassandra还是云端的Astra DB，然后设置相应的连接参数和密钥。

import getpass

database_mode = (input("\n(C)assandra or (A)stra DB? ")).upper()
keyspace_name = input("\nKeyspace name? ")

if database_mode == "A":
    ASTRA_DB_APPLICATION_TOKEN = getpass.getpass('\nAstra DB Token ("AstraCS:...") ')
    ASTRA_DB_SECURE_BUNDLE_PATH = input("Full path to your Secure Connect Bundle? ")
elif database_mode == "C":
    CASSANDRA_CONTACT_POINTS = input("Contact points? (comma-separated, empty for localhost) ").strip()

创建数据库连接

根据选择的数据库模式，创建相应的数据库连接"Session"对象。

from cassandra.auth import PlainTextAuthProvider
from cassandra.cluster import Cluster

if database_mode == "C":
    if CASSANDRA_CONTACT_POINTS:
        cluster = Cluster([cp.strip() for cp in CASSANDRA_CONTACT_POINTS.split(",") if cp.strip()])
    else:
        cluster = Cluster()
    session = cluster.connect()
elif database_mode == "A":
    ASTRA_DB_CLIENT_ID = "token"
    cluster = Cluster(
        cloud={"secure_connect_bundle": ASTRA_DB_SECURE_BUNDLE_PATH},
        auth_provider=PlainTextAuthProvider(ASTRA_DB_CLIENT_ID, ASTRA_DB_APPLICATION_TOKEN),
    )
    session = cluster.connect()
else:
    raise NotImplementedError

代码示例

from langchain_community.chat_message_histories import CassandraChatMessageHistory

message_history = CassandraChatMessageHistory(
    session_id="test-session",
    session=session,
    keyspace=keyspace_name,
)

message_history.add_user_message("hi!")
message_history.add_ai_message("whats up?")