using Yw.DAL.PostgreSql;

namespace HStation.DAL.PostgreSql
{
    /// <summary>
    /// 
    /// </summary>
    public partial class XhsPumpPhartMapping : BaseDAL_Sorter<Entity.XhsPumpPhartMapping>, IXhsPumpPhartMapping
    {
        /// <summary>
        /// 
        /// </summary>
        public override ConnectionConfig ConnectionConfig
        {
            get { return Xhs.ConfigHelper.PostgreSqlConnectionConfig; }
        }


        /// <summary>
        /// 插入拓展
        /// </summary>
        public long InsertEx(Entity.XhsPumpPhartMapping mapping, Entity.XhsPhartDiagram diagram, List<Entity.XhsPhartGraph> graphList)
        {
            if (mapping == null)
            {
                return default;
            }
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                try
                {
                    db.BeginTran();
                    long diagramId = mapping.DiagramID;
                    if (diagram != null)
                    {
                        if (diagram.ID > 0)
                        {
                            var bol = db.Insertable(diagram).ExecuteCommand() > 0;
                            if (bol)
                            {
                                diagramId = diagram.ID;
                            }
                        }
                        else
                        {
                            diagramId = db.Insertable(diagram).ExecuteReturnSnowflakeId();
                        }
                    }

                    if (diagramId < 1)
                    {
                        db.RollbackTran();
                        return default;
                    }

                    if (graphList != null && graphList.Any())
                    {
                        List<long> graphIds = null;
                        graphList.ForEach(x => x.DiagramID = diagramId);
                        if (graphList.Exists(x => x.ID > 0))
                        {
                            graphList.ForEach(x =>
                            {
                                if (x.ID < 1)
                                {
                                    x.ID = SnowFlakeSingle.instance.NextId();
                                }
                            });
                            if (db.Insertable(graphList).ExecuteCommand() > 0)
                            {
                                graphIds = graphList.Select(x => x.ID).Distinct().ToList();
                            }
                        }
                        else
                        {
                            graphIds = db.Insertable(graphList).ExecuteReturnSnowflakeIdList();
                        }
                        if (graphIds == null || !graphIds.Any())
                        {
                            db.RollbackTran();
                            return default;
                        }
                    }
                    mapping.DiagramID = diagramId;
                    long mappingId = 0;
                    if (mapping.ID > 0)
                    {
                        var bol = db.Insertable(mapping).ExecuteCommand() > 0;
                        if (bol)
                        {
                            mappingId = mapping.ID;
                        }
                    }
                    else
                    {
                        mappingId = db.Insertable(mapping).ExecuteReturnSnowflakeId();
                    }
                    if (mappingId < 1)
                    {
                        db.RollbackTran();
                        return default;
                    }
                    db.CommitTran();
                    return mappingId;
                }
                catch
                {
                    db.RollbackTran();
                    throw;
                }
            }
        }

        /// <summary>
        /// 更新拓展
        /// </summary>
        public bool UpdateEx(Entity.XhsPumpPhartMapping mapping, Entity.XhsPhartDiagram diagram, List<Entity.XhsPhartGraph> graphList)
        {
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                try
                {
                    db.BeginTran();
                    var bol = db.Updateable(mapping).ExecuteCommandHasChange();
                    if (!bol)
                    {
                        db.RollbackTran();
                        return false;
                    }
                    if (diagram != null)
                    {
                        if (diagram.ID != mapping.DiagramID)
                        {
                            db.RollbackTran();
                            return false;
                        }
                        bol = db.Updateable(diagram).ExecuteCommandHasChange();
                        if (!bol)
                        {
                            db.RollbackTran();
                            return false;
                        }
                    }
                    if (graphList != null && graphList.Count > 0)
                    {
                        if (graphList.Exists(x => x.DiagramID != mapping.DiagramID))
                        {
                            db.RollbackTran();
                            return false;
                        }
                        bol = db.Updateable(graphList).ExecuteCommandHasChange();
                        if (!bol)
                        {
                            db.RollbackTran();
                            return false;
                        }
                    }
                    db.CommitTran();
                    return true;
                }
                catch
                {
                    db.RollbackTran();
                    throw;
                }
            }
        }

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

        /// <summary>
        /// 通过 PumpIds 获取
        /// </summary>
        public List<Entity.XhsPumpPhartMapping> GetByPumpIds(List<long> PumpIds)
        {
            if (PumpIds == null || PumpIds.Count < 1)
            {
                return default;
            }
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Queryable<Entity.XhsPumpPhartMapping>()
                    .Where(x => PumpIds.Contains(x.PumpID))
                    .OrderBy(x => x.SortCode).ToList();
            }
        }

        /// <summary>
        /// 通过 DiagramID 获取
        /// </summary>
        public List<Entity.XhsPumpPhartMapping> GetByDiagramID(long DiagramID)
        {
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Queryable<Entity.XhsPumpPhartMapping>()
                    .Where(x => x.DiagramID == DiagramID)
                    .OrderBy(x => x.SortCode).ToList();
            }
        }

        /// <summary>
        /// 通过 DiagramIds 获取
        /// </summary>
        public List<Entity.XhsPumpPhartMapping> GetByDiagramIds(List<long> DiagramIds)
        {
            if (DiagramIds == null || DiagramIds.Count < 1)
            {
                return default;
            }
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Queryable<Entity.XhsPumpPhartMapping>()
                    .Where(x => DiagramIds.Contains(x.DiagramID))
                    .OrderBy(x => x.SortCode).ToList();
            }
        }

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

        /// <summary>
        /// 更新 Importance
        /// </summary>
        public bool UpdateImportance(long ID, int Importance)
        {
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Updateable<Entity.XhsPumpPhartMapping>()
                    .SetColumns(x => x.Importance == Importance)
                    .Where(x => x.ID == ID)
                    .ExecuteCommandHasChange();
            }
        }

        /// <summary>
        /// 通过 PumpID 删除
        /// </summary>
        public bool DeleteByPumpID(long PumpID)
        {
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                return db.Deleteable<Entity.XhsPumpPhartMapping>()
                    .Where(x => x.PumpID == PumpID)
                    .ExecuteCommandHasChange();
            }
        }

        /// <summary>
        /// 通过 DiagramID 删除所有映射、图表、图形
        /// </summary>
        public bool DeleteByDiagramID(long DiagramID)
        {
            using (var db = new SqlSugarClient(ConnectionConfig))
            {
                try
                {
                    db.BeginTran();
                    var bol = db.Deleteable<Entity.XhsPumpPhartMapping>().Where(x => x.DiagramID == DiagramID).ExecuteCommandHasChange();
                    if (!bol)
                    {
                        db.RollbackTran();
                        return false;
                    }

                    bol = db.Deleteable<Entity.XhsPhartDiagram>().Where(x => x.ID == DiagramID).ExecuteCommandHasChange();
                    if (!bol)
                    {
                        db.RollbackTran();
                        return false;
                    }

                    bol = db.Deleteable<Entity.XhsPhartGraph>().Where(x => x.DiagramID == DiagramID).ExecuteCommandHasChange();
                    db.CommitTran();
                    return true;
                }
                catch
                {
                    db.RollbackTran();
                    throw;
                }
            }
        }













    }
}