Skip to content

Commit

Permalink
DataNodeRef with single lifetime
Browse files Browse the repository at this point in the history
  • Loading branch information
rwestphal committed Oct 20, 2024
1 parent d6a8299 commit 5d4060d
Show file tree
Hide file tree
Showing 3 changed files with 65 additions and 55 deletions.
94 changes: 46 additions & 48 deletions src/data.rs
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,8 @@ pub struct DataTree<'a> {

/// YANG data node reference.
#[derive(Clone, Debug)]
pub struct DataNodeRef<'a, 'b> {
tree: &'a DataTree<'b>,
pub struct DataNodeRef<'a> {
tree: &'a DataTree<'a>,
raw: *mut ffi::lyd_node,
}

Expand All @@ -42,8 +42,8 @@ pub struct DataNodeRef<'a, 'b> {
/// attributes. In JSON, they are represented as JSON elements starting with the
/// '@' character (for more information, see the YANG metadata RFC).
#[derive(Clone, Debug)]
pub struct Metadata<'a, 'b> {
dnode: &'a DataNodeRef<'a, 'b>,
pub struct Metadata<'a> {
dnode: &'a DataNodeRef<'a>,
raw: *mut ffi::lyd_meta,
}

Expand Down Expand Up @@ -229,7 +229,7 @@ pub trait Data<'a> {
/// the form `leaf-list[.='val']`, these instances are found using hashes
/// with constant (*O(1)*) complexity (unless they are defined in
/// top-level). Other predicates can still follow the aforementioned ones.
fn find_xpath(&self, xpath: &str) -> Result<Set<'_, DataNodeRef<'_, 'a>>> {
fn find_xpath(&'a self, xpath: &str) -> Result<Set<'a, DataNodeRef<'a>>> {
let xpath = CString::new(xpath).unwrap();
let mut set = std::ptr::null_mut();
let set_ptr = &mut set;
Expand All @@ -256,7 +256,7 @@ pub trait Data<'a> {
/// The expected format of the expression is JSON, meaning the first node in
/// every path must have its module name as prefix or be the special `*`
/// value for all the nodes.
fn find_path(&self, path: &str) -> Result<DataNodeRef<'_, 'a>> {
fn find_path(&'a self, path: &str) -> Result<DataNodeRef<'a>> {
let path = CString::new(path).unwrap();
let mut rnode = std::ptr::null_mut();
let rnode_ptr = &mut rnode;
Expand Down Expand Up @@ -532,7 +532,7 @@ impl<'a> DataTree<'a> {

/// Returns a reference to the fist top-level data node, unless the data
/// tree is empty.
pub fn reference<'b>(&'b self) -> Option<DataNodeRef<'b, 'a>> {
pub fn reference(&self) -> Option<DataNodeRef<'_>> {
if self.raw.is_null() {
None
} else {
Expand Down Expand Up @@ -563,7 +563,7 @@ impl<'a> DataTree<'a> {
path: &str,
value: Option<&str>,
output: bool,
) -> Result<Option<DataNodeRef<'_, '_>>> {
) -> Result<Option<DataNodeRef<'_>>> {
let path = CString::new(path).unwrap();
let mut rnode_root = std::ptr::null_mut();
let mut rnode = std::ptr::null_mut();
Expand Down Expand Up @@ -752,7 +752,7 @@ impl<'a> DataTree<'a> {

/// Returns an iterator over all elements in the data tree and its sibling
/// trees (depth-first search algorithm).
pub fn traverse<'b>(&'b self) -> impl Iterator<Item = DataNodeRef<'b, 'a>> {
pub fn traverse(&self) -> impl Iterator<Item = DataNodeRef<'_>> {
let top = Siblings::new(self.reference());
top.flat_map(|dnode| dnode.traverse())
}
Expand Down Expand Up @@ -791,7 +791,7 @@ impl Drop for DataTree<'_> {

// ===== impl DataNodeRef =====

impl<'a, 'b> DataNodeRef<'a, 'b> {
impl<'a> DataNodeRef<'a> {
/// Schema definition of this node.
pub fn schema(&self) -> SchemaNode<'_> {
let raw = unsafe { (*self.raw).schema };
Expand All @@ -807,46 +807,46 @@ impl<'a, 'b> DataNodeRef<'a, 'b> {
}

/// Returns an iterator over the ancestor data nodes.
pub fn ancestors(&self) -> Ancestors<'a, DataNodeRef<'a, 'b>> {
pub fn ancestors(&self) -> Ancestors<'a, DataNodeRef<'a>> {
let parent = self.parent();
Ancestors::new(parent)
}

/// Returns an iterator over this data node and its ancestors.
pub fn inclusive_ancestors(&self) -> Ancestors<'a, DataNodeRef<'a, 'b>> {
pub fn inclusive_ancestors(&self) -> Ancestors<'a, DataNodeRef<'a>> {
Ancestors::new(Some(self.clone()))
}

/// Returns an iterator over the sibling data nodes.
pub fn siblings(&self) -> Siblings<'a, DataNodeRef<'a, 'b>> {
pub fn siblings(&self) -> Siblings<'a, DataNodeRef<'a>> {
let sibling = self.next_sibling();
Siblings::new(sibling)
}

/// Returns an iterator over this data node and its siblings.
pub fn inclusive_siblings(&self) -> Siblings<'a, DataNodeRef<'a, 'b>> {
pub fn inclusive_siblings(&self) -> Siblings<'a, DataNodeRef<'a>> {
Siblings::new(Some(self.clone()))
}

/// Returns an iterator over the child data nodes.
pub fn children(&self) -> Siblings<'a, DataNodeRef<'a, 'b>> {
pub fn children(&self) -> Siblings<'a, DataNodeRef<'a>> {
let child = self.first_child();
Siblings::new(child)
}

/// Returns an iterator over all elements in the data tree (depth-first
/// search algorithm).
pub fn traverse(&self) -> Traverse<'a, DataNodeRef<'a, 'b>> {
pub fn traverse(&self) -> Traverse<'a, DataNodeRef<'a>> {
Traverse::new(self.clone())
}

/// Returns an iterator over the keys of the list.
pub fn list_keys(&self) -> impl Iterator<Item = DataNodeRef<'a, 'b>> {
pub fn list_keys(&self) -> impl Iterator<Item = DataNodeRef<'a>> {
self.children().filter(|dnode| dnode.schema().is_list_key())
}

/// Returns an iterator over all metadata associated to this node.
pub fn meta(&self) -> MetadataList<'_, '_> {
pub fn meta(&self) -> MetadataList<'_> {
let rmeta = unsafe { (*self.raw).meta };
let meta = unsafe { Metadata::from_raw_opt(self, rmeta) };
MetadataList::new(meta)
Expand Down Expand Up @@ -915,7 +915,7 @@ impl<'a, 'b> DataNodeRef<'a, 'b> {
///
/// When the `with_parents` parameter is set, duplicate also all the node
/// parents. Keys are also duplicated for lists.
pub fn duplicate(&self, with_parents: bool) -> Result<DataTree<'b>> {
pub fn duplicate(&self, with_parents: bool) -> Result<DataTree<'a>> {
let mut dup = std::ptr::null_mut();
let dup_ptr = &mut dup;

Expand Down Expand Up @@ -980,7 +980,7 @@ impl<'a, 'b> DataNodeRef<'a, 'b> {
&mut self,
module: Option<&SchemaModule<'_>>,
name: &str,
) -> Result<DataNodeRef<'a, 'b>> {
) -> Result<DataNodeRef<'a>> {
let name_cstr = CString::new(name).unwrap();
let mut rnode = std::ptr::null_mut();
let rnode_ptr = &mut rnode;
Expand Down Expand Up @@ -1015,7 +1015,7 @@ impl<'a, 'b> DataNodeRef<'a, 'b> {
module: Option<&SchemaModule<'_>>,
name: &str,
keys: &str,
) -> Result<DataNodeRef<'a, 'b>> {
) -> Result<DataNodeRef<'a>> {
let name_cstr = CString::new(name).unwrap();
let keys_cstr = CString::new(keys).unwrap();
let mut rnode = std::ptr::null_mut();
Expand Down Expand Up @@ -1053,7 +1053,7 @@ impl<'a, 'b> DataNodeRef<'a, 'b> {
module: Option<&SchemaModule<'_>>,
name: &str,
keys: &[impl AsRef<str>],
) -> Result<DataNodeRef<'a, 'b>> {
) -> Result<DataNodeRef<'a>> {
let name_cstr = CString::new(name).unwrap();
let mut rnode = std::ptr::null_mut();
let rnode_ptr = &mut rnode;
Expand Down Expand Up @@ -1132,7 +1132,7 @@ impl<'a, 'b> DataNodeRef<'a, 'b> {
}
}

impl<'a> Data<'a> for DataNodeRef<'_, 'a> {
impl<'a> Data<'a> for DataNodeRef<'a> {
fn tree(&self) -> &DataTree<'a> {
self.tree
}
Expand All @@ -1142,32 +1142,32 @@ impl<'a> Data<'a> for DataNodeRef<'_, 'a> {
}
}

unsafe impl<'a, 'b> Binding<'a> for DataNodeRef<'a, 'b> {
unsafe impl<'a> Binding<'a> for DataNodeRef<'a> {
type CType = ffi::lyd_node;
type Container = DataTree<'b>;
type Container = DataTree<'a>;

unsafe fn from_raw(
tree: &'a DataTree<'b>,
tree: &'a DataTree<'a>,
raw: *mut ffi::lyd_node,
) -> DataNodeRef<'a, 'b> {
) -> DataNodeRef<'a> {
DataNodeRef { tree, raw }
}
}

impl<'a, 'b> NodeIterable<'a> for DataNodeRef<'a, 'b> {
fn parent(&self) -> Option<DataNodeRef<'a, 'b>> {
impl<'a> NodeIterable<'a> for DataNodeRef<'a> {
fn parent(&self) -> Option<DataNodeRef<'a>> {
// NOTE: can't use lyd_parent() since it's an inline function.
let rparent =
unsafe { &mut (*(*self.raw).parent).__bindgen_anon_1.node };
unsafe { DataNodeRef::from_raw_opt(self.tree, rparent) }
}

fn next_sibling(&self) -> Option<DataNodeRef<'a, 'b>> {
fn next_sibling(&self) -> Option<DataNodeRef<'a>> {
let rsibling = unsafe { (*self.raw).next };
unsafe { DataNodeRef::from_raw_opt(self.tree, rsibling) }
}

fn first_child(&self) -> Option<DataNodeRef<'a, 'b>> {
fn first_child(&self) -> Option<DataNodeRef<'a>> {
// NOTE: can't use lyd_child() since it's an inline function.
let snode = unsafe { (*self.raw).schema };
if snode.is_null() {
Expand All @@ -1192,18 +1192,18 @@ impl<'a, 'b> NodeIterable<'a> for DataNodeRef<'a, 'b> {
}
}

impl PartialEq for DataNodeRef<'_, '_> {
fn eq(&self, other: &DataNodeRef<'_, '_>) -> bool {
impl PartialEq for DataNodeRef<'_> {
fn eq(&self, other: &DataNodeRef<'_>) -> bool {
self.raw == other.raw
}
}

unsafe impl Send for DataNodeRef<'_, '_> {}
unsafe impl Sync for DataNodeRef<'_, '_> {}
unsafe impl Send for DataNodeRef<'_> {}
unsafe impl Sync for DataNodeRef<'_> {}

// ===== impl Metadata =====

impl<'a, 'b> Metadata<'a, 'b> {
impl<'a> Metadata<'a> {
/// Metadata name.
pub fn name(&self) -> &str {
char_ptr_to_str(unsafe { (*self.raw).name })
Expand All @@ -1226,32 +1226,32 @@ impl<'a, 'b> Metadata<'a, 'b> {

/// Next metadata.
#[doc(hidden)]
pub(crate) fn next(&self) -> Option<Metadata<'a, 'b>> {
pub(crate) fn next(&self) -> Option<Metadata<'a>> {
let rnext = unsafe { (*self.raw).next };
unsafe { Metadata::from_raw_opt(self.dnode, rnext) }
}
}

unsafe impl<'a, 'b> Binding<'a> for Metadata<'a, 'b> {
unsafe impl<'a> Binding<'a> for Metadata<'a> {
type CType = ffi::lyd_meta;
type Container = DataNodeRef<'a, 'b>;
type Container = DataNodeRef<'a>;

unsafe fn from_raw(
dnode: &'a DataNodeRef<'a, 'b>,
dnode: &'a DataNodeRef<'a>,
raw: *mut ffi::lyd_meta,
) -> Metadata<'a, 'b> {
) -> Metadata<'a> {
Metadata { dnode, raw }
}
}

impl PartialEq for Metadata<'_, '_> {
fn eq(&self, other: &Metadata<'_, '_>) -> bool {
impl PartialEq for Metadata<'_> {
fn eq(&self, other: &Metadata<'_>) -> bool {
self.raw == other.raw
}
}

unsafe impl Send for Metadata<'_, '_> {}
unsafe impl Sync for Metadata<'_, '_> {}
unsafe impl Send for Metadata<'_> {}
unsafe impl Sync for Metadata<'_> {}

// ===== impl DataDiff =====

Expand All @@ -1276,9 +1276,7 @@ impl<'a> DataDiff<'a> {
}

/// Returns an iterator over the data changes.
pub fn iter(
&self,
) -> impl Iterator<Item = (DataDiffOp, DataNodeRef<'_, '_>)> {
pub fn iter(&self) -> impl Iterator<Item = (DataDiffOp, DataNodeRef<'_>)> {
self.tree.traverse().filter_map(|dnode| {
match dnode.meta().find(|meta| meta.name() == "operation") {
Some(meta) => match meta.value() {
Expand Down
14 changes: 7 additions & 7 deletions src/iter.rs
Original file line number Diff line number Diff line change
Expand Up @@ -121,8 +121,8 @@ pub struct SchemaModules<'a> {

/// An iterator over a list of metadata.
#[derive(Debug)]
pub struct MetadataList<'a, 'b> {
next: Option<Metadata<'a, 'b>>,
pub struct MetadataList<'a> {
next: Option<Metadata<'a>>,
}

// ===== impl Siblings =====
Expand Down Expand Up @@ -414,16 +414,16 @@ impl<'a> Iterator for SchemaModules<'a> {

// ===== impl MetadataList =====

impl<'a, 'b> MetadataList<'a, 'b> {
pub fn new(next: Option<Metadata<'a, 'b>>) -> MetadataList<'a, 'b> {
impl<'a> MetadataList<'a> {
pub fn new(next: Option<Metadata<'a>>) -> MetadataList<'a> {
MetadataList { next }
}
}

impl<'a, 'b> Iterator for MetadataList<'a, 'b> {
type Item = Metadata<'a, 'b>;
impl<'a> Iterator for MetadataList<'a> {
type Item = Metadata<'a>;

fn next(&mut self) -> Option<Metadata<'a, 'b>> {
fn next(&mut self) -> Option<Metadata<'a>> {
let meta = self.next.clone();
if let Some(next) = &self.next {
self.next = next.next();
Expand Down
12 changes: 12 additions & 0 deletions tests/data.rs
Original file line number Diff line number Diff line change
Expand Up @@ -176,6 +176,18 @@ macro_rules! assert_data_eq {
};
}

use std::sync::{Arc, LazyLock as Lazy, OnceLock};
static YANG_CTX: OnceLock<Arc<Context>> = OnceLock::new();
fn get_configuration(config: &DataTree<'_>, path: &str) -> DataTree<'static> {
let yang_ctx = YANG_CTX.get().unwrap();
let mut dtree = DataTree::new(yang_ctx);
for dnode in config.find_xpath(path).unwrap() {
let subtree = dnode.duplicate(true).unwrap();
dtree.merge(&subtree).unwrap();
}
dtree
}

fn create_context() -> Context {
// Initialize context.
let mut ctx = Context::new(ContextFlags::NO_YANGLIBRARY)
Expand Down

0 comments on commit 5d4060d

Please sign in to comment.