Commit 688c8b92 authored by Kefeng Wang's avatar Kefeng Wang Committed by Jens Axboe
Browse files

blk-cgroup: use group allocation/free of per-cpu counters API 



Use group allocation/free of per-cpu counters api to accelerate
blkg_rwstat_init/exit() and simplify code.

Signed-off-by: default avatarKefeng Wang <wangkefeng.wang@huawei.com>
Link: https://lore.kernel.org/r/20240325035955.50019-1-wangkefeng.wang@huawei.com


Signed-off-by: default avatarJens Axboe <axboe@kernel.dk>
parent fa1af65b

block/blk-cgroup-rwstat.c

+6 −12
Original line number Diff line number Diff line
@@ -9,25 +9,19 @@ int blkg_rwstat_init(struct blkg_rwstat *rwstat, gfp_t gfp) 
{

	int i, ret;



	for (i = 0; i < BLKG_RWSTAT_NR; i++) {

		ret = percpu_counter_init(&rwstat->cpu_cnt[i], 0, gfp);

		if (ret) {

			while (--i >= 0)

				percpu_counter_destroy(&rwstat->cpu_cnt[i]);

	ret = percpu_counter_init_many(rwstat->cpu_cnt, 0, gfp, BLKG_RWSTAT_NR);

	if (ret)

		return ret;

		}



	for (i = 0; i < BLKG_RWSTAT_NR; i++)

		atomic64_set(&rwstat->aux_cnt[i], 0);

	}

	return 0;

}

EXPORT_SYMBOL_GPL(blkg_rwstat_init);



void blkg_rwstat_exit(struct blkg_rwstat *rwstat)

{

	int i;



	for (i = 0; i < BLKG_RWSTAT_NR; i++)

		percpu_counter_destroy(&rwstat->cpu_cnt[i]);

	percpu_counter_destroy_many(rwstat->cpu_cnt, BLKG_RWSTAT_NR);

}

EXPORT_SYMBOL_GPL(blkg_rwstat_exit);