108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package idec
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"gitrepo.ru/neonxp/idecnode/pkg/model"
|
|
"go.etcd.io/bbolt"
|
|
)
|
|
|
|
func (i *IDEC) GetMessagesByEcho(echoID string, offset, limit int) ([]string, error) {
|
|
messages := make([]string, 0)
|
|
|
|
return messages, i.db.View(func(tx *bbolt.Tx) error {
|
|
if _, ok := i.config.Echos[echoID]; !ok {
|
|
return nil
|
|
}
|
|
|
|
bEcho := tx.Bucket([]byte(echoID))
|
|
if bEcho == nil {
|
|
return nil
|
|
}
|
|
|
|
cur := bEcho.Cursor()
|
|
cur.First()
|
|
all := bEcho.Stats().KeyN
|
|
if limit == 0 {
|
|
limit = all
|
|
}
|
|
if offset < 0 {
|
|
offset = max(0, all+offset-1)
|
|
}
|
|
for i := 0; i < offset; i++ {
|
|
// skip offset entries
|
|
cur.Next()
|
|
}
|
|
for i := 0; i < limit; i++ {
|
|
_, v := cur.Next()
|
|
if v == nil {
|
|
break
|
|
}
|
|
messages = append(messages, string(v))
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (i *IDEC) GetMessage(messageID string) (*model.Message, error) {
|
|
var msg *model.Message
|
|
return msg, i.db.View(func(tx *bbolt.Tx) error {
|
|
bucket := tx.Bucket([]byte(msgBucket))
|
|
if bucket == nil {
|
|
return ErrMessageNotFound
|
|
}
|
|
b := bucket.Get([]byte(messageID))
|
|
var err error
|
|
msg, err = model.MessageFromBundle(messageID, string(b))
|
|
|
|
return err
|
|
})
|
|
}
|
|
|
|
func (i *IDEC) SavePointMessage(point string, rawMessage string) error {
|
|
rawMessage = strings.NewReplacer("-", "+", "_", "/").Replace(rawMessage)
|
|
|
|
messageBody, err := base64.StdEncoding.DecodeString(rawMessage)
|
|
if err != nil {
|
|
return ErrWrongMessageFormat
|
|
}
|
|
|
|
msg, err := model.MessageFromPointMessage(i.config.Node, point, string(messageBody))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return i.SaveMessage(msg)
|
|
}
|
|
|
|
func (i *IDEC) SaveBundleMessage(msgID, text string) error {
|
|
msg, err := model.MessageFromBundle(msgID, text)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return i.SaveMessage(msg)
|
|
}
|
|
|
|
func (i *IDEC) SaveMessage(msg *model.Message) error {
|
|
return i.db.Update(func(tx *bbolt.Tx) error {
|
|
bMessages, err := tx.CreateBucketIfNotExists([]byte(msgBucket))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := bMessages.Put([]byte(msg.ID), []byte(msg.Bundle())); err != nil {
|
|
return err
|
|
}
|
|
bEcho, err := tx.CreateBucketIfNotExists([]byte(msg.EchoArea))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
bucketKey := fmt.Sprintf("%s.%s", msg.Date.Format("2006.01.02.15.04.05"), msg.ID)
|
|
|
|
return bEcho.Put([]byte(bucketKey), []byte(msg.ID))
|
|
})
|
|
}
|