use criterion::{black_box, criterion_group, criterion_main, Criterion};

fn criterion_benchmark(c: &mut Criterion) {

    c.bench_function("learn pub decode", |b| {
        let example = vec![0; 65535];
        let mut decode = Parser::new();
        let mut buf_example = Vec::new();
        buf_example.extend_from_slice(b"PUB FOO 65535\r\n");
        buf_example.extend_from_slice(example.as_slice());
        buf_example.extend_from_slice(b"\r\n");

        b.iter(|| {
            let result = decode.parse(buf_example.as_slice());

            assert!(result.is_ok());
            if let Ok(r) = result {
                match r.0 {
                    ParseResult::Pub(p) => {
                        assert_eq!(p.subject, "FOO");
                        assert_eq!(p.size_buf, "65535");
                        assert_eq!(p.msg, example.as_slice());
                    }
                    _ => panic!("message parse error"),
                }
            } else {
                panic!("message parse error");
            }

            decode.clear_msg_buf();
        });
    });

    c.bench_function("learn sub decode", |b| {
        let mut decode = Parser::new();

        b.iter(|| {
            // 忽略订阅队列
            let result = decode.parse(b"SUB asdfasd sdfaf\n");

            assert!(result.is_ok());

            if let Ok(r) = result {
                match r.0 {
                    ParseResult::Sub(sub) => {
                        assert_eq!(sub.subject, "asdfasd");
                        assert_eq!(sub.sid, "sdfaf");
                        assert_eq!(sub.queue, None);
                    }
                    _ => panic!("message parse error"),
                }
            } else {
                panic!("message parse error");
            }

            decode.clear_msg_buf();
        });
    });

}

criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);

    c.bench_function("pub decode", |b| {
        let example = vec![0; 65535];
        let mut decode = Decode::new(65535 + 15);
        let mut buf_example = Vec::new();
        buf_example.extend_from_slice(b"PUB FOO 65535\r\n");
        buf_example.extend_from_slice(example.as_slice());
        buf_example.extend_from_slice(b"\r\n");

        b.iter(|| {
            decode.set_buff(buf_example.as_slice());
            let result = decode.decode();

            if let Ok(Poll::Ready(Message::Pub(subject, reply, content))) = result {
                assert_eq!(subject, "FOO");
                assert_eq!(reply, None);
                assert_eq!(content.as_bytes(), example.as_slice());
            } else {
                panic!("message parse error");
            }

            decode.reset();
        });
    });

    c.bench_function("sub decode", |b| {
        let mut decode = Decode::new(512);

        b.iter(|| {
            // 忽略订阅队列
            decode.set_buff(b"SUB asdfasd sdfaf\n");

            let result = decode.decode();

            if let Ok(Poll::Ready(Message::Sub(subject, group, sid))) = result {
                assert_eq!(subject, "asdfasd");
                assert!(group.is_none());
                assert_eq!(sid.as_bytes(), b"sdfaf");
            } else {
                panic!("message parse error");
            }

            decode.reset();
        });
    });