Merge pull request #61 from dscsnu/db-cache-layer

Db cache layer

Author: RachitKumar205

backend/go.mod

@@ -23,6 +23,7 @@ require (
 	github.com/jackc/puddle/v2 v2.2.2 // indirect
 	github.com/josharian/intern v1.0.0 // indirect
 	github.com/mailru/easyjson v0.9.0 // indirect
+	github.com/patrickmn/go-cache v2.1.0+incompatible // indirect
 	github.com/stretchr/testify v1.10.0 // indirect
 	github.com/swaggo/files v0.0.0-20220610200504-28940afbdbfe // indirect
 	github.com/swaggo/files/v2 v2.0.0 // indirect

backend/go.sum

@@ -34,6 +34,8 @@ github.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFF
 github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
 github.com/mailru/easyjson v0.9.0 h1:PrnmzHw7262yW8sTBwxi1PdJA3Iw/EKBa8psRf7d9a4=
 github.com/mailru/easyjson v0.9.0/go.mod h1:1+xMtQp2MRNVL/V1bOzuP3aP8VNwRW55fQUto+XFtTU=
+github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc=
+github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/rs/cors v1.11.1 h1:eU3gRzXLRK57F5rKMGMZURNdIG4EoAmX8k94r9wXWHA=

backend/internal/cache/cache.go

@@ -0,0 +1,45 @@
+package cache
+
+import (
+	"time"
+
+	"github.com/patrickmn/go-cache"
+)
+
+type Cache struct {
+	c *cache.Cache
+}
+
+func New(defaultExpiration, cleanupInterval time.Duration) *Cache {
+	return &Cache{
+		c: cache.New(defaultExpiration, cleanupInterval),
+	}
+}
+
+func (c *Cache) Set(key string, value interface{}, expiration time.Duration) {
+	c.c.Set(key, value, expiration)
+}
+
+func (c *Cache) Get(key string) (interface{}, bool) {
+	return c.c.Get(key)
+}
+
+func (c *Cache) Delete(key string) {
+	c.c.Delete(key)
+}
+
+func (c *Cache) Flush() {
+	c.c.Flush()
+}
+
+func (c *Cache) SetTTL(defaultExpiration, cleanupInterval time.Duration) {
+	c.c = cache.New(defaultExpiration, cleanupInterval)
+}
+
+func DefaultExpiration() time.Duration {
+	return cache.DefaultExpiration
+}
+
+func NoExpiration() time.Duration {
+	return cache.NoExpiration
+}

backend/internal/database/database.go

@@ -3,7 +3,9 @@ package database
 import (
 	"context"
 	"crypto/rand"
+	"labyrinth/internal/cache"
 	"math/big"
+	"time"
 
 	"github.com/google/uuid"
 	"github.com/jackc/pgx/v5/pgtype"
@@ -30,7 +32,8 @@ func genRand() (string, error) {
 }
 
 type PostgresDriver struct {
-	pool *pgxpool.Pool
+	pool  *pgxpool.Pool
+	cache *cache.Cache
 }
 
 func (pd *PostgresDriver) Close() {
@@ -42,5 +45,11 @@ func CreatePostgresDriver(connectionURL string) (*PostgresDriver, error) {
 	if err != nil {
 		return nil, err
 	}
-	return &PostgresDriver{pool: pool}, nil
+
+	c := cache.New(5*time.Minute, 10*time.Minute)
+
+	return &PostgresDriver{
+		pool:  pool,
+		cache: c,
+	}, nil
 }

backend/internal/database/team_database.go

@@ -85,6 +85,14 @@ func (pd *PostgresDriver) LeaveTeamMember(ctx context.Context, teamId string, us
 }
 
 func (pd *PostgresDriver) GetTeamByID(ctx context.Context, teamId string) (types.Team, error) {
+	cacheKey := "team:" + teamId
+
+	if cachedData, found := pd.cache.Get(cacheKey); found {
+		if team, ok := cachedData.(types.Team); ok {
+			return team, nil
+		}
+	}
+
 	team := types.Team{}
 
 	err := pd.pool.QueryRow(ctx, "SELECT id, name FROM team WHERE team.id=$1", teamId).Scan(&team.ID, &team.Name)
@@ -114,8 +122,10 @@ func (pd *PostgresDriver) GetTeamByID(ctx context.Context, teamId string) (types
 	}
 
 	team.Members = members
-	return team, nil
 
+	pd.cache.Set(cacheKey, team, 5*time.Minute)
+
+	return team, nil
 }
 
 func (pd *PostgresDriver) GetTeamByUserId(ctx context.Context, userId uuid.UUID) (types.Team, error) {