diff --git a/examples/client.go b/examples/client.go index 7162408..b34dcd2 100644 --- a/examples/client.go +++ b/examples/client.go @@ -31,6 +31,8 @@ import ( ) var cmd, executor, url, clustersConfig, clusterName, updateId, username, password, zkUrl, hostList, role string +var caCertsPath string +var clientKey, clientCert string var CONNECTION_TIMEOUT = 20000 @@ -46,6 +48,9 @@ func init() { flag.StringVar(&zkUrl, "zkurl", "", "zookeeper url") flag.StringVar(&hostList, "hostList", "", "Comma separated list of hosts to operate on") flag.StringVar(&role, "role", "", "owner role to use") + flag.StringVar(&caCertsPath, "caCertsPath", "", "Path to CA certs on local machine.") + flag.StringVar(&clientCert, "clientCert", "", "Client certificate to use to connect to Aurora.") + flag.StringVar(&clientKey, "clientKey", "", "Client key to use to connect to Aurora.") flag.Parse() @@ -102,6 +107,10 @@ func main() { clientOptions = append(clientOptions, realis.SchedulerUrl(url)) } + if clientKey != "" || clientCert != "" || caCertsPath != "" { + clientOptions = append(clientOptions, realis.Certspath(caCertsPath), realis.ClientCerts(clientKey, clientCert)) + } + r, err = realis.NewRealisClient(clientOptions...) if err != nil { fmt.Println(err)