|
|
|
package repository
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"fmt"
|
|
|
|
"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 ArticleSecurityRepository struct {
|
|
|
|
*cache.CachedRepository
|
|
|
|
}
|
|
|
|
|
|
|
|
func (repository *ArticleSecurityRepository) Insert(ctx context.Context, conn transaction.Conn, dm *domain.ArticleSecurity) (*domain.ArticleSecurity, error) {
|
|
|
|
var (
|
|
|
|
err error
|
|
|
|
m = &models.ArticleSecurity{}
|
|
|
|
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 *ArticleSecurityRepository) Update(ctx context.Context, conn transaction.Conn, dm *domain.ArticleSecurity) (*domain.ArticleSecurity, error) {
|
|
|
|
var (
|
|
|
|
err error
|
|
|
|
m *models.ArticleSecurity
|
|
|
|
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 *ArticleSecurityRepository) UpdateWithVersion(ctx context.Context, transaction transaction.Conn, dm *domain.ArticleSecurity) (*domain.ArticleSecurity, error) {
|
|
|
|
var (
|
|
|
|
err error
|
|
|
|
m *models.ArticleSecurity
|
|
|
|
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 *ArticleSecurityRepository) Delete(ctx context.Context, conn transaction.Conn, dm *domain.ArticleSecurity) (*domain.ArticleSecurity, error) {
|
|
|
|
var (
|
|
|
|
tx = conn.DB()
|
|
|
|
m = &models.ArticleSecurity{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 *ArticleSecurityRepository) FindOne(ctx context.Context, conn transaction.Conn, id int64) (*domain.ArticleSecurity, error) {
|
|
|
|
var (
|
|
|
|
err error
|
|
|
|
tx = conn.DB()
|
|
|
|
m = new(models.ArticleSecurity)
|
|
|
|
)
|
|
|
|
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.ArticleSecurity)
|
|
|
|
cacheModel.Id = id
|
|
|
|
if err = repository.QueryCache(cacheModel.CacheKeyFunc, m, queryFunc); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return repository.ModelToDomainModel(m)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (repository *ArticleSecurityRepository) Find(ctx context.Context, conn transaction.Conn, queryOptions map[string]interface{}) (int64, []*domain.ArticleSecurity, error) {
|
|
|
|
var (
|
|
|
|
tx = conn.DB()
|
|
|
|
ms []*models.ArticleSecurity
|
|
|
|
dms = make([]*domain.ArticleSecurity, 0)
|
|
|
|
total int64
|
|
|
|
)
|
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
|
tx = tx.Model(&ms).Order("id desc")
|
|
|
|
if v, ok := queryOptions["authorName"]; ok {
|
|
|
|
tx.Where("author_name like ? ", fmt.Sprintf("%%%v%%", v))
|
|
|
|
}
|
|
|
|
if v, ok := queryOptions["contentType"]; ok {
|
|
|
|
tx.Where("content_type = ? ", v)
|
|
|
|
}
|
|
|
|
if v, ok := queryOptions["reviewStatus"]; ok {
|
|
|
|
tx.Where("review_status = ? ", v)
|
|
|
|
}
|
|
|
|
if v, ok := queryOptions["suggest"]; ok {
|
|
|
|
tx.Where("suggest = ? ", v)
|
|
|
|
}
|
|
|
|
if v, ok := queryOptions["beginTime"]; ok {
|
|
|
|
tx.Where("created_at >= ?", v)
|
|
|
|
}
|
|
|
|
if v, ok := queryOptions["endTime"]; ok {
|
|
|
|
tx.Where("created_at < ?", v)
|
|
|
|
}
|
|
|
|
|
|
|
|
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 *ArticleSecurityRepository) ModelToDomainModel(from *models.ArticleSecurity) (*domain.ArticleSecurity, error) {
|
|
|
|
to := &domain.ArticleSecurity{}
|
|
|
|
err := copier.Copy(to, from)
|
|
|
|
return to, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (repository *ArticleSecurityRepository) DomainModelToModel(from *domain.ArticleSecurity) (*models.ArticleSecurity, error) {
|
|
|
|
to := &models.ArticleSecurity{}
|
|
|
|
err := copier.Copy(to, from)
|
|
|
|
return to, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewArticleSecurityRepository(cache *cache.CachedRepository) domain.ArticleSecurityRepository {
|
|
|
|
return &ArticleSecurityRepository{CachedRepository: cache}
|
|
|
|
} |
...
|
...
|
|
