xref: /openbmc/linux/io_uring/alloc_cache.h (revision 278d3ba6)
1 #ifndef IOU_ALLOC_CACHE_H
2 #define IOU_ALLOC_CACHE_H
3 
4 /*
5  * Don't allow the cache to grow beyond this size.
6  */
7 #define IO_ALLOC_CACHE_MAX	512
8 
9 struct io_cache_entry {
10 	struct hlist_node	node;
11 };
12 
13 static inline bool io_alloc_cache_put(struct io_alloc_cache *cache,
14 				      struct io_cache_entry *entry)
15 {
16 	if (cache->nr_cached < IO_ALLOC_CACHE_MAX) {
17 		cache->nr_cached++;
18 		hlist_add_head(&entry->node, &cache->list);
19 		return true;
20 	}
21 	return false;
22 }
23 
24 static inline struct io_cache_entry *io_alloc_cache_get(struct io_alloc_cache *cache)
25 {
26 	if (!hlist_empty(&cache->list)) {
27 		struct hlist_node *node = cache->list.first;
28 
29 		hlist_del(node);
30 		return container_of(node, struct io_cache_entry, node);
31 	}
32 
33 	return NULL;
34 }
35 
36 static inline void io_alloc_cache_init(struct io_alloc_cache *cache)
37 {
38 	INIT_HLIST_HEAD(&cache->list);
39 	cache->nr_cached = 0;
40 }
41 
42 static inline void io_alloc_cache_free(struct io_alloc_cache *cache,
43 					void (*free)(struct io_cache_entry *))
44 {
45 	while (!hlist_empty(&cache->list)) {
46 		struct hlist_node *node = cache->list.first;
47 
48 		hlist_del(node);
49 		free(container_of(node, struct io_cache_entry, node));
50 	}
51 	cache->nr_cached = 0;
52 }
53 #endif
54