openpql-runner 0.1.4

A high-performance Rust implementation of Poker Query Language (PQL), enabling SQL-like queries for poker analysis and calculations. This project is a spiritual successor to the original Java implementation developed by Odds Oracle.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

use super::*;

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum VmBinOpLogic {
    And,
    Or,
}

impl VmBinOpLogic {
    #[allow(clippy::unnecessary_wraps)]
    pub fn execute(self, ctx: &mut VmExecContext) -> Result<(), PQLErrorKind> {
        let rhs = ctx.stack.downcast_pop::<PQLBoolean>();
        let lhs = ctx.stack.downcast_pop::<PQLBoolean>();

        ctx.stack.push(
            match self {
                Self::And => lhs && rhs,
                Self::Or => lhs || rhs,
            }
            .into(),
        );

        Ok(())
    }

    #[allow(clippy::unused_self)]
    pub fn resolve_type(
        self,
        lhs_type: PQLType,
        rhs_type: PQLType,
    ) -> Result<PQLType, PQLErrorKind> {
        if lhs_type == PQLType::BOOLEAN && rhs_type == PQLType::BOOLEAN {
            Ok(PQLType::BOOLEAN)
        } else {
            Err(PQLErrorKind::LogicalOperationUnsupported {
                lhs: lhs_type,
                rhs: rhs_type,
            })
        }
    }
}

#[cfg(test)]
#[cfg_attr(coverage_nightly, coverage(off))]
mod tests {
    use super::*;

    fn assert_logic<I, T>(vals: I, op: VmBinOpLogic, expected: bool)
    where
        I: IntoIterator<Item = T>,
        VmStackValue: From<T>,
    {
        let mut ctx = VmExecContext::default();
        for v in vals {
            ctx.stack.push(v.into());
        }

        let ins = VmInstruction::BinOp(op.into());
        ins.execute(&mut ctx).unwrap();

        assert_eq!(ctx.stack.pop().unwrap(), expected.into());
    }

    #[test]
    fn test_and() {
        assert_logic([true, true], VmBinOpLogic::And, true);
        assert_logic([true, false], VmBinOpLogic::And, false);
        assert_logic([false, true], VmBinOpLogic::And, false);
        assert_logic([false, false], VmBinOpLogic::And, false);
    }

    #[test]
    fn test_or() {
        assert_logic([true, true], VmBinOpLogic::Or, true);
        assert_logic([true, false], VmBinOpLogic::Or, true);
        assert_logic([false, true], VmBinOpLogic::Or, true);
        assert_logic([false, false], VmBinOpLogic::Or, false);
    }

    #[test]
    fn test_resolve_type_err() {
        assert_eq!(
            VmBinOpLogic::And.resolve_type(PQLType::LONG, PQLType::BOOLEAN),
            Err(PQLErrorKind::LogicalOperationUnsupported {
                lhs: PQLType::LONG,
                rhs: PQLType::BOOLEAN,
            }),
        );
    }
}