1
1
mirror of https://github.com/MichaelMure/git-bug.git synced 2024-12-18 14:24:21 +03:00
git-bug/cache/multi_repo_cache.go
Michael Muré 0ac39a7ab5
WIP
2022-11-28 17:20:25 +01:00

77 lines
1.7 KiB
Go

package cache
import (
"fmt"
"github.com/MichaelMure/git-bug/repository"
)
const lockfile = "lock"
const defaultRepoName = "__default"
// MultiRepoCache is the root cache, holding multiple RepoCache.
type MultiRepoCache struct {
repos map[string]*RepoCache
}
func NewMultiRepoCache() *MultiRepoCache {
return &MultiRepoCache{
repos: make(map[string]*RepoCache),
}
}
// RegisterRepository register a named repository. Use this for multi-repo setup
func (c *MultiRepoCache) RegisterRepository(ref string, repo repository.ClockedRepo) (*RepoCache, error) {
r, err := NewRepoCache(repo)
if err != nil {
return nil, err
}
c.repos[ref] = r
return r, nil
}
// RegisterDefaultRepository register an unnamed repository. Use this for mono-repo setup
func (c *MultiRepoCache) RegisterDefaultRepository(repo repository.ClockedRepo) (*RepoCache, error) {
r, err := NewRepoCache(repo)
if err != nil {
return nil, err
}
c.repos[defaultRepoName] = r
return r, nil
}
// DefaultRepo retrieve the default repository
func (c *MultiRepoCache) DefaultRepo() (*RepoCache, error) {
if len(c.repos) != 1 {
return nil, fmt.Errorf("repository is not unique")
}
for _, r := range c.repos {
return r, nil
}
panic("unreachable")
}
// ResolveRepo retrieve a repository with a reference
func (c *MultiRepoCache) ResolveRepo(ref string) (*RepoCache, error) {
r, ok := c.repos[ref]
if !ok {
return nil, fmt.Errorf("unknown repo")
}
return r, nil
}
// Close will do anything that is needed to close the cache properly
func (c *MultiRepoCache) Close() error {
for _, cachedRepo := range c.repos {
err := cachedRepo.Close()
if err != nil {
return err
}
}
return nil
}