aboutsummaryrefslogtreecommitdiffstats
path: root/identity/resolver.go
blob: 8e066e9db00c013187c59bfb8744cb849aff5776 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package identity

import (
	"sync"

	"github.com/MichaelMure/git-bug/entity"
	"github.com/MichaelMure/git-bug/repository"
)

// Resolver define the interface of an Identity resolver, able to load
// an identity from, for example, a repo or a cache.
type Resolver interface {
	ResolveIdentity(id entity.Id) (Interface, error)
}

// SimpleResolver is a Resolver loading Identities directly from a Repo
type SimpleResolver struct {
	repo repository.Repo
}

func NewSimpleResolver(repo repository.Repo) *SimpleResolver {
	return &SimpleResolver{repo: repo}
}

func (r *SimpleResolver) ResolveIdentity(id entity.Id) (Interface, error) {
	return ReadLocal(r.repo, id)
}

// StubResolver is a Resolver that doesn't load anything, only returning IdentityStub instances
type StubResolver struct{}

func NewStubResolver() *StubResolver {
	return &StubResolver{}
}

func (s *StubResolver) ResolveIdentity(id entity.Id) (Interface, error) {
	return &IdentityStub{id: id}, nil
}

// CachedResolver is a resolver ensuring that loading is done only once through another Resolver.
type CachedResolver struct {
	mu         sync.RWMutex
	resolver   Resolver
	identities map[entity.Id]Interface
}

func NewCachedResolver(resolver Resolver) *CachedResolver {
	return &CachedResolver{
		resolver:   resolver,
		identities: make(map[entity.Id]Interface),
	}
}

func (c *CachedResolver) ResolveIdentity(id entity.Id) (Interface, error) {
	c.mu.RLock()
	if i, ok := c.identities[id]; ok {
		c.mu.RUnlock()
		return i, nil
	}
	c.mu.RUnlock()

	c.mu.Lock()
	defer c.mu.Unlock()

	i, err := c.resolver.ResolveIdentity(id)
	if err != nil {
		return nil, err
	}
	c.identities[id] = i
	return i, nil
}