-
Notifications
You must be signed in to change notification settings - Fork 53
/
ident.go
69 lines (60 loc) · 1.51 KB
/
ident.go
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
package ident
import (
"context"
"errors"
"fmt"
"strings"
"sync"
)
type Provider interface {
UniqueIdentifier(ctx context.Context) (string, error)
}
type NamedProvider interface {
Provider
Name() string
}
type namedProviderImpl struct {
Provider
name string
}
func (np *namedProviderImpl) Name() string {
return np.name
}
func namedProvider(name string, provider Provider) NamedProvider {
return &namedProviderImpl{
Provider: provider,
name: name,
}
}
var (
identProviders = make(map[string]func() Provider)
identProvidersMu = &sync.Mutex{}
ErrInvalidProviderName = errors.New("invalid or empty ident provider name")
ErrProviderAlreadyExists = errors.New("ident provider already exists")
ErrNilProvider = errors.New("ident provider is nil")
ErrProviderNotFound = errors.New("ident provider not found")
)
func RegisterProvider(name string, provider func() Provider) error {
identProvidersMu.Lock()
defer identProvidersMu.Unlock()
name = strings.TrimSpace(name)
if len(name) == 0 {
return ErrInvalidProviderName
}
if _, ok := identProviders[name]; ok {
return fmt.Errorf("%w: %s", ErrProviderAlreadyExists, name)
}
if provider == nil {
return ErrNilProvider
}
identProviders[name] = provider
return nil
}
func GetProvider(name string) (NamedProvider, error) {
identProvidersMu.Lock()
defer identProvidersMu.Unlock()
if p, ok := identProviders[name]; ok {
return namedProvider(name, p()), nil
}
return nil, fmt.Errorf("%w: %s", ErrProviderNotFound, name)
}