com.tencent.devops.common.webhook.util.EventCacheUtil.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of biz-common-webhook Show documentation

Tencent blueking ci project

The newest version!

package com.tencent.devops.common.webhook.util

import com.tencent.devops.common.webhook.service.code.pojo.EventRepositoryCache
import com.tencent.devops.repository.pojo.Repository

/**
 * 如果同一个仓库配置了多个事件触发插件，那么每个插件都会调用一次scm接口，可以将返回值线程级缓存，减少调用次数
 */
object EventCacheUtil {

    private val eventThreadLocalCache =
        ThreadLocal>()

    fun initEventCache() = eventThreadLocalCache.set(mutableMapOf())

    fun getEventCache(projectId: String, repo: Repository): EventRepositoryCache? =
        eventThreadLocalCache.get()?.get("${projectId}_${repo.aliasName}")

    fun putIfAbsentEventCache(projectId: String, repo: Repository, eventCache: EventRepositoryCache) =
        eventThreadLocalCache.get()?.putIfAbsent("${projectId}_${repo.aliasName}", eventCache)

    /**
     * 如果不存在,则初始化空的仓库缓存
     */
    fun getOrInitRepoCache(projectId: String, repo: Repository): EventRepositoryCache? {
        return eventThreadLocalCache.get()?.let {
            getEventCache(projectId, repo) ?: run {
                val repoCache = EventRepositoryCache()
                eventThreadLocalCache.get()["${projectId}_${repo.aliasName}"] = repoCache
                repoCache
            }
        }
    }

    fun remove() = eventThreadLocalCache.remove()

    fun getAll(): Map = eventThreadLocalCache.get()
}