crush/CrushWrapper: rule_has_take

Signed-off-by: Sage Weil <sage@redhat.com>
ceph · Sep 6, 2017 · ef140de · ef140de
1 parent 60b9cfa
commit ef140de
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/crush/CrushWrapper.h b/src/crush/CrushWrapper.h
@@ -976,6 +976,17 @@ class CrushWrapper {
       return true;
     return false;
   }
+  bool rule_has_take(unsigned ruleno, int take) const {
+    if (!crush) return false;
+    crush_rule *rule = get_rule(ruleno);
+    for (unsigned i = 0; i < rule->len; ++i) {
+      if (rule->steps[i].op == CRUSH_RULE_TAKE &&
+	  rule->steps[i].arg1 == take) {
+	return true;
+      }
+    }
+    return false;
+  }
   int get_rule_len(unsigned ruleno) const {
     crush_rule *r = get_rule(ruleno);
     if (IS_ERR(r)) return PTR_ERR(r);