init
This commit is contained in:
shuhongfan
@@ -0,0 +1,77 @@
|
||||
<?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-jax-rs-quarkus-adapter-deployment</artifactId>
|
||||
<name>sentinel-jax-rs-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>io.quarkus</groupId>
|
||||
<artifactId>quarkus-resteasy-server-common-deployment</artifactId>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>com.alibaba.csp</groupId>
|
||||
<artifactId>sentinel-jax-rs-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>io.quarkus</groupId>
|
||||
<artifactId>quarkus-resteasy-deployment</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>io.rest-assured</groupId>
|
||||
<artifactId>rest-assured</artifactId>
|
||||
<scope>test</scope>
|
||||
</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,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
|
||||
*
|
||||
* 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.jaxrs.deployment;
|
||||
|
||||
import io.quarkus.deployment.annotations.BuildProducer;
|
||||
import io.quarkus.deployment.annotations.BuildStep;
|
||||
import io.quarkus.deployment.builditem.FeatureBuildItem;
|
||||
import org.jboss.logging.Logger;
|
||||
|
||||
/**
|
||||
* @author sea
|
||||
*/
|
||||
class SentinelJaxRsQuarkusAdapterProcessor {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(SentinelJaxRsQuarkusAdapterProcessor.class);
|
||||
|
||||
private static final String FEATURE_JAX_RS = "sentinel-jax-rs";
|
||||
|
||||
@BuildStep
|
||||
void feature(BuildProducer<FeatureBuildItem> featureProducer) {
|
||||
featureProducer.produce(new FeatureBuildItem(FEATURE_JAX_RS));
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,242 @@
|
||||
/*
|
||||
* 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.jaxrs.deployment;
|
||||
|
||||
import com.alibaba.csp.sentinel.Constants;
|
||||
import com.alibaba.csp.sentinel.adapter.jaxrs.config.SentinelJaxRsConfig;
|
||||
import com.alibaba.csp.sentinel.adapter.jaxrs.fallback.SentinelJaxRsFallback;
|
||||
import com.alibaba.csp.sentinel.adapter.jaxrs.request.RequestOriginParser;
|
||||
import com.alibaba.csp.sentinel.node.ClusterNode;
|
||||
import com.alibaba.csp.sentinel.node.EntranceNode;
|
||||
import com.alibaba.csp.sentinel.node.Node;
|
||||
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
|
||||
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.StringUtil;
|
||||
import io.quarkus.test.QuarkusUnitTest;
|
||||
import io.restassured.response.Response;
|
||||
import org.jboss.shrinkwrap.api.ShrinkWrap;
|
||||
import org.jboss.shrinkwrap.api.spec.JavaArchive;
|
||||
import org.junit.jupiter.api.AfterEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.junit.jupiter.api.extension.RegisterExtension;
|
||||
|
||||
import javax.ws.rs.container.ContainerRequestContext;
|
||||
import javax.ws.rs.core.MediaType;
|
||||
import java.util.Collections;
|
||||
import java.util.concurrent.Callable;
|
||||
import java.util.concurrent.Future;
|
||||
import java.util.concurrent.FutureTask;
|
||||
|
||||
import static io.restassured.RestAssured.given;
|
||||
import static org.hamcrest.CoreMatchers.equalTo;
|
||||
import static org.junit.jupiter.api.Assertions.*;
|
||||
|
||||
/**
|
||||
* @author sea
|
||||
*/
|
||||
public class SentinelJaxRsQuarkusAdapterTest {
|
||||
|
||||
private static final String HELLO_STR = "Hello!";
|
||||
|
||||
@RegisterExtension
|
||||
static final QuarkusUnitTest TEST = new QuarkusUnitTest()
|
||||
.setArchiveProducer(() -> ShrinkWrap
|
||||
.create(JavaArchive.class)
|
||||
.addClasses(TestResource.class));
|
||||
|
||||
@AfterEach
|
||||
public void cleanUp() {
|
||||
FlowRuleManager.loadRules(null);
|
||||
ClusterBuilderSlot.resetClusterNodes();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetHello() {
|
||||
String url = "/test/hello";
|
||||
String resourceName = "GET:" + url;
|
||||
Response response = given().get(url);
|
||||
response.then().statusCode(200).body(equalTo(HELLO_STR));
|
||||
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertNotNull(cn);
|
||||
assertEquals(1, cn.passQps(), 0.01);
|
||||
|
||||
String context = "";
|
||||
for (Node n : Constants.ROOT.getChildList()) {
|
||||
if (n instanceof EntranceNode) {
|
||||
String id = ((EntranceNode) n).getId().getName();
|
||||
if (url.equals(id)) {
|
||||
context = ((EntranceNode) n).getId().getName();
|
||||
}
|
||||
}
|
||||
}
|
||||
assertEquals("", context);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAsyncGetHello() {
|
||||
String url = "/test/async-hello";
|
||||
String resourceName = "GET:" + url;
|
||||
Response response = given().get(url);
|
||||
response.then().statusCode(200).body(equalTo(HELLO_STR));
|
||||
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertNotNull(cn);
|
||||
assertEquals(1, cn.passQps(), 0.01);
|
||||
|
||||
String context = "";
|
||||
for (Node n : Constants.ROOT.getChildList()) {
|
||||
if (n instanceof EntranceNode) {
|
||||
String id = ((EntranceNode) n).getId().getName();
|
||||
if (url.equals(id)) {
|
||||
context = ((EntranceNode) n).getId().getName();
|
||||
}
|
||||
}
|
||||
}
|
||||
assertEquals("", context);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUrlPathParam() {
|
||||
String url = "/test/hello/{name}";
|
||||
String resourceName = "GET:" + url;
|
||||
|
||||
String url1 = "/test/hello/abc";
|
||||
Response response1 = given().get(url1);
|
||||
response1.then().statusCode(200).body(equalTo("Hello abc !"));
|
||||
|
||||
String url2 = "/test/hello/def";
|
||||
Response response2 = given().get(url2);
|
||||
response2.then().statusCode(200).body(equalTo("Hello def !"));
|
||||
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertNotNull(cn);
|
||||
assertEquals(2, cn.passQps(), 0.01);
|
||||
|
||||
assertNull(ClusterBuilderSlot.getClusterNode("GET:" + url1));
|
||||
assertNull(ClusterBuilderSlot.getClusterNode("GET:" + url2));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDefaultFallback() {
|
||||
String url = "/test/hello";
|
||||
String resourceName = "GET:" + url;
|
||||
configureRulesFor(resourceName, 0);
|
||||
Response response = given().get(url);
|
||||
response.then().statusCode(javax.ws.rs.core.Response.Status.TOO_MANY_REQUESTS.getStatusCode())
|
||||
.body(equalTo("Blocked by Sentinel (flow limiting)"));
|
||||
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertNotNull(cn);
|
||||
assertEquals(0, cn.passQps(), 0.01);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCustomFallback() {
|
||||
String url = "/test/hello";
|
||||
String resourceName = "GET:" + url;
|
||||
SentinelJaxRsConfig.setJaxRsFallback(new SentinelJaxRsFallback() {
|
||||
@Override
|
||||
public javax.ws.rs.core.Response fallbackResponse(String route, Throwable cause) {
|
||||
return javax.ws.rs.core.Response.status(javax.ws.rs.core.Response.Status.OK)
|
||||
.entity("Blocked by Sentinel (flow limiting)")
|
||||
.type(MediaType.APPLICATION_JSON_TYPE)
|
||||
.build();
|
||||
}
|
||||
|
||||
@Override
|
||||
public Future<javax.ws.rs.core.Response> fallbackFutureResponse(final String route, final Throwable cause) {
|
||||
return new FutureTask<>(new Callable<javax.ws.rs.core.Response>() {
|
||||
@Override
|
||||
public javax.ws.rs.core.Response call() throws Exception {
|
||||
return fallbackResponse(route, cause);
|
||||
}
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
|
||||
configureRulesFor(resourceName, 0);
|
||||
Response response = given().get(url);
|
||||
response.then().statusCode(javax.ws.rs.core.Response.Status.OK.getStatusCode())
|
||||
.body(equalTo("Blocked by Sentinel (flow limiting)"));
|
||||
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertNotNull(cn);
|
||||
assertEquals(0, cn.passQps(), 0.01);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCustomRequestOriginParser() {
|
||||
String url = "/test/hello";
|
||||
String resourceName = "GET:" + url;
|
||||
|
||||
String limitOrigin = "appB";
|
||||
final String headerName = "X-APP";
|
||||
configureRulesFor(resourceName, 0, limitOrigin);
|
||||
|
||||
SentinelJaxRsConfig.setRequestOriginParser(new RequestOriginParser() {
|
||||
@Override
|
||||
public String parseOrigin(ContainerRequestContext request) {
|
||||
String origin = request.getHeaderString(headerName);
|
||||
return origin != null ? origin : "";
|
||||
}
|
||||
});
|
||||
|
||||
Response response = given()
|
||||
.header(headerName, "appA").get(url);
|
||||
response.then().statusCode(200).body(equalTo(HELLO_STR));
|
||||
|
||||
Response blockedResp = given()
|
||||
.header(headerName, "appB")
|
||||
.get(url);
|
||||
blockedResp.then().statusCode(javax.ws.rs.core.Response.Status.TOO_MANY_REQUESTS.getStatusCode())
|
||||
.body(equalTo("Blocked by Sentinel (flow limiting)"));
|
||||
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertNotNull(cn);
|
||||
assertEquals(1, cn.passQps(), 0.01);
|
||||
assertEquals(1, cn.blockQps(), 0.01);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testExceptionMapper() {
|
||||
String url = "/test/ex";
|
||||
String resourceName = "GET:" + url;
|
||||
Response response = given().get(url);
|
||||
response.then().statusCode(javax.ws.rs.core.Response.Status.INTERNAL_SERVER_ERROR.getStatusCode()).body(equalTo("test exception mapper"));
|
||||
|
||||
ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
|
||||
assertNotNull(cn);
|
||||
}
|
||||
|
||||
private void configureRulesFor(String resource, int count) {
|
||||
configureRulesFor(resource, count, "default");
|
||||
}
|
||||
|
||||
private void configureRulesFor(String resource, int count, String limitApp) {
|
||||
FlowRule rule = new FlowRule()
|
||||
.setCount(count)
|
||||
.setGrade(RuleConstant.FLOW_GRADE_QPS);
|
||||
rule.setResource(resource);
|
||||
if (StringUtil.isNotBlank(limitApp)) {
|
||||
rule.setLimitApp(limitApp);
|
||||
}
|
||||
FlowRuleManager.loadRules(Collections.singletonList(rule));
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,90 @@
|
||||
/*
|
||||
* 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.jaxrs.deployment;
|
||||
|
||||
|
||||
import javax.ws.rs.*;
|
||||
import javax.ws.rs.container.AsyncResponse;
|
||||
import javax.ws.rs.container.Suspended;
|
||||
import javax.ws.rs.core.MediaType;
|
||||
import javax.ws.rs.core.Response;
|
||||
import java.util.concurrent.ExecutorService;
|
||||
import java.util.concurrent.Executors;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
/**
|
||||
* @author sea
|
||||
*/
|
||||
@Path("/test")
|
||||
public class TestResource {
|
||||
|
||||
ExecutorService executor = Executors.newFixedThreadPool(5);
|
||||
|
||||
@Path("/hello")
|
||||
@GET
|
||||
@Produces({ MediaType.APPLICATION_JSON })
|
||||
public String sayHello() {
|
||||
return "Hello!";
|
||||
}
|
||||
|
||||
@Path("/async-hello")
|
||||
@GET
|
||||
@Produces({ MediaType.APPLICATION_JSON })
|
||||
public void asyncSayHello(@Suspended final AsyncResponse asyncResponse) {
|
||||
executor.submit(new Runnable() {
|
||||
@Override
|
||||
public void run() {
|
||||
try {
|
||||
TimeUnit.MILLISECONDS.sleep(100);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
asyncResponse.resume("Hello!");
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
@Path("/hello/{name}")
|
||||
@GET
|
||||
@Produces({ MediaType.APPLICATION_JSON })
|
||||
public String sayHelloWithName(@PathParam(value = "name") String name) {
|
||||
return "Hello " + name + " !";
|
||||
}
|
||||
|
||||
@Path("/ex")
|
||||
@GET
|
||||
@Produces({ MediaType.APPLICATION_JSON })
|
||||
public String exception() {
|
||||
throw new RuntimeException("test exception mapper");
|
||||
}
|
||||
|
||||
@Path("/400")
|
||||
@GET
|
||||
@Produces({ MediaType.APPLICATION_JSON })
|
||||
public String badRequest() {
|
||||
throw new WebApplicationException(Response.status(Response.Status.BAD_REQUEST)
|
||||
.entity("test return 400")
|
||||
.build());
|
||||
}
|
||||
|
||||
@Path("/delay/{seconds}")
|
||||
@GET
|
||||
@Produces({ MediaType.APPLICATION_JSON })
|
||||
public String delay(@PathParam(value = "seconds") long seconds) throws InterruptedException {
|
||||
TimeUnit.SECONDS.sleep(seconds);
|
||||
return "finish";
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user