using SqlSugar;
using System;
using System.Collections.Generic;
using System.Data;
using System.Linq;
using System.Transactions;

namespace IStation.DAL
{
    /// <summary>
    /// 泵曲线映射
    /// </summary>
    public partial class PumpCurveMapping : CorpDAL_Sorter<Entity.PumpCurveMapping>
    {
        /// <summary>
        /// 
        /// </summary>
        public override ConnectionConfig ConnectionConfig
        {
            get { return ConfigHelper.DefaultConnectionConfig; }

        }

        /// <summary>
        /// 插入
        /// </summary>
        public long InsertEx(Entity.PumpCurve entity_curve, Entity.PumpCurveMapping entity_map)
        {
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                try
                {
                    db.BeginTran();
                    var curveId = db.Insertable(entity_curve).ExecuteReturnSnowflakeId();
                    if (curveId < 1)
                    {
                        db.RollbackTran();
                        return default;
                    }
                    entity_map.CurveID = curveId;
                    var mapId = db.Insertable(entity_map).ExecuteReturnSnowflakeId();
                    if (mapId < 1)
                    {
                        db.RollbackTran();
                        return default;
                    }
                    db.CommitTran();
                    return mapId;
                }
                catch
                {
                    db.RollbackTran();
                    throw;
                }
            }
        }

