60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
// Copyright 2025 Cloudbase Solutions SRL
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
// not use this file except in compliance with the License. You may obtain
|
|
// a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
// License for the specific language governing permissions and limitations
|
|
// under the License.
|
|
package cache
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/cloudbase/garm/runner/common"
|
|
)
|
|
|
|
var ghClientCache *GithubClientCache
|
|
|
|
type GithubClientCache struct {
|
|
mux sync.Mutex
|
|
|
|
cache map[string]common.GithubClient
|
|
}
|
|
|
|
func init() {
|
|
clientCache := &GithubClientCache{
|
|
cache: make(map[string]common.GithubClient),
|
|
}
|
|
ghClientCache = clientCache
|
|
}
|
|
|
|
func (g *GithubClientCache) SetClient(entityID string, client common.GithubClient) {
|
|
g.mux.Lock()
|
|
defer g.mux.Unlock()
|
|
|
|
g.cache[entityID] = client
|
|
}
|
|
|
|
func (g *GithubClientCache) GetClient(entityID string) (common.GithubClient, bool) {
|
|
g.mux.Lock()
|
|
defer g.mux.Unlock()
|
|
|
|
if client, ok := g.cache[entityID]; ok {
|
|
return client, true
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
func SetGithubClient(entityID string, client common.GithubClient) {
|
|
ghClientCache.SetClient(entityID, client)
|
|
}
|
|
|
|
func GetGithubClient(entityID string) (common.GithubClient, bool) {
|
|
return ghClientCache.GetClient(entityID)
|
|
}
|