feat(model): add bandwidth trace replay model (#13)

Author: Lethe10137

.github/workflows/cargo-test.yml

@@ -30,7 +30,7 @@ jobs:
         uses: Swatinem/rust-cache@v1
       - name: cargo test
         run: |
-          cargo test --no-fail-fast --verbose --features mahimahi --features serde
+          cargo test --no-fail-fast --verbose --features mahimahi --features serde --features truncated-normal
 
   cargo-test-full:
     runs-on: ubuntu-latest
@@ -50,3 +50,61 @@ jobs:
       - name: cargo test
         run: |
           cargo test --no-fail-fast --verbose --features full
+
+  cargo-test-serde:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+
+      - name: Install latest nightly
+        uses: actions-rs/toolchain@v1
+        with:
+            toolchain: stable
+            override: true
+            components: rustfmt, clippy
+      - name: deal with rust cache
+        uses: Swatinem/rust-cache@v1
+      - name: cargo test
+        run: |
+          cargo test --no-fail-fast --verbose --features serde
+
+  cargo-test-mahimahi:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+
+      - name: Install latest nightly
+        uses: actions-rs/toolchain@v1
+        with:
+            toolchain: stable
+            override: true
+            components: rustfmt, clippy
+      - name: deal with rust cache
+        uses: Swatinem/rust-cache@v1
+      - name: cargo test
+        run: |
+          cargo test --no-fail-fast --verbose --features mahimahi --features serde
+          
+
+  cargo-test-truncated-normal:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+
+      - name: Install latest nightly
+        uses: actions-rs/toolchain@v1
+        with:
+            toolchain: stable
+            override: true
+            components: rustfmt, clippy
+      - name: deal with rust cache
+        uses: Swatinem/rust-cache@v1
+      - name: cargo test
+        run: |
+          cargo test --no-fail-fast --verbose --features truncated-normal --features serde

Cargo.toml

@@ -24,10 +24,12 @@ serde = { version = "1.0", features = ["derive"], optional = true }
 typetag = { version = "0.2.5", optional = true }
 humantime-serde = { version = "1.1.1", optional = true }
 human-bandwidth = { version = "0.1.1", optional = true }
+statrs = { version = "0.17.1", optional = true}
 
 [dev-dependencies]
 serde_json = "1.0"
 
+
 [features]
 default = ["model"]
 model = ["bw-model", "delay-model", "loss-model", "duplicate-model"]
@@ -38,7 +40,8 @@ duplicate-model = ["dep:dyn-clone"]
 serde = ["dep:serde", "dep:typetag", "bandwidth/serde"]
 mahimahi = ["dep:itertools"]
 human = ["serde", "dep:humantime-serde", "dep:human-bandwidth", "human-bandwidth/serde"]
-full = ["model", "mahimahi", "human"]
+full = ["model", "mahimahi", "human", "truncated-normal"]
+truncated-normal = ["statrs"]
 
 [package.metadata.docs.rs]
 all-features = true

src/lib.rs

@@ -144,9 +144,9 @@ pub type LossPattern = Vec<f64>;
 /// The meaning of the duplicate_pattern sequence is:
 ///
 /// - The probability on index 0 describes how likely a packet will be duplicated
-/// **if the previous packet was transmitted normally**.
+///   **if the previous packet was transmitted normally**.
 /// - The probability on index 1 describes how likely a packet will be duplicated
-/// **if the previous packet was duplicated**.
+///   **if the previous packet was duplicated**.
 /// - ...
 ///
 /// For example, if the duplicate_pattern is [0.8, 0.1], and packet 100 is not duplicated, then the
@@ -216,6 +216,8 @@ pub trait DuplicateTrace: Send {
 
 #[cfg(test)]
 mod test {
+    use model::TraceBwConfig;
+
     use self::model::bw::Forever;
 
     use super::*;
@@ -346,6 +348,48 @@ mod test {
         );
     }
 
+    #[test]
+    fn test_trace_bw() {
+        let mut trace_bw = TraceBwConfig::new()
+            .pattern(vec![
+                (
+                    Duration::from_millis(1),
+                    vec![
+                        Bandwidth::from_kbps(29123),
+                        Bandwidth::from_kbps(41242),
+                        Bandwidth::from_kbps(7395),
+                    ],
+                ),
+                (
+                    Duration::from_millis(2),
+                    vec![Bandwidth::from_mbps(1), Bandwidth::from_kbps(8542)],
+                ),
+            ])
+            .build();
+
+        assert_eq!(
+            trace_bw.next_bw(),
+            Some((Bandwidth::from_bps(29123000), Duration::from_millis(1)))
+        );
+        assert_eq!(
+            trace_bw.next_bw(),
+            Some((Bandwidth::from_bps(41242000), Duration::from_millis(1)))
+        );
+        assert_eq!(
+            trace_bw.next_bw(),
+            Some((Bandwidth::from_bps(7395000), Duration::from_millis(1)))
+        );
+        assert_eq!(
+            trace_bw.next_bw(),
+            Some((Bandwidth::from_bps(1000000), Duration::from_millis(2)))
+        );
+        assert_eq!(
+            trace_bw.next_bw(),
+            Some((Bandwidth::from_bps(8542000), Duration::from_millis(2)))
+        );
+        assert_eq!(trace_bw.next_bw(), None);
+    }
+
     #[test]
     #[cfg(feature = "serde")]
     fn test_model_serde() {