diff --git a/client.go b/client.go index 700955e..eca9526 100644 --- a/client.go +++ b/client.go @@ -75,6 +75,15 @@ func (h *PrivilegeHandler) Fetch() error { return nil } +func (h *PrivilegeHandler) Setup() []goamqp.Setup { + return []goamqp.Setup{ + goamqp.TransientEventStreamConsumer("User.Added", h.Process, UserAdded{}), + goamqp.TransientEventStreamConsumer("User.Removed", h.Process, UserRemoved{}), + goamqp.TransientEventStreamConsumer("Privilege.Added", h.Process, PrivilegeAdded{}), + goamqp.TransientEventStreamConsumer("Privilege.Removed", h.Process, PrivilegeRemoved{}), + } +} + // Process privilege-related events and update the internal state func (h *PrivilegeHandler) Process(msg interface{}, _ goamqp.Headers) (interface{}, error) { switch ev := msg.(type) {