grafana/pkg/services/thumbs/dashboardthumbsimpl/dashboard_thumbs.go
Kristin Laemmert 05709ce411
chore: remove sqlstore & mockstore dependencies from (most) packages (#57087)
* chore: add alias for InitTestDB and Session

Adds an alias for the sqlstore InitTestDB and Session, and updates tests using these to reduce dependencies on the sqlstore.Store.

* next pass of removing sqlstore imports
* last little bit
* remove mockstore where possible
2022-10-19 09:02:15 -04:00

44 lines
1.2 KiB
Go

package dashboardthumbsimpl
import (
"context"
"github.com/grafana/grafana/pkg/infra/db"
"github.com/grafana/grafana/pkg/services/thumbs"
)
type Service struct {
store store
}
func ProvideService(db db.DB) thumbs.DashboardThumbService {
return &Service{
store: &xormStore{db: db},
}
}
func (s *Service) GetThumbnail(ctx context.Context, query *thumbs.GetDashboardThumbnailCommand) (*thumbs.DashboardThumbnail, error) {
dt, err := s.store.Get(ctx, query)
return dt, err
}
func (s *Service) SaveThumbnail(ctx context.Context, cmd *thumbs.SaveDashboardThumbnailCommand) (*thumbs.DashboardThumbnail, error) {
dt, err := s.store.Save(ctx, cmd)
return dt, err
}
func (s *Service) UpdateThumbnailState(ctx context.Context, cmd *thumbs.UpdateThumbnailStateCommand) error {
err := s.store.UpdateState(ctx, cmd)
return err
}
func (s *Service) FindThumbnailCount(ctx context.Context, cmd *thumbs.FindDashboardThumbnailCountCommand) (int64, error) {
n, err := s.store.Count(ctx, cmd)
return n, err
}
func (s *Service) FindDashboardsWithStaleThumbnails(ctx context.Context, cmd *thumbs.FindDashboardsWithStaleThumbnailsCommand) ([]*thumbs.DashboardWithStaleThumbnail, error) {
thumbs, err := s.store.FindDashboardsWithStaleThumbnails(ctx, cmd)
return thumbs, err
}