mirror of
https://github.com/cberner/raptorq.git
synced 2024-07-01 03:01:48 +00:00
43 lines
1.5 KiB
Rust
43 lines
1.5 KiB
Rust
use rand::Rng;
|
|
use rand::seq::SliceRandom;
|
|
use raptorq::{Decoder, Encoder, EncodingPacket};
|
|
|
|
fn main() {
|
|
// Generate some random data to send
|
|
let mut data: Vec<u8> = vec![0; 10_000];
|
|
for i in 0..data.len() {
|
|
data[i] = rand::thread_rng().gen();
|
|
}
|
|
|
|
// Create the Encoder, with an MTU of 1400 (common for Ethernet)
|
|
let encoder = Encoder::with_defaults(&data, 1400);
|
|
|
|
// Perform the encoding, and serialize to Vec<u8> for transmission
|
|
let mut packets: Vec<Vec<u8>> = encoder.get_encoded_packets(15).iter()
|
|
.map(|packet| packet.serialize())
|
|
.collect();
|
|
|
|
// Here we simulate losing 10 of the packets randomly. Normally, you would send them over
|
|
// (potentially lossy) network here.
|
|
packets.shuffle(&mut rand::thread_rng());
|
|
// Erase 10 packets at random
|
|
let length = packets.len();
|
|
packets.truncate(length - 10);
|
|
|
|
// The Decoder MUST be constructed with the configuration of the Encoder.
|
|
// The ObjectTransmissionInformation configuration should be transmitted over a reliable
|
|
// channel
|
|
let mut decoder = Decoder::new(encoder.get_config());
|
|
|
|
// Perform the decoding
|
|
let mut result = None;
|
|
while !packets.is_empty() {
|
|
result = decoder.decode(EncodingPacket::deserialize(&packets.pop().unwrap()));
|
|
if result != None {
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Check that even though some of the data was lost we are able to reconstruct the original message
|
|
assert_eq!(result.unwrap(), data);
|
|
} |