diff --git a/docker-compose.yml b/docker-compose.yml index 36ce414..275c6df 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -13,7 +13,7 @@ services: WAKAPI_DB_PASSWORD: "asdfasdfasdf" WAKAPI_DB_HOST: "db" WAKAPI_DB_PORT: "5432" - ENV: "dev" + ENVIRONMENT: "dev" db: image: postgres:12.3 diff --git a/services/custom_rule.go b/services/custom_rule.go index be0784d..58c0079 100644 --- a/services/custom_rule.go +++ b/services/custom_rule.go @@ -4,17 +4,21 @@ import ( "github.com/jinzhu/gorm" "github.com/muety/wakapi/config" "github.com/muety/wakapi/models" + "github.com/patrickmn/go-cache" + "time" ) type CustomRuleService struct { Config *config.Config Db *gorm.DB + cache *cache.Cache } func NewCustomRuleService(db *gorm.DB) *CustomRuleService { return &CustomRuleService{ Config: config.Get(), Db: db, + cache: cache.New(1*time.Hour, 2*time.Hour), } } @@ -28,12 +32,16 @@ func (srv *CustomRuleService) GetCustomRuleById(customRuleId uint) (*models.Cust func (srv *CustomRuleService) GetCustomRuleForUser(userId string) ([]*models.CustomRule, error) { var rules []*models.CustomRule + if rules, found := srv.cache.Get(userId); found { + return rules.([]*models.CustomRule), nil; + } + if err := srv.Db. Where(&models.CustomRule{UserID: userId}). Find(&rules).Error; err != nil { return rules, err } - + srv.cache.Set(userId, rules, cache.DefaultExpiration) return rules, nil } @@ -42,6 +50,7 @@ func (srv *CustomRuleService) Create(rule *models.CustomRule) (*models.CustomRul if err := result.Error; err != nil { return nil, err } + srv.cache.Delete(rule.UserID) return rule, nil } @@ -51,4 +60,5 @@ func (srv *CustomRuleService) Delete(rule *models.CustomRule) { Where("id = ?", rule.ID). Where("user_id = ?", rule.UserID). Delete(models.CustomRule{}) + srv.cache.Delete(rule.UserID) }