diff options
Diffstat (limited to 'rules_manager.go')
| -rw-r--r-- | rules_manager.go | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/rules_manager.go b/rules_manager.go index 2e32573..1ddde5e 100644 --- a/rules_manager.go +++ b/rules_manager.go @@ -80,7 +80,7 @@ type RulesDatabase struct { } type RulesManager interface { - AddRule(context context.Context, rule Rule) (RowID, error) + AddRule(context context.Context, rule Rule) (Rule, error) GetRule(id RowID) (Rule, bool) UpdateRule(context context.Context, id RowID, rule Rule) (bool, error) DeleteRule(context context.Context, id RowID) error @@ -150,7 +150,7 @@ func LoadRulesManager(storage Storage, flagRegex string) (RulesManager, error) { return &rulesManager, nil } -func (rm *rulesManagerImpl) AddRule(context context.Context, rule Rule) (RowID, error) { +func (rm *rulesManagerImpl) AddRule(context context.Context, rule Rule) (Rule, error) { rm.mutex.Lock() rule.ID = CustomRowID(uint64(len(rm.rules)), time.Now()) @@ -158,7 +158,7 @@ func (rm *rulesManagerImpl) AddRule(context context.Context, rule Rule) (RowID, if err := rm.validateAndAddRuleLocal(&rule); err != nil { rm.mutex.Unlock() - return EmptyRowID(), err + return Rule{}, err } if err := rm.generateDatabase(rule.ID); err != nil { @@ -171,7 +171,7 @@ func (rm *rulesManagerImpl) AddRule(context context.Context, rule Rule) (RowID, log.WithError(err).WithField("rule", rule).Panic("failed to insert rule on database") } - return rule.ID, nil + return rule, nil } func (rm *rulesManagerImpl) GetRule(id RowID) (Rule, bool) { |