packer-cn/packer/rpc/hook.go

71 lines
1.5 KiB
Go
Raw Normal View History

2013-05-11 12:51:49 -04:00
package rpc
import (
2017-04-04 16:39:01 -04:00
"github.com/hashicorp/packer/packer"
2013-08-31 02:03:43 -04:00
"log"
2013-05-11 12:51:49 -04:00
"net/rpc"
)
// An implementation of packer.Hook where the hook is actually executed
// over an RPC connection.
type hook struct {
client *rpc.Client
2014-09-02 17:23:06 -04:00
mux *muxBroker
2013-05-11 12:51:49 -04:00
}
// HookServer wraps a packer.Hook implementation and makes it exportable
// as part of a Golang RPC server.
type HookServer struct {
hook packer.Hook
2014-09-02 17:23:06 -04:00
mux *muxBroker
2013-05-11 12:51:49 -04:00
}
type HookRunArgs struct {
2013-12-10 14:50:30 -05:00
Name string
Data interface{}
StreamId uint32
2013-05-11 12:51:49 -04:00
}
func (h *hook) Run(name string, ui packer.Ui, comm packer.Communicator, data interface{}) error {
2013-12-10 14:50:30 -05:00
nextId := h.mux.NextId()
server := newServerWithMux(h.mux, nextId)
2013-12-10 14:50:30 -05:00
server.RegisterCommunicator(comm)
server.RegisterUi(ui)
go server.Serve()
2013-05-11 12:51:49 -04:00
2013-12-10 14:50:30 -05:00
args := HookRunArgs{
Name: name,
Data: data,
StreamId: nextId,
}
return h.client.Call("Hook.Run", &args, new(interface{}))
2013-05-11 12:51:49 -04:00
}
2013-08-30 20:03:55 -04:00
func (h *hook) Cancel() {
2013-08-31 02:03:43 -04:00
err := h.client.Call("Hook.Cancel", new(interface{}), new(interface{}))
if err != nil {
log.Printf("Hook.Cancel error: %s", err)
}
2013-08-30 20:03:55 -04:00
}
2013-05-11 12:51:49 -04:00
func (h *HookServer) Run(args *HookRunArgs, reply *interface{}) error {
client, err := newClientWithMux(h.mux, args.StreamId)
2013-05-11 12:51:49 -04:00
if err != nil {
2013-12-10 14:50:30 -05:00
return NewBasicError(err)
2013-05-11 12:51:49 -04:00
}
2013-12-10 14:50:30 -05:00
defer client.Close()
2013-05-11 12:51:49 -04:00
2013-12-10 14:50:30 -05:00
if err := h.hook.Run(args.Name, client.Ui(), client.Communicator(), args.Data); err != nil {
return NewBasicError(err)
}
2013-05-11 12:51:49 -04:00
*reply = nil
return nil
}
2013-08-31 02:03:43 -04:00
func (h *HookServer) Cancel(args *interface{}, reply *interface{}) error {
h.hook.Cancel()
return nil
}