Fixing Improper-sqlite-logs

pkg/gofr/datasource/sql/sql.go

@@ -35,6 +35,24 @@ type DBConfig struct {
 	MaxOpenConn int
 }
 
+func printConnectionSuccessLog(status string, dbconfig *DBConfig, logger datasource.Logger) {
+	if dbconfig.Dialect == sqlite {
+		logger.Debugf("%s to '%s' database", status, dbconfig.Database)
+	} else {
+		logger.Debugf("%s to '%s' user to '%s' database at '%s:%s'", status, dbconfig.User,
+			dbconfig.Database, dbconfig.HostName, dbconfig.Port)
+	}
+}
+
+func printConnectionFailureLog(dbconfig *DBConfig, logger datasource.Logger, err error) {
+	if dbconfig.Dialect == sqlite {
+		logger.Errorf("could not connect to database '%s', error: %v", dbconfig.Database, err)
+	} else {
+		logger.Errorf("could not connect '%s' user to '%s' database at '%s:%s', error: %v",
+			dbconfig.User, dbconfig.Database, dbconfig.HostName, dbconfig.Port, err)
+	}
+}
+
 func NewSQL(configs config.Config, logger datasource.Logger, metrics Metrics) *DB {
 	logger.Debugf("reading database configurations from config file")
 
@@ -64,13 +82,16 @@ func NewSQL(configs config.Config, logger datasource.Logger, metrics Metrics) *D
 
 	database := &DB{config: dbConfig, logger: logger, metrics: metrics}
 
-	logger.Debugf("connecting to '%s' user to '%s' database at '%s:%s'", database.config.User,
-		database.config.Database, database.config.HostName, database.config.Port)
+	printConnectionSuccessLog("connecting", database.config, logger)
 
 	database.DB, err = sql.Open(otelRegisteredDialect, dbConnectionString)
 	if err != nil {
-		database.logger.Errorf("could not open connection with '%s' user to '%s' database at '%s:%s', error: %v",
-			database.config.User, database.config.Database, database.config.HostName, database.config.Port, err)
+		if database.config.Dialect == sqlite {
+			database.logger.Errorf("could not open connection '%s' database, error: %v", database.config.Database, err)
+		} else {
+			database.logger.Errorf("could not open connection with '%s' user to '%s' database at '%s:%s', error: %v",
+				database.config.User, database.config.Database, database.config.HostName, database.config.Port, err)
+		}
 
 		return database
 	}
@@ -94,14 +115,12 @@ func NewSQL(configs config.Config, logger datasource.Logger, metrics Metrics) *D
 
 func pingToTestConnection(database *DB) *DB {
 	if err := database.DB.Ping(); err != nil {
-		database.logger.Errorf("could not connect with '%s' user to '%s' database at '%s:%s', error: %v",
-			database.config.User, database.config.Database, database.config.HostName, database.config.Port, err)
+		printConnectionFailureLog(database.config, database.logger, err)
 
 		return database
 	}
 
-	database.logger.Logf("connected to '%s' database at '%s:%s'", database.config.Database,
-		database.config.HostName, database.config.Port)
+	printConnectionSuccessLog("connected", database.config, database.logger)
 
 	return database
 }
@@ -116,14 +135,12 @@ func retryConnection(database *DB) {
 			for {
 				err := database.DB.Ping()
 				if err == nil {
-					database.logger.Logf("connected to '%s' database at '%s:%s'", database.config.Database,
-						database.config.HostName, database.config.Port)
+					printConnectionSuccessLog("connected", database.config, database.logger)
 
 					break
 				}
 
-				database.logger.Debugf("could not connect with '%s' user to '%s' database at '%s:%s', error: %v",
-					database.config.User, database.config.Database, database.config.HostName, database.config.Port, err)
+				printConnectionFailureLog(database.config, database.logger, err)
 
 				time.Sleep(connRetryFrequencyInSeconds * time.Second)
 			}

pkg/gofr/datasource/sql/sql_test.go

@@ -1,6 +1,7 @@
 package sql
 
 import (
+	"errors"
 	"fmt"
 	"testing"
 	"time"
@@ -256,6 +257,56 @@ func Test_NewSQLMockWithConfig(t *testing.T) {
 	assert.NotNil(t, mockMetric)
 }
 
+var errSqliteConnection = errors.New("connection failed")
+
+func Test_sqliteSuccessfulConnLogs(t *testing.T) {
+	tests := []struct {
+		desc        string
+		status      string
+		expectedLog string
+	}{
+		{"sqlite connection in process", "connecting", `connecting to 'test' database`},
+		{"sqlite connected successfully", "connected", `connected to 'test' database`},
+	}
+
+	for _, test := range tests {
+		logs := testutil.StdoutOutputForFunc(func() {
+			mockLogger := logging.NewMockLogger(logging.DEBUG)
+			mockConfig := &DBConfig{
+				Dialect:  sqlite,
+				Database: "test",
+			}
+
+			printConnectionSuccessLog(test.status, mockConfig, mockLogger)
+		})
+
+		assert.Contains(t, logs, test.expectedLog)
+	}
+}
+
+func Test_sqliteErrConnLogs(t *testing.T) {
+	test := struct {
+		desc        string
+		err         error
+		expectedLog string
+	}{
+		"sqlite connection failure", errSqliteConnection,
+		`could not connect to database 'test', error: connection failed`,
+	}
+
+	logs := testutil.StderrOutputForFunc(func() {
+		mockLogger := logging.NewMockLogger(logging.DEBUG)
+		mockConfig := &DBConfig{
+			Dialect:  sqlite,
+			Database: "test",
+		}
+
+		printConnectionFailureLog(mockConfig, mockLogger, test.err)
+	})
+
+	assert.Contains(t, logs, test.expectedLog)
+}
+
 func Test_SQLRetryConnectionInfoLog(t *testing.T) {
 	logs := testutil.StdoutOutputForFunc(func() {
 		ctrl := gomock.NewController(t)