123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177 |
- #![feature(test)]
- use std::collections::BTreeSet;
- use synapse::push::{
- evaluator::PushRuleEvaluator, Condition, EventMatchCondition, FilteredPushRules, PushRules,
- };
- use test::Bencher;
- extern crate test;
- #[bench]
- fn bench_match_exact(b: &mut Bencher) {
- let flattened_keys = [
- ("type".to_string(), "m.text".to_string()),
- ("room_id".to_string(), "!room:server".to_string()),
- ("content.body".to_string(), "test message".to_string()),
- ]
- .into_iter()
- .collect();
- let eval = PushRuleEvaluator::py_new(
- flattened_keys,
- false,
- BTreeSet::new(),
- false,
- 10,
- Some(0),
- Default::default(),
- Default::default(),
- true,
- vec![],
- false,
- )
- .unwrap();
- let condition = Condition::Known(synapse::push::KnownCondition::EventMatch(
- EventMatchCondition {
- key: "room_id".into(),
- pattern: Some("!room:server".into()),
- pattern_type: None,
- },
- ));
- let matched = eval.match_condition(&condition, None, None).unwrap();
- assert!(matched, "Didn't match");
- b.iter(|| eval.match_condition(&condition, None, None).unwrap());
- }
- #[bench]
- fn bench_match_word(b: &mut Bencher) {
- let flattened_keys = [
- ("type".to_string(), "m.text".to_string()),
- ("room_id".to_string(), "!room:server".to_string()),
- ("content.body".to_string(), "test message".to_string()),
- ]
- .into_iter()
- .collect();
- let eval = PushRuleEvaluator::py_new(
- flattened_keys,
- false,
- BTreeSet::new(),
- false,
- 10,
- Some(0),
- Default::default(),
- Default::default(),
- true,
- vec![],
- false,
- )
- .unwrap();
- let condition = Condition::Known(synapse::push::KnownCondition::EventMatch(
- EventMatchCondition {
- key: "content.body".into(),
- pattern: Some("test".into()),
- pattern_type: None,
- },
- ));
- let matched = eval.match_condition(&condition, None, None).unwrap();
- assert!(matched, "Didn't match");
- b.iter(|| eval.match_condition(&condition, None, None).unwrap());
- }
- #[bench]
- fn bench_match_word_miss(b: &mut Bencher) {
- let flattened_keys = [
- ("type".to_string(), "m.text".to_string()),
- ("room_id".to_string(), "!room:server".to_string()),
- ("content.body".to_string(), "test message".to_string()),
- ]
- .into_iter()
- .collect();
- let eval = PushRuleEvaluator::py_new(
- flattened_keys,
- false,
- BTreeSet::new(),
- false,
- 10,
- Some(0),
- Default::default(),
- Default::default(),
- true,
- vec![],
- false,
- )
- .unwrap();
- let condition = Condition::Known(synapse::push::KnownCondition::EventMatch(
- EventMatchCondition {
- key: "content.body".into(),
- pattern: Some("foobar".into()),
- pattern_type: None,
- },
- ));
- let matched = eval.match_condition(&condition, None, None).unwrap();
- assert!(!matched, "Didn't match");
- b.iter(|| eval.match_condition(&condition, None, None).unwrap());
- }
- #[bench]
- fn bench_eval_message(b: &mut Bencher) {
- let flattened_keys = [
- ("type".to_string(), "m.text".to_string()),
- ("room_id".to_string(), "!room:server".to_string()),
- ("content.body".to_string(), "test message".to_string()),
- ]
- .into_iter()
- .collect();
- let eval = PushRuleEvaluator::py_new(
- flattened_keys,
- false,
- BTreeSet::new(),
- false,
- 10,
- Some(0),
- Default::default(),
- Default::default(),
- true,
- vec![],
- false,
- )
- .unwrap();
- let rules = FilteredPushRules::py_new(
- PushRules::new(Vec::new()),
- Default::default(),
- false,
- false,
- false,
- false,
- false,
- );
- b.iter(|| eval.run(&rules, Some("bob"), Some("person")));
- }
|