mirror of
https://github.com/status-im/consul.git
synced 2025-01-10 13:55:55 +00:00
11a3ce0bdd
Client works for RPC; will honor CONSUL_RPC_ADDR. HTTP works via consul/api; honors CONSUL_HTTP_ADDR. The format of a Unix socket in configuration data is: "unix://[/path/to/socket];[username or uid];[gid];[mode]" Obviously, the user must have appropriate permissions to create the socket file in the given path and assign the requested uid/gid. Also note that Go does not support gid lookups from group name, so gid must be numeric. See https://codereview.appspot.com/101310044 When connecting from the client, the format is just the first part of the above line: "unix://[/path/to/socket]" This code is copyright 2014 Akamai Technologies, Inc. <opensource@akamai.com>
55 lines
1.6 KiB
Go
55 lines
1.6 KiB
Go
package command
|
|
|
|
import (
|
|
"flag"
|
|
"os"
|
|
|
|
consulapi "github.com/hashicorp/consul/api"
|
|
"github.com/hashicorp/consul/command/agent"
|
|
)
|
|
|
|
// RPCAddrEnvName defines an environment variable name which sets
|
|
// an RPC address if there is no -rpc-addr specified.
|
|
const RPCAddrEnvName = "CONSUL_RPC_ADDR"
|
|
|
|
// RPCAddrFlag returns a pointer to a string that will be populated
|
|
// when the given flagset is parsed with the RPC address of the Consul.
|
|
func RPCAddrFlag(f *flag.FlagSet) *string {
|
|
defaultRPCAddr := os.Getenv(RPCAddrEnvName)
|
|
if defaultRPCAddr == "" {
|
|
defaultRPCAddr = "127.0.0.1:8400"
|
|
}
|
|
return f.String("rpc-addr", defaultRPCAddr,
|
|
"RPC address of the Consul agent")
|
|
}
|
|
|
|
// RPCClient returns a new Consul RPC client with the given address.
|
|
func RPCClient(addr string) (*agent.RPCClient, error) {
|
|
return agent.NewRPCClient(addr)
|
|
}
|
|
|
|
// HTTPAddrFlag returns a pointer to a string that will be populated
|
|
// when the given flagset is parsed with the HTTP address of the Consul.
|
|
func HTTPAddrFlag(f *flag.FlagSet) *string {
|
|
return f.String("http-addr", "127.0.0.1:8500",
|
|
"HTTP address of the Consul agent")
|
|
}
|
|
|
|
// HTTPClient returns a new Consul HTTP client with the given address.
|
|
func HTTPClient(addr string) (*consulapi.Client, error) {
|
|
return HTTPClientDC(addr, "")
|
|
}
|
|
|
|
// HTTPClientDC returns a new Consul HTTP client with the given address and datacenter
|
|
func HTTPClientDC(addr, dc string) (*consulapi.Client, error) {
|
|
conf := consulapi.DefaultConfig()
|
|
switch {
|
|
case len(os.Getenv("CONSUL_HTTP_ADDR")) > 0:
|
|
conf.Address = os.Getenv("CONSUL_HTTP_ADDR")
|
|
default:
|
|
conf.Address = addr
|
|
}
|
|
conf.Datacenter = dc
|
|
return consulapi.NewClient(conf)
|
|
}
|