mirror of https://github.com/hashicorp/consul
43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
|
// Package msgpackrpc implements a MessagePack-RPC ClientCodec and ServerCodec
|
||
|
|
// for the rpc package, using the same API as the Go standard library
|
||
|
|
// for jsonrpc.
|
||
|
|
package msgpackrpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"net"
|
||
|
|
"net/rpc"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Dial connects to a MessagePack-RPC server at the specified network address.
|
||
|
|
func Dial(network, address string) (*rpc.Client, error) {
|
||
|
|
conn, err := net.Dial(network, address)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return NewClient(conn), err
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewClient returns a new rpc.Client to handle requests to the set of
|
||
|
|
// services at the other end of the connection.
|
||
|
|
func NewClient(conn io.ReadWriteCloser) *rpc.Client {
|
||
|
|
return rpc.NewClientWithCodec(NewClientCodec(conn))
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewClientCodec returns a new rpc.ClientCodec using MessagePack-RPC on conn.
|
||
|
|
func NewClientCodec(conn io.ReadWriteCloser) rpc.ClientCodec {
|
||
|
|
return NewCodec(true, true, conn)
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewServerCodec returns a new rpc.ServerCodec using MessagePack-RPC on conn.
|
||
|
|
func NewServerCodec(conn io.ReadWriteCloser) rpc.ServerCodec {
|
||
|
|
return NewCodec(true, true, conn)
|
||
|
|
}
|
||
|
|
|
||
|
|
// ServeConn runs the MessagePack-RPC server on a single connection. ServeConn
|
||
|
|
// blocks, serving the connection until the client hangs up. The caller
|
||
|
|
// typically invokes ServeConn in a go statement.
|
||
|
|
func ServeConn(conn io.ReadWriteCloser) {
|
||
|
|
rpc.ServeCodec(NewServerCodec(conn))
|
||
|
|
}
|