add ab protocol to ci

Author: KoRaghav

.github/workflows/rust.yml

@@ -70,3 +70,16 @@ jobs:
               echo "make job failed"
               exit $status
           fi
+
+    - name: Run AB
+      run: |
+          cd examples/ab
+          make node &
+          sleep 1
+          
+          timeout 6s make job || echo "make job terminated after timeout"
+          status=$?
+          if [ $status -ne 0 ] && [ $status -ne 124 ]; then
+              echo "make job failed"
+              exit $status
+          fi

Cargo.lock

@@ -13,14 +13,8 @@ crate-type = ["cdylib", "rlib"]
 [dependencies]
 tokio = { version = "1", features = ["full"] }
 bincode = { version = "2.0.0", features = ["serde"] }
-tokio-util = { version = "0.7", features = ["full"] }
 reactor-actor = { path = "../../actor" }
 reactor-macros = { path = "../../macros" }
 lazy_static = "1.5.0"
-env_logger = "0.11"
 serde_json = "1.0.140"
-rand = "0.9.1"
-log = "0.4.27"
-
-[features]
-chaos = []
+log = "0.4.27"

examples/ab/Cargo.toml

@@ -2,7 +2,7 @@ GENERIC_NCTRL = ../../generic_nctrl
 GENERIC_JCTRL = ../../generic_jctrl
 TARGET_SO = ../../target/debug/lib_ab_actor.so
 
-.PHONY: install install_node install_jobc chaos kill_node node job clean TARGET_SO
+.PHONY: install install_node install_jobc kill_node node job clean TARGET_SO
 
 
 install_node: $(GENERIC_NCTRL)
@@ -11,9 +11,6 @@ install_node: $(GENERIC_NCTRL)
 install_jobc: $(GENERIC_JCTRL)
 	cargo install --debug --path $(GENERIC_JCTRL)
 
-chaos:
-	cargo build --features chaos
-
 kill_node:
 	@echo "Killing process on port 3000 if any..."
 	@lsof -ti :3000 | xargs --no-run-if-empty kill

examples/ab/Makefile

@@ -8,8 +8,37 @@ use std::collections::HashMap;
 use std::time::Duration;
 use std::vec;
 
-type Data = char;
-type Bit = bool;
+#[derive(Encode, Decode, Debug, Clone, Copy)]
+struct Data(char);
+#[derive(Encode, Decode, Debug, Clone, Eq, PartialEq, Copy)]
+struct Bit(bool);
+
+impl Data {
+    const MIN: Data = Data('A');
+    const MAX: Data = Data('Z');
+}
+
+impl Iterator for Data {
+    type Item = Data;
+
+    fn next(&mut self) -> Option<Self::Item> {
+        if self.0 < Data::MAX.0 {
+            self.0 = ((self.0 as u8) + 1) as char;
+        } else {
+            self.0 = Data::MIN.0;
+        }
+        Some(self.clone())
+    }
+}
+
+impl Bit {
+    const INIT: Bit = Bit(true);
+
+    fn negate(&mut self) -> Bit {
+        self.0 = !self.0;
+        self.clone()
+    }
+}
 
 #[derive(Encode, Decode, Debug, Clone, DefaultPrio, DeriveMsg)]
 enum ABMsg {
@@ -18,31 +47,13 @@ enum ABMsg {
     GeneratorMsg,
 }
 
-struct GeneratorIter {
-    current: u8,
-}
-impl GeneratorIter {
-    const MIN: u8 = 0;
-    const MAX: u8 = 25;
-
-    fn new() -> Self {
-        GeneratorIter {
-            current: GeneratorIter::MIN,
-        }
-    }
-}
+struct GeneratorIter;
 impl Iterator for GeneratorIter {
     type Item = ABMsg;
 
     fn next(&mut self) -> Option<Self::Item> {
         std::thread::sleep(Duration::from_secs(4));
-
-        if self.current <= GeneratorIter::MAX {
-            self.current += 1;
-            Some(ABMsg::GeneratorMsg)
-        } else {
-            None
-        }
+        Some(ABMsg::GeneratorMsg)
     }
 }
 
@@ -53,8 +64,8 @@ struct Writer {
 impl Writer {
     fn new() -> Self {
         Writer {
-            data: 'A',
-            bit: true,
+            data: Data::MIN,
+            bit: Bit::INIT,
         }
     }
 }
@@ -73,8 +84,8 @@ impl reactor_actor::ActorProcess for Writer {
             ABMsg::Ack(bit) => {
                 info!("Writer: Recv: {input:?}");
                 if bit == self.bit {
-                    self.data = ((self.data as u8) + 1) as char;
-                    self.bit = !self.bit;
+                    self.data.next();
+                    self.bit.negate();
                 }
                 vec![]
             }
@@ -90,8 +101,8 @@ struct Reader {
 impl Reader {
     fn new() -> Self {
         Reader {
-            data: 'A',
-            bit: true,
+            data: Data::MIN,
+            bit: Bit::INIT,
         }
     }
 }
@@ -151,7 +162,7 @@ pub fn writer(ctx: RuntimeCtx, mut payload: HashMap<String, serde_json::Value>)
     RUNTIME.spawn(async move {
         BehaviourBuilder::new(Writer::new(), BincodeCodec::default())
             .send(Sender::new(other_addr))
-            .generator(GeneratorIter::new())
+            .generator(GeneratorIter {})
             .build()
             .run(ctx)
             .await
@@ -171,7 +182,7 @@ pub fn reader(ctx: RuntimeCtx, mut payload: HashMap<String, serde_json::Value>)
     RUNTIME.spawn(async move {
         BehaviourBuilder::new(Reader::new(), BincodeCodec::default())
             .send(Sender::new(other_addr))
-            .generator(GeneratorIter::new())
+            .generator(GeneratorIter {})
             .build()
             .run(ctx)
             .await