commitquery: Reorganize

1 files changed, 22 insertions, 0 deletions

diff --git a/commitquery/query_set_clear_marks.go b/commitquery/query_set_clear_marks.go
new file mode 100644
index 00000000..b9619338
--- /dev/null
+++ b/commitquery/query_set_clear_marks.go
@@ -0,0 +1,22 @@
+package commitquery
+
+// setMarks ORs one set of mark bits into one internal node.
+func (query *query) setMarks(idx nodeIndex, bits markBits) {
+	newBits := bits &^ query.nodes[idx].marks
+	if newBits == 0 {
+		return
+	}
+
+	query.trackTouched(idx)
+	query.nodes[idx].marks |= bits
+}
+
+// clearMarks removes one set of mark bits from one internal node.
+func (query *query) clearMarks(idx nodeIndex, bits markBits) {
+	if query.nodes[idx].marks&bits == 0 {
+		return
+	}
+
+	query.trackTouched(idx)
+	query.nodes[idx].marks &^= bits
+}