369 lines
13 KiB
C#
369 lines
13 KiB
C#
using CsharpHttpHelper;
|
||
using PCRobot.Pack;
|
||
using PCRobot.PCWechat;
|
||
using PCRobot.Utils;
|
||
using SuperSocket.ClientEngine;
|
||
using SuperSocket.ProtoBase;
|
||
using System;
|
||
using System.Linq;
|
||
using System.Net;
|
||
using System.Text;
|
||
using System.Text.RegularExpressions;
|
||
using System.Threading;
|
||
|
||
namespace PCRobot
|
||
{
|
||
public class EasySoc
|
||
{
|
||
#region 字段信息
|
||
static EasySoc _soc;
|
||
|
||
public EasyClient client;
|
||
|
||
public string key { get; private set; }
|
||
|
||
string ip;
|
||
|
||
int port;
|
||
|
||
public bool Stop { get; set; }
|
||
|
||
public bool IsConnected
|
||
{
|
||
get
|
||
{
|
||
return client != null && client.IsConnected;
|
||
}
|
||
}
|
||
|
||
readonly static object session_lock = new object();
|
||
|
||
IPEndPoint endPoint;
|
||
|
||
Action<string, string> ReceiveDataAction;
|
||
|
||
Action<string> StateAction;
|
||
|
||
Action<Exception> SocketErrorAction;
|
||
|
||
long i = 0;
|
||
|
||
//Thread threadConnection;
|
||
|
||
static bool lock_connection = false;
|
||
#endregion
|
||
|
||
public static EasySoc GetSocket()
|
||
{
|
||
if (_soc == null) _soc = new EasySoc();
|
||
return _soc;
|
||
}
|
||
private System.Threading.Timer timer = null;
|
||
public EasySoc()
|
||
{
|
||
//timer = new Timer(new TimerCallback(timerCall), null, 0, 1000 * 5);
|
||
var thread = new Thread(() =>
|
||
{
|
||
while (true)
|
||
{
|
||
Thread.Sleep(1000 * 5);
|
||
timerCall(null);
|
||
}
|
||
});
|
||
thread.IsBackground = true;
|
||
thread.Start();
|
||
//FluentScheduler.JobManager.AddJob(heartBeat, t =>
|
||
//{
|
||
// t.WithName("心跳").ToRunNow().AndEvery(5).Seconds();
|
||
//});
|
||
}
|
||
|
||
|
||
private void timerCall(object state)
|
||
{
|
||
if (string.IsNullOrEmpty(ip) || string.IsNullOrEmpty(key) || port == 0) return;
|
||
|
||
//throw new NotImplementedException();
|
||
var sb = new StringBuilder();
|
||
try
|
||
{
|
||
sb.AppendLine("AAA");
|
||
if (Stop)
|
||
{
|
||
sb.AppendLine("BBB");
|
||
if (IsConnected) Close();
|
||
sb.AppendLine("ccc");
|
||
StateAction?.Invoke($"您已断开与服务器{ip}失联！");
|
||
return;
|
||
}
|
||
sb.AppendLine("DDD");
|
||
if (!IsConnected)
|
||
{
|
||
sb.AppendLine("EEE");
|
||
StateAction?.Invoke($"与服务器{ip}失联，重连中...");
|
||
|
||
if (WechatClient.IsLog)
|
||
{
|
||
sb.AppendLine("FFF");
|
||
LogHelper.GetSingleObj().Info("失联", $"与服务器{ip}失联，重连中...");
|
||
}
|
||
Connection(ip, port, ReceiveDataAction, SocketErrorAction, key, StateAction);
|
||
sb.AppendLine("GGG");
|
||
}
|
||
else
|
||
{
|
||
sb.AppendLine("HHH");
|
||
i++;
|
||
//Console.WriteLine($"第{i}次测试,key={this.key}");
|
||
|
||
if (i == 1 || i % 6 == 0)
|
||
{
|
||
sb.AppendLine("TTT");
|
||
Send(new CommonResult()
|
||
{
|
||
Cmd = PCRobotCMD.heartBreak,
|
||
RobotType = RobotType.客户端系统,
|
||
RobotUsername = string.Empty,
|
||
RobotUsernick = "测试-nick",
|
||
Data = Common.GetDevice()
|
||
});
|
||
Thread.Sleep(50);
|
||
sb.AppendLine("LLL");
|
||
var robots = WechatClient.Users.Values.ToList();
|
||
foreach (var item in robots)
|
||
{
|
||
sb.AppendLine("MMM");
|
||
if (item.HeartBreakTime.AddMinutes(6) > DateTime.Now)
|
||
{
|
||
var msg = new WechatStatus()
|
||
{
|
||
Cmd = item.Type == WechatType.Xiaoxie_QY ? PCRobotCMD.offline_workWeChat : PCRobotCMD.offline,
|
||
RobotUsername = item.Wxid,
|
||
RobotUsernick = item.Nickname,//Regex.Replace(item.Nickname, @"\s", "", RegexOptions.IgnoreCase),
|
||
RobotType = item.Type == WechatType.Xiaoxie_QY ? RobotType.客户端企业微信 : RobotType.客户端微信,
|
||
Status = Status.在线,
|
||
Uin = item.Uin,
|
||
Device = Common.GetDevice()
|
||
};
|
||
Send(msg);
|
||
Thread.Sleep(50);
|
||
}
|
||
else
|
||
{
|
||
sb.AppendLine("NNN");
|
||
item.HeartBreakTime = DateTime.Now;
|
||
Close();
|
||
if (!IsConnected) Connection(ip, port, ReceiveDataAction, SocketErrorAction, key, StateAction);
|
||
}
|
||
}
|
||
}
|
||
sb.AppendLine("ZZZ");
|
||
StateAction?.Invoke($"与服务器: {ip} 连接成功.");
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
if (WechatClient.IsLog)
|
||
LogHelper.GetSingleObj().Info("心跳包异常1", $"{ex.Message}");
|
||
StateAction?.Invoke($"心跳包异常: {ex.Message} {ex.StackTrace} +{sb.ToString()}");
|
||
}
|
||
}
|
||
|
||
public void Send(BaseMsg msg)
|
||
{
|
||
try
|
||
{
|
||
if (IsConnected)
|
||
{
|
||
try
|
||
{
|
||
msg.Key = this.key;
|
||
var json = HttpHelper.ObjectToJson(msg);
|
||
var data = PackTool.CompressString(json);
|
||
var send = msg.Cmd + " " + data + "\r\n";
|
||
var data_bytes = Encoding.UTF8.GetBytes(send);
|
||
if (WechatClient.IsDebug) LogHelper.GetSingleObj().Debug("往服务器发送", $"{msg.Cmd} - {json} - {data.Length}");
|
||
|
||
//var senddata = new byte[_receiveBufferSize];
|
||
//for (int i = 0; i < data_bytes.Length; i++)
|
||
//{
|
||
// senddata[i] = data_bytes[i];
|
||
//}
|
||
|
||
client.Send(data_bytes);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
StateAction?.Invoke($"发送消息失败: {ex.Message}");
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
if (WechatClient.IsLog)
|
||
LogHelper.GetSingleObj().Info("Send ERROR", $"{ex.Message}");
|
||
LogHelper.GetSingleObj().Error("Send ERROR:", $"{ex.Message} - {ex.StackTrace}");
|
||
}
|
||
}
|
||
|
||
private static int _receiveBufferSize = 1024 * 1024 * 10;
|
||
public void Connection(string ip, int port, Action<string, string> receiveDataAction, Action<Exception> socketErrorAction, string key, Action<string> stateAction)
|
||
{
|
||
if (Stop) return;
|
||
if (lock_connection) return;
|
||
if (key.Length != 16) throw new Exception("秘钥必须 == 16位！");
|
||
try
|
||
{
|
||
lock_connection = true;
|
||
if (this.IsConnected) this.Close();
|
||
//Console.WriteLine("连接中...");
|
||
this.ip = ip;
|
||
this.port = port;
|
||
if (receiveDataAction != null)
|
||
{
|
||
this.ReceiveDataAction = receiveDataAction;
|
||
}
|
||
if (stateAction != null)
|
||
{
|
||
this.StateAction = stateAction;
|
||
}
|
||
if (socketErrorAction != null)
|
||
{
|
||
this.SocketErrorAction = socketErrorAction;
|
||
}
|
||
|
||
this.key = key.Substring(0, 16);
|
||
this.endPoint = new IPEndPoint(IPAddress.Parse(ip), port);
|
||
lock (session_lock)
|
||
{
|
||
client = new EasyClient();
|
||
client.Initialize(new MyReceiveFilter(), clientRcv);
|
||
client.ReceiveBufferSize = _receiveBufferSize;
|
||
|
||
|
||
//LogHelper.GetSingleObj().Info("", "client.ReceiveBufferSize = " + client.ReceiveBufferSize);
|
||
|
||
//client.DataReceived += Client_DataReceived;
|
||
|
||
client.Connected += Client_Connected;
|
||
client.Closed += Client_Closed;
|
||
client.Error += Client_Error;
|
||
var rst = client.ConnectAsync(endPoint).Result;
|
||
Thread.Sleep(500);
|
||
if (!client.IsConnected)
|
||
{
|
||
rst = client.ConnectAsync(endPoint).Result;
|
||
if (!client.IsConnected)
|
||
throw new Exception("连接服务器失败,请检查是否填写正确!");
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
if (WechatClient.IsLog)
|
||
LogHelper.GetSingleObj().Info("断开", $"连接服务器失败:{ex.Message}");
|
||
StateAction?.Invoke($"连接服务器失败：{ex.Message} - {ex.StackTrace}");
|
||
}
|
||
finally
|
||
{
|
||
lock_connection = false;
|
||
}
|
||
}
|
||
|
||
private void clientRcv(StringPackageInfo obj)
|
||
{
|
||
if (WechatClient.IsDebug)
|
||
LogHelper.GetSingleObj().Debug("", $"收到服务器消息:{obj.Key} - {obj.Body}");
|
||
|
||
if (obj.Key.ToLower().Trim().StartsWith("offline".ToLower()))
|
||
{
|
||
try
|
||
{
|
||
var json = HttpExtend.JsonToDictionary(obj.Body);
|
||
if (json != null)
|
||
{
|
||
var client = WechatClient.Users.Values.ToList().FirstOrDefault(f => f.Wxid == json["RobotUsername"].ToString());
|
||
if (client != null)
|
||
client.HeartBreakTime = DateTime.Now;
|
||
}
|
||
else
|
||
{
|
||
var text = PackTool.DecompressString(obj.Body);
|
||
var json1 = HttpExtend.JsonToDictionary(text);
|
||
if (json1 != null)
|
||
{
|
||
var client = WechatClient.Users.Values.ToList().FirstOrDefault(f => f.Wxid == json1["RobotUsername"].ToString());
|
||
if (client != null)
|
||
client.HeartBreakTime = DateTime.Now;
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{ }
|
||
return;
|
||
}
|
||
//Console.WriteLine(obj.Body);
|
||
//ReceiveDataAction.BeginInvoke(Encoding.UTF8.GetString(new_data), null, null);
|
||
ReceiveDataAction.BeginInvoke(obj.Key, obj.Body, null, null);
|
||
}
|
||
|
||
public void Close(bool isQuit = false)
|
||
{
|
||
try
|
||
{
|
||
lock (session_lock)
|
||
{
|
||
if (client != null)
|
||
{
|
||
if (isQuit)
|
||
{
|
||
FluentScheduler.JobManager.RemoveJob("心跳");
|
||
}
|
||
|
||
//client.DataReceived -= Client_DataReceived;
|
||
client.Connected -= Client_Connected;
|
||
client.Closed -= Client_Closed;
|
||
client.Error -= Client_Error;
|
||
client.Close();
|
||
client = null;
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
if (WechatClient.IsLog)
|
||
LogHelper.GetSingleObj().Info("Close ERROR:", $"{ex.Message}");
|
||
|
||
LogHelper.GetSingleObj().Error("Close ERROR:", $"{ex.Message} - {ex.StackTrace}");
|
||
}
|
||
}
|
||
|
||
private void Client_Error(object sender, ErrorEventArgs e)
|
||
{
|
||
if (SocketErrorAction != null)
|
||
{
|
||
if (WechatClient.IsLog)
|
||
LogHelper.GetSingleObj().Info("Client_Error", $"{e.Exception.Message}");
|
||
SocketErrorAction(e.Exception);
|
||
}
|
||
}
|
||
|
||
private void Client_Closed(object sender, EventArgs e)
|
||
{
|
||
if (WechatClient.IsLog)
|
||
LogHelper.GetSingleObj().Info("断开", $"与服务器{ip},{e} - 连接断开！！");
|
||
StateAction?.Invoke($"与服务器{ip}连接断开！");
|
||
}
|
||
|
||
private void Client_Connected(object sender, EventArgs e)
|
||
{
|
||
if (WechatClient.IsLog)
|
||
{
|
||
LogHelper.GetSingleObj().Info("连接成功", $"与服务器{ip}连接成功！");
|
||
}
|
||
StateAction?.Invoke($"与服务器{ip}连接成功！");
|
||
}
|
||
|
||
}
|
||
}
|