From c15f3f70abf81f52e76ba446282e8dc70e08c754 Mon Sep 17 00:00:00 2001
From: Gabriel Simmer <gabriel@gitgalaxy.com>
Date: Wed, 6 Jul 2022 11:39:41 +0100
Subject: [PATCH] Properly close store

---
 main.go           | 1 +
 store/database.go | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/main.go b/main.go
index 82cfca6..2e2a35c 100644
--- a/main.go
+++ b/main.go
@@ -14,6 +14,7 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
+	defer db.Close()
 	handlers := transport.New(db)
 	// Auth endpoints
 	mux.Group(func(mux *flow.Mux) {
diff --git a/store/database.go b/store/database.go
index 2ed2176..4175c00 100644
--- a/store/database.go
+++ b/store/database.go
@@ -34,6 +34,7 @@ type Storer interface {
 	SessionUser(token string) (User, error)
 	SaveOauthState(state OauthState) error
 	OauthState(id string) (OauthState, error)
+	Close() error
 }
 
 type OauthState struct {
@@ -349,3 +350,7 @@ func (s *Store) OauthState(id string) (OauthState, error) {
 	}
 	return state, nil
 }
+
+func (s *Store) Close() error {
+	return s.database.Close()
+}