using System.Collections.Concurrent;
using System.ComponentModel;
using System.Reflection;
using Dapper;
using FreeRedis;
using Npgsql;
using ServiceCenter.Logs;
using ServiceCenter.SqlSugars;
using SqlSugar;
using WCS.Core;
using LogInfo = ServiceCenter.Logs.LogInfo;
namespace WCS.WorkEngineering.Worlds;
///
/// 主世界,所有的系统(交互点)默认在该世界下执行。
/// 如有系统需独立,请自行增加对应世界
/// 新增世界应当继承此世界,而不是直接继承World
///
[Description("主世界")]
public class MainWorld : World
{
///
/// 数据队列
///
public static ConcurrentQueue DataQueue = new();
///
/// redis链接
///
public static RedisClient Redis = new(Configs.DebugRedisUrl);
///
/// 日志队列
///
protected ConcurrentQueue Logs = new();
///
/// 构造函数
///
public MainWorld()
{
}
///
/// 世界执行周期间隔
/// 单位:毫秒
///
protected override int Interval => 500;
///
/// 更新前执行,重写改方法后请自行添加执行内容
/// 执行内容:清空日志队列
///
protected override void BeforeUpdate(List list)
{
// 清空日志队列,确保日志队列中只会有当前周期日志
Logs.Clear();
}
///
/// 更新后执行,重写改方法后请自行添加执行内容
/// 执行内容:清空日志队列
///
protected override void AfterUpdate(List list)
{
//LogHub.WorldPublish(Logs, this.GetType().Name);
using (var conn = new NpgsqlConnection(Configs.QdbConnString))
{
var frameStr = Frame.ToString("yyyy-MM-dd HH:mm:ss.ffffff");
conn.Open();
var trans = conn.BeginTransaction();
try
{
if (DataQueue.Count > 0)
{
var gs = DataQueue.GroupBy(v => new { type = v.ProtocolDataType, itype = v.ProtocolType }).ToList();
foreach (var g in gs)
{
var tableName = ((SugarTable)g.Key.type.GetCustomAttribute(typeof(SugarTable))).TableName;
var cmd = $"insert into {tableName} values('{frameStr}',@Code,";
cmd += string.Join(',', g.Key.itype.GetProperties().Select(v => $"@{v.Name}")) + ")";
var arr = g.ToArray();
conn.Execute(cmd, arr);
}
}
if (DataQueue.Count > 0) conn.Execute($"insert into Frames(Frame) values('{frameStr}')");
trans.Commit();
}
catch (Exception ex)
{
trans.Rollback();
throw;
}
DataQueue.Clear();
}
LogHub.WorldPublish(Logs, GetType().Name);
}
///
/// 异常处理,重写改方法后请自行添加执行内容
/// 执行内容:Exception as KnownException并添加至日志队列
///
///
///
///
protected override void OnError(Channel channel, Exception exception)
{
if (exception is KnownException)
{
var ex = exception as KnownException;
var log = new LogInfo
{ Level = ex.Level, Type = ErrorTypeEnum.Kown, LogUpLoad = ex.logUpLoad, Message = ex.Message };
Logs.Enqueue(new KeyLog { Channel = channel, Log = log, Time = DateTime.Now });
}
else
{
var log = new LogInfo
{
Level = LogLevelEnum.High, Type = ErrorTypeEnum.Unkown, LogUpLoad = LogUpLoadEnum.UpLoadWMS,
Message = exception.Message
};
Logs.Enqueue(new KeyLog { Channel = channel, Log = log, Time = DateTime.Now });
}
}
///
/// 日志处理,重写改方法后请自行添加执行内容
/// 执行内容:LogInfo as KeyLog并添加至日志队列
///
///
///
///
protected override void OnLog(Channel channel, object logObj)
{
if (channel == null) return;
if (logObj.GetType() == typeof(string))
{
Logs.Enqueue(new KeyLog
{
Channel = channel,
Log = new LogInfo
{
Level = LogLevelEnum.High,
LogUpLoad = LogUpLoadEnum.UpLoadWMS,
Message = logObj as string
},
Time = DateTime.Now
});
}
else
{
var log = (LogInfo)logObj;
Logs.Enqueue(new KeyLog { Channel = channel, Log = log, Time = DateTime.Now });
}
}
///
/// 日志处理,重写改方法后请自行添加执行内容
///
///
///
///
protected override void OnInternalLog(Channel channel, string msg)
{
var log = new LogInfo { Level = LogLevelEnum.Low, Message = msg };
if (msg != "开始" && msg != "结束") Logs.Enqueue(new KeyLog { Channel = channel, Log = log, Time = DateTime.Now });
}
///
/// 获取日志,重写改方法后请自行添加执行内容
///
///
///
///
protected override IEnumerable GetChannelMsg(Channel channel)
{
return Logs.Where(v => v.Channel.ToString() == channel.ToString()).Select(v => v.Log.ToString());
}
}