|
|
|
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 UserRoleRepository struct {
|
|
|
|
*cache.CachedRepository
|
|
|
|
}
|
|
|
|
|
|
|
|
func (repository *UserRoleRepository) Insert(ctx context.Context, conn transaction.Conn, dm *domain.UserRole) (*domain.UserRole, error) {
|
|
|
|
var (
|
|
|
|
err error
|
|
|
|
m = &models.UserRole{}
|
|
|
|
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 *UserRoleRepository) Update(ctx context.Context, conn transaction.Conn, dm *domain.UserRole) (*domain.UserRole, error) {
|
|
|
|
var (
|
|
|
|
err error
|
|
|
|
m *models.UserRole
|
|
|
|
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 *UserRoleRepository) UpdateWithVersion(ctx context.Context, transaction transaction.Conn, dm *domain.UserRole) (*domain.UserRole, error) {
|
|
|
|
var (
|
|
|
|
err error
|
|
|
|
m *models.UserRole
|
|
|
|
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 *UserRoleRepository) Delete(ctx context.Context, conn transaction.Conn, dm *domain.UserRole) (*domain.UserRole, error) {
|
|
|
|
var (
|
|
|
|
tx = conn.DB()
|
|
|
|
m = &models.UserRole{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 *UserRoleRepository) DeleteByUser(ctx context.Context, conn transaction.Conn, userId int64) error {
|
|
|
|
var (
|
|
|
|
tx = conn.DB()
|
|
|
|
m = &models.UserRole{}
|
|
|
|
)
|
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
|
tx = tx.Where("user_id = ?", userId).Delete(m)
|
|
|
|
return m, tx.Error
|
|
|
|
}
|
|
|
|
if _, err := repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (repository *UserRoleRepository) DeleteRole(ctx context.Context, conn transaction.Conn, roleId int64) error {
|
|
|
|
var (
|
|
|
|
tx = conn.DB()
|
|
|
|
m = &models.UserRole{}
|
|
|
|
)
|
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
|
tx = tx.Where("role_id = ?", roleId).Delete(m)
|
|
|
|
return m, tx.Error
|
|
|
|
}
|
|
|
|
if _, err := repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (repository *UserRoleRepository) DeleteByUserAndRole(ctx context.Context, conn transaction.Conn, userId int64, roleId int64) error {
|
|
|
|
var (
|
|
|
|
tx = conn.DB()
|
|
|
|
m = &models.UserRole{}
|
|
|
|
)
|
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
|
tx = tx.Where("user_id = ?", userId).Where("role_id = ?", roleId).Delete(m)
|
|
|
|
return m, tx.Error
|
|
|
|
}
|
|
|
|
if _, err := repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (repository *UserRoleRepository) FindOne(ctx context.Context, conn transaction.Conn, id int64) (*domain.UserRole, error) {
|
|
|
|
var (
|
|
|
|
err error
|
|
|
|
tx = conn.DB()
|
|
|
|
m = new(models.UserRole)
|
|
|
|
)
|
|
|
|
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.UserRole)
|
|
|
|
cacheModel.Id = id
|
|
|
|
if err = repository.QueryCache(cacheModel.CacheKeyFunc, m, queryFunc); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return repository.ModelToDomainModel(m)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (repository *UserRoleRepository) Find(ctx context.Context, conn transaction.Conn, queryOptions map[string]interface{}) (int64, []*domain.UserRole, error) {
|
|
|
|
var (
|
|
|
|
tx = conn.DB()
|
|
|
|
ms []*models.UserRole
|
|
|
|
dms = make([]*domain.UserRole, 0)
|
|
|
|
total int64
|
|
|
|
)
|
|
|
|
queryFunc := func() (interface{}, error) {
|
|
|
|
tx = tx.Model(&ms).Order("id desc")
|
|
|
|
if v, ok := queryOptions["companyId"]; ok {
|
|
|
|
tx.Where("company_id = ?", v)
|
|
|
|
}
|
|
|
|
if v, ok := queryOptions["roleId"]; ok {
|
|
|
|
tx.Where("role_id = ?", 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 *UserRoleRepository) ModelToDomainModel(from *models.UserRole) (*domain.UserRole, error) {
|
|
|
|
to := &domain.UserRole{}
|
|
|
|
err := copier.Copy(to, from)
|
|
|
|
return to, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (repository *UserRoleRepository) DomainModelToModel(from *domain.UserRole) (*models.UserRole, error) {
|
|
|
|
to := &models.UserRole{}
|
|
|
|
err := copier.Copy(to, from)
|
|
|
|
return to, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewUserRoleRepository(cache *cache.CachedRepository) domain.UserRoleRepository {
|
|
|
|
return &UserRoleRepository{CachedRepository: cache}
|
|
|
|
} |
...
|
...
|
|
