fix: can now close connection

mgjules · May 20, 2022 · 01dede5 · 01dede5
1 parent b165533
commit 01dede5
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/repo/inmemory/inmemory.go b/repo/inmemory/inmemory.go
@@ -66,3 +66,10 @@ func (r *Repository) Migrate(_ context.Context) error {
 
 	return nil
 }
+
+// Close closes any external connection in the repository.
+func (r *Repository) Close(_ context.Context) error {
+	r.log.Debug("in-memory repository has no external connection to close")
+
+	return nil
+}
diff --git a/repo/postgres/postgres.go b/repo/postgres/postgres.go
@@ -80,3 +80,13 @@ func (r *Repository) Migrate(_ context.Context) error {
 
 	return nil
 }
+
+// Close closes any external connection in the repository.
+func (r *Repository) Close(_ context.Context) error {
+	db, err := r.db.DB()
+	if err != nil {
+		return fmt.Errorf("get db instance: %w", err)
+	}
+
+	return db.Close()
+}
diff --git a/repo/repo.go b/repo/repo.go
@@ -16,6 +16,7 @@ type Repository interface {
 	Get(context.Context, string) (*deck.Deck, error)
 	Save(context.Context, *deck.Deck) error
 	Migrate(context.Context) error
+	Close(ctx context.Context) error
 }
 
 // NewRepository returns a new repository.