@@ -42,7 +42,7 @@ func insertQuayData(db *sql.DB, datetime string, count int, kind string) error {
42
42
return fmt .Errorf ("invalid input: datetime=%v, kind=%v, count=%d (datetime/kind cannot be empty, count cannot be negative)" , datetime , kind , count )
43
43
44
44
}
45
-
45
+
46
46
parsedDate , err := time .Parse ("Mon, 02 Jan 2006 15:04:05 -0700" , datetime )
47
47
if err != nil {
48
48
return fmt .Errorf ("invalid datetime format: %v, expected YYYY-MM-DD" , datetime )
@@ -67,7 +67,7 @@ func insertQuayData(db *sql.DB, datetime string, count int, kind string) error {
67
67
func pingDB (db * sql.DB ) error {
68
68
logrus .Info ("Pinging the database to verify connection..." )
69
69
if err := db .Ping (); err != nil {
70
- db .Close ()
70
+ _ = db .Close ()
71
71
return fmt .Errorf ("database ping failed: %w" , err )
72
72
}
73
73
logrus .Info ("Database connection verified successfully." )
@@ -185,7 +185,11 @@ func initDBAWS() (*sql.DB, error) {
185
185
}
186
186
187
187
// Close the initial connection and reconnect with the specified database.
188
- db .Close ()
188
+ err = db .Close ()
189
+ if err != nil {
190
+ return nil , fmt .Errorf ("failed to close initial connection: %w" , err )
191
+ }
192
+
189
193
DBUsername := os .Getenv ("DB_USER" )
190
194
DBPassword := os .Getenv ("DB_PASSWORD" )
191
195
DBURL := os .Getenv ("DB_URL" )
@@ -230,7 +234,11 @@ func ConnectToLocalDB() (*sql.DB, error) {
230
234
if err != nil {
231
235
return nil , fmt .Errorf ("failed to connect to MySQL server: %w" , err )
232
236
}
233
- defer db .Close ()
237
+ defer func () {
238
+ if err := db .Close (); err != nil {
239
+ logrus .Errorf ("failed to close MySQL connection: %v" , err )
240
+ }
241
+ }()
234
242
235
243
// Check if the database exists
236
244
var exists int
0 commit comments