public async void UnRegister(ConsulRegisteration item)
{
var client = new ConsulClient(configuration =>
{
configuration.Address = new Uri(GlobalConfig.Consule);
}); // uses default host:port which is localhost:8500
string service = item.ServiceId;
await client.Agent.ServiceDeregister(service);
// await client.Agent.CheckDeregister(item.ServiceId);
}
/// <summary>
/// 任务的卸载
/// </summary>
/// <param name="taskid"></param>
/// <returns></returns>
public bool Uninstall(int taskid)
{
var taskruntimeinfo = TaskPoolManager.CreateInstance().Get(taskid.ToString());
if (taskruntimeinfo == null)
{
throw new Exception("任务不在运行中");
}
bool r;
if (taskruntimeinfo.TaskModel.task_type == TaskType.Service.Code ||
(taskruntimeinfo.TaskModel.task_type == TaskType.Task.Code && taskruntimeinfo.TaskModel.ServiceFlag == "cron"))
{
try
{
KillProcess(taskid.ToString(), taskruntimeinfo);
if (taskruntimeinfo.TaskModel.task_type == TaskType.Service.Code &&
taskruntimeinfo.TaskModel.IsRegister == 1)
{
ConsulRegisteration item = taskruntimeinfo.RegistService;// _consulRegisterMgr.Parse(taskruntimeinfo.TaskModel);
_consulRegisterMgr.UnRegister(item);
}
r = true;
}
catch (Exception ex)
{
r = false;
}
LogHelper.AddTaskLog("节点卸载任务成功", taskid);
}
else
{
r = DisposeTask(taskid, taskruntimeinfo, true);
}
SqlHelper.ExcuteSql(GlobalConfig.TaskDataBaseConnectString, (c) =>
{
tb_task_dal taskdal = new tb_task_dal();
taskdal.UpdateTaskState(c, taskid, (int)EnumTaskState.Stop);
//update proc id
taskdal.UpdateProcess(c, taskid, -1);
});
LogHelper.AddTaskLog("节点卸载任务成功", taskid);
return(r);
}
public async void Register(ConsulRegisteration item, string url = "")
{
if (string.IsNullOrEmpty(url))
{
url = GlobalConfig.Consule;
}
var client = new ConsulClient(configuration =>
{
configuration.Address = new Uri(url);
}); // uses default host:port which is localhost:8500
//SpinWait.SpinUntil(() => (taskruntimeinfo.Process.MainWindowHandle != IntPtr.Zero));
//while (taskruntimeinfo.Process.MainWindowHandle == IntPtr.Zero)
// Application.DoEvents();
//修改title
//WinApi.SetWindowText(taskruntimeinfo.Process.Handle, item.ServiceId);
var reg = new AgentCheckRegistration()
{
Name = item.Service,
ServiceID = item.ServiceId,
Interval = TimeSpan.FromMinutes(2),
DeregisterCriticalServiceAfter = TimeSpan.FromMinutes(5),
TCP = $"{item.Host}:{item.Port}"
};
var agentReg = new AgentServiceRegistration()
{
Checks = new[] { reg },
Address = item.Host,
ID = item.ServiceId,
Name = item.Service,
Port = item.Port
};
// await client.Agent.CheckRegister(reg);
await client.Agent.ServiceRegister(agentReg);
// WriteResult reult = client.Agent.ServiceRegister(agentReg).Result;
// Console.WriteLine(reult.StatusCode);
}
public ConsulRegisteration Parse(tb_task_model taskruntimeinfoTaskModel
)
{
ConsulRegisteration item = new ConsulRegisteration();
TaskAppConfigInfo config = taskruntimeinfoTaskModel
.taskappconfigjson.FromJson <TaskAppConfigInfo>();
Uri service = new Uri(config["service_url"]);
item.Host = service.Host;
item.Port = service.Port;
string serviceNames = Path.GetFileNameWithoutExtension(taskruntimeinfoTaskModel.taskmainclassdllfilename);
string ext = Path.GetExtension(taskruntimeinfoTaskModel.taskmainclassdllfilename);
item.Service = serviceNames;
item.ServiceId = $"{serviceNames}_{item.Host}_{item.Port}";
return(item);
}
/// <summary>
/// 任务的开启
/// </summary>
/// <param name="taskid"></param>
/// <returns></returns>
public bool Start(int taskid)
{
var taskruntimeinfo = TaskPoolManager.CreateInstance().Get(taskid.ToString());
if (taskruntimeinfo != null)
{
throw new Exception("任务已在运行中");
}
taskruntimeinfo = new NodeTaskRuntimeInfo();
taskruntimeinfo.TaskLock = new TaskLock();
SqlHelper.ExcuteSql(GlobalConfig.TaskDataBaseConnectString, (c) =>
{
tb_task_dal taskdal = new tb_task_dal();
taskruntimeinfo.TaskModel = taskdal.Get(c, taskid);
tb_version_dal versiondal = new tb_version_dal();
taskruntimeinfo.TaskVersionModel = versiondal.GetCurrentVersion(c, taskid, taskruntimeinfo.TaskModel.taskversion);
//taskruntimeinfo.ProcessId=taskdal.GetProcess(c, taskid);
});
//如果异常退出，进程后没有更新
/*if (taskruntimeinfo.TaskModel.task_type == TaskType.Service.Code)
* {
*
* }*/
string filelocalcachepath = AppDomain.CurrentDomain.BaseDirectory.TrimEnd('\\') + "\\" + GlobalConfig.TaskDllCompressFileCacheDir + @"\" + taskruntimeinfo.TaskModel.id + @"\" + taskruntimeinfo.TaskModel.taskversion + @"\" +
taskruntimeinfo.TaskVersionModel.zipfilename;
string fileinstallpath = AppDomain.CurrentDomain.BaseDirectory.TrimEnd('\\') + "\\" + GlobalConfig.TaskDllDir + @"\" + taskruntimeinfo.TaskModel.id;
string fileinstallmainclassdllpath = fileinstallpath + @"\" + taskruntimeinfo.TaskModel.taskmainclassdllfilename;
string taskshareddlldir = AppDomain.CurrentDomain.BaseDirectory.TrimEnd('\\') + "\\" + GlobalConfig.TaskSharedDllsDir;
string shelldlldir = AppDomain.CurrentDomain.BaseDirectory.TrimEnd('\\') + "\\" + GlobalConfig.TaskShellDir;
XXF.Common.IOHelper.CreateDirectory(filelocalcachepath);
XXF.Common.IOHelper.CreateDirectory(fileinstallpath);
// File.WriteAllBytes(filelocalcachepath, taskruntimeinfo.TaskVersionModel.zipfile);
FileStream fs = new FileStream(filelocalcachepath, FileMode.Create, FileAccess.ReadWrite);
SqlHelper.ExcuteSql(GlobalConfig.TaskDataBaseConnectString, (c) =>
{
tb_version_dal versiondal = new tb_version_dal();
versiondal.FillByteToFile(c, taskid, taskruntimeinfo.TaskModel.taskversion, fs);
});
fs.Close();
if (Directory.Exists(fileinstallpath))
{
File.SetAttributes(fileinstallpath, FileAttributes.Normal);
Directory.Delete(fileinstallpath, true);
}
CompressHelper.UnCompress(filelocalcachepath, fileinstallpath);
//拷贝共享程序集
XXF.Common.IOHelper.CopyDirectory(taskshareddlldir, fileinstallpath);
//如果是module
if (taskruntimeinfo.TaskModel.IsModule)
{
XXF.Common.IOHelper.CopyDirectory(shelldlldir, fileinstallpath);
}
LogHelper.AddTaskLog($"原程序集版本：{taskruntimeinfo.TaskVersionModel.assemblyversion}", taskid);
//LogHelper.AddTaskLog($"程序集文件：{fileinstallmainclassdllpath}",taskid);
string assemblyVersion = null;
byte[] bytes = Encoding.Default.GetBytes(taskruntimeinfo.TaskModel
.taskappconfigjson);
string jsonConfig = Convert.ToBase64String(bytes);
bool r = false;
if (taskruntimeinfo.TaskModel.task_type == TaskType.Service.Code)
{
bool is_module = taskruntimeinfo.TaskModel.IsModule;
IProcessBuilder builder;
//当
//
try
{
// string assemblyVersion = GetAssemblyVersion(fileinstallmainclassdllpath);
Process result;
string flag = taskruntimeinfo.TaskModel.ServiceFlag;
var startupParam = new ProcessStartupParam()
{
Flag = flag, FileName = fileinstallmainclassdllpath,
FilePatten = taskruntimeinfo.TaskModel.taskmainclassdllfilename,
Config = jsonConfig, WorkDir = fileinstallpath,
Cron = taskruntimeinfo.TaskModel.taskcron,
NameSpace = taskruntimeinfo.TaskModel.taskmainclassnamespace,
TaskDbConnection = GlobalConfig.TaskDataBaseConnectString,
TaskModel = taskruntimeinfo.TaskModel,
AppConfig = new XXF.Serialization.JsonHelper().Deserialize <TaskAppConfigInfo>(taskruntimeinfo
.TaskModel
.taskappconfigjson)
};
builder = ProcessStart.GetInstance().GetBuilder(startupParam);
builder.GetMainFileName();
assemblyVersion = builder.GetAssemblyVersion();
// taskruntimeinfo.TaskModel.taskmainclassdllfilename = builder.StartupParam.FileName;
result = builder.StartProcess();
r = StartProcess(taskid, taskruntimeinfo, result);
}
catch (Exception ex)
{
LogHelper.AddTaskLog($"节点开启任务失败{ex.Message}", taskid);
throw;
}
// bool r = TaskPoolManager.CreateInstance().Add(taskid.ToString(), taskruntimeinfo);
SqlHelper.ExcuteSql(GlobalConfig.TaskDataBaseConnectString, (c) =>
{
tb_task_dal taskdal = new tb_task_dal();
//更新类型
taskdal.Edit(c, taskruntimeinfo.TaskModel);
taskdal.UpdateTaskState(c, taskid, (int)EnumTaskState.Running);
taskdal.UpdateProcess(c, taskid, taskruntimeinfo.Process.Id);
//程序集版本更新
if (!string.IsNullOrEmpty(assemblyVersion))
{
if (taskruntimeinfo.TaskVersionModel.assemblyversion != assemblyVersion)
{
taskruntimeinfo.TaskVersionModel.assemblyversion = assemblyVersion;
tb_version_dal versiondal = new tb_version_dal();
versiondal.UpdateAssemblyVersion(c, taskruntimeinfo.TaskVersionModel.id, assemblyVersion);
}
}
});
//
if (taskruntimeinfo.TaskModel.IsRegister == 1)
{
ConsulRegisteration item = _consulRegisterMgr.Parse(taskruntimeinfo.TaskModel);
item.Service = builder.GetService();
taskruntimeinfo.RegistService = item;
_consulRegisterMgr.Register(item);
}
LogHelper.AddTaskLog("节点开启任务成功", taskid);
return(r);
}
else
{
//以dll加载的方式处理，基本已废弃
assemblyVersion = GetAssemblyVersion(fileinstallmainclassdllpath);
if (!string.IsNullOrEmpty(taskruntimeinfo.TaskModel.ServiceFlag))
{
return(StartTaskProcess(taskid, taskruntimeinfo, fileinstallmainclassdllpath,
fileinstallpath, jsonConfig, assemblyVersion));
}
else
{
try
{
var dlltask = new AppDomainLoader <BaseDllTask>().Load(fileinstallmainclassdllpath,
taskruntimeinfo.TaskModel.taskmainclassnamespace, out taskruntimeinfo.Domain);
// dlltask.Domain = taskruntimeinfo.Domain;
var sdktaskmodel = new XXF.BaseService.TaskManager.model.tb_task_model();
PropertyHelper.Copy(taskruntimeinfo.TaskModel, sdktaskmodel);
dlltask.SystemRuntimeInfo = new TaskSystemRuntimeInfo()
{
TaskConnectString = GlobalConfig.TaskDataBaseConnectString,
TaskModel = sdktaskmodel
};
dlltask.AppConfig = new TaskAppConfigInfo();
if (!string.IsNullOrEmpty(taskruntimeinfo.TaskModel.taskappconfigjson))
{
dlltask.AppConfig =
new XXF.Serialization.JsonHelper().Deserialize <TaskAppConfigInfo>(taskruntimeinfo
.TaskModel
.taskappconfigjson);
}
taskruntimeinfo.DllTask = dlltask;
/* if (dlltask is IMicroService)
* {
* taskruntimeinfo.TaskModel.task_type = TaskType.Service.Code;
* }
* else
* {
* taskruntimeinfo.TaskModel.task_type = TaskType.Task.Code;
*
* }*/
r = TaskPoolManager.CreateInstance().Add(taskid.ToString(), taskruntimeinfo);
SqlHelper.ExcuteSql(GlobalConfig.TaskDataBaseConnectString, (c) =>
{
tb_task_dal taskdal = new tb_task_dal();
//更新类型
taskdal.Edit(c, taskruntimeinfo.TaskModel);
taskdal.UpdateTaskState(c, taskid, (int)EnumTaskState.Running);
//程序集版本更新
if (!string.IsNullOrEmpty(assemblyVersion))
{
if (taskruntimeinfo.TaskVersionModel.assemblyversion != assemblyVersion)
{
taskruntimeinfo.TaskVersionModel.assemblyversion = assemblyVersion;
tb_version_dal versiondal = new tb_version_dal();
versiondal.UpdateAssemblyVersion(c, taskruntimeinfo.TaskVersionModel.id,
assemblyVersion);
}
}
});
LogHelper.AddTaskLog("节点开启任务成功", taskid);
return(r);
}
catch (Exception exp)
{
DisposeTask(taskid, taskruntimeinfo, true);
throw exp;
}
}
}
}
