package gormstore import ( "context" "errors" "git.itzana.me/strafesnet/maps-service/pkg/datastore" "git.itzana.me/strafesnet/maps-service/pkg/model" "gorm.io/gorm" "gorm.io/gorm/clause" ) type Mapfixes struct { db *gorm.DB } func (env *Mapfixes) Get(ctx context.Context, id int64) (model.Mapfix, error) { var mapfix model.Mapfix if err := env.db.First(&mapfix, id).Error; err != nil { if errors.Is(err, gorm.ErrRecordNotFound) { return mapfix, datastore.ErrNotExist } return mapfix, err } return mapfix, nil } func (env *Mapfixes) GetList(ctx context.Context, id []int64) ([]model.Mapfix, error) { var mapList []model.Mapfix if err := env.db.Find(&mapList, "id IN ?", id).Error; err != nil { return mapList, err } return mapList, nil } func (env *Mapfixes) Create(ctx context.Context, smap model.Mapfix) (model.Mapfix, error) { if err := env.db.Create(&smap).Error; err != nil { return smap, err } return smap, nil } func (env *Mapfixes) Update(ctx context.Context, id int64, values datastore.OptionalMap) error { if err := env.db.Model(&model.Mapfix{}).Where("id = ?", id).Updates(values.Map()).Error; err != nil { if err == gorm.ErrRecordNotFound { return datastore.ErrNotExist } return err } return nil } // the update can only occur if the status matches one of the provided values. func (env *Mapfixes) IfStatusThenUpdate(ctx context.Context, id int64, statuses []model.MapfixStatus, values datastore.OptionalMap) error { if err := env.db.Model(&model.Mapfix{}).Where("id = ?", id).Where("status_id IN ?", statuses).Updates(values.Map()).Error; err != nil { if err == gorm.ErrRecordNotFound { return datastore.ErrNotExist } return err } return nil } // the update can only occur if the status matches one of the provided values. // returns the updated value func (env *Mapfixes) IfStatusThenUpdateAndGet(ctx context.Context, id int64, statuses []model.MapfixStatus, values datastore.OptionalMap) (model.Mapfix, error) { var mapfix model.Mapfix result := env.db.Model(&mapfix). Clauses(clause.Returning{}). Where("id = ?", id). Where("status_id IN ?", statuses). Updates(values.Map()) if result.Error != nil { if result.Error == gorm.ErrRecordNotFound { return mapfix, datastore.ErrNotExist } return mapfix, result.Error } if result.RowsAffected == 0 { return mapfix, datastore.ErroNoRowsAffected } return mapfix, nil } func (env *Mapfixes) Delete(ctx context.Context, id int64) error { if err := env.db.Delete(&model.Mapfix{}, id).Error; err != nil { if err == gorm.ErrRecordNotFound { return datastore.ErrNotExist } return err } return nil } func (env *Mapfixes) List(ctx context.Context, filters datastore.OptionalMap, page model.Page, sort datastore.ListSort) ([]model.Mapfix, error) { var maps []model.Mapfix db := env.db switch sort { case datastore.ListSortDisabled: // No sort break case datastore.ListSortDisplayNameAscending: db=db.Order("display_name ASC") break case datastore.ListSortDisplayNameDescending: db=db.Order("display_name DESC") break case datastore.ListSortDateAscending: db=db.Order("created_at ASC") break case datastore.ListSortDateDescending: db=db.Order("created_at DESC") break default: return nil, datastore.ErrInvalidListSort } if err := db.Where(filters.Map()).Offset(int((page.Number - 1) * page.Size)).Limit(int(page.Size)).Find(&maps).Error; err != nil { return nil, err } return maps, nil }