python - 使用 python 驱动程序在 Cassandra 中插入一个元组字段

Question

各位

我在通过 python-driver 将数据插入 Cassandra 表上的元组值字段时遇到问题由 DataStax 提供。

要么我不知道如何正确地将元组参数传递给 Session.execute命令或驱动程序在将元组分派(dispatch)给 Cassandra 时以错误的方式在内部转换元组 - 因为在 cqlsh 中执行时相同的插入工作正常 session 。

不涉及元组而是涉及列表的等价插入在代码中和 cqlsh 中都可以正常工作.

我正在使用 Python 2.7.10 和 cassandra-driver 3.7.1。在 Python 执行中出现的错误是 InvalidRequest: Error from server: code=2200 [Invalid query] message="Invalid list literal for tuplefield of type frozen<tuple<int, int, int>>" .

我在下面粘贴了一个最小的工作代码，它重现了我所看到的问题。谁能帮我弄清楚我是否做错了什么？

(注意:我已经尝试将普通列表传递给 Session.execute 来代替元组参数，但也没有成功。)

非常感谢。

'''
    Run with:
        python 2.7.10
        cassandra-driver==3.7.1 installed
'''

from cassandra.cluster import Cluster

if __name__=='__main__':

    serverAddress='SERVER_ADDRESS'
    keyspacename='tupletests'
    tablecreation='''create table tabletest (
                            rowid int,
                            tuplefield tuple < int, int, int >,
                            listfield list < int >,
                        primary key (rowid)
                    );'''

    # connect and create keyspace
    clu=Cluster([serverAddress])
    session=clu.connect()
    session.execute("create keyspace %s with replication = \
        {'class': 'SimpleStrategy', 'replication_factor': 1};" % keyspacename)

    # use keyspace; create a sample table
    session.set_keyspace(keyspacename)
    session.execute(tablecreation)

    # insert a row with rowid,listfield
    session.execute('insert into tabletest (rowid, listfield) values (%s,%s)', \
        [999, [10,11,12]])
    # succeeds

    # insert a row with rowid,tuplefield
    session.execute('insert into tabletest (rowid, tuplefield) values (%s,%s)', \
        [111, tuple([6,5,4])])
    # raises: *** InvalidRequest: Error from server: code=2200 [Invalid query]
    # message="Invalid list literal for tuplefield of type frozen<tuple<int, int, int>>"

    # Compare with analogous statements in cqlsh, which *both* succeed:
    # List:    insert into tabletest (rowid, listfield) values (999, [21,22,23]);
    # Tuple:   insert into tabletest (rowid, tuplefield) values (765, (121,122,123));

    # delete test keyspace
    session.execute("drop keyspace %s;" % keyspacename)    
    print "Test Finished."

Answer 1

好的，我刚刚解决了这个问题。在这里为大家发布解决方案。为了让驱动程序知道在将元组传递给 Cassandra 时应该如何格式化，如果坚持使用非准备语句进行插入，则必须在调用插入之前指定解析元组时使用的编码器:

session.encoder.mapping[tuple] = session.encoder.cql_encode_tuple

在这条语句之后，出现在 Session.execute insert 语句中的元组自动以正确的方式为 Cassandra 转换，Cassandra 不再提示。

(据我所知，最好使用 Prepared 语句，它包含足够的信息，使单独指定编码器变得多余)。