xref: /aosp_15_r20/external/ksp/common-util/src/main/kotlin/com/google/devtools/ksp/KSObjectCacheManager.kt (revision af87fb4bb8e3042070d2a054e912924f599b22b7)
1 /*
2  * Copyright 2020 Google LLC
3  * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17 
18 package com.google.devtools.ksp
19 
20 class KSObjectCacheManager {
21     companion object {
22         val caches = arrayListOf<KSObjectCache<*, *>>()
23 
registernull24         fun register(cache: KSObjectCache<*, *>) = caches.add(cache)
25         fun clear() = caches.forEach { it.clear() }
26     }
27 }
28 
29 abstract class KSObjectCache<K, V> {
30     val cache = mutableMapOf<K, V>()
31 
32     init {
33         KSObjectCacheManager.register(this)
34     }
35 
clearnull36     open fun clear() = cache.clear()
37 }
38