feat(cognitarium): implement base insert logic

amimart · amimart · commit 79dbc9bf7304 · 2023-06-01T11:38:55.000+02:00
diff --git a/contracts/okp4-cognitarium/src/contract.rs b/contracts/okp4-cognitarium/src/contract.rs
@@ -1,11 +1,14 @@
+use crate::contract::execute::insert;
 #[cfg(not(feature = "library"))]
 use cosmwasm_std::entry_point;
-use cosmwasm_std::{Binary, Deps, DepsMut, Env, MessageInfo, Response, StdError, StdResult};
+use cosmwasm_std::{
+    Binary, Deps, DepsMut, Env, MessageInfo, Response, StdError, StdResult, Uint128,
+};
 use cw2::set_contract_version;
 
 use crate::error::ContractError;
 use crate::msg::{ExecuteMsg, InstantiateMsg, QueryMsg};
-use crate::state::{Store, STORE};
+use crate::state::{Store, STORE, TRIPLE_KEY_INCREMENT};
 
 // version info for migration info
 const CONTRACT_NAME: &str = concat!("crates.io:", env!("CARGO_PKG_NAME"));
@@ -20,25 +23,69 @@ pub fn instantiate(
 ) -> Result<Response, ContractError> {
     set_contract_version(deps.storage, CONTRACT_NAME, CONTRACT_VERSION)?;
 
-    STORE.save(
-        deps.storage,
-        &Store {
-            owner: info.sender,
-            limits: msg.limits.into(),
-        },
-    )?;
+    STORE.save(deps.storage, &Store::new(info.sender, msg.limits.into()))?;
+    TRIPLE_KEY_INCREMENT.save(deps.storage, &Uint128::zero())?;
 
     Ok(Response::default())
 }
 
 #[cfg_attr(not(feature = "library"), entry_point)]
 pub fn execute(
-    _deps: DepsMut,
+    deps: DepsMut,
     _env: Env,
     _info: MessageInfo,
-    _msg: ExecuteMsg,
+    msg: ExecuteMsg,
 ) -> Result<Response, ContractError> {
-    Err(ContractError::NotImplemented)
+    match msg {
+        ExecuteMsg::InsertData { input } => insert(deps, input),
+    }
+}
+
+pub mod execute {
+    use super::*;
+    use crate::error::StoreError;
+    use crate::msg::DataInput;
+    use crate::rdf;
+    use crate::state::{triples, Triple};
+
+    pub fn insert(deps: DepsMut, graph: DataInput) -> Result<Response, ContractError> {
+        let mut store = STORE.load(deps.storage)?;
+
+        let mut pk = TRIPLE_KEY_INCREMENT.load(deps.storage)?;
+        let old_count = store.stat.triples_count;
+        rdf::parse_triples(
+            graph,
+            |triple| -> Result<Triple, ContractError> { Ok(triple.try_into()?) },
+            |res| -> Result<(), ContractError> {
+                res.and_then(|triple| {
+                    pk += Uint128::one();
+                    store.stat.triples_count += Uint128::one();
+
+                    store
+                        .limits
+                        .max_triple_count
+                        .filter(|&max| max < store.stat.triples_count)
+                        .map(|max| {
+                            Err(ContractError::from(StoreError::MaxTriplesLimitExceeded(
+                                max,
+                            )))
+                        })
+                        .unwrap_or(Ok(()))?;
+
+                    triples()
+                        .save(deps.storage, pk.u128(), &triple)
+                        .map_err(ContractError::Std)
+                })
+            },
+        )?;
+
+        TRIPLE_KEY_INCREMENT.save(deps.storage, &pk)?;
+        STORE.save(deps.storage, &store)?;
+
+        Ok(Response::new()
+            .add_attribute("action", "insert")
+            .add_attribute("inserted_count", store.stat.triples_count - old_count))
+    }
 }
 
 #[cfg_attr(not(feature = "library"), entry_point)]
@@ -52,7 +99,6 @@ mod tests {
     use crate::msg::StoreLimitsInput;
     use crate::state;
     use cosmwasm_std::testing::{mock_dependencies, mock_env, mock_info};
-    use cosmwasm_std::Uint128;
 
     #[test]
     fn proper_initialization() {
@@ -88,5 +134,16 @@ mod tests {
                 max_insert_data_triple_count: Uint128::from(7u128),
             }
         );
+        assert_eq!(
+            store.stat,
+            state::StoreStat {
+                triples_count: Uint128::zero(),
+            }
+        );
+
+        assert_eq!(
+            TRIPLE_KEY_INCREMENT.load(&deps.storage),
+            Ok(Uint128::zero())
+        );
     }
 }
diff --git a/contracts/okp4-cognitarium/src/error.rs b/contracts/okp4-cognitarium/src/error.rs
@@ -16,9 +16,6 @@ pub enum ContractError {
 
     #[error("Only the owner can perform this operation.")]
     Unauthorized {},
-
-    #[error("Not implemented.")]
-    NotImplemented,
 }
 
 impl From<RdfXmlError> for ContractError {
diff --git a/contracts/okp4-cognitarium/src/state/store.rs b/contracts/okp4-cognitarium/src/state/store.rs
@@ -9,6 +9,17 @@ pub const STORE: Item<Store> = Item::new("store");
 pub struct Store {
     pub owner: Addr,
     pub limits: StoreLimits,
+    pub stat: StoreStat,
+}
+
+impl Store {
+    pub fn new(owner: Addr, limits: StoreLimits) -> Store {
+        Store {
+            owner,
+            limits,
+            stat: StoreStat::default(),
+        }
+    }
 }
 
 #[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq)]
@@ -49,3 +60,8 @@ impl From<StoreLimits> for msg::StoreLimits {
         }
     }
 }
+
+#[derive(Serialize, Deserialize, Default, Clone, Debug, PartialEq, Eq)]
+pub struct StoreStat {
+    pub triples_count: Uint128,
+}
diff --git a/contracts/okp4-cognitarium/src/state/triples.rs b/contracts/okp4-cognitarium/src/state/triples.rs
@@ -1,9 +1,15 @@
 use crate::rdf::explode_iri;
 use cosmwasm_std::{StdError, Uint128};
-use cw_storage_plus::{Index, IndexList, IndexedMap, MultiIndex};
+use cw_storage_plus::{Index, IndexList, IndexedMap, Item, MultiIndex};
 use rio_api::model::NamedNode;
 use serde::{Deserialize, Serialize};
 
+/// Triple primary key as [Uint128] auto-increment.
+///
+/// Note: Considering the maximum value of [Uint128] there is no need to manage any re-usability of
+/// keys in case of triple removal.
+pub const TRIPLE_KEY_INCREMENT: Item<Uint128> = Item::new("triple-key");
+
 pub struct TripleIndexes<'a> {
     subject_and_predicate: MultiIndex<'a, (Subject, Predicate), Triple, Uint128>,
     predicate_and_object: MultiIndex<'a, (Predicate, Object), Triple, Uint128>,

Original file line number	Diff line number	Diff line change
`@@ -16,9 +16,6 @@ pub enum ContractError {`
`16`	`16`
`17`	`17`	`#[error("Only the owner can perform this operation.")]`
`18`	`18`	`Unauthorized {},`
`19`		`-`
`20`		`- #[error("Not implemented.")]`
`21`		`- NotImplemented,`
`22`	`19`	`}`
`23`	`20`
`24`	`21`	`impl From<RdfXmlError> for ContractError {`