Commit d022d18c authored by Ming Lei's avatar Ming Lei Committed by Martin K. Petersen

scsi: blk-mq: Add callbacks for storing & retrieving budget token

Since SCSI is the only driver which requires dispatch budget move the token
from struct request to struct scsi_cmnd.

Link: https://lore.kernel.org/r/20210122023317.687987-8-ming.lei@redhat.com
Cc: Omar Sandoval <osandov@fb.com>
Cc: Kashyap Desai <kashyap.desai@broadcom.com>
Cc: Sumanesh Samanta <sumanesh.samanta@broadcom.com>
Cc: Ewan D. Milne <emilne@redhat.com>
Cc: Hannes Reinecke <hare@suse.de>
Tested-by: default avatarSumanesh Samanta <sumanesh.samanta@broadcom.com>
Reviewed-by: default avatarHannes Reinecke <hare@suse.de>
Signed-off-by: default avatarMing Lei <ming.lei@redhat.com>
Signed-off-by: default avatarMartin K. Petersen <martin.petersen@oracle.com>
parent 2d13b1ea
...@@ -1642,6 +1642,20 @@ static bool scsi_mq_get_budget(struct request_queue *q) ...@@ -1642,6 +1642,20 @@ static bool scsi_mq_get_budget(struct request_queue *q)
return false; return false;
} }
static void scsi_mq_set_rq_budget_token(struct request *req, int token)
{
struct scsi_cmnd *cmd = blk_mq_rq_to_pdu(req);
cmd->budget_token = token;
}
static int scsi_mq_get_rq_budget_token(struct request *req)
{
struct scsi_cmnd *cmd = blk_mq_rq_to_pdu(req);
return cmd->budget_token;
}
static blk_status_t scsi_queue_rq(struct blk_mq_hw_ctx *hctx, static blk_status_t scsi_queue_rq(struct blk_mq_hw_ctx *hctx,
const struct blk_mq_queue_data *bd) const struct blk_mq_queue_data *bd)
{ {
...@@ -1856,6 +1870,8 @@ static const struct blk_mq_ops scsi_mq_ops_no_commit = { ...@@ -1856,6 +1870,8 @@ static const struct blk_mq_ops scsi_mq_ops_no_commit = {
.cleanup_rq = scsi_cleanup_rq, .cleanup_rq = scsi_cleanup_rq,
.busy = scsi_mq_lld_busy, .busy = scsi_mq_lld_busy,
.map_queues = scsi_map_queues, .map_queues = scsi_map_queues,
.set_rq_budget_token = scsi_mq_set_rq_budget_token,
.get_rq_budget_token = scsi_mq_get_rq_budget_token,
}; };
...@@ -1884,6 +1900,8 @@ static const struct blk_mq_ops scsi_mq_ops = { ...@@ -1884,6 +1900,8 @@ static const struct blk_mq_ops scsi_mq_ops = {
.cleanup_rq = scsi_cleanup_rq, .cleanup_rq = scsi_cleanup_rq,
.busy = scsi_mq_lld_busy, .busy = scsi_mq_lld_busy,
.map_queues = scsi_map_queues, .map_queues = scsi_map_queues,
.set_rq_budget_token = scsi_mq_set_rq_budget_token,
.get_rq_budget_token = scsi_mq_get_rq_budget_token,
}; };
struct request_queue *scsi_mq_alloc_queue(struct scsi_device *sdev) struct request_queue *scsi_mq_alloc_queue(struct scsi_device *sdev)
......
...@@ -313,6 +313,15 @@ struct blk_mq_ops { ...@@ -313,6 +313,15 @@ struct blk_mq_ops {
*/ */
void (*put_budget)(struct request_queue *); void (*put_budget)(struct request_queue *);
/*
* @set_rq_budget_toekn: store rq's budget token
*/
void (*set_rq_budget_token)(struct request *, int);
/*
* @get_rq_budget_toekn: retrieve rq's budget token
*/
int (*get_rq_budget_token)(struct request *);
/** /**
* @timeout: Called on request timeout. * @timeout: Called on request timeout.
*/ */
......
...@@ -75,6 +75,8 @@ struct scsi_cmnd { ...@@ -75,6 +75,8 @@ struct scsi_cmnd {
int eh_eflags; /* Used by error handlr */ int eh_eflags; /* Used by error handlr */
int budget_token;
/* /*
* This is set to jiffies as it was when the command was first * This is set to jiffies as it was when the command was first
* allocated. It is used to time how long the command has * allocated. It is used to time how long the command has
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment