perms-system-server/internal/model/dept/sysDeptModel.go

package dept

import (
	"context"
	"fmt"
	"strings"

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

var _ SysDeptModel = (*customSysDeptModel)(nil)

type (
	SysDeptModel interface {
		sysDeptModel
		FindAll(ctx context.Context) ([]*SysDept, error)
		FindByParentId(ctx context.Context, parentId int64) ([]*SysDept, error)
		FindByPathPrefix(ctx context.Context, pathPrefix string) ([]*SysDept, error)
	}

	customSysDeptModel struct {
		*defaultSysDeptModel
	}
)

func NewSysDeptModel(conn sqlx.SqlConn, c cache.CacheConf, cachePrefix string, opts ...cache.Option) SysDeptModel {
	return &customSysDeptModel{
		defaultSysDeptModel: newSysDeptModel(conn, c, cachePrefix, opts...),
	}
}

func (m *customSysDeptModel) FindAll(ctx context.Context) ([]*SysDept, error) {
	var list []*SysDept
	query := fmt.Sprintf("SELECT %s FROM %s ORDER BY `sort` ASC, `id` ASC", sysDeptRows, m.table)
	if err := m.QueryRowsNoCacheCtx(ctx, &list, query); err != nil {
		return nil, err
	}
	return list, nil
}

func (m *customSysDeptModel) FindByParentId(ctx context.Context, parentId int64) ([]*SysDept, error) {
	var list []*SysDept
	query := fmt.Sprintf("SELECT %s FROM %s WHERE `parentId` = ? ORDER BY `sort` ASC, `id` ASC", sysDeptRows, m.table)
	if err := m.QueryRowsNoCacheCtx(ctx, &list, query, parentId); err != nil {
		return nil, err
	}
	return list, nil
}

func (m *customSysDeptModel) FindByPathPrefix(ctx context.Context, pathPrefix string) ([]*SysDept, error) {
	escaped := strings.NewReplacer("%", "\\%", "_", "\\_").Replace(pathPrefix)
	var list []*SysDept
	query := fmt.Sprintf("SELECT %s FROM %s WHERE `path` LIKE ? ORDER BY `sort` ASC, `id` ASC", sysDeptRows, m.table)
	if err := m.QueryRowsNoCacheCtx(ctx, &list, query, escaped+"%"); err != nil {
		return nil, err
	}
	return list, nil
}