perms-system-server/internal/model/userperm/sysUserPermModel.go

package userperm

import (
	"context"
	"fmt"

	"github.com/zeromicro/go-zero/core/stores/cache"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

var _ SysUserPermModel = (*customSysUserPermModel)(nil)

type (
	SysUserPermModel interface {
		sysUserPermModel
		FindByUserId(ctx context.Context, userId int64) ([]*SysUserPerm, error)
		FindPermIdsByUserIdAndEffect(ctx context.Context, userId int64, effect string) ([]int64, error)
		FindPermIdsByUserIdAndEffectForProduct(ctx context.Context, userId int64, effect string, productCode string) ([]int64, error)
		DeleteByUserId(ctx context.Context, userId int64) error
		DeleteByUserIdTx(ctx context.Context, session sqlx.Session, userId int64) error
		DeleteByUserIdForProduct(ctx context.Context, userId int64, productCode string) error
		DeleteByUserIdForProductTx(ctx context.Context, session sqlx.Session, userId int64, productCode string) error
	}

	customSysUserPermModel struct {
		*defaultSysUserPermModel
	}
)

func NewSysUserPermModel(conn sqlx.SqlConn, c cache.CacheConf, cachePrefix string, opts ...cache.Option) SysUserPermModel {
	return &customSysUserPermModel{
		defaultSysUserPermModel: newSysUserPermModel(conn, c, cachePrefix, opts...),
	}
}

func (m *customSysUserPermModel) FindByUserId(ctx context.Context, userId int64) ([]*SysUserPerm, error) {
	var list []*SysUserPerm
	query := fmt.Sprintf("SELECT %s FROM %s WHERE `userId` = ?", sysUserPermRows, m.table)
	if err := m.QueryRowsNoCacheCtx(ctx, &list, query, userId); err != nil {
		return nil, err
	}
	return list, nil
}

func (m *customSysUserPermModel) FindPermIdsByUserIdAndEffect(ctx context.Context, userId int64, effect string) ([]int64, error) {
	var ids []int64
	query := fmt.Sprintf("SELECT `permId` FROM %s WHERE `userId` = ? AND `effect` = ?", m.table)
	if err := m.QueryRowsNoCacheCtx(ctx, &ids, query, userId, effect); err != nil {
		return nil, err
	}
	return ids, nil
}

func (m *customSysUserPermModel) FindPermIdsByUserIdAndEffectForProduct(ctx context.Context, userId int64, effect string, productCode string) ([]int64, error) {
	var ids []int64
	query := fmt.Sprintf(
		"SELECT up.`permId` FROM %s up INNER JOIN `sys_perm` p ON up.`permId` = p.`id` WHERE up.`userId` = ? AND up.`effect` = ? AND p.`productCode` = ?",
		m.table)
	if err := m.QueryRowsNoCacheCtx(ctx, &ids, query, userId, effect, productCode); err != nil {
		return nil, err
	}
	return ids, nil
}

func (m *customSysUserPermModel) DeleteByUserId(ctx context.Context, userId int64) error {
	query := fmt.Sprintf("DELETE FROM %s WHERE `userId` = ?", m.table)
	_, err := m.ExecNoCacheCtx(ctx, query, userId)
	return err
}

func (m *customSysUserPermModel) DeleteByUserIdTx(ctx context.Context, session sqlx.Session, userId int64) error {
	query := fmt.Sprintf("DELETE FROM %s WHERE `userId` = ?", m.table)
	_, err := session.ExecCtx(ctx, query, userId)
	return err
}

func (m *customSysUserPermModel) DeleteByUserIdForProduct(ctx context.Context, userId int64, productCode string) error {
	query := fmt.Sprintf("DELETE FROM %s WHERE `userId` = ? AND `permId` IN (SELECT `id` FROM `sys_perm` WHERE `productCode` = ?)", m.table)
	_, err := m.ExecNoCacheCtx(ctx, query, userId, productCode)
	return err
}

func (m *customSysUserPermModel) DeleteByUserIdForProductTx(ctx context.Context, session sqlx.Session, userId int64, productCode string) error {
	query := fmt.Sprintf("DELETE FROM %s WHERE `userId` = ? AND `permId` IN (SELECT `id` FROM `sys_perm` WHERE `productCode` = ?)", m.table)
	_, err := session.ExecCtx(ctx, query, userId, productCode)
	return err
}