|
|
package repository
|
|
|
|
|
|
import (
|
|
|
"context"
|
|
|
"github.com/jinzhu/copier"
|
|
|
"github.com/pkg/errors"
|
|
|
"github.com/tiptok/gocomm/pkg/cache"
|
|
|
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/db/models"
|
|
|
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/db/transaction"
|
|
|
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/domain"
|
|
|
"gorm.io/gorm"
|
|
|
)
|
|
|
|
|
|
type UserFollowRepository struct {
|
|
|
*cache.CachedRepository
|
|
|
}
|
|
|
|
|
|
func (repository *UserFollowRepository) Insert(ctx context.Context, conn transaction.Conn, dm *domain.UserFollow) (*domain.UserFollow, error) {
|
|
|
var (
|
|
|
err error
|
|
|
m = &models.UserFollow{}
|
|
|
tx = conn.DB()
|
|
|
)
|
|
|
if m, err = repository.DomainModelToModel(dm); err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
if tx = tx.Model(m).Save(m); tx.Error != nil {
|
|
|
return nil, tx.Error
|
|
|
}
|
|
|
dm.Id = m.Id
|
|
|
return repository.ModelToDomainModel(m)
|
|
|
|
|
|
}
|
|
|
|
|
|
func (repository *UserFollowRepository) Update(ctx context.Context, conn transaction.Conn, dm *domain.UserFollow) (*domain.UserFollow, error) {
|
|
|
var (
|
|
|
err error
|
|
|
m *models.UserFollow
|
|
|
tx = conn.DB()
|
|
|
)
|
|
|
if m, err = repository.DomainModelToModel(dm); err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
tx = tx.Model(m).Updates(m)
|
|
|
return nil, tx.Error
|
|
|
}
|
|
|
if _, err = repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
return repository.ModelToDomainModel(m)
|
|
|
}
|
|
|
|
|
|
func (repository *UserFollowRepository) UpdateWithVersion(ctx context.Context, transaction transaction.Conn, dm *domain.UserFollow) (*domain.UserFollow, error) {
|
|
|
var (
|
|
|
err error
|
|
|
m *models.UserFollow
|
|
|
tx = transaction.DB()
|
|
|
)
|
|
|
if m, err = repository.DomainModelToModel(dm); err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
oldVersion := dm.Version
|
|
|
m.Version += 1
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
tx = tx.Model(m).Select("*").Where("id = ?", m.Id).Where("version = ?", oldVersion).Updates(m)
|
|
|
if tx.RowsAffected == 0 {
|
|
|
return nil, domain.ErrUpdateFail
|
|
|
}
|
|
|
return nil, tx.Error
|
|
|
}
|
|
|
if _, err = repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
return repository.ModelToDomainModel(m)
|
|
|
}
|
|
|
|
|
|
func (repository *UserFollowRepository) Delete(ctx context.Context, conn transaction.Conn, dm *domain.UserFollow) (*domain.UserFollow, error) {
|
|
|
var (
|
|
|
tx = conn.DB()
|
|
|
m = &models.UserFollow{Id: dm.Identify().(int64)}
|
|
|
)
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
tx = tx.Where("id = ?", m.Id).Delete(m)
|
|
|
return m, tx.Error
|
|
|
}
|
|
|
if _, err := repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
|
|
|
return dm, err
|
|
|
}
|
|
|
return repository.ModelToDomainModel(m)
|
|
|
}
|
|
|
|
|
|
func (repository *UserFollowRepository) FindOne(ctx context.Context, conn transaction.Conn, id int64) (*domain.UserFollow, error) {
|
|
|
var (
|
|
|
err error
|
|
|
tx = conn.DB()
|
|
|
m = new(models.UserFollow)
|
|
|
)
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
tx = tx.Model(m).Where("id = ?", id).First(m)
|
|
|
if errors.Is(tx.Error, gorm.ErrRecordNotFound) {
|
|
|
return nil, domain.ErrNotFound
|
|
|
}
|
|
|
return m, tx.Error
|
|
|
}
|
|
|
cacheModel := new(models.UserFollow)
|
|
|
cacheModel.Id = id
|
|
|
if err = repository.QueryCache(cacheModel.CacheKeyFunc, m, queryFunc); err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
return repository.ModelToDomainModel(m)
|
|
|
}
|
|
|
|
|
|
func (repository *UserFollowRepository) Find(ctx context.Context, conn transaction.Conn, queryOptions map[string]interface{}) (int64, []*domain.UserFollow, error) {
|
|
|
var (
|
|
|
tx = conn.DB()
|
|
|
ms []*models.UserFollow
|
|
|
dms = make([]*domain.UserFollow, 0)
|
|
|
total int64
|
|
|
)
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
tx = tx.Model(&ms).Order("id desc")
|
|
|
if total, tx = transaction.PaginationAndCount(ctx, tx, queryOptions, &ms); tx.Error != nil {
|
|
|
return dms, tx.Error
|
|
|
}
|
|
|
return dms, nil
|
|
|
}
|
|
|
|
|
|
if _, err := repository.Query(queryFunc); err != nil {
|
|
|
return 0, nil, err
|
|
|
}
|
|
|
|
|
|
for _, item := range ms {
|
|
|
if dm, err := repository.ModelToDomainModel(item); err != nil {
|
|
|
return 0, dms, err
|
|
|
} else {
|
|
|
dms = append(dms, dm)
|
|
|
}
|
|
|
}
|
|
|
return total, dms, nil
|
|
|
}
|
|
|
|
|
|
func (repository *UserFollowRepository) ModelToDomainModel(from *models.UserFollow) (*domain.UserFollow, error) {
|
|
|
to := &domain.UserFollow{}
|
|
|
err := copier.Copy(to, from)
|
|
|
return to, err
|
|
|
}
|
|
|
|
|
|
func (repository *UserFollowRepository) DomainModelToModel(from *domain.UserFollow) (*models.UserFollow, error) {
|
|
|
to := &models.UserFollow{}
|
|
|
err := copier.Copy(to, from)
|
|
|
return to, err
|
|
|
}
|
|
|
|
|
|
func NewUserFollowRepository(cache *cache.CachedRepository) domain.UserFollowRepository {
|
|
|
return &UserFollowRepository{CachedRepository: cache}
|
|
|
} |
...
|
...
|
|