namerena-rs/src/main.rs

mod data_struct;
mod name;

use base16384::Base16384Utf8;
use clap::Parser;
use tracing::info;

/// 根据 u64 生成对应的 name
/// 转换成 base 16384
/// 禁用:
/// U00 ~ U1F ，换行，制表符 等
/// ? , 问号
/// U2000 - U202F , unicode特殊空格 等
/// 不可以空格开头
pub fn gen_name(id: u64) -> String {
    // u64 -> [u8]
    let id_bytes = id.to_be_bytes();
    Base16384Utf8::encode(id_bytes.as_slice())
}

pub fn show_name(namer: &name::Namer) -> String {
    // var attributeNames = ["HP", "攻", "防", "速", "敏", "魔", "抗", "智"]
    format!(
        "HP|{} 攻|{} 防|{} 速|{} 敏|{} 魔|{} 抗|{} 智|{} 八围:{}",
        namer.name_prop[7],
        namer.name_prop[0],
        namer.name_prop[1],
        namer.name_prop[2],
        namer.name_prop[3],
        namer.name_prop[4],
        namer.name_prop[5],
        namer.name_prop[6],
        namer.get_property()
    )
}

#[allow(non_upper_case_globals)]
const allow_d: u32 = 10;

#[allow(non_upper_case_globals)]
const report_interval: u32 = 1_000_0000;

#[derive(Parser, Debug)]
pub struct Command {
    #[arg(long, default_value_t = 0)]
    pub start: u64,
    #[arg(long, default_value_t = u64::MAX)]
    pub end: u64,
    #[arg(long, short = 't', default_value_t = 10)]
    pub thread_count: u32,
    #[arg(long, default_value_t = 750)]
    pub top: u32,
    #[arg(long)]
    pub team: String,
}

fn cacl(max: u64, step: usize, top: u32, id: u64, team: &String) {
    let mut start_time = std::time::Instant::now();
    let mut k: u64 = 0;
    let mut top = top;
    for i in (0+id..max).step_by(step) {
        let name = gen_name(i as u64);
        // let full_name = format!("{}@shenjack", name);
        // let namer = name::Namer::new(&full_name);
        let namer = name::Namer::new_raw(name, team);
        if let Some(namer) = namer {
            let prop = namer.get_property();
            if (prop + allow_d as f32) > top as f32 {
                if prop > top as f32 {
                    info!("新的最高属性 {}", prop);
                    top = prop as u32;
                }
                let name = gen_name(i as u64);
                let full_name = format!("{}@{}", name, team);
                info!("{:>20}|{}|{}", i, full_name, show_name(&namer));
            }
        }
        k += 1;
        if k > report_interval as u64 {
            let now = std::time::Instant::now();
            info!("{:>20} {} {}/s", i, id, k / now.duration_since(start_time).as_secs());
            start_time = now;
            k = 0;
        }
    }
}

fn main() {
    tracing_subscriber::fmt::init();
    let mut cli_arg = Command::parse();

    // 将数据量处理成可被 thread_count 整除
    let left = cli_arg.start % cli_arg.thread_count as u64;
    cli_arg.end += left;

    
    let mut n = 0;
    let mut threads = Vec::with_capacity(cli_arg.thread_count as usize);
    for i in 0..cli_arg.thread_count {
        let top = cli_arg.top;
        let max = cli_arg.end / cli_arg.thread_count as u64;
        n += 1;
        let thread_name = format!("thread_{}", i);
        let thread_count = cli_arg.thread_count;
        let team = cli_arg.team.clone();
        threads.push(std::thread::spawn(move || {
            info!("线程 {} 开始计算", thread_name);
            cacl(max as u64, thread_count as usize, top as u32, n as u64, &team);
            info!("线程 {} 结束计算", thread_name);
        }));
    }
    info!("开始计算");

    for t in threads {
        t.join().unwrap();
    }
}