use crate::cassandra::iterator::ColumnIterator;
use crate::cassandra::iterator::FieldIterator;
use crate::cassandra::schema::column_meta::ColumnMeta;
use crate::cassandra::util::{Protected, ProtectedInner};
use crate::cassandra::value::Value;
use crate::cassandra_sys::cass_iterator_columns_from_table_meta;
use crate::cassandra_sys::cass_iterator_fields_from_table_meta;
use crate::cassandra_sys::cass_table_meta_clustering_key;
use crate::cassandra_sys::cass_table_meta_clustering_key_count;
use crate::cassandra_sys::cass_table_meta_column;
use crate::cassandra_sys::cass_table_meta_column_by_name;
use crate::cassandra_sys::cass_table_meta_column_count;
use crate::cassandra_sys::cass_table_meta_field_by_name;
use crate::cassandra_sys::cass_table_meta_name;
use crate::cassandra_sys::cass_table_meta_partition_key;
use crate::cassandra_sys::cass_table_meta_partition_key_count;
use crate::cassandra_sys::CassTableMeta as _CassTableMeta;
use std::marker::PhantomData;
use std::os::raw::c_char;
use std::slice;
use std::str;
#[derive(Debug)]
pub struct TableMeta<'a>(*const _CassTableMeta, PhantomData<&'a _CassTableMeta>);
impl ProtectedInner<*const _CassTableMeta> for TableMeta<'_> {
fn inner(&self) -> *const _CassTableMeta {
self.0
}
}
impl Protected<*const _CassTableMeta> for TableMeta<'_> {
fn build(inner: *const _CassTableMeta) -> Self {
if inner.is_null() {
panic!("Unexpected null pointer")
};
TableMeta(inner, PhantomData)
}
}
impl<'a> TableMeta<'a> {
pub fn field_iter(&self) -> FieldIterator<'a> {
unsafe { FieldIterator::build(cass_iterator_fields_from_table_meta(self.0)) }
}
pub fn columns_iter(&self) -> ColumnIterator<'a> {
unsafe { ColumnIterator::build(cass_iterator_columns_from_table_meta(self.0)) }
}
pub fn column_by_name(&self, name: &str) -> ColumnMeta<'a> {
unsafe {
ColumnMeta::build(cass_table_meta_column_by_name(
self.0,
name.as_ptr() as *const c_char,
))
}
}
pub fn get_name(&self) -> String {
let mut name = std::ptr::null();
let mut name_length = 0;
unsafe {
cass_table_meta_name(self.0, &mut name, &mut name_length);
str::from_utf8(slice::from_raw_parts(name as *const u8, name_length))
.expect("must be utf8")
.to_owned()
}
}
pub fn column_count(&self) -> usize {
unsafe { cass_table_meta_column_count(self.0) }
}
pub fn column(&self, index: usize) -> ColumnMeta<'a> {
unsafe { ColumnMeta::build(cass_table_meta_column(self.0, index)) }
}
pub fn partition_key_count(&self) -> usize {
unsafe { cass_table_meta_partition_key_count(self.0) }
}
pub fn partition_key(&self, index: usize) -> Option<ColumnMeta<'a>> {
unsafe {
let key = cass_table_meta_partition_key(self.0, index);
if key.is_null() {
None
} else {
Some(ColumnMeta::build(key))
}
}
}
pub fn clustering_key_count(&self) -> usize {
unsafe { cass_table_meta_clustering_key_count(self.0) }
}
pub fn cluster_key(&self, index: usize) -> Option<ColumnMeta<'a>> {
unsafe {
let key = cass_table_meta_clustering_key(self.0, index);
if key.is_null() {
None
} else {
Some(ColumnMeta::build(key))
}
}
}
pub fn field_by_name(&self, name: &str) -> Option<Value<'a>> {
unsafe {
let value = cass_table_meta_field_by_name(self.0, name.as_ptr() as *const c_char);
if value.is_null() {
None
} else {
Some(Value::build(value))
}
}
}
}