xet-runtime 1.5.2

Async runtime, configuration, logging, and utility infrastructure for the Hugging Face Xet client tools.
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
32
33
34
35
36
37
38

use std::env;
use std::time::Duration;

use tracing::info;
use xet_runtime::logging::{LogDirConfig, LoggingConfig, LoggingMode, init, wait_for_log_directory_cleanup};

fn main() {
    let args: Vec<String> = env::args().collect();
    if args.len() != 3 {
        eprintln!("Usage: {} <log_directory> <num_lines>", args[0]);
        std::process::exit(1);
    }

    let log_directory = &args[1];
    let num_lines: usize = args[2].parse().expect("num_lines must be a number");

    // Initialize logging to the specified directory
    let config = LoggingConfig {
        logging_mode: LoggingMode::Directory(log_directory.into()),
        use_json: true,
        enable_log_dir_cleanup: true,
        version: "test".to_string(),
        log_dir_config: LogDirConfig::default(),
    };

    init(config);

    // Generate log messages with 5ms delay between each
    for i in 0..num_lines {
        info!("Test log message number {} - this is a dummy log message for testing purposes", i + 1);

        // Wait for 50 microseconds between each log message just to spread it out a little...
        std::thread::sleep(Duration::from_micros(50));
    }

    // Wait for background cleanup to complete before exiting
    wait_for_log_directory_cleanup();
}