Function custom_key_fn

nodedb/src/control/trigger/batch/partition.rs:181–206

()

    #[test]
    fn custom_key_fn() {
        let rows = vec![
            row_with_field("r1", "score", "10"),
            row_with_field("r2", "score", "90"),
            row_with_field("r3", "score", "50"),
        ];

        match partition_batch(&rows, 32, |row| {
            let score: i64 = row
                .new_fields()
                .and_then(|m| m.get("score"))
                .and_then(|v| v.as_str())
                .and_then(|s| s.parse().ok())
                .unwrap_or(0);
            if score >= 50 {
                "high".into()
            } else {
                "low".into()
            }
        }) {
            PartitionResult::Partitions(parts) => {
                assert_eq!(parts.len(), 2);
            }
            PartitionResult::CircuitBroken { .. } => panic!("unexpected"),
        }
    }
}