crush: allow "osd crush set-device-class" to create class automatically

Signed-off-by: xie xingguo <xie.xingguo@zte.com.cn>

diff --git a/src/crush/CrushWrapper.cc b/src/crush/CrushWrapper.cc
index 6a0017df96d7cc3b7150a29a6ca38b2fbee4fdf8..1f4b3f5219348d913d5718039362b75ee51fdd0c 100644 (file)
--- a/src/crush/CrushWrapper.cc
+++ b/src/crush/CrushWrapper.cc
@@ -1562,11 +1562,7 @@ int CrushWrapper::update_device_class(int id,
                                       const string& name,
                                       ostream *ss)
 {
-  int class_id = get_class_id(class_name);
-  if (class_id < 0) {
-    *ss << "class " << class_name << " does not exist";
-    return -ENOENT;
-  }
+  int class_id = get_or_create_class_id(class_name);
   if (id < 0) {
     *ss << name << " id " << id << " is negative";
     return -EINVAL;