[PATCH v2 6/8] btrfs: extent-tree: Use btrfs_ref to refactor add_pinned_bytes()

[Qu Wenruo]

Since add_pinned_bytes() only needs to know if the extent is metadata
and if it's a chunk tree extent, btrfs_ref is a perfect match for it, as
we don't need various owner/level trick to determine extent type.

Signed-off-by: Qu Wenruo <w...@suse.com>
Reviewed-by: Nikolay Borisov <nbori...@suse.com>
---
 fs/btrfs/extent-tree.c | 26 ++++++++++----------------
 1 file changed, 10 insertions(+), 16 deletions(-)

diff --git a/fs/btrfs/extent-tree.c b/fs/btrfs/extent-tree.c
index 6e0706195279..83662e9bde0f 100644
--- a/fs/btrfs/extent-tree.c
+++ b/fs/btrfs/extent-tree.c
@@ -738,14 +738,15 @@ static struct btrfs_space_info *__find_space_info(struct 
btrfs_fs_info *info,
        return NULL;
 }
 
-static void add_pinned_bytes(struct btrfs_fs_info *fs_info, s64 num_bytes,
-                            bool metadata, u64 root_objectid)
+static void add_pinned_bytes(struct btrfs_fs_info *fs_info,
+                            struct btrfs_ref *ref)
 {
        struct btrfs_space_info *space_info;
+       s64 num_bytes = -ref->len;
        u64 flags;
 
-       if (metadata) {
-               if (root_objectid == BTRFS_CHUNK_TREE_OBJECTID)
+       if (ref->type == BTRFS_REF_METADATA) {
+               if (ref->tree_ref.root == BTRFS_CHUNK_TREE_OBJECTID)
                        flags = BTRFS_BLOCK_GROUP_SYSTEM;
                else
                        flags = BTRFS_BLOCK_GROUP_METADATA;
@@ -2052,11 +2053,8 @@ int btrfs_inc_extent_ref(struct btrfs_trans_handle 
*trans,
 
        btrfs_ref_tree_mod(fs_info, &generic_ref);
 
-       if (ret == 0 && old_ref_mod < 0 && new_ref_mod >= 0) {
-               bool metadata = owner < BTRFS_FIRST_FREE_OBJECTID;
-
-               add_pinned_bytes(fs_info, -num_bytes, metadata, root_objectid);
-       }
+       if (ret == 0 && old_ref_mod < 0 && new_ref_mod >= 0)
+               add_pinned_bytes(fs_info, &generic_ref);
 
        return ret;
 }
@@ -7054,8 +7052,7 @@ void btrfs_free_tree_block(struct btrfs_trans_handle 
*trans,
        }
 out:
        if (pin)
-               add_pinned_bytes(fs_info, buf->len, true,
-                                root->root_key.objectid);
+               add_pinned_bytes(fs_info, &generic_ref);
 
        if (last_ref) {
                /*
@@ -7106,11 +7103,8 @@ int btrfs_free_extent(struct btrfs_trans_handle *trans,
        if (root_objectid != BTRFS_TREE_LOG_OBJECTID)
                btrfs_ref_tree_mod(fs_info, &generic_ref);
 
-       if (ret == 0 && old_ref_mod >= 0 && new_ref_mod < 0) {
-               bool metadata = owner < BTRFS_FIRST_FREE_OBJECTID;
-
-               add_pinned_bytes(fs_info, num_bytes, metadata, root_objectid);
-       }
+       if (ret == 0 && old_ref_mod >= 0 && new_ref_mod < 0)
+               add_pinned_bytes(fs_info, &generic_ref);
 
        return ret;
 }
-- 
2.19.2