[PATCH V2 3/5] loop: move command blkcg/memcg initialization into loop_queue_work

[Ming Lei]

Move loop command blkcg/memcg initialization into loop_queue_work,
and prepare for supporting to handle loop io command by IOCB_NOWAIT.

Signed-off-by: Ming Lei <ming....@redhat.com>
---
 drivers/block/loop.c | 32 +++++++++++++++++---------------
 1 file changed, 17 insertions(+), 15 deletions(-)

diff --git a/drivers/block/loop.c b/drivers/block/loop.c
index 3f921d9d6de6..81f6ba9bc911 100644
--- a/drivers/block/loop.c
+++ b/drivers/block/loop.c
@@ -878,11 +878,28 @@ static inline int queue_on_root_worker(struct 
cgroup_subsys_state *css)
 
 static void loop_queue_work(struct loop_device *lo, struct loop_cmd *cmd)
 {
+       struct request __maybe_unused *rq = blk_mq_rq_from_pdu(cmd);
        struct rb_node **node, *parent = NULL;
        struct loop_worker *cur_worker, *worker = NULL;
        struct work_struct *work;
        struct list_head *cmd_list;
 
+       /* always use the first bio's css */
+       cmd->blkcg_css = NULL;
+       cmd->memcg_css = NULL;
+#ifdef CONFIG_BLK_CGROUP
+       if (rq->bio) {
+               cmd->blkcg_css = bio_blkcg_css(rq->bio);
+#ifdef CONFIG_MEMCG
+               if (cmd->blkcg_css) {
+                       cmd->memcg_css =
+                               cgroup_get_e_css(cmd->blkcg_css->cgroup,
+                                               &memory_cgrp_subsys);
+               }
+#endif
+       }
+#endif
+
        spin_lock_irq(&lo->lo_work_lock);
 
        if (queue_on_root_worker(cmd->blkcg_css))
@@ -1920,21 +1937,6 @@ static blk_status_t loop_queue_rq(struct blk_mq_hw_ctx 
*hctx,
                break;
        }
 
-       /* always use the first bio's css */
-       cmd->blkcg_css = NULL;
-       cmd->memcg_css = NULL;
-#ifdef CONFIG_BLK_CGROUP
-       if (rq->bio) {
-               cmd->blkcg_css = bio_blkcg_css(rq->bio);
-#ifdef CONFIG_MEMCG
-               if (cmd->blkcg_css) {
-                       cmd->memcg_css =
-                               cgroup_get_e_css(cmd->blkcg_css->cgroup,
-                                               &memory_cgrp_subsys);
-               }
-#endif
-       }
-#endif
        loop_queue_work(lo, cmd);
 
        return BLK_STS_OK;
-- 
2.47.0