librbd: Add set_op_flags in AioWrite to set op_flags for rbd write.

Signed-off-by: Jianpeng Ma <jianpeng.ma@intel.com>

diff --git a/src/librbd/AioRequest.h b/src/librbd/AioRequest.h
index 31cfee9a2965413f5d0f7ea7874786c6b27674e7..3dbf84c64bf742bf80afbfea95654439e79d7656 100644 (file)
--- a/src/librbd/AioRequest.h
+++ b/src/librbd/AioRequest.h
@@ -165,6 +165,9 @@ namespace librbd {
     }
     virtual ~AioWrite() {}
 
+    void set_op_flags(int op_flags) {
+      m_write.set_op_flags2(op_flags);
+    }
   protected:
     virtual void add_copyup_ops() {
       add_write_ops(m_copyup);

diff --git a/src/librbd/internal.cc b/src/librbd/internal.cc
index 2e63941714910753c9d563f7be266035980acf53..2ff935237a06a1aded08b7fcc61bee8dd2b35142 100644 (file)
--- a/src/librbd/internal.cc
+++ b/src/librbd/internal.cc
@@ -3019,6 +3019,8 @@ reprotect_and_return_err:
                                     objectx, object_overlap,
                                     bl, snapc, snap_id, req_comp);
        c->add_request();
+
+       req->set_op_flags(op_flags);
        r = req->send();
        if (r < 0)
          goto done;