GH-1793 Fix BinderErrorChannel subscriber count

Also, set component name on BinderErrorChannel object
so it doesn't show up as "unknown.channel.name"
Fix build

Resolves #1793
Resolves #1796
This commit is contained in:
Walliee
2019-08-26 23:14:11 -04:00
committed by Oleg Zhurakousky
parent db0fff9d32
commit ea924ffb95
3 changed files with 90 additions and 2 deletions

View File

@@ -635,7 +635,9 @@ public abstract class AbstractMessageChannelBinder<C extends ConsumerProperties,
errorChannel = (SubscribableChannel) errorChannelObject;
}
else {
errorChannel = new BinderErrorChannel();
BinderErrorChannel binderErrorChannel = new BinderErrorChannel();
binderErrorChannel.setComponentName(errorChannelName);
errorChannel = binderErrorChannel;
((GenericApplicationContext) getApplicationContext()).registerBean(
errorChannelName, SubscribableChannel.class, () -> errorChannel);

View File

@@ -45,7 +45,7 @@ class BinderErrorChannel extends PublishSubscribeChannel
"Only one LastSubscriberMessageHandler is allowed");
}
if (this.finalHandler != null) {
unsubscribe(this.finalHandler);
super.unsubscribe(this.finalHandler);
}
boolean result = super.subscribe(handler);
if (this.finalHandler != null) {

View File

@@ -0,0 +1,86 @@
/*
* Copyright 2017-2019 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.cloud.stream.binder;
import org.junit.Test;
import org.springframework.beans.DirectFieldAccessor;
import org.springframework.integration.dispatcher.AbstractDispatcher;
import org.springframework.messaging.MessageHandler;
import static org.assertj.core.api.Assertions.assertThat;
/**
* @author Anshul Mehra
*/
public class BinderErrorChannelTests {
private static final LastSubscriberMessageHandler FINAL_HANDLER = message -> {
};
private static final MessageHandler FIRST_HANDLER = message -> {
};
private static final MessageHandler SECOND_HANDLER = message -> {
};
@Test
public void testSubscribeUnsubscribe() {
BinderErrorChannel channel = new BinderErrorChannel();
DirectFieldAccessor fieldAccessor = new DirectFieldAccessor(channel);
AbstractDispatcher dispatcher = (AbstractDispatcher) fieldAccessor
.getPropertyValue("dispatcher");
assertThat(dispatcher).isNotNull();
assertThat(channel.subscribers()).isEqualTo(0);
assertThat(dispatcher.getHandlerCount()).isEqualTo(0);
channel.subscribe(FINAL_HANDLER);
assertThat(channel.subscribers()).isEqualTo(1);
assertThat(dispatcher.getHandlerCount()).isEqualTo(1);
channel.subscribe(FIRST_HANDLER);
assertThat(channel.subscribers()).isEqualTo(2);
assertThat(dispatcher.getHandlerCount()).isEqualTo(2);
channel.subscribe(SECOND_HANDLER);
assertThat(channel.subscribers()).isEqualTo(3);
assertThat(dispatcher.getHandlerCount()).isEqualTo(3);
channel.unsubscribe(FIRST_HANDLER);
assertThat(channel.subscribers()).isEqualTo(2);
assertThat(dispatcher.getHandlerCount()).isEqualTo(2);
channel.unsubscribe(FINAL_HANDLER);
assertThat(channel.subscribers()).isEqualTo(1);
assertThat(dispatcher.getHandlerCount()).isEqualTo(1);
channel.unsubscribe(SECOND_HANDLER);
assertThat(channel.subscribers()).isEqualTo(0);
assertThat(dispatcher.getHandlerCount()).isEqualTo(0);
}
}