Utility methodsv0.0.1

1 files changed, 55 insertions, 0 deletions

diff --git a/model/condition.go b/model/condition.go
new file mode 100644
index 0000000..0b558b4
--- /dev/null
+++ b/model/condition.go
@@ -0,0 +1,55 @@
+package model
+
+// Compare current node with another node
+func (n *Node) Compare(op Operand, node *Node) bool {
+	switch op {
+	case OpEq:
+		return n.Value() == node.Value()
+	case OpNeq:
+		return n.Value() != node.Value()
+	case OpLess:
+		return less(n, node)
+	case OpGt:
+		return less(node, n)
+	case OpLessEq:
+		return less(n, node) || n.Value() == node.Value()
+	case OpGtEq:
+		return less(node, n) || n.Value() == node.Value()
+	case OpIn:
+		if n.Type != ArrayNode {
+			return false
+		}
+		for _, v := range n.arrayValue {
+			if v.Value() == node.Value() {
+				return true
+			}
+		}
+	}
+	return false
+}
+
+func less(n1 *Node, n2 *Node) bool {
+	if n1.Type != n2.Type {
+		return false
+	}
+	switch n1.Type {
+	case NumberNode:
+		return n1.numberValue < n2.numberValue
+	case StringNode:
+		return n1.stringValue < n2.stringValue
+	default:
+		return false
+	}
+}
+
+type Operand int
+
+const (
+	OpEq Operand = iota
+	OpNeq
+	OpLess
+	OpLessEq
+	OpGt
+	OpGtEq
+	OpIn
+)