This commit is contained in:
Muhammed Efe Çetin
2022-11-03 23:36:06 +03:00
parent 90e5588e5d
commit 1fc6144cc0
3 changed files with 15 additions and 1 deletions

View File

@@ -17,6 +17,7 @@ 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() error
func (s *Storage) Conn() *pgxpool.Pool
```
### 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

@@ -160,6 +160,11 @@ func (s *Storage) Close() error {
return nil
}
// Return database client
func (s *Storage) Conn() *pgxpool.Pool {
return s.db
}
// gcTicker starts the gc ticker
func (s *Storage) gcTicker() {
ticker := time.NewTicker(s.gcInterval)

View File

@@ -3,6 +3,7 @@ package v2
import (
"context"
"database/sql"
"os"
"testing"
"time"
@@ -10,7 +11,10 @@ import (
)
var testStore = New(Config{
Reset: true,
Database: os.Getenv("POSTGRES_DATABASE"),
Username: os.Getenv("POSTGRES_USERNAME"),
Password: os.Getenv("POSTGRES_PASSWORD"),
Reset: true,
})
func Test_Postgres_Set(t *testing.T) {
@@ -167,6 +171,10 @@ func Test_SslRequiredMode(t *testing.T) {
})
}
func Test_Postgres_Conn(t *testing.T) {
utils.AssertEqual(t, true, testStore.Conn() != nil)
}
func Test_Postgres_Close(t *testing.T) {
utils.AssertEqual(t, nil, testStore.Close())
}