129 lines
2.7 KiB
Go
129 lines
2.7 KiB
Go
package rpc
|
|
|
|
import (
|
|
"github.com/mitchellh/packer/packer"
|
|
"net/rpc"
|
|
)
|
|
|
|
// An implementation of packer.Build where the build is actually executed
|
|
// over an RPC connection.
|
|
type build struct {
|
|
client *rpc.Client
|
|
}
|
|
|
|
// BuildServer wraps a packer.Build implementation and makes it exportable
|
|
// as part of a Golang RPC server.
|
|
type BuildServer struct {
|
|
build packer.Build
|
|
}
|
|
|
|
type BuildRunArgs struct {
|
|
UiRPCAddress string
|
|
}
|
|
|
|
func Build(client *rpc.Client) *build {
|
|
return &build{client}
|
|
}
|
|
|
|
func (b *build) Name() (result string) {
|
|
b.client.Call("Build.Name", new(interface{}), &result)
|
|
return
|
|
}
|
|
|
|
func (b *build) Prepare(v map[string]string) (err error) {
|
|
if cerr := b.client.Call("Build.Prepare", v, &err); cerr != nil {
|
|
return cerr
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (b *build) Run(ui packer.Ui, cache packer.Cache) ([]packer.Artifact, error) {
|
|
// Create and start the server for the UI
|
|
server := rpc.NewServer()
|
|
RegisterCache(server, cache)
|
|
RegisterUi(server, ui)
|
|
args := &BuildRunArgs{serveSingleConn(server)}
|
|
|
|
var result []string
|
|
if err := b.client.Call("Build.Run", args, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
artifacts := make([]packer.Artifact, len(result))
|
|
for i, addr := range result {
|
|
client, err := rpc.Dial("tcp", addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
artifacts[i] = Artifact(client)
|
|
}
|
|
|
|
return artifacts, nil
|
|
}
|
|
|
|
func (b *build) SetDebug(val bool) {
|
|
if err := b.client.Call("Build.SetDebug", val, new(interface{})); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (b *build) SetForce(val bool) {
|
|
if err := b.client.Call("Build.SetForce", val, new(interface{})); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (b *build) Cancel() {
|
|
if err := b.client.Call("Build.Cancel", new(interface{}), new(interface{})); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (b *BuildServer) Name(args *interface{}, reply *string) error {
|
|
*reply = b.build.Name()
|
|
return nil
|
|
}
|
|
|
|
func (b *BuildServer) Prepare(v map[string]string, reply *error) error {
|
|
*reply = b.build.Prepare(v)
|
|
return nil
|
|
}
|
|
|
|
func (b *BuildServer) Run(args *BuildRunArgs, reply *[]string) error {
|
|
client, err := rpc.Dial("tcp", args.UiRPCAddress)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
artifacts, err := b.build.Run(&Ui{client}, Cache(client))
|
|
if err != nil {
|
|
return NewBasicError(err)
|
|
}
|
|
|
|
*reply = make([]string, len(artifacts))
|
|
for i, artifact := range artifacts {
|
|
server := rpc.NewServer()
|
|
RegisterArtifact(server, artifact)
|
|
(*reply)[i] = serveSingleConn(server)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (b *BuildServer) SetDebug(val *bool, reply *interface{}) error {
|
|
b.build.SetDebug(*val)
|
|
return nil
|
|
}
|
|
|
|
func (b *BuildServer) SetForce(val *bool, reply *interface{}) error {
|
|
b.build.SetForce(*val)
|
|
return nil
|
|
}
|
|
|
|
func (b *BuildServer) Cancel(args *interface{}, reply *interface{}) error {
|
|
b.build.Cancel()
|
|
return nil
|
|
}
|