1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use crate::cassandra::data_type::ConstDataType;
use crate::cassandra::iterator::FieldIterator;

use crate::cassandra::schema::function_meta::FunctionMeta;
use crate::cassandra::util::Protected;
use crate::cassandra::value::Value;

use crate::cassandra_sys::cass_aggregate_meta_argument_count;
use crate::cassandra_sys::cass_aggregate_meta_argument_type;
use crate::cassandra_sys::cass_aggregate_meta_field_by_name;
use crate::cassandra_sys::cass_aggregate_meta_final_func;
use crate::cassandra_sys::cass_aggregate_meta_full_name;
use crate::cassandra_sys::cass_aggregate_meta_init_cond;
use crate::cassandra_sys::cass_aggregate_meta_name;
use crate::cassandra_sys::cass_aggregate_meta_return_type;
use crate::cassandra_sys::cass_aggregate_meta_state_func;
use crate::cassandra_sys::cass_aggregate_meta_state_type;
use crate::cassandra_sys::cass_iterator_fields_from_aggregate_meta;
use crate::cassandra_sys::raw2utf8;
use crate::cassandra_sys::CassAggregateMeta as _CassAggregateMeta;
use std::ffi::CString;
use std::mem;

/// Metadata about a cassandra aggregate
#[derive(Debug)]
pub struct AggregateMeta(*const _CassAggregateMeta);

impl Protected<*const _CassAggregateMeta> for AggregateMeta {
    fn inner(&self) -> *const _CassAggregateMeta {
        self.0
    }
    fn build(inner: *const _CassAggregateMeta) -> Self {
        if inner.is_null() {
            panic!("Unexpected null pointer")
        };
        AggregateMeta(inner)
    }
}

impl AggregateMeta {
    /// An iterator over the fields of an aggregate
    pub fn fields_iter(&self) -> FieldIterator {
        unsafe { FieldIterator::build(cass_iterator_fields_from_aggregate_meta(self.0)) }
    }

    /// Gets the name of the aggregate.
    pub fn get_name(&self) -> String {
        unsafe {
            let mut name = mem::zeroed();
            let mut name_length = mem::zeroed();
            cass_aggregate_meta_name(self.0, &mut name, &mut name_length);
            raw2utf8(name, name_length).expect("must be utf8")
        }
    }

    /// Gets the full name of the aggregate.
    pub fn full_name(&self) -> String {
        unsafe {
            let mut name = mem::zeroed();
            let mut name_length = mem::zeroed();
            cass_aggregate_meta_full_name(self.0, &mut name, &mut name_length);
            raw2utf8(name, name_length).expect("must be utf8")
        }
    }

    /// Gets the number of arguments this aggregate takes.
    pub fn argument_count(&self) -> usize {
        unsafe { cass_aggregate_meta_argument_count(self.0) }
    }

    /// Gets the aggregate's argument type for the provided index.
    pub fn argument_type(&self, index: usize) -> ConstDataType {
        // TODO: can return NULL
        unsafe { ConstDataType::build(cass_aggregate_meta_argument_type(self.0, index)) }
    }

    /// Gets the aggregate's argument return type.
    pub fn return_type(&self) -> ConstDataType {
        unsafe { ConstDataType::build(cass_aggregate_meta_return_type(self.0)) }
    }

    /// Gets the aggregate's argument state type.
    pub fn state_type(&self) -> ConstDataType {
        unsafe { ConstDataType::build(cass_aggregate_meta_state_type(self.0)) }
    }

    /// Gets the function metadata for the aggregate's state function.
    pub fn state_func(&self) -> FunctionMeta {
        unsafe { FunctionMeta::build(cass_aggregate_meta_state_func(self.0)) }
    }

    /// Gets the function metadata for the aggregates's final function.
    pub fn final_func(&self) -> FunctionMeta {
        unsafe { FunctionMeta::build(cass_aggregate_meta_final_func(self.0)) }
    }

    ///  Gets the initial condition value for the aggregate.
    pub fn init_cond(&self) -> Value {
        unsafe { Value::build(cass_aggregate_meta_init_cond(self.0)) }
    }

    ///  Gets a metadata field for the provided name. Metadata fields allow direct
    /// access to the column data found in the underlying "aggregates" metadata table.
    pub fn field_by_name(&self, name: &str) -> Option<Value> {
        unsafe {
            let name_cstr = CString::new(name).expect("must be utf8");
            let agg = cass_aggregate_meta_field_by_name(self.0, name_cstr.as_ptr());
            if agg.is_null() {
                None
            } else {
                Some(Value::build(agg))
            }
        }
    }
}