/
state.go
66 lines (53 loc) · 1.69 KB
/
state.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
// Copyright 2015 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.
package imagemetadatamanager
import (
"github.com/juju/errors"
"github.com/juju/names/v5"
"github.com/juju/juju/environs/config"
"github.com/juju/juju/state"
"github.com/juju/juju/state/cloudimagemetadata"
)
type metadataAccess interface {
FindMetadata(cloudimagemetadata.MetadataFilter) (map[string][]cloudimagemetadata.Metadata, error)
SaveMetadata([]cloudimagemetadata.Metadata) error
DeleteMetadata(imageId string) error
ModelConfig() (*config.Config, error)
ControllerTag() names.ControllerTag
Model() (Model, error)
}
type Model interface {
CloudRegion() string
}
var getState = func(st *state.State) metadataAccess {
return stateShim{st}
}
type stateShim struct {
*state.State
}
func (s stateShim) FindMetadata(f cloudimagemetadata.MetadataFilter) (map[string][]cloudimagemetadata.Metadata, error) {
return s.State.CloudImageMetadataStorage.FindMetadata(f)
}
func (s stateShim) SaveMetadata(m []cloudimagemetadata.Metadata) error {
return s.State.CloudImageMetadataStorage.SaveMetadata(m)
}
func (s stateShim) DeleteMetadata(imageId string) error {
return s.State.CloudImageMetadataStorage.DeleteMetadata(imageId)
}
// Model returns the Model for this state.
func (s stateShim) Model() (Model, error) {
return s.State.Model()
}
// ModelConfig implements the metadataAccess method as an expedient until the
// State is replaced with its underlying Model.
func (s stateShim) ModelConfig() (*config.Config, error) {
model, err := s.State.Model()
if err != nil {
return nil, errors.Trace(err)
}
cfg, err := model.Config()
if err != nil {
return nil, errors.Trace(err)
}
return cfg, nil
}