kble-tfsync 0.1.0

Virtual Harness Toolkit
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use anyhow::Result;
use bytes::BytesMut;
use futures::{SinkExt, StreamExt};
use kble_tfsync::AosTransferFrameCodec;
use tokio_util::codec::Decoder;
use tracing_subscriber::{prelude::*, EnvFilter};

#[tokio::main]
async fn main() -> Result<()> {
    tracing_subscriber::registry()
        .with(
            tracing_subscriber::fmt::layer()
                .with_ansi(false)
                .with_writer(std::io::stderr),
        )
        .with(EnvFilter::from_default_env())
        .init();
    let (mut tx, mut rx) = kble_socket::from_stdio().await;
    let mut buf = BytesMut::new();
    let mut codec = AosTransferFrameCodec::new();
    loop {
        let Some(chunk) = rx.next().await else {
            break;
        };
        buf.extend_from_slice(&chunk?);
        while let Some(frame) = codec.decode(&mut buf)? {
            tx.send(frame.into()).await?;
        }
    }
    Ok(())
}