lsst.dax.apdb g5ea96fc03c+341ea1ce94
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions | Protected Attributes | Static Protected Attributes | List of all members
lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema Class Reference
Inheritance diagram for lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema:
lsst.dax.apdb.apdbSchema.ApdbSchema

Public Member Functions

 __init__ (self, cassandra.cluster.Session session, str keyspace, str schema_file, str schema_name="ApdbSchema", str prefix="", bool time_partition_tables=False, bool enable_replica=False)
 
bool has_replica_chunks (self)
 
bool empty (self)
 
str tableName (self, ApdbTables|ExtraTables table_name)
 
str keyspace (self)
 
Mapping[str, schema_model.ColumngetColumnMap (self, ApdbTables|ExtraTables table_name)
 
list[str] apdbColumnNames (self, ApdbTables|ExtraTables table_name)
 
list[str] partitionColumns (self, ApdbTables|ExtraTables table_name)
 
list[str] clusteringColumns (self, ApdbTables|ExtraTables table_name)
 
None makeSchema (self, *bool drop=False, tuple[int, int]|None part_range=None, int|None replication_factor=None)
 
- Public Member Functions inherited from lsst.dax.apdb.apdbSchema.ApdbSchema
type|str column_dtype (self, felis.datamodel.DataType|ExtraDataTypes felis_type)
 
VersionTuple schemaVersion (self)
 

Protected Member Functions

Mapping[ApdbTables, schema_model.Table_apdb_tables_schema (self, bool time_partition_tables)
 
Mapping[ExtraTables, schema_model.Table_extra_tables_schema (self)
 
bool _check_replica_chunks (self)
 
None _makeTableSchema (self, ApdbTables|ExtraTables table, bool drop=False, tuple[int, int]|None part_range=None)
 
list[str] _tableColumns (self, ApdbTables|ExtraTables table_name)
 
schema_model.Table _table_schema (self, ApdbTables|ExtraTables table)
 
- Protected Member Functions inherited from lsst.dax.apdb.apdbSchema.ApdbSchema
tuple[Mapping[ApdbTables, Table], VersionTuple|None] _buildSchemas (cls, str schema_file, str schema_name="ApdbSchema")
 

Protected Attributes

 _session
 
 _keyspace
 
 _prefix
 
 _time_partition_tables
 
 _enable_replica
 
 _apdb_tables
 
 _extra_tables
 
 _has_replica_chunks
 
- Protected Attributes inherited from lsst.dax.apdb.apdbSchema.ApdbSchema
 _schemaVersion
 

Static Protected Attributes

dict _type_map
 
list _time_partitioned_tables
 
list _spatially_partitioned_tables = [ApdbTables.DiaObjectLast]
 

Additional Inherited Members

- Public Attributes inherited from lsst.dax.apdb.apdbSchema.ApdbSchema
 tableSchemas
 

Detailed Description

Class for management of APDB schema.

Parameters
----------
session : `cassandra.cluster.Session`
    Cassandra session object
keyspace : `str`
    Keyspace name for all tables.
schema_file : `str`
    Name of the YAML schema file.
schema_name : `str`, optional
    Name of the schema in YAML files.
prefix : `str`, optional
    Prefix to add to all schema elements.
time_partition_tables : `bool`
    If `True` then schema will have a separate table for each time
    partition.
enable_replica : `bool`, optional
    If `True` then use additional tables for replica chunks.

Constructor & Destructor Documentation

◆ __init__()

lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.__init__ ( self,
cassandra.cluster.Session session,
str keyspace,
str schema_file,
str schema_name = "ApdbSchema",
str prefix = "",
bool time_partition_tables = False,
bool enable_replica = False )

Member Function Documentation

◆ _apdb_tables_schema()

Mapping[ApdbTables, schema_model.Table] lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema._apdb_tables_schema ( self,
bool time_partition_tables )
protected
Generate schema for regular APDB tables.

◆ _check_replica_chunks()

bool lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema._check_replica_chunks ( self)
protected
Check whether database has tables for tracking insert IDs.

◆ _extra_tables_schema()

Mapping[ExtraTables, schema_model.Table] lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema._extra_tables_schema ( self)
protected
Generate schema for extra tables.

◆ _table_schema()

schema_model.Table lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema._table_schema ( self,
ApdbTables | ExtraTables table )
protected
Return schema definition for a table.

◆ _tableColumns()

list[str] lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema._tableColumns ( self,
ApdbTables | ExtraTables table_name )
protected
Return set of columns in a table

Parameters
----------
table_name : `ApdbTables`
    Name of the table.

Returns
-------
column_defs : `list`
    List of strings in the format "column_name type".

◆ apdbColumnNames()

list[str] lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.apdbColumnNames ( self,
ApdbTables | ExtraTables table_name )
Return a list of columns names for a table as defined in APDB
schema.

Parameters
----------
table_name : `ApdbTables` or `ExtraTables`
    Enum for a table in APDB schema.

Returns
-------
columns : `list` of `str`
    Names of regular columns in the table.

◆ clusteringColumns()

list[str] lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.clusteringColumns ( self,
ApdbTables | ExtraTables table_name )
Return a list of columns used for clustering.

Parameters
----------
table_name : `ApdbTables`
    Table name in APDB schema

Returns
-------
columns : `list` of `str`
    Names of columns for used for clustering.

◆ empty()

bool lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.empty ( self)
Return True if database schema is empty.

Returns
-------
empty : `bool`
    `True` if none of the required APDB tables exist in the database,
    `False` if all required tables exist.

Raises
------
InconsistentSchemaError
    Raised when some of the required tables exist but not all.

◆ getColumnMap()

Mapping[str, schema_model.Column] lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.getColumnMap ( self,
ApdbTables | ExtraTables table_name )
Return mapping of column names to Column definitions.

Parameters
----------
table_name : `ApdbTables`
    One of known APDB table names.

Returns
-------
column_map : `dict`
    Mapping of column names to `ColumnDef` instances.

◆ has_replica_chunks()

bool lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.has_replica_chunks ( self)
Whether insert ID tables are to be used (`bool`).

◆ keyspace()

str lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.keyspace ( self)
Return Cassandra keyspace for APDB tables.

◆ makeSchema()

None lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.makeSchema ( self,
*bool drop = False,
tuple[int, int] | None part_range = None,
int | None replication_factor = None )
Create or re-create all tables.

Parameters
----------
drop : `bool`
    If True then drop tables before creating new ones. Note that
    only tables are dropped and not the whole keyspace.
part_range : `tuple` [ `int` ] or `None`
    Start and end partition number for time partitions, end is not
    inclusive. Used to create per-partition DiaObject, DiaSource, and
    DiaForcedSource tables. If `None` then per-partition tables are
    not created.
replication_factor : `int`, optional
    Replication factor used when creating new keyspace, if keyspace
    already exists its replication factor is not changed.

◆ partitionColumns()

list[str] lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.partitionColumns ( self,
ApdbTables | ExtraTables table_name )
Return a list of columns used for table partitioning.

Parameters
----------
table_name : `ApdbTables`
    Table name in APDB schema

Returns
-------
columns : `list` of `str`
    Names of columns used for partitioning.

◆ tableName()

str lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema.tableName ( self,
ApdbTables | ExtraTables table_name )
Return Cassandra table name for APDB table.

Member Data Documentation

◆ _time_partitioned_tables

list lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema._time_partitioned_tables
staticprotected
Initial value:
= [
ApdbTables.DiaObject,
ApdbTables.DiaSource,
ApdbTables.DiaForcedSource,
]

◆ _type_map

dict lsst.dax.apdb.cassandra.apdbCassandraSchema.ApdbCassandraSchema._type_map
staticprotected
Initial value:
= {
felis.datamodel.DataType.double: "DOUBLE",
felis.datamodel.DataType.float: "FLOAT",
felis.datamodel.DataType.timestamp: "TIMESTAMP",
felis.datamodel.DataType.long: "BIGINT",
felis.datamodel.DataType.int: "INT",
felis.datamodel.DataType.short: "INT",
felis.datamodel.DataType.byte: "TINYINT",
felis.datamodel.DataType.binary: "BLOB",
felis.datamodel.DataType.char: "TEXT",
felis.datamodel.DataType.string: "TEXT",
felis.datamodel.DataType.unicode: "TEXT",
felis.datamodel.DataType.text: "TEXT",
felis.datamodel.DataType.boolean: "BOOLEAN",
schema_model.ExtraDataTypes.UUID: "UUID",
}

The documentation for this class was generated from the following file: