private void RegisterMessageHandler(Type handlerType, string filteredMessageHandlerCategory) { if (handlerType.IsInterface || handlerType.IsAbstract || handlerType.IsGenericType) { return; } var currentHandlerInterfaceTypes = handlerType.GetInterfaces() .Where(m => m.IsGenericType && !m.IsGenericTypeDefinition && m.GetGenericTypeDefinition() == typeof(IMessageHandler <>)) .ToArray(); var messageHandlerCategory = MessageHandlerTypeAttribute.GetCategory(handlerType); if (string.IsNullOrEmpty(filteredMessageHandlerCategory) || messageHandlerCategory == filteredMessageHandlerCategory) { foreach (var currentHandlerInterfaceType in currentHandlerInterfaceTypes) { var messageType = currentHandlerInterfaceType.GenericTypeArguments[0]; var messageTypeName = MessageTypeAttribute.GetTypeName(messageType); if (!_messageTypes.ContainsKey(messageTypeName)) { _messageTypes.Add(messageTypeName, messageType); } if (!_messageHandlers.ContainsKey(messageTypeName)) { _messageHandlers.Add(messageTypeName, (IMessageHandler)Activator.CreateInstance(handlerType)); } else { var ex = new MessageHandlerDuplicateRegistrationException(messageTypeName, handlerType, this); _logger.Error(ex.Message); throw ex; } } } }
/// <summary> /// 处理消息队列的消息 /// </summary> /// <param name="currentConsumer"></param> /// <param name="props"></param> /// <param name="exchangeName"></param> /// <param name="queueName"></param> /// <param name="routingKey"></param> /// <param name="body"></param> /// <param name="deliveryTag"></param> /// <param name="redelivered"></param> /// <param name="dropIfMessageHandlerNotRegistered">消息处理器未注册时直接丢弃</param> protected void HandleMessage(IBasicConsumer currentConsumer, IBasicProperties props, string exchangeName, string queueName, string routingKey, byte[] body, ulong deliveryTag, bool redelivered, bool dropIfMessageHandlerNotRegistered) { var currentChannel = currentConsumer.Model; var messageTypeName = props.Type; if (string.IsNullOrEmpty(props.Type)) { // 如果未传入类型,则以队列名作为类型名 if (queueName.StartsWith("rpc.")) { messageTypeName = queueName.Substring(4); } else { messageTypeName = queueName; } } var context = new MessageHandlingTransportationContext(exchangeName, queueName, new Dictionary <string, object> { { MessagePropertyConstants.MESSAGE_ID, string.IsNullOrEmpty(props.MessageId) ? Guid.NewGuid().ToString() : props.MessageId }, { MessagePropertyConstants.MESSAGE_TYPE, messageTypeName }, { MessagePropertyConstants.TIMESTAMP, props.Timestamp.UnixTime == 0 ? DateTime.Now : DateTime2UnixTime.FromUnixTime(props.Timestamp.UnixTime) }, { MessagePropertyConstants.CONTENT_TYPE, string.IsNullOrEmpty(props.ContentType) ? Serializer.ContentType : props.ContentType }, { MessagePropertyConstants.PAYLOAD, Serializer.GetString(body) }, { MessagePropertyConstants.ROUTING_KEY, routingKey }, { MessagePropertyConstants.REPLY_TO, props.ReplyTo }, { MessagePropertyConstants.CORRELATION_ID, props.CorrelationId } }); var messageHandlerCategory = string.Empty; try { if (_messageTypes.ContainsKey(messageTypeName)) { object message = null; try { message = Serializer.Deserialize(body, _messageTypes[messageTypeName]); } catch { // 消费端无法反序列化消息,则直接丢弃 _logger.Error($"Message [\"{context.GetMessageType()}\"] \"{context.GetPayload()}\" deserialized fail in queue \"{queueName}\"."); LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicReject(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } if (message == null) { return; } var messageHandler = GetMessageHandler(messageTypeName); if (messageHandler != null) { messageHandlerCategory = MessageHandlerTypeAttribute.GetCategory(messageHandler.GetType()); TriggerOnMessageReceived(new MessageReceivedEventArgs(messageHandlerCategory, this.GetReceiverType(), context)); RetryPolicy.Retry(() => { try { messageHandler.GetType().InvokeMember("Handle", BindingFlags.InvokeMethod | BindingFlags.Public | BindingFlags.Instance, null, messageHandler, new object[] { message, context }); } catch (Exception handleEx) { context.LastException = handleEx is TargetInvocationException ? handleEx.InnerException : handleEx; throw handleEx; } }, cancelOnFailed: (retryCount, retryException) => { if (retryCount >= MaxRetryCount) { return(true); } context.LastException = retryException is TargetInvocationException ? retryException.InnerException : retryException; context.RetryCount = retryCount + 1; return(false); }); TriggerOnMessageHandlingSucceeded(new MessageHandlingSucceededEventArgs(messageHandlerCategory, this.GetReceiverType(), context)); LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicAck(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } else if (dropIfMessageHandlerNotRegistered) { // 消费端无对应的消息处理器,则忽略,直接响应成功 LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicAck(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } else { context.LastException = new MessageHandlerNotRegisteredException(messageTypeName); if (TriggerOnMessageHandlingFailed(new MessageHandlingFailedEventArgs(messageHandlerCategory, this.GetReceiverType(), context))) { LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicAck(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } else { LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicReject(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } } } else if (dropIfMessageHandlerNotRegistered) { // 消费端无对应的消息处理器,则忽略,直接响应成功 LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicAck(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } else { context.LastException = new MessageHandlerNotRegisteredException(messageTypeName); if (TriggerOnMessageHandlingFailed(new MessageHandlingFailedEventArgs(messageHandlerCategory, this.GetReceiverType(), context))) { LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicAck(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } else { LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicReject(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } } } catch (Exception ex) { // 消费端处理其他异常 context.LastException = ex is TargetInvocationException ? ex.InnerException : ex; if (TriggerOnMessageHandlingFailed(new MessageHandlingFailedEventArgs(messageHandlerCategory, this.GetReceiverType(), context))) { LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicAck(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } else { LockerExecuter.Execute(currentChannel, () => { RetryPolicy.Retry(() => currentChannel.BasicReject(deliveryTag, false), retryCondition: retryEx => IOHelper.IsIOError(retryEx) || RabbitMQExceptionHelper.IsChannelError(retryEx), retryTimeInterval: 1000); }); } } }