stateful

Cargo.toml

@@ -1,5 +1,5 @@
 [workspace]
-members = ["unit_struct", "build_script"]
+members = ["unit_struct", "build_script", "stateful"]
 resolver = "2"
 
 [workspace.dependencies]

build_script/build.rs

@@ -31,5 +31,5 @@ fn main() {
     // https://docs.rs/prost-build/latest/prost_build/struct.Config.html#method.compile_protos
     let out_dir = std::env::var("OUT_DIR").unwrap();
     let path = Path::new(&out_dir).join("counter.rs");
-    std::fs::write(&path, BINDING).unwrap();
+    std::fs::write(path, BINDING).unwrap();
 }

dfx.json

@@ -13,6 +13,13 @@
             "candid": "counter.did",
             "wasm": "target/wasm32-unknown-unknown/release/build_script.wasm",
             "shrink": true
+        },
+        "stateful": {
+            "type": "custom",
+            "build": "sh build.sh stateful",
+            "candid": "counter.did",
+            "wasm": "target/wasm32-unknown-unknown/release/stateful.wasm",
+            "shrink": true
         }
     }
 }

stateful/Cargo.toml

@@ -0,0 +1,12 @@
+[package]
+name = "stateful"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+ic-cdk.workspace = true
+once_cell  = "1"
+[lib]
+crate-type = ["cdylib"]

stateful/src/lib.rs

@@ -0,0 +1,46 @@
+// ic_cdk::generate!({path: "../../counter.did", service : Canister, trait: Counter});
+
+// Expand Start
+trait Counter {
+    fn inc(&mut self);
+    fn read(&self) -> u64;
+}
+
+#[export_name = "canister_update inc"]
+fn __canister_method_inc() {
+    ic_cdk::setup();
+    ic_cdk::spawn(async {
+        let _result = CANISTER.lock().unwrap().inc();
+        ic_cdk::api::call::reply(())
+    });
+}
+
+#[export_name = "canister_query read"]
+fn __canister_method_read() {
+    ic_cdk::setup();
+    ic_cdk::spawn(async {
+        let result = CANISTER.lock().unwrap().read();
+        ic_cdk::api::call::reply((result,))
+    });
+}
+
+use once_cell::sync::Lazy;
+use std::sync::Mutex;
+static CANISTER: Lazy<Mutex<Canister>> = Lazy::new(|| Mutex::new(Default::default()));
+
+// Expand End
+
+#[derive(Default)]
+struct Canister {
+    counter: u64,
+}
+
+impl Counter for Canister {
+    fn inc(&mut self) {
+        self.counter += 1;
+    }
+
+    fn read(&self) -> u64 {
+        self.counter
+    }
+}