Merge branch 'notarock/62'

services/custom_rule.go

@@ -0,0 +1,64 @@
+package services
+
+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),
+	}
+}
+
+func (srv *CustomRuleService) GetCustomRuleById(customRuleId uint) (*models.CustomRule, error) {
+	r := &models.CustomRule{}
+	if err := srv.Db.Where(&models.CustomRule{ID: customRuleId}).First(r).Error; err != nil {
+		return r, err
+	}
+	return r, nil
+}
+
+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
+}
+
+func (srv *CustomRuleService) Create(rule *models.CustomRule) (*models.CustomRule, error) {
+	result := srv.Db.Create(rule)
+	if err := result.Error; err != nil {
+		return nil, err
+	}
+	srv.cache.Delete(rule.UserID)
+
+	return rule, nil
+}
+
+func (srv *CustomRuleService) Delete(rule *models.CustomRule) {
+	srv.Db.
+		Where("id = ?", rule.ID).
+		Where("user_id = ?", rule.UserID).
+		Delete(models.CustomRule{})
+	srv.cache.Delete(rule.UserID)
+}

services/summary.go

@@ -17,20 +17,22 @@ import (
 const HeartbeatDiffThreshold = 2 * time.Minute
 
 type SummaryService struct {
-	Config           *config.Config
-	Cache            *cache.Cache
-	Db               *gorm.DB
-	HeartbeatService *HeartbeatService
-	AliasService     *AliasService
+	Config            *config.Config
+	Cache             *cache.Cache
+	Db                *gorm.DB
+	HeartbeatService  *HeartbeatService
+	AliasService      *AliasService
+	CustomRuleService *CustomRuleService
 }
 
-func NewSummaryService(db *gorm.DB, heartbeatService *HeartbeatService, aliasService *AliasService) *SummaryService {
+func NewSummaryService(db *gorm.DB, heartbeatService *HeartbeatService, aliasService *AliasService, customRuleService *CustomRuleService) *SummaryService {
 	return &SummaryService{
-		Config:           config.Get(),
-		Cache:            cache.New(24*time.Hour, 24*time.Hour),
-		Db:               db,
-		HeartbeatService: heartbeatService,
-		AliasService:     aliasService,
+		Config:            config.Get(),
+		Cache:             cache.New(24*time.Hour, 24*time.Hour),
+		Db:                db,
+		HeartbeatService:  heartbeatService,
+		AliasService:      aliasService,
+		CustomRuleService: customRuleService,
 	}
 }