JLH/JLHHJSvr/Helper/LockHelper.cs

using JLHHJSvr.BLL;
using JLHHJSvr.Com.Model;
using JLHHJSvr.LJException;
using LJLib.DAL.SQL;
using System;
using System.Collections.Generic;
using System.Data.SqlClient;
using System.Data;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.ComponentModel;
using NPOI.SS.Formula;
using System.Web.Configuration;

namespace JLHHJSvr.Helper
{
    internal class LockHelper : HelperBase
    {
        /// <summary>
        /// 加锁单据
        /// </summary>
        /// <param name="keyword"></param>
        /// <param name="billid"></param>
        public static void LockBill(SqlCommand cmd, string keyword, int billid, string opemp, int lockMinutes = 30)
        {
            try
            {
                cmd.CommandText = "sp_lock_record";
                cmd.CommandType = CommandType.StoredProcedure;
                cmd.Parameters.Clear();
                cmd.Parameters.AddWithValue("@keyword", keyword);
                cmd.Parameters.AddWithValue("@billid", billid);
                cmd.Parameters.AddWithValue("@opemp", opemp);
                cmd.Parameters.AddWithValue("@lockMinutes", lockMinutes);
                using (var reader = cmd.ExecuteReader())
                {
                    if (reader.Read())
                    {
                        var arg_rslt = Convert.ToInt32(reader["arg_rslt"]);
                        var arg_msg = Convert.ToString(reader["arg_msg"]);

                        if (arg_rslt != 1) throw new LJCommonException(arg_msg);
                    }
                }
            }
            finally
            {
                cmd.CommandType = CommandType.Text;
                cmd.CommandText = string.Empty;
            }
        }

