Increase and decrease the task metrics counter with events

This commit is contained in:
kolaente 2021-02-01 21:27:37 +01:00
parent 56c712e6b6
commit dfbb3fe6f4
Signed by: konrad
GPG Key ID: F40E70337AB24C9B
3 changed files with 50 additions and 17 deletions

View File

@ -42,6 +42,17 @@ func (t *TaskUpdatedEvent) TopicName() string {
return "task.updated"
}
// TaskDeletedEvent represents a TaskDeletedEvent event
type TaskDeletedEvent struct {
Task *Task
Doer *user.User
}
// TopicName defines the name for TaskDeletedEvent
func (t *TaskDeletedEvent) TopicName() string {
return "task.deleted"
}
type TaskAssigneeCreatedEvent struct {
Task *Task
Assignee *user.User

View File

@ -17,35 +17,50 @@
package models
import (
"code.vikunja.io/api/pkg/metrics"
"encoding/json"
"code.vikunja.io/api/pkg/events"
"code.vikunja.io/api/pkg/metrics"
"github.com/ThreeDotsLabs/watermill/message"
)
func RegisterListeners() {
events.RegisterListener((&TaskCreatedEvent{}).TopicName(), &SendTaskCreatedNotification{})
events.RegisterListener((&ListCreatedEvent{}).TopicName(), &IncreaseListCounter{})
events.RegisterListener((&ListDeletedEvent{}).TopicName(), &DecreaseListCounter{})
events.RegisterListener((&NamespaceCreatedEvent{}).TopicName(), &IncreaseNamespaceCounter{})
events.RegisterListener((&NamespaceDeletedEvent{}).TopicName(), &DecreaseNamespaceCounter{})
events.RegisterListener((&TaskCreatedEvent{}).TopicName(), &IncreaseTaskCounter{})
events.RegisterListener((&TaskDeletedEvent{}).TopicName(), &DecreaseTaskCounter{})
}
type SendTaskCreatedNotification struct {
//////
// Task Events
// IncreaseTaskCounter represents a listener
type IncreaseTaskCounter struct {
}
func (s *SendTaskCreatedNotification) Name() string {
return "task.created.notification"
// Name defines the name for the IncreaseTaskCounter listener
func (s *IncreaseTaskCounter) Name() string {
return "increase.task.counter"
}
func (s *SendTaskCreatedNotification) Handle(payload message.Payload) (err error) {
event := &TaskCreatedEvent{}
err = json.Unmarshal(payload, event)
if err != nil {
return err
}
// Hanlde is executed when the event IncreaseTaskCounter listens on is fired
func (s *IncreaseTaskCounter) Handle(payload message.Payload) (err error) {
metrics.UpdateCount(1, metrics.TaskCountKey)
return nil
}
// DecreaseTaskCounter represents a listener
type DecreaseTaskCounter struct {
}
// Name defines the name for the DecreaseTaskCounter listener
func (s *DecreaseTaskCounter) Name() string {
return "decrease.task.counter"
}
// Hanlde is executed when the event DecreaseTaskCounter listens on is fired
func (s *DecreaseTaskCounter) Handle(payload message.Payload) (err error) {
metrics.UpdateCount(-1, metrics.TaskCountKey)
return nil
}

View File

@ -26,7 +26,6 @@ import (
"code.vikunja.io/api/pkg/db"
"code.vikunja.io/api/pkg/config"
"code.vikunja.io/api/pkg/metrics"
"code.vikunja.io/api/pkg/user"
"code.vikunja.io/api/pkg/utils"
"code.vikunja.io/web"
@ -825,8 +824,6 @@ func createTask(s *xorm.Session, t *Task, a web.Auth, updateAssignees bool) (err
return err
}
metrics.UpdateCount(1, metrics.TaskCountKey)
t.setIdentifier(l)
err = events.Publish(&TaskCreatedEvent{
@ -1198,7 +1195,17 @@ func (t *Task) Delete(s *xorm.Session, a web.Auth) (err error) {
return err
}
metrics.UpdateCount(-1, metrics.TaskCountKey)
doer, err := user.GetFromAuth(a)
if err != nil {
return err
}
err = events.Publish(&TaskDeletedEvent{
Task: t,
Doer: doer,
})
if err != nil {
return
}
err = updateListLastUpdated(s, &List{ID: t.ListID})
return