use std::{thread::current, time::Instant};

use channel_server::{prelude::*, Response};

#[derive(Debug, Serialize, Deserialize)]
struct User {
    name: String,
}

#[handler]
fn hello(name: String) -> String {
    let res = format!("hello: {}", name);
    println!("tid:{:?} {}", current().id(), &res);
    res
}

#[handler]
fn hello_json(user: ReqParam<User>, json: Json<String>, data: Data<&i32>) -> String {
    let res = format!("user: {:?}, json: {}, data: {}", user, json.0, data.0);
    println!("tid:{:?} {}", current().id(), &res);
    res
}

#[test]
fn test_basic() -> Result<(), ChannelError> {
    let uri1 = "/hello";
    let uri2 = "/hello_json";
    let topic_uri = "/topic/video";

    let ep = Route::new().at(uri1, hello).at(uri2, hello_json).data(1);

    let (mut client, topic) = ChannelService::start(ep);

    println!("执行开始");
    let start_time = Instant::now();

    // 订阅主题, 用于接收 Topic 发布的数据
    client.subject(topic_uri);

    // 发送请求
    client.req_with_body(uri1, Body::from_string("this is a string".to_string()))?;

    // 发送请求
    client.req_with_param_body(
        uri2,
        Param::from_obj(User {
            name: "maxu".to_string(),
        }),
        Body::from_string(serde_json::to_string("this is a json string").unwrap()),
    )?;

    let mut res1_ok = false;
    let mut res2_ok = false;
    let mut topic_ok = false;

    // 发布两条数据
    topic.publish(Response::topic(topic_uri).body("hello".as_bytes().into()));
    topic.publish(Response::topic(topic_uri).body("hello world".as_bytes().into()));

    loop {
        // 运行队列, 接收 response
        if client.run_once() {
            // 查询执行结果
            if let Some(res) = client.fetch(&uri1) {
                println!("{:?}", res);
                // 如果执行成功, 清除响应
                if res.is_ok() {
                    client.clean(&uri1);
                    res1_ok = true;
                }
            }

            if let Some(res) = client.fetch(&uri2) {
                println!("{:?}", res);
                // 如果执行成功, 清除响应
                if res.is_ok() {
                    client.clean(&uri2);
                    res2_ok = true;
                }
            }

            if let Some(data) = client.fetch_topic(topic_uri) {
                for d in data {
                    println!("\t{:?}", &d);
                }
                topic_ok = true;
            }

            if res1_ok && res2_ok && topic_ok {
                println!("执行完成");
                break;
            }
        }
        std::thread::yield_now();
    }

    println!("执行时间: {:?}", start_time.elapsed());

    Ok(())
}