欢迎访问宙启技术站
智能推送

使用Python的connect()函数连接到Cassandra数据库

发布时间:2023-12-28 01:11:29

要使用Python连接到Cassandra数据库,我们需要安装cassandra-driver库。可以使用pip命令来安装它:

pip install cassandra-driver

接下来,我们可以使用以下代码示例来连接到Cassandra数据库:

from cassandra.cluster import Cluster

# 创建一个集群对象并连接到Cassandra节点
cluster = Cluster(['127.0.0.1'])
session = cluster.connect()

# 创建一个新的keyspace
session.execute("""
    CREATE KEYSPACE IF NOT EXISTS mykeyspace 
    WITH replication = { 'class': 'SimpleStrategy', 'replication_factor': '3' }
""")

# 使用新的keyspace
session.set_keyspace('mykeyspace')

# 创建一个新的表
session.execute("""
    CREATE TABLE IF NOT EXISTS users (
        id INT PRIMARY KEY,
        name TEXT,
        age INT
    )
""")

# 插入一些数据
session.execute("""
    INSERT INTO users (id, name, age)
    VALUES (1, 'John', 25)
""")

session.execute("""
    INSERT INTO users (id, name, age)
    VALUES (2, 'Jane', 32)
""")

# 查询数据
result = session.execute("SELECT * FROM users")
for row in result:
    print(row.id, row.name, row.age)

# 关闭连接
session.shutdown()
cluster.shutdown()

在上述代码中,我们首先通过提供Cassandra节点的IP地址,创建了一个Cluster对象并连接到Cassandra数据库。然后我们创建了一个名为mykeyspace的新的keyspace,并在其中创建了一个名为users的表来存储用户数据。我们使用session.execute()方法执行CQL(Cassandra Query Language)语句。

接下来,我们插入了一些用户数据,并从表中检索数据。最后,我们关闭了会话和集群连接。

请注意,您需要根据您自己的环境进行适当的修改,例如Cassandra节点的IP地址和使用的keyspace。