92 lines
1.9 KiB
Go
92 lines
1.9 KiB
Go
|
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
|
||
|
|
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(a packer.Artifact) (packer.Artifact, error) {
|
||
|
|
server := rpc.NewServer()
|
||
|
|
RegisterArtifact(server, a)
|
||
|
|
|
||
|
|
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)
|
||
|
|
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(Artifact(client))
|
||
|
|
if err == nil && artifact != nil {
|
||
|
|
server := rpc.NewServer()
|
||
|
|
RegisterArtifact(server, artifact)
|
||
|
|
responseAddress = serveSingleConn(server)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
err = NewBasicError(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
*reply = PostProcessorProcessResponse{
|
||
|
|
Err: err,
|
||
|
|
RPCAddress: responseAddress,
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|