54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
|
package document
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"forge.cadoles.com/arcad/edge/pkg/storage"
|
||
|
|
"forge.cadoles.com/arcad/edge/pkg/storage/filter"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
type QueryDocumentsArgs struct {
|
||
|
|
Collection string
|
||
|
|
Filter map[string]any
|
||
|
|
Options *storage.QueryOptions
|
||
|
|
}
|
||
|
|
|
||
|
|
type QueryDocumentsReply struct {
|
||
|
|
Documents []storage.Document
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) QueryDocuments(ctx context.Context, args QueryDocumentsArgs, reply *QueryDocumentsReply) error {
|
||
|
|
var (
|
||
|
|
argsFilter *filter.Filter
|
||
|
|
err error
|
||
|
|
)
|
||
|
|
|
||
|
|
if args.Filter != nil {
|
||
|
|
argsFilter, err = filter.NewFrom(args.Filter)
|
||
|
|
if err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
documents, err := s.store.Query(ctx, args.Collection, argsFilter, withQueryOptions(args.Options))
|
||
|
|
if err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
*reply = QueryDocumentsReply{
|
||
|
|
Documents: documents,
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func withQueryOptions(opts *storage.QueryOptions) storage.QueryOptionFunc {
|
||
|
|
return func(o *storage.QueryOptions) {
|
||
|
|
o.Limit = opts.Limit
|
||
|
|
o.Offset = opts.Offset
|
||
|
|
o.OrderBy = opts.OrderBy
|
||
|
|
o.OrderDirection = opts.OrderDirection
|
||
|
|
}
|
||
|
|
}
|