znyc.recruitment.admin.api/Znyc.Recruitment.Admin.Security.Core/Repositories/Recruitment/MenuRepository.cs

using Dapper;
using System.Collections.Generic;
using Znyc.Recruitment.Admin.Commons.IDbContext;
using Znyc.Recruitment.Admin.Commons.Repositories;
using Znyc.Recruitment.Admin.Security.Entitys;
using Znyc.Recruitment.Admin.Security.IRepositories;

namespace Znyc.Recruitment.Admin.Security.Repositories
{
    public class MenuRepository : BaseRepository<MenuEntity, long>, IMenuRepository
    {
        public MenuRepository()
        {
        }

        public MenuRepository(IDbContextCore dbContext) : base(dbContext)
        {
        }

        /// <summary>
        ///     根据角色ID字符串（逗号分开)和系统类型ID，获取对应的操作功能列表
        /// </summary>
        /// <param name="roleIds">角色ID</param>
        /// <param name="typeID">系统类型ID</param>
        /// <param name="isMenu">是否是菜单</param>
        /// <returns></returns>
        public IEnumerable<MenuEntity> GetFunctions(string roleIds, string typeID, bool isMenu = false)
        {
            string sql =
                "SELECT DISTINCT b.* FROM sys_menu as b INNER JOIN Sys_RoleAuthorize as a On b.Id = a.ItemId  WHERE ObjectId IN (" +
                roleIds + ")";
            if (roleIds == "")
            {
                sql = "SELECT DISTINCT b.* FROM sys_menu as b where 1=1 ";
            }

            if (isMenu)
            {
                sql = sql + "and menutype in('M','C')";
            }

            if (!string.IsNullOrEmpty(typeID))
            {
                sql = sql + string.Format(" AND SystemTypeId='{0}' ", typeID);
            }

            return DapperConnRead.Query<MenuEntity>(sql);
        }

        /// <summary>
        ///     根据系统类型ID，获取对应的操作功能列表
        /// </summary>
        /// <param name="typeID">系统类型ID</param>
        /// <returns></returns>
        public IEnumerable<MenuEntity> GetFunctions(string typeID)
        {
            string sql = "SELECT DISTINCT b.* FROM sys_menu as b ";
            if (!string.IsNullOrEmpty(typeID))
            {
                sql = sql + string.Format(" Where SystemTypeId='{0}' ", typeID);
            }

            return DapperConnRead.Query<MenuEntity>(sql);
        }
    }
}