summaryrefslogtreecommitdiffstats
path: root/main.go
diff options
context:
space:
mode:
Diffstat (limited to 'main.go')
-rw-r--r--main.go64
1 files changed, 49 insertions, 15 deletions
diff --git a/main.go b/main.go
index 1c0d886..197541d 100644
--- a/main.go
+++ b/main.go
@@ -29,6 +29,7 @@ import (
"text/tabwriter"
"github.com/Sirupsen/logrus"
+ "github.com/iotbzh/xds-agent/lib/xaapiv1"
common "github.com/iotbzh/xds-common/golib"
"github.com/joho/godotenv"
socketio_client "github.com/sebd71/go-socket.io-client"
@@ -99,9 +100,9 @@ func main() {
Setting of global options is driven either by environment variables or by command
line options or using a config file knowning that the following priority order is used:
1. use option value (for example --url option),
- 2. else use variable 'XDS_xxx' (for example 'XDS_SERVER_URL' variable) when a
+ 2. else use variable 'XDS_xxx' (for example 'XDS_AGENT_URL' variable) when a
config file is specified with '--config|-c' option,
- 3. else use 'XDS_xxx' (for example 'XDS_SERVER_URL') environment variable.
+ 3. else use 'XDS_xxx' (for example 'XDS_AGENT_URL') environment variable.
Examples:
# Get help of 'projects' sub-command
@@ -163,9 +164,15 @@ func main() {
},
cli.StringFlag{
Name: "url, u",
+ EnvVar: "XDS_AGENT_URL",
+ Value: "localhost:8800",
+ Usage: "local XDS agent url",
+ },
+ cli.StringFlag{
+ Name: "url-server, us",
EnvVar: "XDS_SERVER_URL",
- Value: "localhost:8000",
- Usage: "remote XDS server url",
+ Value: "",
+ Usage: "overwrite remote XDS server url (default value set in xds-agent-config.json file)",
},
cli.BoolFlag{
Name: "timestamp, ts",
@@ -265,20 +272,26 @@ func XdsConnInit(ctx *cli.Context) error {
var err error
// Define HTTP and WS url
- baseURL := ctx.String("url")
+ agentURL := ctx.String("url")
+ serverURL := ctx.String("url-server")
// Allow to only set port number
- if match, _ := regexp.MatchString("^([0-9]+)$", baseURL); match {
- baseURL = "http://localhost:" + ctx.String("url")
+ if match, _ := regexp.MatchString("^([0-9]+)$", agentURL); match {
+ agentURL = "http://localhost:" + ctx.String("url")
+ }
+ if match, _ := regexp.MatchString("^([0-9]+)$", serverURL); match {
+ serverURL = "http://localhost:" + ctx.String("url-server")
}
-
// Add http prefix if missing
- if !strings.HasPrefix(baseURL, "http://") {
- baseURL = "http://" + baseURL
+ if agentURL != "" && !strings.HasPrefix(agentURL, "http://") {
+ agentURL = "http://" + agentURL
+ }
+ if serverURL != "" && !strings.HasPrefix(serverURL, "http://") {
+ serverURL = "http://" + serverURL
}
// Create HTTP client
- Log.Debugln("Connect HTTP client on ", baseURL)
+ Log.Debugln("Connect HTTP client on ", agentURL)
conf := common.HTTPClientConfig{
URLPrefix: "/api/v1",
HeaderClientKeyName: "Xds-Agent-Sid",
@@ -288,12 +301,12 @@ func XdsConnInit(ctx *cli.Context) error {
LogLevel: common.HTTPLogLevelWarning,
}
- HTTPCli, err = common.HTTPNewClient(baseURL, conf)
+ HTTPCli, err = common.HTTPNewClient(agentURL, conf)
if err != nil {
errmsg := err.Error()
if m, err := regexp.MatchString("Get http.?://", errmsg); m && err == nil {
i := strings.LastIndex(errmsg, ":")
- errmsg = "Cannot connection to " + baseURL + errmsg[i:]
+ errmsg = "Cannot connection to " + agentURL + errmsg[i:]
}
return cli.NewExitError(errmsg, 1)
}
@@ -301,7 +314,7 @@ func XdsConnInit(ctx *cli.Context) error {
Log.Infoln("HTTP session ID : ", HTTPCli.GetClientID())
// Create io Websocket client
- Log.Debugln("Connecting IO.socket client on ", baseURL)
+ Log.Debugln("Connecting IO.socket client on ", agentURL)
opts := &socketio_client.Options{
Transport: "websocket",
@@ -309,7 +322,7 @@ func XdsConnInit(ctx *cli.Context) error {
}
opts.Header["XDS-AGENT-SID"] = []string{HTTPCli.GetClientID()}
- IOsk, err = socketio_client.NewClient(baseURL, opts)
+ IOsk, err = socketio_client.NewClient(agentURL, opts)
if err != nil {
return cli.NewExitError("IO.socket connection error: "+err.Error(), 1)
}
@@ -321,6 +334,27 @@ func XdsConnInit(ctx *cli.Context) error {
ctx.App.Metadata["httpCli"] = HTTPCli
ctx.App.Metadata["ioskCli"] = IOsk
+ // Display version in logs (debug helpers)
+ ver := xaapiv1.XDSVersion{}
+ if err := XdsVersionGet(&ver); err != nil {
+ return cli.NewExitError("ERROR while retrieving XDS version: "+err.Error(), 1)
+ }
+ Log.Infof("XDS Agent/Server version: %v", ver)
+
+ // Get current config and update connection to server when needed
+ xdsConf := xaapiv1.APIConfig{}
+ if err := XdsConfigGet(&xdsConf); err != nil {
+ return cli.NewExitError("ERROR while getting XDS config: "+err.Error(), 1)
+ }
+ svrCfg := xdsConf.Servers[XdsServerIndexGet()]
+ if serverURL != "" && (svrCfg.URL != serverURL || !svrCfg.Connected) {
+ svrCfg.URL = serverURL
+ svrCfg.ConnRetry = 10
+ if err := XdsConfigSet(xdsConf); err != nil {
+ return cli.NewExitError("ERROR while updating XDS server URL: "+err.Error(), 1)
+ }
+ }
+
return nil
}