idr: Return the deleted entry from idr_remove

It is a relatively common idiom (8 instances) to first look up an IDR
entry, and then remove it from the tree if it is found, possibly doing
further operations upon the entry afterwards.  If we change idr_remove()
to return the removed object, all of these users can save themselves a
walk of the IDR tree.

Signed-off-by: Matthew Wilcox <mawilcox@microsoft.com>

1 files changed, 1 insertions, 3 deletions

diff --git a/drivers/target/target_core_user.c b/drivers/target/target_core_user.c
index 8041710b6972..18f0ec2e1f9c 100644
--- a/drivers/target/target_core_user.c
+++ b/drivers/target/target_core_user.c
@@ -642,9 +642,7 @@ static unsigned int tcmu_handle_completions(struct tcmu_dev *udev)
                 WARN_ON(tcmu_hdr_get_op(entry->hdr.len_op) != TCMU_OP_CMD);
 
                 spin_lock(&udev->commands_lock);
-                cmd = idr_find(&udev->commands, entry->hdr.cmd_id);
-                if (cmd)
-                        idr_remove(&udev->commands, cmd->cmd_id);
+                cmd = idr_remove(&udev->commands, entry->hdr.cmd_id);
                 spin_unlock(&udev->commands_lock);
 
                 if (!cmd) {