Package com.dtsx.astra.sdk.cassio

Class ClusteredCassandraTable

java.lang.Object

com.dtsx.astra.sdk.cassio.AbstractCassandraTable<ClusteredCassandraTable.Record>

com.dtsx.astra.sdk.cassio.ClusteredCassandraTable

public class ClusteredCassandraTable
extends AbstractCassandraTable<ClusteredCassandraTable.Record>

Table representing persistence for LangChain operations

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	ClusteredCassandraTable.Record	Represents a row of the Table

Field Summary

Fields inherited from class com.dtsx.astra.sdk.cassio.AbstractCassandraTable

ATTRIBUTES_BLOB, BODY_BLOB, COLUMN_SIMILARITY, cqlSession, DEFAULT_RECORD_COUNT, keyspaceName, METADATA_S, PARTITION_ID, ROW_ID, SAI_INDEX_CLASSNAME, tableName, VECTOR

Constructor Summary

Constructors

Constructor	Description
ClusteredCassandraTable(@NonNull com.datastax.oss.driver.api.core.CqlSession session, @NonNull String keyspaceName, @NonNull String tableName)	Constructor with the mandatory parameters.

Method Summary

Modifier and Type	Method	Description
void	createSchema()	Create table if not exist.
void	delete(@NonNull String partitionId, @NonNull UUID rowId)	Delete one row.
void	deletePartition(@NonNull String partitionId)	Delete Partition.
Optional<ClusteredCassandraTable.Record>	findById(String partition, UUID rowId)	Find a row by its id.
List<ClusteredCassandraTable.Record>	findPartition(@NonNull String partitionDd)	Find a partition.
void	insert(@NonNull String partitionId, @NonNull UUID rowId, @NonNull String bodyBlob)	Insert Row.
ClusteredCassandraTable.Record	mapRow(@NonNull com.datastax.oss.driver.api.core.cql.Row row)	Should be table to map from a Cassandra row to a record.
void	put(ClusteredCassandraTable.Record row)	Upsert a row of the table.
void	upsertPartition(List<ClusteredCassandraTable.Record> rows)	Update the history in one go.

Methods inherited from class com.dtsx.astra.sdk.cassio.AbstractCassandraTable

clear, delete, putAsync

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

ClusteredCassandraTable

public ClusteredCassandraTable(@NonNull
 @NonNull com.datastax.oss.driver.api.core.CqlSession session,
 @NonNull
 @NonNull String keyspaceName,
 @NonNull
 @NonNull String tableName)

Constructor with the mandatory parameters.

Parameters:

session - cassandra Session

keyspaceName - keyspace name

tableName - table name

Method Details

createSchema

public void createSchema()

Description copied from class: AbstractCassandraTable

Create table if not exist.

Specified by:

createSchema in class AbstractCassandraTable<ClusteredCassandraTable.Record>

put

public void put(@NonNull
 ClusteredCassandraTable.Record row)

Upsert a row of the table.

Specified by:

put in class AbstractCassandraTable<ClusteredCassandraTable.Record>

Parameters:

row - current row

mapRow

public ClusteredCassandraTable.Record mapRow(@NonNull
 @NonNull com.datastax.oss.driver.api.core.cql.Row row)

Should be table to map from a Cassandra row to a record.

Specified by:

mapRow in class AbstractCassandraTable<ClusteredCassandraTable.Record>

Parameters:

row - current cassandra row

Returns:

current record

findPartition

public List<ClusteredCassandraTable.Record> findPartition(@NonNull
 @NonNull String partitionDd)

Find a partition.

Parameters:

partitionDd - partition id

Returns:

list of rows

upsertPartition

public void upsertPartition(List<ClusteredCassandraTable.Record> rows)

Update the history in one go.

Parameters:

rows - current rows.

findById

public Optional<ClusteredCassandraTable.Record> findById(String partition,
 UUID rowId)

Find a row by its id.

Parameters:

partition - partition id

rowId - row id

Returns:

record if exists

deletePartition

public void deletePartition(@NonNull
 @NonNull String partitionId)

Delete Partition.

Parameters:

partitionId - delete the whole partition

delete

public void delete(@NonNull
 @NonNull String partitionId,
 @NonNull
 @NonNull UUID rowId)

Delete one row.

Parameters:

partitionId - current session

rowId - message id

insert

public void insert(@NonNull
 @NonNull String partitionId,
 @NonNull
 @NonNull UUID rowId,
 @NonNull
 @NonNull String bodyBlob)

Insert Row.

Parameters:

partitionId - partition id

rowId - rowId

bodyBlob - body