diff --git a/client.go b/client.go index 0f9b23b..5a8e321 100644 --- a/client.go +++ b/client.go @@ -10,8 +10,9 @@ import ( ) type Client struct { - db *sqlx.DB - history []string + db *sqlx.DB + history []string + connectionString string } type Row []interface{} @@ -22,13 +23,14 @@ type Result struct { } func NewClient() (*Client, error) { - db, err := sqlx.Open("postgres", getConnectionString()) + str := getConnectionString() + db, err := sqlx.Open("postgres", str) if err != nil { return nil, err } - return &Client{db: db}, nil + return &Client{db: db, connectionString: str}, nil } func NewClientFromUrl(url string) (*Client, error) { @@ -38,7 +40,7 @@ func NewClientFromUrl(url string) (*Client, error) { return nil, err } - return &Client{db: db}, nil + return &Client{db: db, connectionString: url}, nil } func (client *Client) Test() error { diff --git a/main.go b/main.go index 759f0e7..d1fca4a 100644 --- a/main.go +++ b/main.go @@ -107,6 +107,10 @@ func initClient() { exitWithMessage(err.Error()) } + if options.Debug { + fmt.Println("Server connection string:", client.connectionString) + } + fmt.Println("Connecting to server...") err = client.Test() if err != nil {