130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
// Copyright (c) Mainflux
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package notifiers
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/mainflux/mainflux"
|
|
"github.com/mainflux/mainflux/consumers"
|
|
"github.com/mainflux/mainflux/pkg/errors"
|
|
"github.com/mainflux/mainflux/pkg/messaging"
|
|
)
|
|
|
|
var (
|
|
// ErrMessage indicates an error converting a message to Mainflux message.
|
|
ErrMessage = errors.New("failed to convert to Mainflux message")
|
|
)
|
|
|
|
// Service reprents a notification service.
|
|
type Service interface {
|
|
// CreateSubscription persists a subscription.
|
|
// Successful operation is indicated by non-nil error response.
|
|
CreateSubscription(ctx context.Context, token string, sub Subscription) (string, error)
|
|
|
|
// ViewSubscription retrieves the subscription for the given user and id.
|
|
ViewSubscription(ctx context.Context, token, id string) (Subscription, error)
|
|
|
|
// ListSubscriptions lists subscriptions having the provided user token and search params.
|
|
ListSubscriptions(ctx context.Context, token string, pm PageMetadata) (Page, error)
|
|
|
|
// RemoveSubscription removes the subscription having the provided identifier.
|
|
RemoveSubscription(ctx context.Context, token, id string) error
|
|
|
|
consumers.Consumer
|
|
}
|
|
|
|
var _ Service = (*notifierService)(nil)
|
|
|
|
type notifierService struct {
|
|
auth mainflux.AuthServiceClient
|
|
subs SubscriptionsRepository
|
|
idp mainflux.IDProvider
|
|
notifier Notifier
|
|
from string
|
|
}
|
|
|
|
// New instantiates the subscriptions service implementation.
|
|
func New(auth mainflux.AuthServiceClient, subs SubscriptionsRepository, idp mainflux.IDProvider, notifier Notifier, from string) Service {
|
|
return ¬ifierService{
|
|
auth: auth,
|
|
subs: subs,
|
|
idp: idp,
|
|
notifier: notifier,
|
|
from: from,
|
|
}
|
|
}
|
|
|
|
func (ns *notifierService) CreateSubscription(ctx context.Context, token string, sub Subscription) (string, error) {
|
|
res, err := ns.auth.Identify(ctx, &mainflux.Token{Value: token})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
sub.ID, err = ns.idp.ID()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
sub.OwnerID = res.GetId()
|
|
return ns.subs.Save(ctx, sub)
|
|
}
|
|
|
|
func (ns *notifierService) ViewSubscription(ctx context.Context, token, id string) (Subscription, error) {
|
|
if _, err := ns.auth.Identify(ctx, &mainflux.Token{Value: token}); err != nil {
|
|
return Subscription{}, err
|
|
}
|
|
|
|
return ns.subs.Retrieve(ctx, id)
|
|
}
|
|
|
|
func (ns *notifierService) ListSubscriptions(ctx context.Context, token string, pm PageMetadata) (Page, error) {
|
|
if _, err := ns.auth.Identify(ctx, &mainflux.Token{Value: token}); err != nil {
|
|
return Page{}, err
|
|
}
|
|
|
|
return ns.subs.RetrieveAll(ctx, pm)
|
|
}
|
|
|
|
func (ns *notifierService) RemoveSubscription(ctx context.Context, token, id string) error {
|
|
if _, err := ns.auth.Identify(ctx, &mainflux.Token{Value: token}); err != nil {
|
|
return err
|
|
}
|
|
|
|
return ns.subs.Remove(ctx, id)
|
|
}
|
|
|
|
func (ns *notifierService) Consume(message interface{}) error {
|
|
msg, ok := message.(messaging.Message)
|
|
if !ok {
|
|
return ErrMessage
|
|
}
|
|
topic := msg.Channel
|
|
if msg.Subtopic != "" {
|
|
topic = fmt.Sprintf("%s.%s", msg.Channel, msg.Subtopic)
|
|
}
|
|
pm := PageMetadata{
|
|
Topic: topic,
|
|
Offset: 0,
|
|
Limit: -1,
|
|
}
|
|
page, err := ns.subs.RetrieveAll(context.Background(), pm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var to []string
|
|
for _, sub := range page.Subscriptions {
|
|
to = append(to, sub.Contact)
|
|
}
|
|
if len(to) > 0 {
|
|
err := ns.notifier.Notify(ns.from, to, msg)
|
|
if err != nil {
|
|
return errors.Wrap(ErrNotify, err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|