131 lines
2.6 KiB
Go
131 lines
2.6 KiB
Go
package elasticsearch
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/Selly-Modules/natsio"
|
|
)
|
|
|
|
// Client ...
|
|
type Client struct {
|
|
Config Config
|
|
natsServer natsio.Server
|
|
natsJetStream natsio.JetStream
|
|
}
|
|
|
|
// NewClient
|
|
// Init client elasticsearch
|
|
func NewClient(config Config) (*Client, error) {
|
|
if config.ApiKey == "" {
|
|
return nil, errors.New("api key is required")
|
|
}
|
|
if config.Nats.URL == "" {
|
|
return nil, errors.New("nats url is required")
|
|
}
|
|
if err := natsio.Connect(config.Nats); err != nil {
|
|
return nil, fmt.Errorf("nats connect failed: %v", err)
|
|
}
|
|
|
|
c := &Client{
|
|
Config: config,
|
|
natsServer: natsio.GetServer(),
|
|
natsJetStream: natsio.GetJetStream(),
|
|
}
|
|
|
|
return c, nil
|
|
}
|
|
|
|
// SyncData
|
|
// Sync data to services ES
|
|
func (c *Client) SyncData(data SyncData) (bool, error) {
|
|
var (
|
|
req = RequestBody{
|
|
ApiKey: c.Config.ApiKey,
|
|
Body: toBytes(data),
|
|
}
|
|
)
|
|
err := c.natsJetStream.Publish(JetStreamSearchService, SubjectSyncData, toBytes(req))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// Search
|
|
// Request search to service es
|
|
func (c *Client) Search(query ESQuery) (*Response, error) {
|
|
var (
|
|
req = RequestBody{
|
|
ApiKey: c.Config.ApiKey,
|
|
Body: toBytes(query),
|
|
}
|
|
res *Response
|
|
)
|
|
msg, err := c.natsServer.Request(SubjectSearch, toBytes(req))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err = json.Unmarshal(msg.Data, &res); err != nil {
|
|
return nil, err
|
|
}
|
|
if res.Message != "" {
|
|
return nil, errors.New(res.Message)
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
// UpdateDocument
|
|
// Insert or update document to ES
|
|
func (c *Client) UpdateDocument(query UpdateDataPayload) (bool, error) {
|
|
var (
|
|
req = RequestBody{
|
|
ApiKey: c.Config.ApiKey,
|
|
Body: toBytes(query),
|
|
}
|
|
)
|
|
err := c.natsJetStream.Publish(JetStreamSearchService, SubjectUpdateDocument, toBytes(req))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// DeleteDocument
|
|
// Delete document to ES
|
|
func (c *Client) DeleteDocument(payload DeleteDataPayload) (bool, error) {
|
|
var (
|
|
req = RequestBody{
|
|
ApiKey: c.Config.ApiKey,
|
|
Body: toBytes(payload),
|
|
}
|
|
)
|
|
err := c.natsJetStream.Publish(JetStreamSearchService, SubjectUpdateDocument, toBytes(req))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// CreateIndex
|
|
// Create index to ES
|
|
func (c *Client) CreateIndex(name string) (bool, error) {
|
|
var (
|
|
req = RequestBody{
|
|
ApiKey: c.Config.ApiKey,
|
|
Body: toBytes(name),
|
|
}
|
|
)
|
|
err := c.natsJetStream.Publish(JetStreamSearchService, SubjectCreateIndex, toBytes(req))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
func toBytes(data interface{}) []byte {
|
|
b, _ := json.Marshal(data)
|
|
return b
|
|
}
|