BuYuHouDuan/Infrastructure/SharpAbp.Infrastructure.Api/Extensions/ControllerExtensions.cs

using System;
using System.Collections.Generic;
using Common;
using Microsoft.AspNetCore.Mvc;

namespace SharpAbp.Infrastructure.Api.Extensions
{
    public static class ControllerExtensions
    {
        public static BaseResponse<T> ToFailResult<T>(
            this Controller controller,
            string info) => new BaseResponse<T>() { Info = info, Code = CommonResponseCodeEnum.ResultFail };

        public static ListResponse<T> ToListFailResult<T>(
            this Controller controller,
            string info) => new ListResponse<T>()
            { Info = info, Code = CommonResponseCodeEnum.ResultFail };

        public static PageResponse<T> ToPageFailResult<T>(
            this Controller controller,
            string info) => new PageResponse<T>()
            {
                Code = CommonResponseCodeEnum.ResultFail,
                Info = info
            };

        public static BaseResponse<T> ToResult<T>(
            this Controller controller,
            T obj, string info = null) => new BaseResponse<T>() 
            { 
                Data = obj, 
                Info = info, 
                Code = CommonResponseCodeEnum.Success 
            };

        public static ListResponse<T> ToListResult<T>(
            this Controller controller,
            IEnumerable<T> list, string info = null) => new ListResponse<T>()
            {
                Data = list,
                Code = CommonResponseCodeEnum.Success,
                Info = info
            };

        public static PageResponse<T> ToPageResult<T>(
            this Controller controller,
            IEnumerable<T> list, int count = 0, string info = null) => new PageResponse<T>()
            {
                Code = CommonResponseCodeEnum.Success,
                Info = info,
                Count = count,
                Data = list
            };

        public static BaseResponse<T> ToNoLoginResult<T>(
            this Controller controller) => new BaseResponse<T>() 
            { 
                Info = "未登录", 
                Code = CommonResponseCodeEnum.NoLogin 
            };

        public static BaseResponse<T> ToTokenInvalidResult<T>(
            this Controller controller) => new BaseResponse<T>() 
            { 
                Info = "token失效", 
                Code = CommonResponseCodeEnum.TokenInvalid 
            };

        public static BaseResponse<T> ToParamErrorResult<T>(
            this Controller controller,
            string info = "请求参数错误") => new BaseResponse<T>() 
            { 
                Info = info, 
                Code = CommonResponseCodeEnum.ParamError 
            };
    }
}