        /// <summary>
        /// 解锁单据
        /// </summary>
        /// <param name="keyword"></param>
        /// <param name="billid"></param>
        public static void UnLockBill(SqlCommand cmd, string keyword, int billid, string opemp, byte is_admin)
        {
            try
            {
                cmd.CommandText = "sp_unlock_record";
                cmd.CommandType = CommandType.StoredProcedure;
                cmd.Parameters.Clear();
                cmd.Parameters.AddWithValue("@keyword", keyword);
                cmd.Parameters.AddWithValue("@billid", billid);
                cmd.Parameters.AddWithValue("@opemp", opemp);
                cmd.Parameters.AddWithValue("@is_admin", is_admin);
                using (var reader = cmd.ExecuteReader())
                {
                    if (reader.Read())
                    {
                        var arg_rslt = Convert.ToInt32(reader["arg_rslt"]);
                        var arg_msg = Convert.ToString(reader["arg_msg"]);

                        if (arg_rslt != 1) throw new LJCommonException(arg_msg);
                    }
                }
            }
            finally
            {
                cmd.CommandType = CommandType.Text;
                cmd.CommandText = string.Empty;
            }
        }
        /// <summary>
        /// 生成锁提示信息
        /// </summary>
        private static string GetLockMessage(SqlCommand cmd, string keyword, int billid, string opEmp)
        {
            if (BillNames.MattressQuote.Equals(keyword))
            {
                var mattress = GetHelper<MattressHelper>(cmd).GetMattress(billid, "mattresscode");
                return $"床垫报价【{mattress.mattresscode}】被用户【{opEmp}】编辑中。";
            }
            if (BillNames.MattressInterface.Equals(keyword))
            {
                var mattress = GetHelper<MattressHelper>(cmd).GetMattress(billid, "mattresscode");
                return $"床垫清单【{mattress.mattresscode}】被用户【{opEmp}】编辑中。";
            }
            if (BillNames.BedNetQuote.Equals(keyword))
            {
                var bednet = GetHelper<BedNetHelper>(cmd).GetBedNet(billid, "bednetcode");
                return $"床网报价【{bednet.bednetcode}】被用户【{opEmp}】编辑中。";
            }
            if (BillNames.SoftBedQuote.Equals(keyword))
            {
                var softbed = GetHelper<SoftBedHelper>(cmd).GetSoftBed(billid, "softbed_code");
                return $"软床报价【{softbed.softbed_code}】正在被用户【{opEmp}】编辑中。";
            }
            return $"单据【{keyword}-{billid}】被用户【{opEmp}】编辑中。";
        }
        #region
        /// <summary>
          /// DFS方式检查单据及其依赖的锁
          /// </summary>
        public static u_lock_table CheckLockAll(SqlCommand cmd, string keyword, int billid, string opemp)
        {
            var visited = new Dictionary<string, int>(); // 防止循环依赖
            return GetLockBill(cmd, keyword, billid, opemp, visited);
        }
        /// <summary>
        /// 检查单据是否上锁
        /// </summary>
        /// <param name="cmd"></param>
        /// <param name="keyword"></param>
        /// <param name="billid"></param>
        /// <param name="opemp"></param>
        /// <param name="arg_msg"></param>
        /// <returns></returns>
        public static bool CheckIsLock(SqlCommand cmd, string keyword, int billid, string opemp,out string arg_msg)
        {
            arg_msg = string.Empty;
            var item = CheckLockAll(cmd, keyword, billid, opemp);
            if (item.Code == LockCheckResultCode.LockedByOther)
            {
                arg_msg = item.log_msg;
                return true;
            }
            return !(item.Code == LockCheckResultCode.Success || item.Code == LockCheckResultCode.LockedBySelf);
        }
        /// <summary>
        /// DFS方式检查单据及依赖链上的锁
        /// </summary>
        private static u_lock_table GetLockBill(SqlCommand cmd, string keyword, int billid, string opemp, Dictionary<string, int> visited)
        {
            var key = $"{keyword}:{billid}";
            if (visited.TryGetValue(key, out var state))
            {
                if (state == 1) throw new InvalidOperationException($"检测到循环依赖: {key}");
                if (state == 2) return new u_lock_table { keyword = keyword, billid = billid, Code = LockCheckResultCode.Success };
            }
            visited[key] = 1; // 标记正在访问

            //
            var mainLock = new u_lock_table { keyword = keyword, billid = billid };
            DbSqlHelper.SelectOne(cmd,"u_lock_table","keyword = @keyword AND billid = @billid AND is_locked = 1",
                new Dictionary<string, object> { { "@keyword", keyword }, { "@billid", billid } },
                mainLock, "keyword,billid,opemp,is_locked,lock_time,expire_time");

            if (mainLock.is_locked == 1)
            {
                mainLock.Code = mainLock.opemp.Equals(opemp) ? LockCheckResultCode.LockedBySelf : LockCheckResultCode.LockedByOther;
                mainLock.log_msg = GetLockMessage(cmd, keyword, billid, mainLock.opemp);
                return mainLock;
            }
            visited[key] = 2; // 标记完成

            // 
            var dependencies = BuildDependencies(cmd, keyword, billid);
            foreach (var dep in dependencies)
            {
                var depResult = GetLockBill(cmd, dep.keyword, dep.billid, opemp, visited);
                if (depResult.Code == LockCheckResultCode.LockedBySelf || depResult.Code == LockCheckResultCode.LockedByOther) return depResult;
            }

            mainLock.Code = LockCheckResultCode.Success;
            return mainLock;
        }
        /// <summary>
        /// 根据单据类型构造依赖关系
        /// </summary>
        private static List<u_lock_table> BuildDependencies(SqlCommand cmd, string keyword, int billid)
        {
            var dependencies = new List<u_lock_table>();

            if (BillNames.MattressQuote.Equals(keyword))
            {
                dependencies.Add(new u_lock_table { keyword = BillNames.MattressInterface, billid = billid });

                var bednetList = GetHelper<MattressHelper>(cmd)
                    .GetMattressMxMtrl(billid, new List<string> { "u_mattress_formula.formulatype = 99" });
                foreach (var bednet in bednetList)
                {
                    dependencies.Add(new u_lock_table { keyword = BillNames.BedNetQuote, billid = bednet.mtrlid.Value });
                }
            }
            else if (BillNames.MattressInterface.Equals(keyword))
            {
                dependencies.Add(new u_lock_table { keyword = BillNames.MattressQuote, billid = billid });
            }
            else if (BillNames.BedNetQuote.Equals(keyword))
            {
                var mattressList = GetHelper<MattressHelper>(cmd)
                    .GetMattressMxMtrl(0, new List<string> { $"u_mattress_formula.formulatype = 99 AND vv_bednet.bednetid = {billid}" });
                var idsList = mattressList.GroupBy(t => t.mattressid.Value).Select(t => t.Key).ToList();
                foreach (var id in idsList)
                {
                    dependencies.Add(new u_lock_table { keyword = BillNames.MattressQuote, billid = id });
                }
            }

            return dependencies;
        }

        #endregion
    }
}