forked from juju/juju
-
Notifications
You must be signed in to change notification settings - Fork 0
/
manifold_model.go
77 lines (68 loc) · 2.39 KB
/
manifold_model.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
70
71
72
73
74
75
76
77
// Copyright 2015 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.
package storageprovisioner
import (
"github.com/juju/clock"
"github.com/juju/errors"
"github.com/juju/names/v4"
"github.com/juju/worker/v3"
"github.com/juju/worker/v3/dependency"
"github.com/DavinZhang/juju/api/base"
"github.com/DavinZhang/juju/api/storageprovisioner"
"github.com/DavinZhang/juju/storage"
"github.com/DavinZhang/juju/worker/common"
)
// ModelManifoldConfig defines a storage provisioner's configuration and dependencies.
type ModelManifoldConfig struct {
APICallerName string
StorageRegistryName string
Clock clock.Clock
Model names.ModelTag
StorageDir string
NewCredentialValidatorFacade func(base.APICaller) (common.CredentialAPI, error)
NewWorker func(config Config) (worker.Worker, error)
Logger Logger
}
// ModelManifold returns a dependency.Manifold that runs a storage provisioner.
func ModelManifold(config ModelManifoldConfig) dependency.Manifold {
return dependency.Manifold{
Inputs: []string{config.APICallerName, config.StorageRegistryName},
Start: func(context dependency.Context) (worker.Worker, error) {
var apiCaller base.APICaller
if err := context.Get(config.APICallerName, &apiCaller); err != nil {
return nil, errors.Trace(err)
}
var registry storage.ProviderRegistry
if err := context.Get(config.StorageRegistryName, ®istry); err != nil {
return nil, errors.Trace(err)
}
api, err := storageprovisioner.NewState(apiCaller)
if err != nil {
return nil, errors.Trace(err)
}
credentialAPI, err := config.NewCredentialValidatorFacade(apiCaller)
if err != nil {
return nil, errors.Trace(err)
}
w, err := config.NewWorker(Config{
Model: config.Model,
Scope: config.Model,
StorageDir: config.StorageDir,
Applications: api,
Volumes: api,
Filesystems: api,
Life: api,
Registry: registry,
Machines: api,
Status: api,
Clock: config.Clock,
Logger: config.Logger,
CloudCallContextFunc: common.NewCloudCallContextFunc(credentialAPI),
})
if err != nil {
return nil, errors.Trace(err)
}
return w, nil
},
}
}