109 lines
2.3 KiB
Go
109 lines
2.3 KiB
Go
package rpc
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"net/rpc"
|
|
|
|
"github.com/hashicorp/terraform/terraform"
|
|
"github.com/hashicorp/yamux"
|
|
)
|
|
|
|
// Client connects to a Server in order to request plugin implementations
|
|
// for Terraform.
|
|
type Client struct {
|
|
broker *muxBroker
|
|
control *rpc.Client
|
|
}
|
|
|
|
// Dial opens a connection to a Terraform RPC server and returns a client.
|
|
func Dial(network, address string) (*Client, error) {
|
|
conn, err := net.Dial(network, address)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if tcpConn, ok := conn.(*net.TCPConn); ok {
|
|
// Make sure to set keep alive so that the connection doesn't die
|
|
tcpConn.SetKeepAlive(true)
|
|
}
|
|
|
|
return NewClient(conn)
|
|
}
|
|
|
|
// NewClient creates a client from an already-open connection-like value.
|
|
// Dial is typically used instead.
|
|
func NewClient(conn io.ReadWriteCloser) (*Client, error) {
|
|
// Create the yamux client so we can multiplex
|
|
mux, err := yamux.Client(conn, nil)
|
|
if err != nil {
|
|
conn.Close()
|
|
return nil, err
|
|
}
|
|
|
|
// Connect to the control stream.
|
|
control, err := mux.Open()
|
|
if err != nil {
|
|
mux.Close()
|
|
return nil, err
|
|
}
|
|
|
|
// Create the broker and start it up
|
|
broker := newMuxBroker(mux)
|
|
go broker.Run()
|
|
|
|
// Build the client using our broker and control channel.
|
|
return &Client{
|
|
broker: broker,
|
|
control: rpc.NewClient(control),
|
|
}, nil
|
|
}
|
|
|
|
// Close closes the connection. The client is no longer usable after this
|
|
// is called.
|
|
func (c *Client) Close() error {
|
|
if err := c.control.Close(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.broker.Close()
|
|
}
|
|
|
|
func (c *Client) ResourceProvider() (terraform.ResourceProvider, error) {
|
|
var id uint32
|
|
if err := c.control.Call(
|
|
"Dispenser.ResourceProvider", new(interface{}), &id); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
conn, err := c.broker.Dial(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &ResourceProvider{
|
|
Broker: c.broker,
|
|
Client: rpc.NewClient(conn),
|
|
Name: "ResourceProvider",
|
|
}, nil
|
|
}
|
|
|
|
func (c *Client) ResourceProvisioner() (terraform.ResourceProvisioner, error) {
|
|
var id uint32
|
|
if err := c.control.Call(
|
|
"Dispenser.ResourceProvisioner", new(interface{}), &id); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
conn, err := c.broker.Dial(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &ResourceProvisioner{
|
|
Broker: c.broker,
|
|
Client: rpc.NewClient(conn),
|
|
Name: "ResourceProvisioner",
|
|
}, nil
|
|
}
|