Merge pull request #68 from luizm/feat-add-support-to-gcp-proxy

feat: add support to gcp sqlproxy

Author: brunomanzo

config.yml

@@ -3,3 +3,11 @@ databases:
   url: postgres://localhost:5432/dba?sslmode=disable
 - name: dbb
   url: postgres://user:[email protected]:5432/dbb
+- name: dbc
+  # Used to connect into gcp sql instance using sqlproxy, when defined the URL is not necessary.
+  # https://cloud.google.com/sql/docs/postgres/connect-admin-proxy#go
+  sql:
+    connection_name: gcp-project:region:instance-name
+    database_name: dbc
+    database_user: user
+    database_password: pwd

config/config.go

@@ -8,9 +8,17 @@ import (
 	"gopkg.in/yaml.v2"
 )
 
+type Sql struct {
+	ConnectionName   string `yaml:"connection_name,omitempty"`
+	DatabaseName     string `yaml:"database_name,omitempty"`
+	DatabaseUser     string `yaml:"database_user,omitempty"`
+	DatabasePassword string `yaml:"database_password,omitempty"`
+}
+
 type Database struct {
 	URL  string `yaml:"url,omitempty"`
 	Name string `yaml:"name,omitempty"`
+	Sql  Sql    `yaml:"sql,omitempty"`
 }
 
 type Config struct {
@@ -38,8 +46,8 @@ func validate(config Config) error {
 		if conf.Name == "" {
 			return errors.New("failed to validate configuration. Database name cannot be empty")
 		}
-		if conf.URL == "" {
-			return fmt.Errorf("failed to validate configuration. URL for database '%s' cannot be empty", conf.Name)
+		if conf.URL == "" && conf.Sql.ConnectionName == "" {
+			return fmt.Errorf("failed to validate configuration. URL or sql field cannot be empty in the '%s' database", conf.Name)
 		}
 		if names[conf.Name] {
 			return fmt.Errorf("failed to validate configuration. A database named '%s' has already been declared", conf.Name)

config/testdata/valid-config.yml

@@ -3,3 +3,9 @@ databases:
   url: postgres://localhost:5432/dba?sslmode=disable
 - name: dbb
   url: postgres://user:[email protected]:5432/dbb
+- name: dbc
+  sql:
+    connection_name: gcp-project:region:instance-name
+    database_name: dbc
+    database_user: user
+    database_password: pwd

main.go

@@ -3,11 +3,13 @@ package main
 import (
 	"database/sql"
 	"flag"
+	"fmt"
 	"net/http"
 	"time"
 
 	"github.com/ContaAzul/postgresql_exporter/config"
 	"github.com/ContaAzul/postgresql_exporter/gauges"
+	_ "github.com/GoogleCloudPlatform/cloudsql-proxy/proxy/dialers/postgres"
 	"github.com/apex/httplog"
 	"github.com/apex/log"
 	"github.com/apex/log/handlers/logfmt"
@@ -58,9 +60,25 @@ func main() {
 	for _, con := range config.Databases {
 		var log = log.WithField("db", con.Name)
 		log.Info("started monitoring")
-		db, err := sql.Open("postgres", con.URL)
-		if err != nil {
-			log.WithError(err).Error("failed to open url")
+		var db *sql.DB
+
+		if con.Sql.ConnectionName != "" {
+			log.Info(fmt.Sprintf("Using sqlproxy with connection name: %s", con.Sql.ConnectionName))
+			dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s sslmode=disable",
+				con.Sql.ConnectionName,
+				con.Sql.DatabaseUser,
+				con.Sql.DatabasePassword,
+				con.Sql.DatabaseName)
+
+			db, err = sql.Open("cloudsqlpostgres", dsn)
+			if err != nil {
+				log.WithError(err).Error("failed to open url")
+			}
+		} else {
+			db, err = sql.Open("postgres", con.URL)
+			if err != nil {
+				log.WithError(err).Error("failed to open url")
+			}
 		}
 		if err := db.Ping(); err != nil {
 			log.WithError(err).Error("failed to ping database")