246 lines
9.3 KiB
C#
246 lines
9.3 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
||
using MySqlConnector;
|
||
using Quartz;
|
||
using Quartz.Impl;
|
||
using Quartz.Impl.AdoJobStore.Common;
|
||
using Quartz.Spi;
|
||
using RabbitMQ.Client;
|
||
using SqlSugar;
|
||
using System;
|
||
using System.Collections.Specialized;
|
||
using System.ComponentModel.DataAnnotations;
|
||
using System.ComponentModel.DataAnnotations.Schema;
|
||
using System.Linq;
|
||
using HT.Cloud.Code;
|
||
using HT.Cloud.DataBase;
|
||
using HT.Cloud.Domain.SystemOrganize;
|
||
using HT.Cloud.Service.AutoJob;
|
||
using HT.Cloud.Service.Event;
|
||
|
||
namespace HT.Cloud.Service
|
||
{
|
||
/// <summary>
|
||
/// 服务设置
|
||
/// </summary>
|
||
public static class ServiceSetup
|
||
{
|
||
/// <summary>
|
||
/// SqlSugar设置
|
||
/// </summary>
|
||
/// <param name="services"></param>
|
||
public static IServiceCollection AddSqlSugar(this IServiceCollection services)
|
||
{
|
||
var configList = DBInitialize.GetConnectionConfigs(true);
|
||
SqlSugarScope sqlSugarScope = new SqlSugarScope(configList,
|
||
//全局上下文生效
|
||
db =>
|
||
{
|
||
foreach (var item in configList)
|
||
{
|
||
db.GetConnection(item.ConfigId).DefaultConfig();
|
||
}
|
||
});
|
||
//注入数据库连接
|
||
// 注册 SqlSugar
|
||
services.AddSingleton<ISqlSugarClient>(sqlSugarScope);
|
||
return services;
|
||
}
|
||
|
||
/// <summary>
|
||
/// sqlsugar配置
|
||
/// </summary>
|
||
/// <param name="db"></param>
|
||
public static void DefaultConfig(this ISqlSugarClient db)
|
||
{
|
||
db.Ado.CommandTimeOut = GlobalContext.SystemConfig.DBCommandTimeout;
|
||
db.CurrentConnectionConfig.ConfigureExternalServices = new ConfigureExternalServices()
|
||
{
|
||
DataInfoCacheService = new SqlSugarCache(), //配置我们创建的缓存类
|
||
EntityNameService = (type, entity) =>
|
||
{
|
||
var attributes = type.GetCustomAttributes(true);
|
||
if (attributes.Any(it => it is TableAttribute))
|
||
{
|
||
entity.DbTableName = (attributes.First(it => it is TableAttribute) as TableAttribute).Name;
|
||
}
|
||
},
|
||
EntityService = (property, column) =>
|
||
{
|
||
var attributes = property.GetCustomAttributes(true);//get all attributes
|
||
if (attributes.Any(it => it is KeyAttribute))// by attribute set primarykey
|
||
{
|
||
column.IsPrimarykey = true; //有哪些特性可以看 1.2 特性明细
|
||
}
|
||
if (attributes.Any(it => it is ColumnAttribute))
|
||
{
|
||
column.DbColumnName = (attributes.First(it => it is ColumnAttribute) as ColumnAttribute).Name;
|
||
}
|
||
if (attributes.Any(it => it is SugarColumn) && column.DataType == "longtext" && db.CurrentConnectionConfig.DbType == DbType.SqlServer)
|
||
{
|
||
column.DataType = "nvarchar(4000)";
|
||
}
|
||
}
|
||
};
|
||
db.Aop.OnLogExecuted = (sql, pars) => //SQL执行完
|
||
{
|
||
if (sql.StartsWith("SELECT"))
|
||
{
|
||
Console.ForegroundColor = ConsoleColor.Green;
|
||
Console.WriteLine("[SELECT]-" + DateTime.Now.ToString("yyyy/MM/dd HH:mm:ss"));
|
||
}
|
||
if (sql.StartsWith("INSERT"))
|
||
{
|
||
Console.ForegroundColor = ConsoleColor.White;
|
||
Console.WriteLine("[INSERT]-" + DateTime.Now.ToString("yyyy/MM/dd HH:mm:ss"));
|
||
}
|
||
if (sql.StartsWith("UPDATE"))
|
||
{
|
||
Console.ForegroundColor = ConsoleColor.Yellow;
|
||
Console.WriteLine("[UPDATE]-" + DateTime.Now.ToString("yyyy/MM/dd HH:mm:ss"));
|
||
}
|
||
if (sql.StartsWith("DELETE"))
|
||
{
|
||
Console.ForegroundColor = ConsoleColor.Red;
|
||
Console.WriteLine("[DELETE]-" + DateTime.Now.ToString("yyyy/MM/dd HH:mm:ss"));
|
||
}
|
||
Console.WriteLine($"执行库{db.CurrentConnectionConfig.ConfigId}");
|
||
Console.WriteLine("NeedTime-" + db.Ado.SqlExecutionTime.ToString());
|
||
//App.PrintToMiniProfiler("SqlSugar", "Info", sql + "\r\n" + db.Utilities.SerializeObject(pars.ToDictionary(it => it.ParameterName, it => it.Value)));
|
||
Console.WriteLine("Content:" + UtilMethods.GetSqlString(db.CurrentConnectionConfig.DbType, sql, pars));
|
||
Console.WriteLine("---------------------------------");
|
||
Console.WriteLine("");
|
||
};
|
||
}
|
||
|
||
/// <summary>
|
||
/// Quartz设置
|
||
/// </summary>
|
||
/// <param name="services"></param>
|
||
public static IServiceCollection AddQuartz(this IServiceCollection services)
|
||
{
|
||
services.AddSingleton<JobExecute>();
|
||
//注册ISchedulerFactory的实例。
|
||
services.AddSingleton<IJobFactory, IOCJobFactory>();
|
||
if (GlobalContext.SystemConfig.IsCluster != true)
|
||
{
|
||
services.AddSingleton<ISchedulerFactory, StdSchedulerFactory>();
|
||
}
|
||
else
|
||
{
|
||
//开启集群模式,具体数据库从官方github下载
|
||
//https://github.com/quartznet/quartznet/blob/main/database/tables/
|
||
services.AddSingleton<ISchedulerFactory>(u =>
|
||
{
|
||
//当前是mysql的例子，其他数据库做相应更改
|
||
DbProvider.RegisterDbMetadata("mysql-custom", new DbMetadata()
|
||
{
|
||
AssemblyName = typeof(MySqlConnection).Assembly.GetName().Name,
|
||
ConnectionType = typeof(MySqlConnection),
|
||
CommandType = typeof(MySqlCommand),
|
||
ParameterType = typeof(MySqlParameter),
|
||
ParameterDbType = typeof(DbType),
|
||
ParameterDbTypePropertyName = "DbType",
|
||
ParameterNamePrefix = "@",
|
||
ExceptionType = typeof(MySqlException),
|
||
BindByName = true
|
||
});
|
||
var properties = new NameValueCollection
|
||
{
|
||
["quartz.jobStore.type"] = "Quartz.Impl.AdoJobStore.JobStoreTX, Quartz", // 配置Quartz以使用JobStoreTx
|
||
//["quartz.jobStore.useProperties"] = "true", // 配置AdoJobStore以将字符串用作JobDataMap值
|
||
["quartz.jobStore.dataSource"] = "myDS", // 配置数据源名称
|
||
["quartz.jobStore.tablePrefix"] = "QRTZ_", // quartz所使用的表，在当前数据库中的表前缀
|
||
["quartz.jobStore.driverDelegateType"] = "Quartz.Impl.AdoJobStore.MySQLDelegate, Quartz", // 配置AdoJobStore使用的DriverDelegate
|
||
["quartz.dataSource.myDS.connectionString"] = AESKeyEncrypt.DecryptDES(GlobalContext.SystemConfig.DBConnectionString), // 配置数据库连接字符串，自己处理好连接字符串，我这里就直接这么写了
|
||
["quartz.dataSource.myDS.provider"] = "mysql-custom", // 配置数据库提供程序（这里是自定义的，定义的代码在上面）
|
||
["quartz.jobStore.lockHandler.type"] = "Quartz.Impl.AdoJobStore.UpdateLockRowSemaphore, Quartz",
|
||
["quartz.serializer.type"] = "json",
|
||
["quartz.jobStore.clustered"] = "true", // 指示Quartz.net的JobStore是应对 集群模式
|
||
["quartz.scheduler.instanceId"] = "AUTO"
|
||
};
|
||
return new StdSchedulerFactory(properties);
|
||
});
|
||
}
|
||
//是否开启后台任务
|
||
if (GlobalContext.SystemConfig.OpenQuartz == true)
|
||
{
|
||
services.AddHostedService<JobCenter>();
|
||
}
|
||
return services;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 重置超管密码
|
||
/// </summary>
|
||
public static IServiceCollection ReviseSuperSysem(this IServiceCollection services)
|
||
{
|
||
var data = GlobalContext.SystemConfig;
|
||
try
|
||
{
|
||
if (data.ReviseSystem == true)
|
||
{
|
||
using (var context = new SqlSugarClient(DBContexHelper.Contex()))
|
||
{
|
||
context.Ado.BeginTran();
|
||
var systemSet = context.Queryable<SystemSetEntity>().First(a => a.F_DbNumber == data.MainDbNumber);
|
||
var user = context.Queryable<UserEntity>().First(a => a.F_CompanyId == systemSet.F_Id && a.F_IsAdmin == true);
|
||
var userinfo = context.Queryable<UserLogOnEntity>().Where(a => a.F_UserId == user.F_Id).First();
|
||
userinfo.F_UserSecretkey = Md5.md5(Utils.CreateNo(), 16).ToLower();
|
||
userinfo.F_UserPassword = Md5.md5(DESEncrypt.Encrypt(Md5.md5(systemSet.F_AdminPassword, 32).ToLower(), userinfo.F_UserSecretkey).ToLower(), 32).ToLower();
|
||
context.Updateable<UserEntity>(a => new UserEntity
|
||
{
|
||
F_Account = systemSet.F_AdminAccount
|
||
}).Where(a => a.F_Id == userinfo.F_Id).ExecuteCommand();
|
||
context.Updateable<UserLogOnEntity>(a => new UserLogOnEntity
|
||
{
|
||
F_UserPassword = userinfo.F_UserPassword,
|
||
F_UserSecretkey = userinfo.F_UserSecretkey
|
||
}).Where(a => a.F_Id == userinfo.F_Id).ExecuteCommand();
|
||
context.Ado.CommitTran();
|
||
CacheHelper.Remove(GlobalContext.SystemConfig.ProjectPrefix + "_operator_" + "info_" + user.F_Id);
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
LogHelper.Write(ex);
|
||
}
|
||
return services;
|
||
}
|
||
/// <summary>
|
||
/// 注册事件总线
|
||
/// </summary>
|
||
/// <param name="services"></param>
|
||
/// <returns></returns>
|
||
public static IServiceCollection AddEventBus(this IServiceCollection services)
|
||
{
|
||
// 注册 EventBus 服务
|
||
return services.AddEventBus(builder =>
|
||
{
|
||
if (GlobalContext.SystemConfig.RabbitMq.Enabled)
|
||
{
|
||
// 创建连接工厂
|
||
var factory = new ConnectionFactory
|
||
{
|
||
HostName= GlobalContext.SystemConfig.RabbitMq.HostName,
|
||
VirtualHost= GlobalContext.SystemConfig.RabbitMq.VirtualHost,
|
||
Port= GlobalContext.SystemConfig.RabbitMq.Port,
|
||
UserName = GlobalContext.SystemConfig.RabbitMq.UserName,
|
||
Password = GlobalContext.SystemConfig.RabbitMq.Password
|
||
};
|
||
// 创建默认内存通道事件源对象，可自定义队列路由key，比如这里是 eventbus
|
||
var rbmqEventSourceStorer = new RabbitMQEventSourceStorer(factory, GlobalContext.SystemConfig.ProjectPrefix
|
||
+ "_eventbus", 3000);
|
||
// 替换默认事件总线存储器
|
||
builder.ReplaceStorer(serviceProvider =>
|
||
{
|
||
return rbmqEventSourceStorer;
|
||
});
|
||
}
|
||
// 注册 ToDo 事件订阅者
|
||
builder.AddSubscriber<LogEventSubscriber>();
|
||
builder.AddSubscriber<MessageEventSubscriber>();
|
||
});
|
||
}
|
||
}
|
||
} |