OA2023/OASystem/OASystem.Infrastructure/Repositories/Resource/CountryFeeRepository.cs

using AutoMapper;
using OASystem.Domain;
using OASystem.Domain.Dtos.Resource;
using OASystem.Domain.Entities.Resource;
using OASystem.Domain.ViewModels.Resource;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace OASystem.Infrastructure.Repositories.Resource
{
    public class CountryFeeRepository : BaseRepository<Res_CountryFeeCost, CountryFeeCostView>
    {
        private readonly IMapper _mapper;
        public CountryFeeRepository(SqlSugarClient sqlSugar, IMapper mapper) : base(sqlSugar)
        {
            _mapper = mapper;
        }

        public async Task<Result> OperationCountryFeeCost(OperationCountryFeeCostDto dto)
        {
            Result result = new Result() { Code = -2, Msg = "未知错误" };
            try
            {
                if (dto.Status == 1)//添加
                {
                    string selectSql = string.Format(@"select * from Res_CountryFeeCost where VisaContinent='{0}' and VisaCountry='{1}' and IsDel='{2}'"
                                                       , dto.VisaContinent,dto.VisaCountry, 0);
                    var CountryFeeCost = await _sqlSugar.SqlQueryable<Res_CountryFeeCost>(selectSql).FirstAsync();//查询是否存在
                    if (CountryFeeCost != null)
                    {
                        return result = new Result() { Code = -1, Msg = "该国家已存在，请勿重复添加！" };

                    }
                    else//不存在，可添加
                    {
                        Res_CountryFeeCost _CountryFeeCost = _mapper.Map<Res_CountryFeeCost>(dto);
                        int id = await AddAsyncReturnId(_CountryFeeCost);
                        if (id == 0)
                        {
                            return result = new Result() { Code = -1, Msg = "添加失败！" };

                        }
                        result = new Result() { Code = 0, Msg = "添加成功！", Data = new { Id = id } };
                    }
                }
                else if (dto.Status == 2)//修改
                {
                    bool res = await UpdateAsync(a => a.Id == dto.Id, a => new Res_CountryFeeCost
                    {
                        VisaContinent = dto.VisaContinent,
                        VisaCountry = dto.VisaCountry,
                        IsVisaExemption = dto.IsVisaExemption,
                        IsVisaOnArrival = dto.IsVisaOnArrival,
                        IsElectronicSignature = dto.IsElectronicSignature,
                        VisaPrice = dto.VisaPrice,
                        VisaPriceDesc = dto.VisaPriceDesc,
                        VisaType = dto.VisaType,
                        VisaTime = dto.VisaTime,
                        IsUrgent = dto.IsUrgent,
                        UrgentTime = dto.UrgentTime,
                        UrgentPrice = dto.UrgentPrice,
                        UrgentPriceDesc = dto.UrgentPriceDesc,
                        VisaAddress = dto.VisaAddress,
                        Remark = dto.Remark
                    });
                    if (!res)
                    {
                        return result = new Result() { Code = -1, Msg = "修改失败！" };
                    }
                    result = new Result() { Code = 0, Msg = "修改成功！" };
                }
                else
                {
                    return result = new Result() { Code = -1, Msg = "请传入Status参数，1添加 2修改！" };
                }
            }
            catch (Exception ex)
            {
                return result = new Result() { Code = -2, Msg = "程序错误！" };
            }
            return result;
        }
    }
}