Add ping to db connection

Piszmog · May 1, 2024 · 588f21d · 588f21d
1 parent b705eb2
commit 588f21d
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/db/database.go b/db/database.go
@@ -23,11 +23,20 @@ type Database interface {
 }
 
 func New(logger *slog.Logger, opts DatabaseOpts) (Database, error) {
+	var db Database
+	var err error
 	if opts.Token == "" {
-		return newLocalDB(logger, opts.URL)
+		db, err = newLocalDB(logger, opts.URL)
 	} else {
-		return newRemoteDB(logger, opts.URL, opts.Token)
+		db, err = newRemoteDB(logger, opts.URL, opts.Token)
 	}
+	if err != nil {
+		return nil, err
+	}
+	if err = db.DB().Ping(); err != nil {
+		return nil, err
+	}
+	return db, nil
 }
 
 type DatabaseOpts struct {