diff --git a/pkg/options/options.go b/pkg/options/options.go index 09296f359d42e838f4389cc106d8d93bc955cfd6..066d3f8aa347a4b08fb901c87e2fc9afee7aa15f 100644 --- a/pkg/options/options.go +++ b/pkg/options/options.go @@ -1,6 +1,10 @@ package options -import "time" +import ( + "time" + + commonpb "git.perx.ru/perxis/perxis-go/proto/common" +) // SortOptions настройки сортировки результатов type SortOptions struct { @@ -82,6 +86,56 @@ func MergeFindOptions(opts ...interface{}) *FindOptions { return fo } +func FindOptionsFromPB(protoOpts *commonpb.FindOptions) *FindOptions { + if protoOpts == nil { + return nil + } + fo := &FindOptions{ + SortOptions: SortOptions{ + Sort: protoOpts.Sort, + }, + PaginationOptions: PaginationOptions{ + Offset: int(protoOpts.Offset), + Limit: int(protoOpts.Limit), + }, + FieldOptions: FieldOptions{ + Fields: protoOpts.Fields, + ExcludeFields: protoOpts.ExcludeFields, + }, + } + + if fo.Offset == 0 && fo.Limit == 0 && fo.PageSize != 0 { + fo.Offset = fo.PageSize * (fo.PageNum - 1) + fo.Limit = fo.PageSize + } + + return fo +} + +func FindOptionsToPB(opts *FindOptions) *commonpb.FindOptions { + if opts == nil { + return nil + } + fo := &commonpb.FindOptions{ + Sort: opts.Sort, + PageNum: int32(opts.PageNum), + PageSize: int32(opts.PageSize), + Offset: int32(opts.Offset), + Limit: int32(opts.Limit), + Fields: opts.Fields, + ExcludeFields: opts.ExcludeFields, + } + if fo.PageSize != 0 && fo.Limit == 0 { + fo.Limit = fo.PageSize + fo.Offset = fo.PageSize * (fo.PageNum - 1) + } + if fo.Limit != 0 && fo.PageSize == 0 { + fo.PageSize = fo.Limit + fo.PageNum = fo.Offset / fo.Limit + } + return fo +} + type TimeFilter struct { Before, After time.Time }