/**
* Copyright 2016 Netflix, Inc.
*
* 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
*
* http://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 com.netflix.hystrix.metric.consumer;
import com.netflix.hystrix.HystrixCollapserKey;
import com.netflix.hystrix.metric.CachedValuesHistogram;
import com.netflix.hystrix.metric.CommandStreamTest;
import com.netflix.hystrix.strategy.concurrency.HystrixRequestContext;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import rx.Subscriber;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
public class RollingCollapserBatchSizeDistributionStreamTest extends CommandStreamTest {
RollingCollapserBatchSizeDistributionStream stream;
HystrixRequestContext context;
@Before
public void setUp() {
context = HystrixRequestContext.initializeContext();
}
@After
public void tearDown() {
stream.unsubscribe();
context.shutdown();
RollingCollapserBatchSizeDistributionStream.reset();
}
@Test
public void testEmptyStreamProducesEmptyDistributions() {
HystrixCollapserKey key = HystrixCollapserKey.Factory.asKey("Collapser-Batch-Size-A");
stream = RollingCollapserBatchSizeDistributionStream.getInstance(key, 10, 100);
stream.startCachingStreamValuesIfUnstarted();
final CountDownLatch latch = new CountDownLatch(1);
stream.observe().skip(10).take(10).subscribe(new Subscriber<CachedValuesHistogram>() {
@Override
public void onCompleted() {
latch.countDown();
}
@Override
public void onError(Throwable e) {
fail(e.getMessage());
}
@Override
public void onNext(CachedValuesHistogram distribution) {
System.out.println("OnNext @ " + System.currentTimeMillis());
assertEquals(0, distribution.getTotalCount());
}
});
//no writes
try {
assertTrue(latch.await(10000, TimeUnit.MILLISECONDS));
} catch (InterruptedException ex) {
fail("Interrupted ex");
}
assertEquals(0, stream.getLatest().getTotalCount());
}
@Test
public void testBatches() {
HystrixCollapserKey key = HystrixCollapserKey.Factory.asKey("Collapser-Batch-Size-B");
stream = RollingCollapserBatchSizeDistributionStream.getInstance(key, 10, 100);
stream.startCachingStreamValuesIfUnstarted();
final CountDownLatch latch = new CountDownLatch(1);
stream.observe().take(10).subscribe(new Subscriber<CachedValuesHistogram>() {
@Override
public void onCompleted() {
latch.countDown();
}
@Override
public void onError(Throwable e) {
fail(e.getMessage());
}
@Override
public void onNext(CachedValuesHistogram distribution) {
System.out.println("OnNext @ " + System.currentTimeMillis());
}
});
Collapser.from(key, 1).observe();
Collapser.from(key, 2).observe();
Collapser.from(key, 3).observe();
try {
Thread.sleep(250);
} catch (InterruptedException ex) {
fail("Interrupted ex");
}
Collapser.from(key, 4).observe();
try {
Thread.sleep(250);
} catch (InterruptedException ex) {
fail("Interrupted ex");
}
Collapser.from(key, 5).observe();
Collapser.from(key, 6).observe();
Collapser.from(key, 7).observe();
Collapser.from(key, 8).observe();
Collapser.from(key, 9).observe();
try {
Thread.sleep(250);
} catch (InterruptedException ex) {
fail("Interrupted ex");
}
Collapser.from(key, 10).observe();
Collapser.from(key, 11).observe();
Collapser.from(key, 12).observe();
try {
assertTrue(latch.await(10000, TimeUnit.MILLISECONDS));
} catch (InterruptedException ex) {
fail("Interrupted ex");
}
//should have 4 batches: 3, 1, 5, 3
assertEquals(4, stream.getLatest().getTotalCount());
assertEquals(3, stream.getLatestMean());
assertEquals(1, stream.getLatestPercentile(0));
assertEquals(5, stream.getLatestPercentile(100));
}
//by doing a take(30), all metrics should fall out of window and we should observe an empty histogram
@Test
public void testBatchesAgeOut() {
HystrixCollapserKey key = HystrixCollapserKey.Factory.asKey("Collapser-Batch-Size-B");
stream = RollingCollapserBatchSizeDistributionStream.getInstance(key, 10, 100);
stream.startCachingStreamValuesIfUnstarted();
final CountDownLatch latch = new CountDownLatch(1);
stream.observe().take(30).subscribe(new Subscriber<CachedValuesHistogram>() {
@Override
public void onCompleted() {
latch.countDown();
}
@Override
public void onError(Throwable e) {
fail(e.getMessage());
}
@Override
public void onNext(CachedValuesHistogram distribution) {
System.out.println("OnNext @ " + System.currentTimeMillis());
}
});
Collapser.from(key, 1).observe();
Collapser.from(key, 2).observe();
Collapser.from(key, 3).observe();
try {
Thread.sleep(200);
} catch (InterruptedException ex) {
fail("Interrupted ex");
}
Collapser.from(key, 4).observe();
try {
Thread.sleep(200);
} catch (InterruptedException ex) {
fail("Interrupted ex");
}
Collapser.from(key, 5).observe();
Collapser.from(key, 6).observe();
Collapser.from(key, 7).observe();
Collapser.from(key, 8).observe();
Collapser.from(key, 9).observe();
try {
Thread.sleep(200);
} catch (InterruptedException ex) {
fail("Interrupted ex");
}
Collapser.from(key, 10).observe();
Collapser.from(key, 11).observe();
Collapser.from(key, 12).observe();
try {
assertTrue(latch.await(10000, TimeUnit.MILLISECONDS));
} catch (InterruptedException ex) {
fail("Interrupted ex");
}
assertEquals(0, stream.getLatest().getTotalCount());
assertEquals(0, stream.getLatestMean());
}
}