package main

//Extended timeout will reproduce for both msyql and postgres

import (
	"context"
	"database/sql"
	"log"
	"time"

	//_ "github.com/go-sql-driver/mysql" //MySQL driver produces same issue
	_ "github.com/lib/pq" //Postgresql driver produces same issue
)

func main() {
	log.Print("Trying to ping database!")

	//Prepare a "context" to execute queries in, this will allow us to use timeouts
	var bgCtx = context.Background()
	var ctx2SecondTimeout, cancelFunc2SecondTimeout = context.WithTimeout(bgCtx, time.Second*2)
	defer cancelFunc2SecondTimeout()

	//Open  database connection
	//db, err := sql.Open("mysql", "root:@tcp(172.1.2.3)/testdb?parseTime=true")
	db, err := sql.Open("postgres", "host=172.1.2.3 user=a password=b dbname=c")
	if err != nil {
		log.Printf("Unable to open db, %s", err.Error())
		return
	}
	log.Print("Successfully called open()")

	//Ping database connection with 2 second timeout
	err = db.PingContext(ctx2SecondTimeout)
	if err != nil {
		log.Printf("Can't ping database server in order to use storage, %s", err.Error())
		return
	}
	log.Print("Successfully pinged database!")
}