packer-cn/packer/rpc/post_processor.go

97 lines
2.1 KiB
Go
Raw Normal View History

2013-06-18 16:44:57 -04:00
package rpc
import (
"github.com/mitchellh/packer/packer"
"net/rpc"
)
// An implementation of packer.PostProcessor where the PostProcessor is actually
// executed over an RPC connection.
type postProcessor struct {
client *rpc.Client
}
// PostProcessorServer wraps a packer.PostProcessor implementation and makes it
// exportable as part of a Golang RPC server.
type PostProcessorServer struct {
p packer.PostProcessor
}
type PostProcessorProcessResponse struct {
Err error
2013-06-18 16:44:57 -04:00
RPCAddress string
}
func PostProcessor(client *rpc.Client) *postProcessor {
return &postProcessor{client}
}
func (p *postProcessor) Configure(raw interface{}) (err error) {
if cerr := p.client.Call("PostProcessor.Configure", &raw, &err); cerr != nil {
err = cerr
}
return
}
func (p *postProcessor) PostProcess(ui packer.Ui, a packer.Artifact) (packer.Artifact, error) {
2013-06-18 16:44:57 -04:00
server := rpc.NewServer()
RegisterArtifact(server, a)
RegisterUi(server, ui)
2013-06-18 16:44:57 -04:00
var response PostProcessorProcessResponse
if err := p.client.Call("PostProcessor.PostProcess", serveSingleConn(server), &response); err != nil {
return nil, err
}
if response.Err != nil {
return nil, response.Err
}
if response.RPCAddress == "" {
return nil, nil
}
client, err := rpc.Dial("tcp", response.RPCAddress)
if err != nil {
return nil, err
}
return Artifact(client), nil
}
func (p *PostProcessorServer) Configure(raw *interface{}, reply *error) error {
*reply = p.p.Configure(*raw)
if *reply != nil {
*reply = NewBasicError(*reply)
}
2013-06-18 16:44:57 -04:00
return nil
}
func (p *PostProcessorServer) PostProcess(address string, reply *PostProcessorProcessResponse) error {
client, err := rpc.Dial("tcp", address)
if err != nil {
return err
}
responseAddress := ""
artifact, err := p.p.PostProcess(&Ui{client}, Artifact(client))
2013-06-18 16:44:57 -04:00
if err == nil && artifact != nil {
server := rpc.NewServer()
RegisterArtifact(server, artifact)
responseAddress = serveSingleConn(server)
}
if err != nil {
err = NewBasicError(err)
}
*reply = PostProcessorProcessResponse{
Err: err,
2013-06-18 16:44:57 -04:00
RPCAddress: responseAddress,
}
return nil
}