40e6581a75
This change renames a lot of variables, types and functions to be more generic. The goal is to allow GARM to add more forges in the future. Signed-off-by: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>
107 lines
2.4 KiB
Go
107 lines
2.4 KiB
Go
package cache
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/cloudbase/garm/params"
|
|
)
|
|
|
|
var credentialsCache *GithubCredentials
|
|
|
|
func init() {
|
|
ghCredentialsCache := &GithubCredentials{
|
|
cache: make(map[uint]params.GithubCredentials),
|
|
}
|
|
credentialsCache = ghCredentialsCache
|
|
}
|
|
|
|
type GithubCredentials struct {
|
|
mux sync.Mutex
|
|
|
|
cache map[uint]params.GithubCredentials
|
|
}
|
|
|
|
func (g *GithubCredentials) SetCredentialsRateLimit(credsID uint, rateLimit params.GithubRateLimit) {
|
|
g.mux.Lock()
|
|
defer g.mux.Unlock()
|
|
|
|
if creds, ok := g.cache[credsID]; ok {
|
|
creds.RateLimit = rateLimit
|
|
g.cache[credsID] = creds
|
|
}
|
|
}
|
|
|
|
func (g *GithubCredentials) SetCredentials(credentials params.GithubCredentials) {
|
|
g.mux.Lock()
|
|
defer g.mux.Unlock()
|
|
|
|
g.cache[credentials.ID] = credentials
|
|
UpdateCredentialsInAffectedEntities(credentials.GetForgeCredentials())
|
|
}
|
|
|
|
func (g *GithubCredentials) GetCredentials(id uint) (params.GithubCredentials, bool) {
|
|
g.mux.Lock()
|
|
defer g.mux.Unlock()
|
|
|
|
if creds, ok := g.cache[id]; ok {
|
|
return creds, true
|
|
}
|
|
return params.GithubCredentials{}, false
|
|
}
|
|
|
|
func (g *GithubCredentials) DeleteCredentials(id uint) {
|
|
g.mux.Lock()
|
|
defer g.mux.Unlock()
|
|
|
|
delete(g.cache, id)
|
|
}
|
|
|
|
func (g *GithubCredentials) GetAllCredentials() []params.GithubCredentials {
|
|
g.mux.Lock()
|
|
defer g.mux.Unlock()
|
|
|
|
creds := make([]params.GithubCredentials, 0, len(g.cache))
|
|
for _, cred := range g.cache {
|
|
creds = append(creds, cred)
|
|
}
|
|
|
|
// Sort the credentials by ID
|
|
sortByID(creds)
|
|
return creds
|
|
}
|
|
|
|
func (g *GithubCredentials) GetAllCredentialsAsMap() map[uint]params.GithubCredentials {
|
|
g.mux.Lock()
|
|
defer g.mux.Unlock()
|
|
|
|
creds := make(map[uint]params.GithubCredentials, len(g.cache))
|
|
for id, cred := range g.cache {
|
|
creds[id] = cred
|
|
}
|
|
|
|
return creds
|
|
}
|
|
|
|
func SetGithubCredentials(credentials params.GithubCredentials) {
|
|
credentialsCache.SetCredentials(credentials)
|
|
}
|
|
|
|
func GetGithubCredentials(id uint) (params.GithubCredentials, bool) {
|
|
return credentialsCache.GetCredentials(id)
|
|
}
|
|
|
|
func DeleteGithubCredentials(id uint) {
|
|
credentialsCache.DeleteCredentials(id)
|
|
}
|
|
|
|
func GetAllGithubCredentials() []params.GithubCredentials {
|
|
return credentialsCache.GetAllCredentials()
|
|
}
|
|
|
|
func SetCredentialsRateLimit(credsID uint, rateLimit params.GithubRateLimit) {
|
|
credentialsCache.SetCredentialsRateLimit(credsID, rateLimit)
|
|
}
|
|
|
|
func GetAllGithubCredentialsAsMap() map[uint]params.GithubCredentials {
|
|
return credentialsCache.GetAllCredentialsAsMap()
|
|
}
|