init
This commit is contained in:
shuhongfan
@@ -0,0 +1,72 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
<parent>
|
||||
<groupId>com.alibaba.csp</groupId>
|
||||
<artifactId>sentinel-quarkus-adapter-parent</artifactId>
|
||||
<version>1.8.3</version>
|
||||
<relativePath>../pom.xml</relativePath>
|
||||
</parent>
|
||||
|
||||
<artifactId>sentinel-annotation-quarkus-adapter-deployment</artifactId>
|
||||
<name>sentinel-annotation-quarkus-adapter-deployment</name>
|
||||
|
||||
<properties>
|
||||
<java.source.version>1.8</java.source.version>
|
||||
<java.target.version>1.8</java.target.version>
|
||||
</properties>
|
||||
|
||||
<dependencies>
|
||||
<dependency>
|
||||
<groupId>io.quarkus</groupId>
|
||||
<artifactId>quarkus-core-deployment</artifactId>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>io.quarkus</groupId>
|
||||
<artifactId>quarkus-arc-deployment</artifactId>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>com.alibaba.csp</groupId>
|
||||
<artifactId>sentinel-annotation-quarkus-adapter</artifactId>
|
||||
<version>${project.version}</version>
|
||||
</dependency>
|
||||
|
||||
<!-- Test dependencies -->
|
||||
<dependency>
|
||||
<groupId>io.quarkus</groupId>
|
||||
<artifactId>quarkus-junit5-internal</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.assertj</groupId>
|
||||
<artifactId>assertj-core</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>io.quarkus</groupId>
|
||||
<artifactId>quarkus-arc-deployment</artifactId>
|
||||
</dependency>
|
||||
|
||||
</dependencies>
|
||||
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<groupId>org.apache.maven.plugins</groupId>
|
||||
<artifactId>maven-compiler-plugin</artifactId>
|
||||
<configuration>
|
||||
<annotationProcessorPaths>
|
||||
<path>
|
||||
<groupId>io.quarkus</groupId>
|
||||
<artifactId>quarkus-extension-processor</artifactId>
|
||||
<version>${quarkus.version}</version>
|
||||
</path>
|
||||
</annotationProcessorPaths>
|
||||
</configuration>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
|
||||
</project>
|
||||
@@ -0,0 +1,46 @@
|
||||
/*
|
||||
* Copyright 1999-2020 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.csp.sentinel.adapter.quarkus.annotation.deployment;
|
||||
|
||||
import com.alibaba.csp.sentinel.annotation.cdi.interceptor.SentinelResourceInterceptor;
|
||||
import io.quarkus.arc.deployment.AdditionalBeanBuildItem;
|
||||
import io.quarkus.deployment.annotations.BuildProducer;
|
||||
import io.quarkus.deployment.annotations.BuildStep;
|
||||
import io.quarkus.deployment.builditem.FeatureBuildItem;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* @author sea
|
||||
*/
|
||||
class SentinelAnnotationQuarkusAdapterProcessor {
|
||||
|
||||
private static final String FEATURE_ANNOTATION = "sentinel-annotation";
|
||||
|
||||
@BuildStep
|
||||
void feature(BuildProducer<FeatureBuildItem> featureProducer) {
|
||||
featureProducer.produce(new FeatureBuildItem(FEATURE_ANNOTATION));
|
||||
}
|
||||
|
||||
@BuildStep
|
||||
List<AdditionalBeanBuildItem> additionalBeans() {
|
||||
return Arrays.asList(
|
||||
new AdditionalBeanBuildItem(SentinelResourceInterceptor.class)
|
||||
);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,84 @@
|
||||
/*
|
||||
* Copyright 1999-2020 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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.alibaba.csp.sentinel.adapter.quarkus.annotation.deployment;
|
||||
import com.alibaba.csp.sentinel.annotation.cdi.interceptor.SentinelResourceBinding;
|
||||
import com.alibaba.csp.sentinel.slots.block.BlockException;
|
||||
|
||||
import javax.enterprise.context.ApplicationScoped;
|
||||
import java.util.concurrent.ThreadLocalRandom;
|
||||
|
||||
/**
|
||||
* @author Eric Zhao
|
||||
* @author sea
|
||||
*/
|
||||
@ApplicationScoped
|
||||
public class FooService {
|
||||
|
||||
@SentinelResourceBinding(value = "apiFoo", blockHandler = "fooBlockHandler",
|
||||
exceptionsToTrace = {IllegalArgumentException.class})
|
||||
public String foo(int i) throws Exception {
|
||||
if (i == 5758) {
|
||||
throw new IllegalAccessException();
|
||||
}
|
||||
if (i == 5763) {
|
||||
throw new IllegalArgumentException();
|
||||
}
|
||||
return "Hello for " + i;
|
||||
}
|
||||
|
||||
@SentinelResourceBinding(value = "apiFooWithFallback", blockHandler = "fooBlockHandler", fallback = "fooFallbackFunc",
|
||||
exceptionsToTrace = {IllegalArgumentException.class})
|
||||
public String fooWithFallback(int i) throws Exception {
|
||||
if (i == 5758) {
|
||||
throw new IllegalAccessException();
|
||||
}
|
||||
if (i == 5763) {
|
||||
throw new IllegalArgumentException();
|
||||
}
|
||||
return "Hello for " + i;
|
||||
}
|
||||
|
||||
@SentinelResourceBinding(value = "apiAnotherFooWithDefaultFallback", defaultFallback = "globalDefaultFallback",
|
||||
fallbackClass = {FooUtil.class})
|
||||
public String anotherFoo(int i) {
|
||||
if (i == 5758) {
|
||||
throw new IllegalArgumentException("oops");
|
||||
}
|
||||
return "Hello for " + i;
|
||||
}
|
||||
|
||||
@SentinelResourceBinding(blockHandler = "globalBlockHandler", blockHandlerClass = FooUtil.class)
|
||||
public int random() {
|
||||
return ThreadLocalRandom.current().nextInt(0, 30000);
|
||||
}
|
||||
|
||||
@SentinelResourceBinding(value = "apiBaz", blockHandler = "bazBlockHandler",
|
||||
exceptionsToIgnore = {IllegalMonitorStateException.class})
|
||||
public String baz(String name) {
|
||||
if (name.equals("fail")) {
|
||||
throw new IllegalMonitorStateException("boom!");
|
||||
}
|
||||
return "cheers, " + name;
|
||||
}
|
||||
|
||||
public String fooBlockHandler(int i, BlockException ex) {
|
||||
return "Oops, " + i;
|
||||
}
|
||||
|
||||
public String fooFallbackFunc(int i) {
|
||||
return "eee...";
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,37 @@
|
||||
/*
|
||||
* Copyright 1999-2020 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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.alibaba.csp.sentinel.adapter.quarkus.annotation.deployment;
|
||||
|
||||
import com.alibaba.csp.sentinel.slots.block.BlockException;
|
||||
|
||||
/**
|
||||
* @author Eric Zhao
|
||||
*/
|
||||
public class FooUtil {
|
||||
|
||||
public static final int BLOCK_FLAG = 88888;
|
||||
public static final String FALLBACK_DEFAULT_RESULT = "fallback";
|
||||
|
||||
public static int globalBlockHandler(BlockException ex) {
|
||||
System.out.println("Oops: " + ex.getClass().getSimpleName());
|
||||
return BLOCK_FLAG;
|
||||
}
|
||||
|
||||
public static String globalDefaultFallback(Throwable t) {
|
||||
System.out.println("Fallback caught: " + t.getClass().getSimpleName());
|
||||
return FALLBACK_DEFAULT_RESULT;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,195 @@
|
||||
/*
|
||||
* Copyright 1999-2020 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* 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 com.alibaba.csp.sentinel.adapter.quarkus.annotation.deployment;
|
||||
|
||||
import com.alibaba.csp.sentinel.node.ClusterNode;
|
||||
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
|
||||
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;
|
||||
import com.alibaba.csp.sentinel.slots.clusterbuilder.ClusterBuilderSlot;
|
||||
import com.alibaba.csp.sentinel.util.MethodUtil;
|
||||
import io.quarkus.arc.ArcUndeclaredThrowableException;
|
||||
import io.quarkus.test.QuarkusUnitTest;
|
||||
import org.jboss.shrinkwrap.api.ShrinkWrap;
|
||||
import org.jboss.shrinkwrap.api.spec.JavaArchive;
|
||||
import org.junit.jupiter.api.AfterEach;
|
||||
import org.junit.jupiter.api.Assertions;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.junit.jupiter.api.extension.RegisterExtension;
|
||||
|
||||
import javax.inject.Inject;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collections;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.assertj.core.api.Assertions.fail;
|
||||
import static org.junit.jupiter.api.Assertions.assertThrows;
|
||||
|
||||
/**
|
||||
* @author sea
|
||||
*/
|
||||
public class SentinelAnnotationQuarkusAdapterTest {
|
||||
|
||||
@RegisterExtension
|
||||
static final QuarkusUnitTest TEST = new QuarkusUnitTest()
|
||||
.setArchiveProducer(() -> ShrinkWrap
|
||||
.create(JavaArchive.class)
|
||||
.addClasses(FooService.class, FooUtil.class)
|
||||
.addPackage("com.alibaba.csp.sentinel.annotation.cdi.interceptor")
|
||||
);
|
||||
|
||||
@Inject
|
||||
FooService fooService;
|
||||
|
||||
@BeforeEach
|
||||
public void setUp() throws Exception {
|
||||
FlowRuleManager.loadRules(new ArrayList<FlowRule>());
|
||||
ClusterBuilderSlot.resetClusterNodes();
|
||||
}
|
||||
|
||||
@AfterEach
|
||||
public void tearDown() throws Exception {
|
||||
FlowRuleManager.loadRules(new ArrayList<FlowRule>());
|
||||
ClusterBuilderSlot.resetClusterNodes();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testForeignBlockHandlerClass() throws Exception {
|
||||
assertThat(fooService.random()).isNotEqualTo(FooUtil.BLOCK_FLAG);
|
||||
String resourceName = MethodUtil.resolveMethodName(FooService.class.getDeclaredMethod("random"));
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertThat(cn).isNotNull();
|
||||
assertThat(cn.passQps()).isPositive();
|
||||
|
||||
FlowRuleManager.loadRules(Collections.singletonList(
|
||||
new FlowRule(resourceName).setCount(0)
|
||||
));
|
||||
assertThat(fooService.random()).isEqualTo(FooUtil.BLOCK_FLAG);
|
||||
assertThat(cn.blockQps()).isPositive();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testBlockHandlerNotFound() {
|
||||
assertThat(fooService.baz("Sentinel")).isEqualTo("cheers, Sentinel");
|
||||
String resourceName = "apiBaz";
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertThat(cn).isNotNull();
|
||||
assertThat(cn.passQps()).isPositive();
|
||||
|
||||
FlowRuleManager.loadRules(Collections.singletonList(
|
||||
new FlowRule(resourceName).setCount(0)
|
||||
));
|
||||
|
||||
assertThrows(ArcUndeclaredThrowableException.class, () -> {
|
||||
fooService.baz("Sentinel");
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAnnotationExceptionsToIgnore() {
|
||||
assertThat(fooService.baz("Sentinel")).isEqualTo("cheers, Sentinel");
|
||||
String resourceName = "apiBaz";
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertThat(cn).isNotNull();
|
||||
assertThat(cn.passQps()).isPositive();
|
||||
assertThrows(IllegalMonitorStateException.class, () -> {
|
||||
fooService.baz("fail");
|
||||
});
|
||||
assertThat(cn.exceptionQps()).isZero();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testFallbackWithNoParams() throws Exception {
|
||||
assertThat(fooService.fooWithFallback(1)).isEqualTo("Hello for 1");
|
||||
String resourceName = "apiFooWithFallback";
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertThat(cn).isNotNull();
|
||||
assertThat(cn.passQps()).isPositive();
|
||||
|
||||
// Fallback should be ignored for this.
|
||||
try {
|
||||
fooService.fooWithFallback(5758);
|
||||
fail("should not reach here");
|
||||
} catch (IllegalAccessException e) {
|
||||
assertThat(cn.exceptionQps()).isZero();
|
||||
}
|
||||
|
||||
// Fallback should take effect.
|
||||
assertThat(fooService.fooWithFallback(5763)).isEqualTo("eee...");
|
||||
assertThat(cn.exceptionQps()).isPositive();
|
||||
assertThat(cn.blockQps()).isZero();
|
||||
|
||||
FlowRuleManager.loadRules(Collections.singletonList(
|
||||
new FlowRule(resourceName).setCount(0)
|
||||
));
|
||||
// Fallback should not take effect for BlockException, as blockHandler is configured.
|
||||
assertThat(fooService.fooWithFallback(2221)).isEqualTo("Oops, 2221");
|
||||
assertThat(cn.blockQps()).isPositive();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDefaultFallbackWithSingleParam() {
|
||||
assertThat(fooService.anotherFoo(1)).isEqualTo("Hello for 1");
|
||||
String resourceName = "apiAnotherFooWithDefaultFallback";
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertThat(cn).isNotNull();
|
||||
assertThat(cn.passQps()).isPositive();
|
||||
|
||||
// Default fallback should take effect.
|
||||
assertThat(fooService.anotherFoo(5758)).isEqualTo(FooUtil.FALLBACK_DEFAULT_RESULT);
|
||||
assertThat(cn.exceptionQps()).isPositive();
|
||||
assertThat(cn.blockQps()).isZero();
|
||||
|
||||
FlowRuleManager.loadRules(Collections.singletonList(
|
||||
new FlowRule(resourceName).setCount(0)
|
||||
));
|
||||
// Default fallback should also take effect for BlockException.
|
||||
assertThat(fooService.anotherFoo(5758)).isEqualTo(FooUtil.FALLBACK_DEFAULT_RESULT);
|
||||
assertThat(cn.blockQps()).isPositive();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testNormalBlockHandlerAndFallback() throws Exception {
|
||||
assertThat(fooService.foo(1)).isEqualTo("Hello for 1");
|
||||
String resourceName = "apiFoo";
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertThat(cn).isNotNull();
|
||||
assertThat(cn.passQps()).isPositive();
|
||||
|
||||
// Test for biz exception.
|
||||
try {
|
||||
fooService.foo(5758);
|
||||
fail("should not reach here");
|
||||
} catch (Exception ex) {
|
||||
// Should not be traced.
|
||||
assertThat(cn.exceptionQps()).isZero();
|
||||
}
|
||||
|
||||
try {
|
||||
fooService.foo(5763);
|
||||
fail("should not reach here");
|
||||
} catch (Exception ex) {
|
||||
assertThat(cn.exceptionQps()).isPositive();
|
||||
}
|
||||
|
||||
// Test for blockHandler
|
||||
FlowRuleManager.loadRules(Collections.singletonList(
|
||||
new FlowRule(resourceName).setCount(0)
|
||||
));
|
||||
assertThat(fooService.foo(1121)).isEqualTo("Oops, 1121");
|
||||
assertThat(cn.blockQps()).isPositive();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user