[query] Add values_id to insert nodes #401

src/agdb/query/query_id.rs

@@ -1,4 +1,23 @@
+#[derive(Clone)]
 pub enum QueryId {
     Id(u64),
     Alias(String),
 }
+
+impl From<&str> for QueryId {
+    fn from(value: &str) -> Self {
+        Self::from(value.to_string())
+    }
+}
+
+impl From<String> for QueryId {
+    fn from(value: String) -> Self {
+        Self::Alias(value)
+    }
+}
+
+impl From<u64> for QueryId {
+    fn from(value: u64) -> Self {
+        Self::Id(value)
+    }
+}

src/agdb/query/query_ids.rs

@@ -2,6 +2,7 @@ use super::query_id::QueryId;
 use super::search_query::SearchQuery;
 
 pub enum QueryIds {
+    Id(QueryId),
     Ids(Vec<QueryId>),
     Search(SearchQuery),
 }

src/agdb/query_builder/insert_node.rs

@@ -2,6 +2,8 @@ use super::insert_node_alias::InsertNodeAlias;
 use super::insert_node_values::InsertNodeValues;
 use crate::db::db_key_value::DbKeyValue;
 use crate::query::insert_nodes_query::InsertNodesQuery;
+use crate::query::query_id::QueryId;
+use crate::query::query_ids::QueryIds;
 use crate::query::query_values::QueryValues;
 use crate::Query;
 
@@ -20,6 +22,12 @@ impl InsertNode {
         InsertNodeValues(self.0)
     }
 
+    pub fn values_id(mut self, id: QueryId) -> InsertNodeValues {
+        self.0.values = QueryValues::Query(QueryIds::Id(id));
+
+        InsertNodeValues(self.0)
+    }
+
     pub fn query(self) -> Query {
         Query::InsertNodes(self.0)
     }

src/agdb/query_builder/insert_node_alias.rs

@@ -1,5 +1,7 @@
 use super::insert_node_values::InsertNodeValues;
 use crate::query::insert_nodes_query::InsertNodesQuery;
+use crate::query::query_id::QueryId;
+use crate::query::query_ids::QueryIds;
 use crate::query::query_values::QueryValues;
 use crate::DbKeyValue;
 use crate::Query;
@@ -16,4 +18,10 @@ impl InsertNodeAlias {
 
         InsertNodeValues(self.0)
     }
+
+    pub fn values_id(mut self, id: QueryId) -> InsertNodeValues {
+        self.0.values = QueryValues::Query(QueryIds::Ids(vec![id]));
+
+        InsertNodeValues(self.0)
+    }
 }

src/agdb/query_builder/insert_nodes.rs

@@ -2,6 +2,8 @@ use super::insert_nodes_aliases::InsertNodesAliases;
 use super::insert_nodes_count::InsertNodesCount;
 use super::insert_nodes_values::InsertNodesValues;
 use crate::query::insert_nodes_query::InsertNodesQuery;
+use crate::query::query_id::QueryId;
+use crate::query::query_ids::QueryIds;
 use crate::query::query_values::QueryValues;
 use crate::DbKeyValue;
 
@@ -25,4 +27,16 @@ impl InsertNodes {
 
         InsertNodesValues(self.0)
     }
+
+    pub fn values_id(mut self, id: QueryId) -> InsertNodesValues {
+        self.0.values = QueryValues::Query(QueryIds::Id(id));
+
+        InsertNodesValues(self.0)
+    }
+
+    pub fn values_ids(mut self, ids: &[QueryId]) -> InsertNodesValues {
+        self.0.values = QueryValues::Query(QueryIds::Ids(ids.to_vec()));
+
+        InsertNodesValues(self.0)
+    }
 }

src/agdb/query_builder/insert_nodes_aliases.rs

@@ -1,4 +1,6 @@
 use crate::query::insert_nodes_query::InsertNodesQuery;
+use crate::query::query_id::QueryId;
+use crate::query::query_ids::QueryIds;
 use crate::query::query_values::QueryValues;
 use crate::DbKeyValue;
 use crate::Query;
@@ -17,4 +19,16 @@ impl InsertNodesAliases {
 
         InsertNodesValues(self.0)
     }
+
+    pub fn values_id(mut self, id: QueryId) -> InsertNodesValues {
+        self.0.values = QueryValues::Query(QueryIds::Id(id));
+
+        InsertNodesValues(self.0)
+    }
+
+    pub fn values_ids(mut self, ids: &[QueryId]) -> InsertNodesValues {
+        self.0.values = QueryValues::Query(QueryIds::Ids(ids.to_vec()));
+
+        InsertNodesValues(self.0)
+    }
 }

tests/insert_node_test.rs

@@ -18,6 +18,14 @@ fn insert_node_values() {
         .query();
 }
 
+#[test]
+fn insert_node_values_id() {
+    let _query = QueryBuilder::insert()
+        .node()
+        .values_id("alias".into())
+        .query();
+}
+
 #[test]
 fn insert_node_alias_values() {
     let _query = QueryBuilder::insert()
@@ -26,3 +34,12 @@ fn insert_node_alias_values() {
         .values(&[("key", "value").into()])
         .query();
 }
+
+#[test]
+fn insert_node_alias_values_id() {
+    let _query = QueryBuilder::insert()
+        .node()
+        .alias("alias1")
+        .values_id("alias2".into())
+        .query();
+}

tests/insert_nodes_test.rs

@@ -20,6 +20,24 @@ fn insert_nodes_aliases_values() {
         .query();
 }
 
+#[test]
+fn insert_nodes_aliases_values_id() {
+    let _query = QueryBuilder::insert()
+        .nodes()
+        .aliases(&["alias1".to_string(), "alias2".to_string()])
+        .values_id("alias3".into())
+        .query();
+}
+
+#[test]
+fn insert_nodes_aliases_values_ids() {
+    let _query = QueryBuilder::insert()
+        .nodes()
+        .aliases(&["alias1".to_string(), "alias2".to_string()])
+        .values_ids(&["alias3".into(), 4.into()])
+        .query();
+}
+
 #[test]
 fn insert_nodes_count() {
     let _query = QueryBuilder::insert().nodes().count(2).query();
@@ -57,3 +75,19 @@ fn insert_nodes_values() {
         ])
         .query();
 }
+
+#[test]
+fn insert_nodes_values_id() {
+    let _query = QueryBuilder::insert()
+        .nodes()
+        .values_id("alias3".into())
+        .query();
+}
+
+#[test]
+fn insert_nodes_values_ids() {
+    let _query = QueryBuilder::insert()
+        .nodes()
+        .values_ids(&["alias3".into(), 4.into()])
+        .query();
+}