/
backend.go
55 lines (48 loc) · 1.36 KB
/
backend.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
package state
import (
log "github.com/sirupsen/logrus"
"github.com/mensaah/reka/config"
)
// Backender interface
// Definitions of set of methods required for all state types
type Backender interface {
Writer
Reader
}
// Reader is the interface for state types that can return state resources
type Reader interface {
GetState() *State
}
// Writer is the interface for state types that can write to state files
type Writer interface {
WriteState(st *State) error
}
// InitBackend initializes the backend for reading and writing to state
// It checks state configuration defined in the config and returns
// The appropriate backend
func InitBackend() Backender {
cfg = config.GetConfig()
s := NewEmptyState()
switch {
case config.Contains(config.RemoteBackendTypes, cfg.StateBackend.Type):
log.Infof("using Remote StateBackend %s://%s/%s", cfg.StateBackend.Type, cfg.StateBackend.Bucket, cfg.StateBackend.Path)
backend = RemoteBackend{
Path: cfg.StateBackend.Path,
state: &s,
Bucket: cfg.StateBackend.Bucket,
BlobType: cfg.StateBackend.Type,
Region: cfg.StateBackend.Region,
}
default:
log.Debugf("using Local State at %s", cfg.StateBackend.Path)
backend = LocalBackend{
Path: cfg.StateBackend.Path,
state: &s,
}
}
return backend
}
// GetBackend gets the current active backend
func GetBackend() Backender {
return backend
}