Increase and decrease the task metrics counter with events
This commit is contained in:
parent
56c712e6b6
commit
dfbb3fe6f4
|
@ -42,6 +42,17 @@ func (t *TaskUpdatedEvent) TopicName() string {
|
||||||
return "task.updated"
|
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 {
|
type TaskAssigneeCreatedEvent struct {
|
||||||
Task *Task
|
Task *Task
|
||||||
Assignee *user.User
|
Assignee *user.User
|
||||||
|
|
|
@ -17,35 +17,50 @@
|
||||||
package models
|
package models
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"code.vikunja.io/api/pkg/metrics"
|
|
||||||
"encoding/json"
|
|
||||||
|
|
||||||
"code.vikunja.io/api/pkg/events"
|
"code.vikunja.io/api/pkg/events"
|
||||||
|
"code.vikunja.io/api/pkg/metrics"
|
||||||
"github.com/ThreeDotsLabs/watermill/message"
|
"github.com/ThreeDotsLabs/watermill/message"
|
||||||
)
|
)
|
||||||
|
|
||||||
func RegisterListeners() {
|
func RegisterListeners() {
|
||||||
events.RegisterListener((&TaskCreatedEvent{}).TopicName(), &SendTaskCreatedNotification{})
|
|
||||||
events.RegisterListener((&ListCreatedEvent{}).TopicName(), &IncreaseListCounter{})
|
events.RegisterListener((&ListCreatedEvent{}).TopicName(), &IncreaseListCounter{})
|
||||||
events.RegisterListener((&ListDeletedEvent{}).TopicName(), &DecreaseListCounter{})
|
events.RegisterListener((&ListDeletedEvent{}).TopicName(), &DecreaseListCounter{})
|
||||||
events.RegisterListener((&NamespaceCreatedEvent{}).TopicName(), &IncreaseNamespaceCounter{})
|
events.RegisterListener((&NamespaceCreatedEvent{}).TopicName(), &IncreaseNamespaceCounter{})
|
||||||
events.RegisterListener((&NamespaceDeletedEvent{}).TopicName(), &DecreaseNamespaceCounter{})
|
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 {
|
// Name defines the name for the IncreaseTaskCounter listener
|
||||||
return "task.created.notification"
|
func (s *IncreaseTaskCounter) Name() string {
|
||||||
|
return "increase.task.counter"
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SendTaskCreatedNotification) Handle(payload message.Payload) (err error) {
|
// Hanlde is executed when the event IncreaseTaskCounter listens on is fired
|
||||||
event := &TaskCreatedEvent{}
|
func (s *IncreaseTaskCounter) Handle(payload message.Payload) (err error) {
|
||||||
err = json.Unmarshal(payload, event)
|
metrics.UpdateCount(1, metrics.TaskCountKey)
|
||||||
if err != nil {
|
return nil
|
||||||
return err
|
}
|
||||||
}
|
|
||||||
|
|
||||||
|
// 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
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -26,7 +26,6 @@ import (
|
||||||
"code.vikunja.io/api/pkg/db"
|
"code.vikunja.io/api/pkg/db"
|
||||||
|
|
||||||
"code.vikunja.io/api/pkg/config"
|
"code.vikunja.io/api/pkg/config"
|
||||||
"code.vikunja.io/api/pkg/metrics"
|
|
||||||
"code.vikunja.io/api/pkg/user"
|
"code.vikunja.io/api/pkg/user"
|
||||||
"code.vikunja.io/api/pkg/utils"
|
"code.vikunja.io/api/pkg/utils"
|
||||||
"code.vikunja.io/web"
|
"code.vikunja.io/web"
|
||||||
|
@ -825,8 +824,6 @@ func createTask(s *xorm.Session, t *Task, a web.Auth, updateAssignees bool) (err
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
metrics.UpdateCount(1, metrics.TaskCountKey)
|
|
||||||
|
|
||||||
t.setIdentifier(l)
|
t.setIdentifier(l)
|
||||||
|
|
||||||
err = events.Publish(&TaskCreatedEvent{
|
err = events.Publish(&TaskCreatedEvent{
|
||||||
|
@ -1198,7 +1195,17 @@ func (t *Task) Delete(s *xorm.Session, a web.Auth) (err error) {
|
||||||
return err
|
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})
|
err = updateListLastUpdated(s, &List{ID: t.ListID})
|
||||||
return
|
return
|
||||||
|
|
Loading…
Reference in New Issue