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) Delete(key string) error
func (s *Storage) Reset() error func (s *Storage) Reset() error
func (s *Storage) Close() error func (s *Storage) Close() error
func (s *Storage) Conn() *pgxpool.Pool
``` ```
### Installation ### 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: 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 nil
} }
// Return database client
func (s *Storage) Conn() *pgxpool.Pool {
return s.db
}
// gcTicker starts the gc ticker // gcTicker starts the gc ticker
func (s *Storage) gcTicker() { func (s *Storage) gcTicker() {
ticker := time.NewTicker(s.gcInterval) ticker := time.NewTicker(s.gcInterval)

View File

@@ -3,6 +3,7 @@ package v2
import ( import (
"context" "context"
"database/sql" "database/sql"
"os"
"testing" "testing"
"time" "time"
@@ -10,6 +11,9 @@ import (
) )
var testStore = New(Config{ var testStore = New(Config{
Database: os.Getenv("POSTGRES_DATABASE"),
Username: os.Getenv("POSTGRES_USERNAME"),
Password: os.Getenv("POSTGRES_PASSWORD"),
Reset: true, Reset: true,
}) })
@@ -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) { func Test_Postgres_Close(t *testing.T) {
utils.AssertEqual(t, nil, testStore.Close()) utils.AssertEqual(t, nil, testStore.Close())
} }