Refactoring

This commit is contained in:
2023-04-13 18:22:17 +03:00
parent f47dbefb67
commit 2b7a33e72d
9 changed files with 73 additions and 43 deletions

View File

@@ -20,19 +20,17 @@ type Pages interface {
GetFile(ctx context.Context, pageID, fileID uuid.UUID) (*entity.File, error)
}
func NewService(pages Pages, ch chan *entity.Page, processor entity.Processor) *Service {
func NewService(pages Pages, ch chan *entity.Page) *Service {
return &Service{
pages: pages,
ch: ch,
processor: processor,
pages: pages,
ch: ch,
}
}
type Service struct {
openapi.UnimplementedHandler
pages Pages
ch chan *entity.Page
processor entity.Processor
pages Pages
ch chan *entity.Page
}
func (s *Service) GetPage(ctx context.Context, params openapi.GetPageParams) (openapi.GetPageRes, error) {
@@ -83,13 +81,6 @@ func (s *Service) AddPage(ctx context.Context, req openapi.OptAddPageReq, params
page := entity.NewPage(url, description, domainFormats...)
page.Status = entity.StatusProcessing
meta, err := s.processor.GetMeta(ctx, page.URL)
if err != nil {
page.Meta.Error = err.Error()
} else {
page.Meta = meta
}
if err := s.pages.Save(ctx, page); err != nil {
return nil, fmt.Errorf("save page: %w", err)
}