✨ feat: added context timeout while trying to connect mysql #6

sivaosorg · Oct 4, 2023 · 8a203c7 · 8a203c7
1 parent 4029f67
commit 8a203c7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mysqlconn/mysqlconn.go b/mysqlconn/mysqlconn.go
@@ -74,7 +74,9 @@ func NewClient(config mysql.MysqlConfig) (*MySql, dbx.Dbx) {
 	client.SetMaxOpenConns(config.MaxOpenConn)
 	client.SetMaxIdleConns(config.MaxIdleConn)
 	client.SetConnMaxLifetime(time.Duration(time.Duration(config.MaxLifeTimeMinutesConn).Minutes()))
-	err = client.PingContext(context.Background())
+	ctx, cancel := context.WithTimeout(context.Background(), config.Timeout)
+	defer cancel()
+	err = client.PingContext(ctx)
 	if err != nil {
 		s.SetConnected(false).SetError(err).SetMessage(err.Error())
 		return &MySql{}, *s