/// <summary>
///
/// </summary>
/// <param name="mq"></param>
/// <param name="type"></param>
/// <returns></returns>
public static Task <IMQ> SubscibeTypeAsync(this IMQ mq, Type type)
{
return(Task.Run(() =>
{
return mq.SubscibeType(type);
}));
}
/// <summary>
/// 扫描该程序集所有实现了<seealso cref="IHandler{TMessage}"/>接口的类并订阅他们
/// </summary>
/// <param name="mq"></param>
/// <param name="assemblys">程序集合</param>
/// <returns></returns>
public static IMQ SubscibeAssembly(this IMQ mq, params Assembly[] assemblys)
{
if (assemblys?.Count() <= 0)
{
assemblys = new Assembly[] { Assembly.GetCallingAssembly() };
}
var handleType = typeof(IHandler);
var method = typeof(MQExtension).GetMethod(nameof(SubscibeType), new Type[] { typeof(IMQ) });
foreach (var assembly in assemblys)
{
foreach (var type in assembly.GetTypes())
{
if (type.IsClass && !type.IsInterface && handleType.IsAssignableFrom(type))
{
var methodInfo = method.MakeGenericMethod(type);
methodInfo.Invoke(null, new object[] { mq });
}
}
}
return(mq);
}
/// <summary>
/// 添加事件处理
/// </summary>
/// <typeparam name="TMessage"></typeparam>
/// <param name="mq"></param>
/// <param name="action"></param>
/// <returns></returns>
public static void Add <TMessage>(this IMQ mq, Action <TMessage> action)
{
Action <TMessage> actionMessage = null;
var key = NameUtils.Get <TMessage>();
object @object;
lock (_store)
{
var dic = GetSotrage(mq);
if (dic.TryGetValue(key, out @object))
{
actionMessage = @object as Action <TMessage>;
actionMessage += action;
}
else
{
actionMessage = action;
}
dic[key] = actionMessage;
//_store.AddOrUpdate(key, actionMessage, (a, b) => actionMessage);
}
}
/// <summary>
///
/// </summary>
/// <param name="mq"></param>
/// <param name="type"></param>
/// <returns></returns>
public static IMQ SubscibeType(this IMQ mq, Type type)
{
var typeHandler = type;
if (typeHandler.IsInterface)
{
return(mq);
}
var handleInterface = typeof(IHandler <>);
var subscibeMethod = typeof(MQExtension).GetMethod(nameof(SubscibeInternelHandle), BindingFlags.NonPublic |
BindingFlags.Static |
BindingFlags.IgnoreCase);
var internalHandleType = typeof(Handle <>);
Func <object> funcObject = () => ServiceFacade.Provider.GetService(typeHandler);
foreach (var @interface in typeHandler.GetInterfaces())
{
if (@interface.IsGenericType && @interface.GetGenericTypeDefinition() == handleInterface)
{
var argumentType = @interface.GenericTypeArguments[0];
var handleType = internalHandleType.MakeGenericType(argumentType);
var internalHandleObject = handleType.GetConstructors().First().Invoke(new object[] { funcObject });
subscibeMethod.MakeGenericMethod(argumentType)
.Invoke(null, new object[] { mq, internalHandleObject });
}
}
return(mq);
}
/// <summary>
/// 获得处理函数
/// </summary>
/// <typeparam name="TMessage"></typeparam>
/// <param name="mq"></param>
/// <returns></returns>
public static Action <TMessage> Get <TMessage>(this IMQ mq)
{
Action <TMessage> actionMessage = null;
var key = NameUtils.Get <TMessage>();
object @object = null;
if (GetSotrage(mq).TryGetValue(key, out @object))
{
actionMessage = @object as Action <TMessage>;
}
if (actionMessage == null)
{
return(m =>
{
var log = ServiceFacade.Provider
.GetService <ILog <IMQ> >();
log?.Info($"订阅函数:\"{key}\"不存在");
});
}
else
{
return(actionMessage);
}
}
public MQTest(ITestOutputHelper output) : base(output)
{
//_mq = Unit.GetProvider(s => s.AddMQ(new TestConfig()))
// .GetService<IMQ>();
_mq = Unit.GetProvider(s => s.AddMemoryMQ())
.GetService <IMQ>();
}
private static Dictionary <string, object> GetSotrage(IMQ mq)
{
Dictionary <string, object> dic;
if (!_store.TryGetValue(mq, out dic))
{
dic = new Dictionary <string, object>();
_store.AddOrUpdate(mq, dic, (a, b) => dic);
}
return(dic);
}
/// <summary>
/// 扫描该程序集所有实现了<seealso cref="IHandler{TMessage}"/>接口的类并订阅他们
/// </summary>
/// <param name="mq"></param>
/// <param name="assemblys">程序集合</param>
/// <returns></returns>
public static Task <IMQ> SubscibeAssemblyAsync(this IMQ mq, params Assembly[] assemblys)
{
if (assemblys?.Count() <= 0)
{
assemblys = new Assembly[] { Assembly.GetCallingAssembly() };
}
return(Task.Run(() =>
{
return mq.SubscibeAssembly(assemblys);
}));
}
public DocumentStoreTest(ITestOutputHelper testOutputHelper) : base(testOutputHelper)
{
var serviceProvider = Unit.GetProvider(s => s.AddMemberDocument()
.AddMemoryPersistenceService()
.AddMemoryMQ()
.AddDefaultHandler()
.AddLogger());
serviceProvider.UseDefaultDocumentPublish();
this._documentStore = serviceProvider.GetService <IDocumentStore>();
this._mq = serviceProvider.GetService <IMQ>();
this._persistenceService = serviceProvider.GetService <IPersistenceService>();
}
/// <summary>
/// 创建一个构建者对象
/// </summary>
/// <param name="mq"></param>
internal RegistrationBuilder(IMQ mq)
{
this.MQ = mq;
this._store = new List <Map>();
}
/// <summary>
/// 订阅<typeparamref name="THandler"/>的所有<seealso cref="IHandler{TMessage}"/>接口的实现
/// </summary>
/// <typeparam name="THandler"></typeparam>
/// <param name="mq"></param>
/// <returns></returns>
public static IMQ SubscibeType <THandler>(this IMQ mq) where THandler : IHandler
{
return(mq.SubscibeType(typeof(THandler)));
}
/// <summary>
/// 订阅<typeparamref name="THandler"/>的所有<seealso cref="IHandler{TMessage}"/>接口的实现
/// </summary>
/// <typeparam name="THandler"></typeparam>
/// <param name="mq"></param>
/// <returns></returns>
public static Task <IMQ> SubscibeTypeAsync <THandler>(this IMQ mq) where THandler : IHandler
{
return(mq.SubscibeTypeAsync(typeof(THandler)));
}
/// <summary>
/// 异步订阅
/// </summary>
/// <typeparam name="TMessage">消息模型</typeparam>
/// <param name="mq"></param>
/// <param name="handler"></param>
/// <returns></returns>
public static Task <IMQ> SubscibeAsync <TMessage>(this IMQ mq, IHandler <TMessage> handler)
{
return(Task.Run(() => mq.Subscibe(handler)));
}
/// <summary>
/// 订阅事件
/// </summary>
/// <typeparam name="TMessage">消息模型</typeparam>
/// <param name="mq"></param>
/// <param name="handler"></param>
/// <returns></returns>
public static IMQ Subscibe <TMessage>(this IMQ mq, IHandler <TMessage> handler)
{
return(mq.Subscibe <TMessage>(m => handler.Handle(m)));
}
/// <summary>
/// 异步订阅
/// </summary>
/// <typeparam name="TMessage"></typeparam>
/// <param name="mq"></param>
/// <param name="action"></param>
/// <returns></returns>
public static Task <IMQ> SubscibeAsync <TMessage>(this IMQ mq, Action <TMessage> action)
{
return(Task.Run(() => mq.Subscibe(action)));
}
private static IMQ SubscibeInternelHandle <TMessage>(this IMQ mq, Handle <TMessage> handle)
{
return(mq.Subscibe <TMessage>(handle.Invoke));;
}
/// <summary>
/// 创建查询构建者
/// </summary>
/// <param name="mq"></param>
/// <returns></returns>
public static IRegistrationBuilder CreateBuilder(this IMQ mq)
{
return(new RegistrationBuilder(mq));
}
/// <summary>
/// 建立映射
/// </summary>
/// <typeparam name="TMessage"></typeparam>
/// <param name="mq"></param>
/// <param name="key"></param>
/// <returns></returns>
public static IMQ Mapping <TMessage>(this IMQ mq, string key) where TMessage : class, new()
{
NameUtils.Add(typeof(TMessage), key);
return(mq);
}
/// <summary>
/// 获得默认的编码
/// </summary>
/// <param name="mq"></param>
/// <returns></returns>
public static Encoding GetEncoding(this IMQ mq)
{
return(Encoding.UTF8);
}
public DocumentPublish(IMQ mq)
{
this._mq = mq;
}
/// <summary>
/// 异步发布
/// </summary>
/// <typeparam name="TMessage"></typeparam>
/// <param name="mq"></param>
/// <param name="message"></param>
/// <returns></returns>
public static async Task <IMQ> PublishAsync <TMessage>(this IMQ mq, TMessage message)
{
return(await Task.Factory.StartNew(() => mq.Publish(message), TaskCreationOptions.AttachedToParent));
}
/// <summary>
/// 消息是否被订阅过,<see cref="bool.TrueString"/>订阅过，<see cref="bool.FalseString"/>尚未订阅
/// </summary>
/// <typeparam name="TMessage"></typeparam>
/// <param name="mq"></param>
/// <returns></returns>
public static bool IsExist <TMessage>(this IMQ mq)
{
var key = NameUtils.Get <TMessage>();
return(_store.ContainsKey(key));
}
public GatewayController(ILogger <GatewayController> logger, IMQ mq)
{
_logger = logger;
_mq = mq;
}
