btrfs_release_block_group(block_group, delalloc);
 }
 
+static void found_extent_clustered(struct find_free_extent_ctl *ffe_ctl,
+                                  struct btrfs_key *ins)
+{
+       struct btrfs_free_cluster *last_ptr = ffe_ctl->last_ptr;
+
+       if (!ffe_ctl->use_cluster && last_ptr) {
+               spin_lock(&last_ptr->lock);
+               last_ptr->window_start = ins->objectid;
+               spin_unlock(&last_ptr->lock);
+       }
+}
+
+static void found_extent(struct find_free_extent_ctl *ffe_ctl,
+                        struct btrfs_key *ins)
+{
+       switch (ffe_ctl->policy) {
+       case BTRFS_EXTENT_ALLOC_CLUSTERED:
+               found_extent_clustered(ffe_ctl, ins);
+               break;
+       default:
+               BUG();
+       }
+}
+
 /*
  * Return >0 means caller needs to re-search for free extent
  * Return 0 means we have the needed free extent.
                return 1;
 
        if (ins->objectid) {
-               if (!use_cluster && last_ptr) {
-                       spin_lock(&last_ptr->lock);
-                       last_ptr->window_start = ins->objectid;
-                       spin_unlock(&last_ptr->lock);
-               }
+               found_extent(ffe_ctl, ins);
                return 0;
        }