using log4net;
|
using SolidWorks.Interop.sldworks;
|
using SolidWorks.Interop.swconst;
|
using SuperSocket.ClientEngine;
|
using System;
|
using System.Diagnostics;
|
using System.Net;
|
using System.Runtime.InteropServices;
|
using System.ServiceProcess;
|
using System.Text;
|
using System.Timers;
|
using System.Windows;
|
using Test1;
|
|
namespace PluginListener
|
{
|
public class Listener : ServiceBase
|
{
|
ILog log = LogManager.GetLogger("Listener");
|
private AsyncTcpSession client;
|
|
private long err_count;
|
|
private System.Timers.Timer socketTimer;
|
private System.Timers.Timer importTimer;
|
private System.Timers.Timer checkInTimer;
|
|
private Alert alert;
|
private volatile int is_open = int.MaxValue;
|
|
private int msgInterval = 5000;
|
|
#region 消息
|
public static IntPtr WTS_CURRENT_SERVER_HANDLE = IntPtr.Zero;
|
|
public void ShowMessageBox(string message, string title)
|
{
|
/*if (is_open == 0)
|
{
|
return;
|
}*/
|
is_open = 0;
|
int res;
|
WTSSendMessage(
|
WTS_CURRENT_SERVER_HANDLE,
|
WTSGetActiveConsoleSessionId(),
|
title, title.Length,
|
message, message.Length,
|
0, 0, out res, true);
|
is_open = res;
|
}
|
|
[DllImport("kernel32.dll", SetLastError = true)]
|
public static extern int WTSGetActiveConsoleSessionId();
|
|
[DllImport("wtsapi32.dll", SetLastError = true)]
|
public static extern bool WTSSendMessage(
|
IntPtr hServer,
|
int SessionId,
|
String pTitle,
|
int TitleLength,
|
String pMessage,
|
int MessageLength,
|
int Style,
|
int Timeout,
|
out int pResponse,
|
bool bWait);
|
#endregion
|
|
|
#region 组件设计器生成的代码
|
/// <summary>
|
/// 必需的设计器变量。
|
/// </summary>
|
private System.ComponentModel.IContainer components = null;
|
|
/// <summary>
|
/// 清理所有正在使用的资源。
|
/// </summary>
|
/// <param name="disposing">如果应释放托管资源，为 true；否则为 false。</param>
|
protected override void Dispose(bool disposing)
|
{
|
if (disposing && (components != null))
|
{
|
components.Dispose();
|
}
|
base.Dispose(disposing);
|
}
|
|
/// <summary>
|
/// 设计器支持所需的方法 - 不要修改
|
/// 使用代码编辑器修改此方法的内容。
|
/// </summary>
|
private void InitializeComponent()
|
{
|
components = new System.ComponentModel.Container();
|
this.ServiceName = "PdmSwPluginLisener";
|
this.CanStop = true;
|
}
|
#endregion
|
|
private void AlertMsg(string msg)
|
{
|
/// 内置WPF的 不能用
|
// alert.Open(msg);
|
|
/*try
|
{
|
// SwApp = (SldWorks)Marshal.GetActiveObject("SldWorks.Application.28");
|
var swApp = (SldWorks)System.Runtime.InteropServices.Marshal.GetActiveObject("SldWorks.Application");
|
swApp.SendMsgToUser2(msg, (int)swMessageBoxIcon_e.swMbStop, (int)swMessageBoxBtn_e.swMbOk);
|
Marshal.ReleaseComObject(swApp);
|
}
|
catch (COMException ex)
|
{
|
log.Error("Attach Sw Failed.", ex);
|
}*/
|
|
WinAPI_Interop.ShowServiceMessage(msg, "提示");
|
// ShowMessageBox(msg, "提示");
|
|
/// 用MessageBox的 不能用
|
/*if (swApp == null)
|
{
|
int res;
|
WTSSendMessage(
|
WTS_CURRENT_SERVER_HANDLE,
|
WTSGetActiveConsoleSessionId(),//获得当前显示的桌面所在的SessionID
|
"提示", "提示".Length,
|
msg, msg.Length,
|
0, 0, out res, false);
|
}
|
else
|
{
|
swApp.SendMsgToUser2(msg, (int)swMessageBoxIcon_e.swMbStop, (int)swMessageBoxBtn_e.swMbOk);
|
}*/
|
|
//
|
|
///启动exe的
|
/*ProcessStartInfo psi = new ProcessStartInfo
|
{
|
FileName = "AlertWindow.exe",
|
UseShellExecute = false,
|
CreateNoWindow = true
|
};
|
Process process = Process.Start(psi);*/
|
}
|
|
public Listener()
|
{
|
alert = new Alert();
|
client = new AsyncTcpSession();
|
client.Connected += Client_Connected;
|
client.Error += Client_Error;
|
client.Closed += Client_Closed;
|
client.DataReceived += Client_DataReceived;
|
InitializeComponent();
|
log.Info("Service Started.");
|
}
|
|
protected override void OnStart(string[] args)
|
{
|
Start();
|
}
|
|
protected override void OnStop()
|
{
|
Stop();
|
}
|
|
public void Start()
|
{
|
InitSocketTimer();
|
}
|
|
public void Stop()
|
{
|
socketTimer?.Stop();
|
StopInterceptTimer();
|
// alert.DoClose();
|
// if (intercepter != null) intercepter.StopIntercept();
|
}
|
|
/// <summary>
|
/// 启动Socket定时器
|
/// </summary>
|
public void InitSocketTimer()
|
{
|
log.Info("Starting Socket Timer...");
|
socketTimer = new System.Timers.Timer(msgInterval);
|
socketTimer.Elapsed += Timer_Elapsed;
|
socketTimer.Enabled = true;
|
socketTimer.Start();
|
log.Info("Started Socket Timer!");
|
}
|
|
/// <summary>
|
/// 启动金蝶拦截任务
|
/// </summary>
|
public void StartInterceptTimer()
|
{
|
// if (intercepter == null) intercepter = new Intercepter(ListenImportHandle, ListenCheckInHandle);
|
if (importTimer == null)
|
{
|
importTimer = new System.Timers.Timer(500);
|
importTimer.Elapsed += ImportTimer_Elapsed;
|
}
|
importTimer.Start();
|
if (checkInTimer == null)
|
{
|
checkInTimer = new System.Timers.Timer(500);
|
checkInTimer.Elapsed += CheckInTimer_Elapsed;
|
}
|
checkInTimer.Start();
|
}
|
|
private void ImportTimer_Elapsed(object sender, ElapsedEventArgs e)
|
{
|
lock (importTimer)
|
{
|
ListenImportHandle();
|
}
|
}
|
|
private void CheckInTimer_Elapsed(object sender, ElapsedEventArgs e)
|
{
|
lock (checkInTimer)
|
{
|
ListenCheckInHandle();
|
}
|
}
|
|
/// <summary>
|
/// 停止金蝶拦截任务
|
/// </summary>
|
private void StopInterceptTimer()
|
{
|
/*if (intercepter != null)
|
{
|
intercepter.StopIntercept();
|
intercepter = null;
|
}*/
|
importTimer?.Stop();
|
checkInTimer?.Stop();
|
}
|
|
/// <summary>
|
/// 插件返回正常时的处理
|
/// </summary>
|
private void HandleNormal()
|
{
|
err_count = 0;
|
StopInterceptTimer();
|
// alert.DoHide();
|
}
|
|
private bool IsSwRunning()
|
{
|
Process[] ps = Process.GetProcessesByName("SLDWORKS");
|
bool IsRunning = (ps != null && ps.Length > 0);
|
log.Debug($"Sw Is Running: {IsRunning}");
|
return IsRunning;
|
}
|
|
/// <summary>
|
/// 插件返回异常时的处理
|
/// </summary>
|
private void HandleErr()
|
{
|
log.Debug($"err_count: {err_count}");
|
if (++err_count > 3)
|
{
|
if (IsSwRunning())
|
{
|
// 如果solidworks启动
|
StartInterceptTimer();
|
log.Debug("Ready to show Msg.");
|
AlertMsg("请启用插件并登录");
|
err_count = 3;
|
}
|
else
|
{
|
HandleNormal();
|
}
|
}
|
}
|
|
|
/// <summary>
|
/// 搜索是否有导入窗口，有就一直最小化，并执行规则检查
|
/// </summary>
|
/// <returns></returns>
|
private void ListenImportHandle()
|
{
|
IntPtr ptr = WindowIntPtrUtil.FindWindow(null, "Everything");//导入文档
|
if (ptr == IntPtr.Zero)
|
{
|
return;
|
}
|
WindowIntPtrUtil.KillProcessByIntPtr(ptr);
|
}
|
|
/// <summary>
|
/// 搜索是否有检入窗口，有就一直最小化，并执行规则检查
|
/// </summary>
|
/// <returns></returns>
|
private void ListenCheckInHandle()
|
{
|
IntPtr ptr = WindowIntPtrUtil.FindWindow(null, "检入文档");
|
if (ptr == IntPtr.Zero)
|
{
|
return;
|
}
|
WindowIntPtrUtil.KillProcessByIntPtr(ptr);
|
}
|
|
private void Timer_Elapsed(object sender, ElapsedEventArgs e)
|
{
|
socketTimer.Stop();
|
try
|
{
|
if (!client.IsConnected)
|
{
|
client.Connect(new IPEndPoint(IPAddress.Parse("127.0.0.1"), 5678));
|
}
|
else
|
{
|
var msg = Encoding.Default.GetBytes("HeatBeat");
|
client.Send(msg, 0, msg.Length);
|
}
|
}
|
catch (Exception ex)
|
{
|
HandleErr();
|
}
|
finally
|
{
|
socketTimer.Start();
|
}
|
}
|
|
/// <summary>
|
/// 接收到消息
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void Client_DataReceived(object sender, DataEventArgs e)
|
{
|
byte flag = e.Data[0];
|
if (flag == 1)
|
{
|
HandleNormal();
|
}
|
else
|
{
|
log.Error("Plugin Return 0.");
|
HandleErr();
|
}
|
}
|
|
/// <summary>
|
/// 客户端关闭，视为异常
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void Client_Closed(object sender, EventArgs e)
|
{
|
HandleErr();
|
}
|
|
/// <summary>
|
/// 客户端异常
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void Client_Error(object sender, ErrorEventArgs e)
|
{
|
HandleErr();
|
}
|
|
/// <summary>
|
/// 连接后的操作
|
/// </summary>
|
/// <param name="sender"></param>
|
/// <param name="e"></param>
|
private void Client_Connected(object sender, EventArgs e)
|
{
|
|
}
|
}
|
}
|
