go: add check connect for postgres

Signed-off-by: Victor Login <batazor@evrone.com>
shortlink-org · Apr 8, 2023 · 82e1781 · 82e1781
1 parent 66e2f81
commit 82e1781
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/internal/pkg/db/postgres/postgres.go b/internal/pkg/db/postgres/postgres.go
@@ -57,6 +57,12 @@ func (p *Store) Init(ctx context.Context) error {
 		return err
 	}
 
+	// Check connect
+	err = p.client.Ping(ctx)
+	if err != nil {
+		return err
+	}
+
 	return nil
 }
 

diff --git a/internal/pkg/db/redis/redis.go b/internal/pkg/db/redis/redis.go
@@ -45,6 +45,7 @@ func (s *Store) Init(ctx context.Context) error {
 		return err
 	}
 
+	// Check connect
 	if _, err := s.client.Ping(ctx).Result(); err != nil {
 		return errors.Wrap(err, "redis ping failed")
 	}