Reverted Close() method signature change

This commit is contained in:
Technerder
2021-07-09 13:29:58 -04:00
parent c4e1c722b1
commit bd837d91ab
3 changed files with 8 additions and 7 deletions

View File

@@ -16,7 +16,7 @@ func (s *Storage) Get(key string) ([]byte, error)
func (s *Storage) Set(key string, val []byte, exp time.Duration) error
func (s *Storage) Delete(key string) error
func (s *Storage) Reset() error
func (s *Storage) Close()
func (s *Storage) Close() error
```
### Installation
Postgres is tested on the 2 last [Go versions](https://golang.org/dl/) with support for modules. So make sure to initialize one first if you didn't do that yet:

View File

@@ -172,10 +172,11 @@ func (s *Storage) Reset() error {
}
// Close the database
func (s *Storage) Close() {
func (s *Storage) Close() error {
s.done <- struct{}{}
s.db.Stat()
s.db.Close()
return nil
}
// gcTicker starts the gc ticker
@@ -208,8 +209,4 @@ func (s *Storage) checkSchema(tableName string) {
if strings.ToLower(string(data)) != "bytea" {
fmt.Printf(checkSchemaMsg, string(data))
}
}
func (s *Storage) DB() *pgxpool.Pool {
return s.db
}
}

View File

@@ -174,3 +174,7 @@ func Test_SslRequiredMode(t *testing.T) {
SslMode: "require",
})
}
func Test_Postgres_Close(t *testing.T) {
utils.AssertEqual(t, nil, testStore.Close())
}