80 lines
1.6 KiB
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
|
||
|
|
}
|