packer-cn/packer/rpc/provisioner.go

75 lines
1.8 KiB
Go
Raw Normal View History

2013-05-22 18:35:52 -04:00
package rpc
import (
2017-04-04 16:39:01 -04:00
"github.com/hashicorp/packer/packer"
"log"
2013-05-22 18:35:52 -04:00
"net/rpc"
)
// An implementation of packer.Provisioner where the provisioner is actually
// executed over an RPC connection.
type provisioner struct {
client *rpc.Client
2014-09-02 17:23:06 -04:00
mux *muxBroker
2013-05-22 18:35:52 -04:00
}
// ProvisionerServer wraps a packer.Provisioner implementation and makes it
// exportable as part of a Golang RPC server.
type ProvisionerServer struct {
2013-12-10 14:56:15 -05:00
p packer.Provisioner
2014-09-02 17:23:06 -04:00
mux *muxBroker
2013-05-22 18:35:52 -04:00
}
type ProvisionerPrepareArgs struct {
Configs []interface{}
2013-05-22 18:35:52 -04:00
}
2013-06-06 20:07:42 -04:00
func (p *provisioner) Prepare(configs ...interface{}) (err error) {
args := &ProvisionerPrepareArgs{configs}
if cerr := p.client.Call("Provisioner.Prepare", args, new(interface{})); cerr != nil {
2013-06-06 20:07:42 -04:00
err = cerr
}
return
2013-05-22 18:35:52 -04:00
}
func (p *provisioner) Provision(ui packer.Ui, comm packer.Communicator) error {
2013-12-10 14:56:15 -05:00
nextId := p.mux.NextId()
server := newServerWithMux(p.mux, nextId)
2013-12-10 14:56:15 -05:00
server.RegisterCommunicator(comm)
server.RegisterUi(ui)
go server.Serve()
2013-05-22 18:35:52 -04:00
2013-12-10 14:56:15 -05:00
return p.client.Call("Provisioner.Provision", nextId, new(interface{}))
2013-05-22 18:35:52 -04:00
}
func (p *provisioner) Cancel() {
err := p.client.Call("Provisioner.Cancel", new(interface{}), new(interface{}))
if err != nil {
log.Printf("Provisioner.Cancel err: %s", err)
}
}
func (p *ProvisionerServer) Prepare(args *ProvisionerPrepareArgs, reply *interface{}) error {
return p.p.Prepare(args.Configs...)
2013-05-22 18:35:52 -04:00
}
2013-12-10 14:56:15 -05:00
func (p *ProvisionerServer) Provision(streamId uint32, reply *interface{}) error {
client, err := newClientWithMux(p.mux, streamId)
2013-05-22 18:35:52 -04:00
if err != nil {
2013-12-10 14:56:15 -05:00
return NewBasicError(err)
2013-05-22 18:35:52 -04:00
}
2013-12-10 14:56:15 -05:00
defer client.Close()
2013-05-22 18:35:52 -04:00
2013-12-10 14:56:15 -05:00
if err := p.p.Provision(client.Ui(), client.Communicator()); err != nil {
return NewBasicError(err)
}
2013-05-22 18:35:52 -04:00
return nil
}
func (p *ProvisionerServer) Cancel(args *interface{}, reply *interface{}) error {
p.p.Cancel()
return nil
}