~rogpeppe/juju-core/azure

« back to all changes in this revision

Viewing changes to environs/cloudinit.go

[r=jtv] Unify providers' user-data composition functions.

All the "physical" providers (EC2, OpenStack, MAAS, Azure) were generating
user data in essentially the same way. The MAAS one looked different but
actually just added an optional bit that was still completely generic. And so
this branch unifies the implementations.

Construction of the cloudinit.MachineConfig was lifted out of the function.
That may seem arbitrary, but it's part of a greater plan: in a later step of
the ongoing refactoring the construction can be lifted out of the providers'
internalStartInstance methods as well, doing away with the EC2 and OpenStack
providers' startInstanceParams structs. This actually shortens some of the
data flows and reduces cognitive load, which can only be good for maintenance.

After that step we can have a fresh look at sanitizing NewMachineConfig (which
arguably should take more parameters, or not exist at all, but is in an
intermediate state that facilitates the changes happening now). We may also
look into re-introducing some sort of dedicated Parameter Object for
internalStartInstance, but by then the cluster of methods Bootstrap /
StartInstance / internalStartInstance will look a bit different and we can
find the most maintainable solution for that new situation.

Show diffs side-by-side

added added

removed removed

Lines of Context:
6
6
import (
7
7
        "fmt"
8
8
 
 
9
        coreCloudinit "launchpad.net/juju-core/cloudinit"
9
10
        "launchpad.net/juju-core/constraints"
10
11
        "launchpad.net/juju-core/environs/cloudinit"
11
12
        "launchpad.net/juju-core/environs/config"
93
94
        mcfg.StateServerKey = key
94
95
        return nil
95
96
}
 
97
 
 
98
// ComposeUserData puts together a binary (gzipped) blob of user data.
 
99
// The additionalScripts are additional command lines that you need cloudinit
 
100
// to run on the instance.  Use with care.
 
101
func ComposeUserData(cfg *cloudinit.MachineConfig, additionalScripts ...string) ([]byte, error) {
 
102
        cloudcfg := coreCloudinit.New()
 
103
        for _, script := range additionalScripts {
 
104
                cloudcfg.AddRunCmd(script)
 
105
        }
 
106
        cloudcfg, err := cloudinit.Configure(cfg, cloudcfg)
 
107
        if err != nil {
 
108
                return nil, err
 
109
        }
 
110
        data, err := cloudcfg.Render()
 
111
        if err != nil {
 
112
                return nil, err
 
113
        }
 
114
        return utils.Gzip(data), nil
 
115
}