        /// <summary>
        /// 批量插入
        /// </summary>
        public bool InsertsEx(Dictionary<Entity.PumpCurve, Entity.PumpCurveMapping> dict)
        {
            if (dict == null || dict.Count < 1)
                return default;
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                try
                {
                    db.BeginTran();
                    foreach (var item in dict)
                    {
                        var curveId = db.Insertable(item.Key).ExecuteReturnSnowflakeId();
                        if (curveId < 1)
                        {
                            db.RollbackTran();
                            return default;
                        }
                        item.Value.CurveID = curveId;
                        var mapId = db.Insertable(item.Value).ExecuteReturnSnowflakeId();
                        if (mapId < 1)
                        {
                            db.RollbackTran();
                            return default;
                        }
                    }
                    db.CommitTran();
                    return true;
                }
                catch
                {
                    db.RollbackTran();
                    throw;
                }
            }

        }

        /// <summary>
        /// 批量插入
        /// </summary>
        public List<long> InsertsREx(Dictionary<Entity.PumpCurve, Entity.PumpCurveMapping> dict)
        {
            if (dict == null || dict.Count < 1)
                return default;
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                try
                {
                    db.BeginTran();
                    var list = new List<long>();
                    foreach (var item in dict)
                    {
                        var curveId = db.Insertable(item.Key).ExecuteReturnSnowflakeId();
                        if (curveId < 1)
                        {
                            db.RollbackTran();
                            return default;
                        }
                        item.Value.CurveID = curveId;
                        var mapId = db.Insertable(item.Value).ExecuteReturnSnowflakeId();
                        if (mapId < 1)
                        {
                            db.RollbackTran();
                            return default;
                        }
                        list.Add(mapId);
                    }
                    db.CommitTran();
                    return list;
                }
                catch
                {
                    db.RollbackTran();
                    throw;
                }
            }

        }

        /// <summary>
        /// 通过 PumpID 获取
        /// </summary>
        public List<Entity.PumpCurveMapping> GetByPumpID(long CorpID, long PumpID)
        {
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Queryable<Entity.PumpCurveMapping>().Where(x => x.CorpID == CorpID && x.PumpID == PumpID).ToList();
            }
        }

        /// <summary>
        /// 通过 CurveID 获取
        /// </summary>
        public List<Entity.PumpCurveMapping> GetByCurveID(long CorpID, long CurveID)
        {
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Queryable<Entity.PumpCurveMapping>().Where(x => x.CorpID == CorpID && x.CurveID == CurveID).ToList();
            }
        }

        /// <summary>
        /// 更新 IsWorking
        /// </summary>
        public bool UpdateIsWorking(long CorpID, long ID, bool IsWorking)
        {
            using (SqlSugarClient db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Updateable<Entity.PumpCurveMapping>()
                    .SetColumns(x => x.IsWorking == IsWorking)
                    .Where(x => x.CorpID == CorpID && x.ID == ID)
                    .ExecuteCommand() > 0;
            }
        }

        /// <summary>
        /// 更新 OtherName
        /// </summary>
        public bool UpdateOtherName(long CorpID, long ID, string OtherName)
        {
            using (SqlSugarClient db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Updateable<Entity.PumpCurveMapping>()
                    .SetColumns(x => x.OtherName == OtherName)
                    .Where(x => x.CorpID == CorpID && x.ID == ID)
                    .ExecuteCommand() > 0;
            }
        }

        /// <summary>
        /// 设置工作曲线
        /// </summary>
        public bool SetWorkingCurve(long CorpID, long PumpID, long ID)
        {
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                try
                {
                    db.BeginTran();
                    var bol = db.Updateable<Entity.PumpCurveMapping>()
                        .SetColumns(x => x.IsWorking == true)
                        .Where(x => x.CorpID == CorpID && x.ID == ID).ExecuteCommand() > 0;
                    if (!bol)
                    {
                        db.RollbackTran();
                        return false;
                    }
                    db.Updateable<Entity.PumpCurveMapping>()
                        .SetColumns(x => x.IsWorking == false)
                        .Where(x => x.CorpID == CorpID && x.PumpID == PumpID && x.ID != ID).ExecuteCommand();
                    db.CommitTran();
                    return true;
                }
                catch
                {
                    db.RollbackTran();
                    throw;
                }
            }
        }

        /// <summary>
        /// 通过 CurveID 删除
        /// </summary>
        public bool DeleteByCurveID(long CorpID, long CurveID)
        {
            using (SqlSugarClient db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Deleteable<Entity.PumpCurveMapping>()
                    .Where(x => x.CorpID == CorpID && x.CurveID == CurveID)
                    .ExecuteCommand() > 0;
            }
        }

        /// <summary>
        /// 通过 PumpID 删除
        /// </summary>
        public bool DeleteByPumpID(long CorpID, long PumpID)
        {
            using (SqlSugarClient db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Deleteable<Entity.PumpCurveMapping>()
                    .Where(x => x.CorpID == CorpID && x.PumpID == PumpID)
                    .ExecuteCommand() > 0;
            }
        }

        /// <summary>
        /// 删除映射同时删除曲线
        /// </summary>
        public bool DeleteWithCurveByID(long CorpID, long ID, long CurveID)
        {
            using (SqlSugarClient db = new SqlSugarClient(ConnectionConfig))
            {
                try
                {
                    db.BeginTran();
                    var bol = db.Deleteable<Entity.PumpCurveMapping>().Where(x => x.CorpID == CorpID && x.ID == ID).ExecuteCommand() > 0;
                    if (!bol)
                    {
                        db.RollbackTran();
                        return default;
                    }
                    bol = db.Deleteable<Entity.PumpCurve>().Where(x => x.CorpID == CorpID && x.ID == CurveID).ExecuteCommand() > 0;
                    if (!bol)
                    {
                        db.RollbackTran();
                        return default;
                    }
                    db.CommitTran();
                    return true;
                }
                catch
                {
                    db.RollbackTran();
                    throw;
                }
            }
        }


        /// <summary>
        /// 通过  CurveID 设置
        /// </summary>
        public bool SetOfObject(long CorpID,  long CurveID,  List<Entity.PumpCurveMapping> list)
        {
            if (list == null || list.Count() < 1)
            {
                using (var db = new SqlSugarClient(ConnectionConfig))
                {
                    return db.Deleteable<Entity.PumpCurveMapping>()
                        .Where(x => x.CorpID == CorpID && x.CurveID == CurveID)
                        .ExecuteCommand() > 0;
                }
            }

            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                try
                {
                    db.BeginTran();
                    var add_list = list.Where(x => x.ID < 1).ToList();
                    var update_list = list.Where(x => x.ID > 0).ToList();
                    var ids_exist = update_list.Select(x => x.ID).Distinct().ToList();
                    db.Deleteable<Entity.PumpCurveMapping>().Where(x => x.CorpID == CorpID && x.CurveID == CurveID && !ids_exist.Contains(x.ID)).ExecuteCommand(); 
                    if (add_list.Count > 0)
                    {
                        var result = db.Insertable(add_list).ExecuteReturnSnowflakeIdList().Count > 0;
                        if (!result)
                        {
                            db.RollbackTran();
                            return default;
                        }
                    }
                    if (update_list.Count > 0)
                    {
                        var result = db.Updateable(update_list).ExecuteCommand() > 0;
                        if (!result)
                        {
                            db.RollbackTran();
                            return default;
                        }
                    } 

                    db.CommitTran();
                    return true;
                }
                catch
                {
                    db.RollbackTran();
                    throw;
                }

            }
        }











    }
}