优化WCSHelper / 序列号生成逻辑 , 添加强制任务处理模块

kazelee

昨天 fc25dda9baf3b5f4df23d35914f3dd343cf492e3

 wms/SYSHelper.cs

@@ -2,48 +2,62 @@

using SqlSugar;

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace HH.WCS.Mobox3.DSZSH.wms {
    /// <summary>
    /// 用于生成序列号，并管理序列号前缀的工具类
    /// [ 序列号生成 ] 帮助类
    /// </summary>
    internal class SYSHelper
    {
        //使用静态 locker 对象实现线程同步
    public class SYSHelper {
        private static object locker = new object();
        internal static int GetSerialNumber(string snType, string prefix) {
        /// <summary>
        /// <c>"{prefix}-{DateTime.Now.ToString(dateFormat)}-{id.ToString().PadLeft(length, '0')}"</c>
        /// </summary>
        /// <param name="snType"></param>
        /// <param name="prefix"></param>
        /// <param name="dateFormat"></param>
        /// <param name="length"></param>
        /// <returns></returns>
        public static string GenerateSN(string snType, string prefix, string dateFormat = "yyMM", int length = 4) {
            var header = $"{prefix}{DateTime.Now.ToString(dateFormat)}-";
            var res = GetSerialNumber(snType, header);
            if (res == 0) return "";
            else {
                return $"{header}{res.ToString().PadLeft(length, '0')}";
            }
        }

        private static int GetSerialNumber(string snType, string prefix) {
            int result = 0;
            lock (locker) {
                var db = new SqlHelper<object>().GetInstance();
                var sId = db.Queryable<OI_SYS_MAXID>().Where(a => a.CN_S_TYPE.Trim() == snType && a.CN_S_PRE.Trim() == prefix).First();
                if (sId != null) {
                    sId.CN_N_MAX++;
                    //if (db.Ado.ExecuteCommand($"update OI_SYS_MAXID set CN_N_MAX={sId.CN_N_MAX} where CN_S_TYPE='{snType}' and CN_S_PRE='{prefix}' ") > 0) {
                    //    result = sId.CN_N_MAX;
                    //}

                    // 参数化查询，避免SQL注入
                    if (db.Updateable<OI_SYS_MAXID>().SetColumns(it => it.CN_N_MAX == sId.CN_N_MAX).Where(it => it.CN_S_TYPE == snType
                        && it.CN_S_PRE == prefix).ExecuteCommand() > 0) {
                    if (db.Ado.ExecuteCommand($"update OI_SYS_MAXID set CN_N_MAX={sId.CN_N_MAX} " +
                        $"where CN_S_TYPE='{snType}' and CN_S_PRE='{prefix}' ") > 0) {
                        result = sId.CN_N_MAX;
                    }

                }
                else {
                    //插入表
                    sId = new OI_SYS_MAXID { CN_S_TYPE = snType, CN_S_PRE = prefix, CN_N_MAX = 1 };
                    result = db.Insertable<OI_SYS_MAXID>(sId).ExecuteCommand() > 0 ? 1 : 0;

                    result = db.Insertable(sId).ExecuteCommand() > 0 ? 1 : 0;
                }
            }
            return result;
        }

        [SugarTable("dbo.OI_SYS_MAXID")]
        public class OI_SYS_MAXID
        {
        public class OI_SYS_MAXID {
            public string CN_S_TYPE { get; set; }
            public string CN_S_PRE { get; set; }
            public int CN_N_MAX { get; set; }
            public DateTime CN_T_LAST { get; set; }
        }
    }
}