82 lines
1.5 KiB
Go
82 lines
1.5 KiB
Go
package search
|
|
|
|
import (
|
|
"encoding/json"
|
|
"voltaserve/infra"
|
|
"voltaserve/model"
|
|
"voltaserve/repo"
|
|
)
|
|
|
|
type GroupSearch struct {
|
|
index string
|
|
search *infra.SearchManager
|
|
groupRepo repo.GroupRepo
|
|
}
|
|
|
|
func NewGroupSearch() *GroupSearch {
|
|
return &GroupSearch{
|
|
index: infra.GroupSearchIndex,
|
|
search: infra.NewSearchManager(),
|
|
groupRepo: repo.NewGroupRepo(),
|
|
}
|
|
}
|
|
|
|
func (s *GroupSearch) Index(groups []model.Group) error {
|
|
if len(groups) == 0 {
|
|
return nil
|
|
}
|
|
var res []infra.SearchModel
|
|
for _, g := range groups {
|
|
res = append(res, g)
|
|
}
|
|
if err := s.search.Index(s.index, res); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *GroupSearch) Update(groups []model.Group) error {
|
|
if len(groups) == 0 {
|
|
return nil
|
|
}
|
|
var res []infra.SearchModel
|
|
for _, g := range groups {
|
|
res = append(res, g)
|
|
}
|
|
if err := s.search.Update(s.index, res); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *GroupSearch) Delete(ids []string) error {
|
|
if len(ids) == 0 {
|
|
return nil
|
|
}
|
|
if err := s.search.Delete(s.index, ids); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *GroupSearch) Query(query string) ([]model.Group, error) {
|
|
hits, err := s.search.Query(s.index, query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var res []model.Group
|
|
for _, v := range hits {
|
|
var b []byte
|
|
b, err = json.Marshal(v)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
group := repo.NewGroup()
|
|
if err = json.Unmarshal(b, &group); err != nil {
|
|
return nil, err
|
|
}
|
|
res = append(res, group)
|
|
}
|
|
return res, nil
|
|
}
|