forked from hashicorp/packer
-
Notifications
You must be signed in to change notification settings - Fork 1
/
hook.go
52 lines (42 loc) · 1.13 KB
/
hook.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
package rpc
import (
"github.com/mitchellh/packer/packer"
"net/rpc"
)
// An implementation of packer.Hook where the hook is actually executed
// over an RPC connection.
type hook struct {
client *rpc.Client
}
// HookServer wraps a packer.Hook implementation and makes it exportable
// as part of a Golang RPC server.
type HookServer struct {
hook packer.Hook
}
type HookRunArgs struct {
Name string
Data interface{}
RPCAddress string
}
func Hook(client *rpc.Client) *hook {
return &hook{client}
}
func (h *hook) Run(name string, ui packer.Ui, comm packer.Communicator, data interface{}) error {
server := rpc.NewServer()
RegisterCommunicator(server, comm)
RegisterUi(server, ui)
address := serveSingleConn(server)
args := &HookRunArgs{name, data, address}
return h.client.Call("Hook.Run", args, new(interface{}))
}
func (h *HookServer) Run(args *HookRunArgs, reply *interface{}) error {
client, err := rpc.Dial("tcp", args.RPCAddress)
if err != nil {
return err
}
if err := h.hook.Run(args.Name, &Ui{client}, Communicator(client), args.Data); err != nil {
return NewBasicError(err)
}
*reply = nil
return nil
}