notification/notification.go

80 lines
1.6 KiB
Go

package notification
import (
"encoding/json"
"errors"
"fmt"
"github.com/Selly-Modules/natsio"
)
const (
SendByTopic = "topic"
SendByUsers = "users"
)
const (
SubjectPushNotification = "push_notification"
SubjectGetNotification = "get_notification"
SubjectReadNotification = "read_notification"
SubjectCountUnreadNotification = "count_unread_notification"
)
// Client ...
type Client struct {
Config Config
natsServer natsio.Server
natsJetStream natsio.JetStream
}
// NewClient ...
func NewClient(cfg Config) (*Client, error) {
if cfg.APIKey == "" {
return nil, errors.New("api key is required")
}
if cfg.Nats.URL == "" {
return nil, errors.New("nats url is required")
}
if err := natsio.Connect(cfg.Nats); err != nil {
return nil, fmt.Errorf("nats connect failed: %v", err)
}
c := &Client{
Config: cfg,
natsServer: natsio.GetServer(),
natsJetStream: natsio.GetJetStream(),
}
return c, nil
}
// PushToUsers ...
func (c *Client) PushToUsers(payload PushRequest) (requestID string, err error) {
p := pushRequest{
APIKey: c.Config.APIKey,
Title: payload.Title,
Body: payload.Body,
Data: payload.Data,
SendBy: SendByUsers,
Users: payload.Users,
Label: payload.Label,
}
msg, err := c.natsServer.Request(SubjectPushNotification, toBytes(p))
if err != nil {
return "", err
}
var res PushResponse
if err := json.Unmarshal(msg.Data, &res); err != nil {
return "", err
}
if res.Error != "" {
return "", errors.New(res.Error)
}
return res.RequestID, nil
}
func toBytes(data interface{}) []byte {
b, _ := json.Marshal(data)
return b
}