From 19a068d7083b1849c088a4c8b9a246f1537649ad Mon Sep 17 00:00:00 2001 From: wankai123 Date: Sun, 20 Jun 2021 07:57:10 +0800 Subject: [PATCH] Feature: support endpoint name grouping by OpenAPI definitions. (#7130) --- CHANGES.md | 1 + .../apm/util/StringFormatGroup.java | 7 +- .../setup/backend/configuration-vocabulary.md | 1 + .../setup/backend/endpoint-grouping-rules.md | 287 +++++++++++++++++- .../src/main/resources/application.yml | 2 + .../serviceA/productAPI-v1.yaml | 215 +++++++++++++ oap-server/server-core/pom.xml | 6 +- .../oap/server/core/CoreModuleConfig.java | 4 + .../oap/server/core/CoreModuleProvider.java | 6 + .../config/group/EndpointNameGrouping.java | 34 ++- .../openapi/EndpointGroupingRule4Openapi.java | 117 +++++++ .../EndpointGroupingRuleReader4Openapi.java | 143 +++++++++ .../EndpointGroupingBenchmark4Openapi.java | 172 +++++++++++ ...ndpointGroupingRuleReader4OpenapiTest.java | 97 ++++++ .../serviceA/customerAPI-v1.yaml | 172 +++++++++++ .../serviceA/productAPI-v1.yaml | 180 +++++++++++ .../serviceA-1/customerAPI-a-1-v1.yaml | 171 +++++++++++ .../serviceB/productAPI-v2.yaml | 210 +++++++++++++ .../server/library/util/ResourceUtils.java | 45 ++- 19 files changed, 1859 insertions(+), 11 deletions(-) create mode 100644 oap-server/server-bootstrap/src/main/resources/openapi-definitions/serviceA/productAPI-v1.yaml create mode 100644 oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRule4Openapi.java create mode 100644 oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRuleReader4Openapi.java create mode 100644 oap-server/server-core/src/test/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingBenchmark4Openapi.java create mode 100644 oap-server/server-core/src/test/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRuleReader4OpenapiTest.java create mode 100644 oap-server/server-core/src/test/resources/openapi-definitions/serviceA/customerAPI-v1.yaml create mode 100644 oap-server/server-core/src/test/resources/openapi-definitions/serviceA/productAPI-v1.yaml create mode 100644 oap-server/server-core/src/test/resources/openapi-definitions/serviceA/serviceA-1/customerAPI-a-1-v1.yaml create mode 100644 oap-server/server-core/src/test/resources/openapi-definitions/serviceB/productAPI-v2.yaml diff --git a/CHANGES.md b/CHANGES.md index 6a81392e18..4a734f7ed2 100644 --- a/CHANGES.md +++ b/CHANGES.md @@ -38,6 +38,7 @@ Release Notes. * Upgrade embed tomcat caused by CVE-2020-13935. * Upgrade commons-lang3 to avoid potential NPE in some JDK versions. * OAL supports generating metrics from events. +* Support endpoint name grouping by OpenAPI definitions. #### UI diff --git a/apm-commons/apm-util/src/main/java/org/apache/skywalking/apm/util/StringFormatGroup.java b/apm-commons/apm-util/src/main/java/org/apache/skywalking/apm/util/StringFormatGroup.java index ab30cdb451..9f22dafd2a 100644 --- a/apm-commons/apm-util/src/main/java/org/apache/skywalking/apm/util/StringFormatGroup.java +++ b/apm-commons/apm-util/src/main/java/org/apache/skywalking/apm/util/StringFormatGroup.java @@ -19,6 +19,7 @@ package org.apache.skywalking.apm.util; import java.util.ArrayList; +import java.util.Comparator; import java.util.List; import java.util.regex.Pattern; import lombok.Getter; @@ -68,6 +69,10 @@ public class StringFormatGroup { return new FormatResult(false, string, string); } + public void sortRules(Comparator comparator) { + rules.sort(comparator); + } + @Getter @RequiredArgsConstructor public static class FormatResult { @@ -78,7 +83,7 @@ public class StringFormatGroup { @Getter @ToString - private static class PatternRule { + public static class PatternRule { private final String name; private final Pattern pattern; diff --git a/docs/en/setup/backend/configuration-vocabulary.md b/docs/en/setup/backend/configuration-vocabulary.md index 5becb12ee5..33f406a2c9 100644 --- a/docs/en/setup/backend/configuration-vocabulary.md +++ b/docs/en/setup/backend/configuration-vocabulary.md @@ -42,6 +42,7 @@ core|default|role|Option values, `Mixed/Receiver/Aggregator`. **Receiver** mode | - | - | maxSizeOfAnalyzeProfileSnapshot|The max number of snapshots analyzed by OAP| - | 12000 | | - | - | syncThreads|The number of threads used to synchronously refresh the metrics data to the storage.| SW_CORE_SYNC_THREADS | 2 | | - | - | maxSyncOperationNum|The maximum number of processes supported for each synchronous storage operation. When the number of the flush data is greater than this value, it will be assigned to multiple cores for execution.| SW_CORE_MAX_SYNC_OPERATION_NUM | 50000 | +| - | - | enableEndpointNameGroupingByOpenapi |Turn it on then automatically grouping endpoint by the given OpenAPI definitions.| SW_CORE_ENABLE_ENDPOINT_NAME_GROUPING_BY_OPAENAPI | true | |cluster|standalone| - | standalone is not suitable for one node running, no available configuration.| - | - | | - | zookeeper|nameSpace|The namespace, represented by root path, isolates the configurations in the zookeeper.|SW_NAMESPACE| `/`, root path| | - | - | hostPort|hosts and ports of Zookeeper Cluster|SW_CLUSTER_ZK_HOST_PORT| localhost:2181| diff --git a/docs/en/setup/backend/endpoint-grouping-rules.md b/docs/en/setup/backend/endpoint-grouping-rules.md index dd623beacb..e4eda52b51 100644 --- a/docs/en/setup/backend/endpoint-grouping-rules.md +++ b/docs/en/setup/backend/endpoint-grouping-rules.md @@ -1,15 +1,294 @@ # Group Parameterized Endpoints -In most cases, the endpoint should be detected automatically through the language agents, service mesh observability solution, +In most cases, the endpoint should be detected automatically through the language agents, service mesh observability solution, or configuration of meter system. -There are some special cases, especially when people use REST style URI, the application codes put the parameter in the endpoint name, +There are some special cases, especially when people use REST style URI, the application codes put the parameter in the endpoint name, such as putting order id in the URI, like `/prod/ORDER123` and `/prod/ORDER123`. But logically, people expect they could have an endpoint name like `prod/{order-id}`. This is the feature of parameterized endpoint grouping designed for. +If the incoming endpoint name hit the rules, SkyWalking will grouping the endpoint by rules. + +SkyWalking provides 2 ways to support endpoint grouping: +1. Endpoint name grouping by OpenAPI definitions. +2. Endpoint name grouping by custom configuration. + +The 2 grouping features can work together in sequence. +## Endpoint name grouping by OpenAPI definitions +The OpenAPI definitions are the documents based on The [OpenAPI Specification (OAS)](https://www.openapis.org/) which used to define a standard, language-agnostic interface for HTTP APIs. +SkyWalking now support `OAS v2.0+`, could parse the documents `(yaml)` and build the grouping rules from them automatically. + + +### How to use +1. Add some `Specification Extensions` for SkyWalking config in the OpenAPI definition documents, otherwise, all configs are default:
+ `${METHOD}` is a reserved placeholder which represents the HTTP method eg. `POST/GET...`
+ `${PATH}` is a reserved placeholder which represents the path eg. `/products/{id}`. + + | Extension Name | Required | Description | Default Value | + |-----|-----|-----|-----| + | x-sw-service-name | false | The service name which these endpoints belong to | The directory name which the OpenAPI definition documents belong to | + | x-sw-endpoint-name-match-rule | false | The rule used to match the endpoint.| `${METHOD}:${PATH}` | + | x-sw-endpoint-name-format | false | The endpoint name after grouping.| `${METHOD}:${PATH}` | + + These extensions are under `OpenAPI Object`. For example, the document below has a full custom config: + +``` yaml +openapi: 3.0.0 +x-sw-service-name: serviceB +x-sw-endpoint-name-match-rule: "${METHOD}:${PATH}" +x-sw-endpoint-name-format: "${METHOD}:${PATH}" + +info: + description: OpenAPI definition for SkyWalking test. + version: v2 + title: Product API + ... +``` + + We highly recommend using the default config, the custom config (`x-sw-endpoint-name-match-rule/x-sw-endpoint-name-format`) would be considered as part of the match rules (regex pattern). + We provide some cases in `org.apache.skywalking.oap.server.core.config.group.openapi.EndpointGroupingRuleReader4OpenapiTest`, you could validate your custom config as well. + +2. All OpenAPI definition documents should be located in the `openapi-definitions` directory, with at most two levels directories. Recommend using the service name as the subDirectory name then you are not required to set `x-sw-service-name`. For example: + ``` +├── openapi-definitions +│   ├── serviceA +│   │   ├── customerAPI-v1.yaml +│   │   └── productAPI-v1.yaml +│   └── serviceB +│   └── productAPI-v2.yaml +``` +3. The feature is enabled by default, you can disable it by setting the `Core Module` configuration `${SW_CORE_ENABLE_ENDPOINT_NAME_GROUPING_BY_OPAENAPI:false}` + +### Rules match priority +We recommend designing the API path as clear as possible. If the API path is fuzzy and an endpoint name might match multiple paths, SkyWalking would follow the match priority to select one as below orders: +1. The exact path matched first. + Eg. `/products or /products/inventory` +2. The path which has the less variables. + Eg. `/products/{var1}/{var2} and /products/{var1}/abc`, endpoint name `/products/123/abc` will match the second one. +3. If the paths have the same number of variables, match the longest path, and the vars are considered to be `1`. + Eg. `/products/abc/{var1} and products/{var12345}/ef`, endpoint name `/products/abc/ef` will match the first one, because `length("abc") = 3` is larger than `length("ef") = 2`. +### Examples +If we have an OpenAPI definition doc `productAPI-v2.yaml` in directory `serviceB` like this: +```yaml + +openapi: 3.0.0 + +info: + description: OpenAPI definition for SkyWalking test. + version: v2 + title: Product API + +tags: + - name: product + description: product + - name: relatedProducts + description: Related Products + +paths: + /products: + get: + tags: + - product + summary: Get all products list + description: Get all products list. + operationId: getProducts + responses: + "200": + description: Success + content: + application/json: + schema: + type: array + items: + $ref: "#/components/schemas/Product" + /products/{region}/{country}: + get: + tags: + - product + summary: Get products regional + description: Get products regional with the given id. + operationId: getProductRegional + parameters: + - name: region + in: path + description: Products region + required: true + schema: + type: string + - name: country + in: path + description: Products country + required: true + schema: + type: string + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/Product" + "400": + description: Invalid parameters supplied + /products/{id}: + get: + tags: + - product + summary: Get product details + description: Get product details with the given id. + operationId: getProduct + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/ProductDetails" + "400": + description: Invalid product id + post: + tags: + - product + summary: Update product details + description: Update product details with the given id. + operationId: updateProduct + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + - name: name + in: query + description: Product name + required: true + schema: + type: string + responses: + "200": + description: successful operation + delete: + tags: + - product + summary: Delete product details + description: Delete product details with the given id. + operationId: deleteProduct + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + /products/{id}/relatedProducts: + get: + tags: + - relatedProducts + summary: Get related products + description: Get related products with the given product id. + operationId: getRelatedProducts + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/RelatedProducts" + "400": + description: Invalid product id + +components: + schemas: + Product: + type: object + description: Product id and name + properties: + id: + type: integer + format: int64 + description: Product id + name: + type: string + description: Product name + required: + - id + - name + ProductDetails: + type: object + description: Product details + properties: + id: + type: integer + format: int64 + description: Product id + name: + type: string + description: Product name + description: + type: string + description: Product description + required: + - id + - name + RelatedProducts: + type: object + description: Related Products + properties: + id: + type: integer + format: int32 + description: Product id + relatedProducts: + type: array + description: List of related products + items: + $ref: "#/components/schemas/Product" + + +``` + +Here are some cases: + + | Incoming Endpiont | Incoming Service | x-sw-service-name | x-sw-endpoint-name-match-rule | x-sw-endpoint-name-format | Matched | Grouping Result | + |-----|-----|-----|-----|-----|-----|-----| + | `GET:/products` | serviceB | default | default | default | true | `GET:/products` | + | `GET:/products/123` | serviceB | default | default | default | true | `GET:/products{id}` | + | `GET:/products/asia/cn` | serviceB | default | default | default | true | `GET:/products/{region}/{country}` | + | `GET:/products/123/abc/efg` | serviceB | default | default | default | false | `GET:/products/123/abc/efg` | + | `:/products/123` | serviceB | default | default | default | false | `:/products/123`| + | `GET:/products/123` | serviceC | default | default | default | false | `GET:/products/123` | + | `GET:/products/123` | serviceC | serviceC | default | default | true | `GET:/products/123` | + | `:/products/123` | serviceB | default | `<${METHOD}>:${PATH}` | `<${METHOD}>:${PATH}` | true | `:/products/{id}` | + | `GET:/products/123` | serviceB | default | default | `${PATH}:<${METHOD}>` | true | `/products/{id}:` | + | `/products/123:` | serviceB | default | `${PATH}:<${METHOD}>` | default | true | `GET:/products/{id}` | + + +## Endpoint name grouping by custom configuration Current, user could set up grouping rules through the static YAML file, named `endpoint-name-grouping.yml`, or use [Dynamic Configuration](dynamic-config.md) to initial and update the endpoint grouping rule. -## Configuration Format +### Configuration Format No matter in static local file or dynamic configuration value, they are sharing the same YAML format. ```yaml @@ -20,4 +299,4 @@ grouping: # Logic name when the regex expression matched. - endpoint-name: /prod/{id} regex: \/prod\/.+ -``` \ No newline at end of file +``` diff --git a/oap-server/server-bootstrap/src/main/resources/application.yml b/oap-server/server-bootstrap/src/main/resources/application.yml index 1bb16ee25a..4c39a68b8e 100755 --- a/oap-server/server-bootstrap/src/main/resources/application.yml +++ b/oap-server/server-bootstrap/src/main/resources/application.yml @@ -108,6 +108,8 @@ core: syncThreads: ${SW_CORE_SYNC_THREADS:2} # The maximum number of processes supported for each synchronous storage operation. When the number of the flush data is greater than this value, it will be assigned to multiple cores for execution. maxSyncOperationNum: ${SW_CORE_MAX_SYNC_OPERATION_NUM:50000} + # Turn it on then automatically grouping endpoint by the given OpenAPI definitions. + enableEndpointNameGroupingByOpenapi: ${SW_CORE_ENABLE_ENDPOINT_NAME_GROUPING_BY_OPAENAPI:true} storage: selector: ${SW_STORAGE:h2} elasticsearch: diff --git a/oap-server/server-bootstrap/src/main/resources/openapi-definitions/serviceA/productAPI-v1.yaml b/oap-server/server-bootstrap/src/main/resources/openapi-definitions/serviceA/productAPI-v1.yaml new file mode 100644 index 0000000000..89e85ec019 --- /dev/null +++ b/oap-server/server-bootstrap/src/main/resources/openapi-definitions/serviceA/productAPI-v1.yaml @@ -0,0 +1,215 @@ +# Licensed to the Apache Software Foundation (ASF) under one or more +# contributor license agreements. See the NOTICE file distributed with +# this work for additional information regarding copyright ownership. +# The ASF licenses this file to You 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. +# +# +# Add some `Specification Extensions` for SkyWalking config in the OpenAPI definition documents, otherwise, all configs are default: +# `${METHOD}` is a reserved placeholder which represents the HTTP method eg. `POST/GET...` +# `${PATH}` is a reserved placeholder which represents the path eg. `/products/{id}`. +# +# | Extension Name | Required | Description | Default Value | +# |-----|-----|-----|-----| +# | x-sw-service-name | false | The service name to which these endpoints belong | The directory name which the definition documents belong to| +# | x-sw-endpoint-name-match-rule | false | The rule used to match the endpoint.| `${METHOD}:${PATH}` | +# | x-sw-endpoint-name-format | false | The endpoint name after grouping.| `${METHOD}:${PATH}` | +# +# These extensions are under `OpenAPI Object`. For example, the document below has a full custom config: +# +#``` yaml +#openapi: 3.0.0 +#x-sw-service-name: serviceB +#x-sw-endpoint-name-match-rule: "<${METHOD}>:${PATH}" +#x-sw-endpoint-name-format: "<${METHOD}>:${PATH}" +# +#info: +# description: OpenAPI definition for SkyWalking test. +# version: v2 +# title: Product API +# ... +#``` +# +# We highly recommend using the default config, the custom config (`x-sw-endpoint-name-match-rule/x-sw-endpoint-name-format`) would be considered as part of the match rules (regex pattern). +# We provide some cases in `org.apache.skywalking.oap.server.core.config.group.openapi.EndpointGroupingRuleReader4OpenapiTest`, you could validate your custom config as well. + + +################################################### +# This is only a example using the default config.# +################################################### + + +#openapi: 3.0.0 +# +#info: +# description: OpenAPI definition for SkyWalking test. +# version: v1 +# title: Product API +# +#tags: +# - name: product +# description: product +# - name: relatedProducts +# description: Related Products +# +#paths: +# /products: +# get: +# tags: +# - product +# summary: Get all products list +# description: Get all products list. +# operationId: getProducts +# responses: +# "200": +# description: Success +# content: +# application/json: +# schema: +# type: array +# items: +# $ref: "#/components/schemas/Product" +# /products/{id}: +# get: +# tags: +# - product +# summary: Get product details +# description: Get product details with the given id. +# operationId: getProduct +# parameters: +# - name: id +# in: path +# description: Product id +# required: true +# schema: +# type: integer +# format: int64 +# responses: +# "200": +# description: successful operation +# content: +# application/json: +# schema: +# $ref: "#/components/schemas/ProductDetails" +# "400": +# description: Invalid product id +# post: +# tags: +# - product +# summary: Update product details +# description: Update product details with the given id. +# operationId: updateProduct +# parameters: +# - name: id +# in: path +# description: Product id +# required: true +# schema: +# type: integer +# format: int64 +# - name: name +# in: query +# description: Product name +# required: true +# schema: +# type: string +# responses: +# "200": +# description: successful operation +# delete: +# tags: +# - product +# summary: Delete product details +# description: Delete product details with the given id. +# operationId: deleteProduct +# parameters: +# - name: id +# in: path +# description: Product id +# required: true +# schema: +# type: integer +# format: int64 +# responses: +# "200": +# description: successful operation +# /products/{id}/relatedProducts: +# get: +# tags: +# - relatedProducts +# summary: Get related products +# description: Get related products with the given product id. +# operationId: getRelatedProducts +# parameters: +# - name: id +# in: path +# description: Product id +# required: true +# schema: +# type: integer +# format: int64 +# responses: +# "200": +# description: successful operation +# content: +# application/json: +# schema: +# $ref: "#/components/schemas/RelatedProducts" +# "400": +# description: Invalid product id +# +#components: +# schemas: +# Product: +# type: object +# description: Product id and name +# properties: +# id: +# type: integer +# format: int64 +# description: Product id +# name: +# type: string +# description: Product name +# required: +# - id +# - name +# ProductDetails: +# type: object +# description: Product details +# properties: +# id: +# type: integer +# format: int64 +# description: Product id +# name: +# type: string +# description: Product name +# description: +# type: string +# description: Product description +# required: +# - id +# - name +# RelatedProducts: +# type: object +# description: Related Products +# properties: +# id: +# type: integer +# format: int32 +# description: Product id +# relatedProducts: +# type: array +# description: List of related products +# items: +# $ref: "#/components/schemas/Product" \ No newline at end of file diff --git a/oap-server/server-core/pom.xml b/oap-server/server-core/pom.xml index 098d2f8e9e..1135336b56 100644 --- a/oap-server/server-core/pom.xml +++ b/oap-server/server-core/pom.xml @@ -82,7 +82,6 @@ io.vavr vavr - org.apache.skywalking server-testing @@ -94,6 +93,11 @@ grpc-testing test + + org.openjdk.jmh + jmh-generator-annprocess + test + diff --git a/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/CoreModuleConfig.java b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/CoreModuleConfig.java index 22f75a9df8..bed117dfe2 100644 --- a/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/CoreModuleConfig.java +++ b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/CoreModuleConfig.java @@ -158,6 +158,10 @@ public class CoreModuleConfig extends ModuleConfig { @Setter private int maxSyncOperationNum = 50000; + @Getter + @Setter + private boolean enableEndpointNameGroupingByOpenapi = false; + public CoreModuleConfig() { this.downsampling = new ArrayList<>(); } diff --git a/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/CoreModuleProvider.java b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/CoreModuleProvider.java index 38929892f9..9e99bd1f09 100755 --- a/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/CoreModuleProvider.java +++ b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/CoreModuleProvider.java @@ -46,6 +46,7 @@ import org.apache.skywalking.oap.server.core.config.ConfigService; import org.apache.skywalking.oap.server.core.config.DownSamplingConfigService; import org.apache.skywalking.oap.server.core.config.IComponentLibraryCatalogService; import org.apache.skywalking.oap.server.core.config.NamingControl; +import org.apache.skywalking.oap.server.core.config.group.openapi.EndpointGroupingRuleReader4Openapi; import org.apache.skywalking.oap.server.core.config.group.EndpointNameGrouping; import org.apache.skywalking.oap.server.core.config.group.EndpointNameGroupingRuleWatcher; import org.apache.skywalking.oap.server.core.management.ui.template.UITemplateInitializer; @@ -159,6 +160,11 @@ public class CoreModuleProvider extends ModuleProvider { try { endpointNameGroupingRuleWatcher = new EndpointNameGroupingRuleWatcher( this, endpointNameGrouping); + + if (moduleConfig.isEnableEndpointNameGroupingByOpenapi()) { + endpointNameGrouping.setEndpointGroupingRule4Openapi( + new EndpointGroupingRuleReader4Openapi("openapi-definitions").read()); + } } catch (FileNotFoundException e) { throw new ModuleStartException(e.getMessage(), e); } diff --git a/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/EndpointNameGrouping.java b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/EndpointNameGrouping.java index da952d67d2..8c460a1f05 100644 --- a/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/EndpointNameGrouping.java +++ b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/EndpointNameGrouping.java @@ -21,20 +21,48 @@ package org.apache.skywalking.oap.server.core.config.group; import lombok.Setter; import lombok.extern.slf4j.Slf4j; import org.apache.skywalking.apm.util.StringFormatGroup; +import org.apache.skywalking.oap.server.core.config.group.openapi.EndpointGroupingRule4Openapi; @Slf4j public class EndpointNameGrouping { @Setter private volatile EndpointGroupingRule endpointGroupingRule; + @Setter + private volatile EndpointGroupingRule4Openapi endpointGroupingRule4Openapi; public String format(String serviceName, String endpointName) { - if (endpointGroupingRule == null) { - return endpointName; + String formattedName = endpointName; + if (endpointGroupingRule4Openapi != null) { + formattedName = formatByOpenapi(serviceName, formattedName); + } + + if (endpointGroupingRule != null) { + formattedName = formatByCustom(serviceName, formattedName); } + + return formattedName; + } + + private String formatByCustom(String serviceName, String endpointName) { final StringFormatGroup.FormatResult formatResult = endpointGroupingRule.format(serviceName, endpointName); if (log.isDebugEnabled() || log.isTraceEnabled()) { if (formatResult.isMatch()) { - log.debug("Endpoint {} of Service {} has been renamed in group {}", + log.debug("Endpoint {} of Service {} has been renamed in group {} by endpointGroupingRule", + endpointName, serviceName, formatResult.getName() + ); + } else { + log.trace("Endpoint {} of Service {} keeps unchanged.", endpointName, serviceName); + } + } + return formatResult.getName(); + } + + private String formatByOpenapi(String serviceName, String endpointName) { + final StringFormatGroup.FormatResult formatResult = endpointGroupingRule4Openapi.format( + serviceName, endpointName); + if (log.isDebugEnabled() || log.isTraceEnabled()) { + if (formatResult.isMatch()) { + log.debug("Endpoint {} of Service {} has been renamed in group {} by endpointGroupingRule4Openapi", endpointName, serviceName, formatResult.getName() ); } else { diff --git a/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRule4Openapi.java b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRule4Openapi.java new file mode 100644 index 0000000000..ea3b0ebb6f --- /dev/null +++ b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRule4Openapi.java @@ -0,0 +1,117 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.skywalking.oap.server.core.config.group.openapi; + +import java.util.Comparator; +import java.util.HashMap; +import java.util.Map; +import lombok.Getter; +import org.apache.skywalking.apm.util.StringFormatGroup; + +public class EndpointGroupingRule4Openapi { + private final Map> directLookup = new HashMap<>(); + @Getter + private final Map> groupedRules = new HashMap<>(); + + void addDirectLookup(String serviceName, String endpointName, String endpointGroupName) { + Map endpointNameLookup = directLookup.computeIfAbsent(serviceName, name -> new HashMap<>()); + endpointNameLookup.put(endpointName, endpointGroupName); + } + + void addGroupedRule(String serviceName, String endpointGroupName, String ruleRegex) { + String rulesGroupkey = getGroupedRulesKey(ruleRegex); + Map rules = groupedRules.computeIfAbsent(serviceName, name -> new HashMap<>()); + StringFormatGroup formatGroup = rules.computeIfAbsent(rulesGroupkey, name -> new StringFormatGroup()); + formatGroup.addRule(endpointGroupName, ruleRegex); + } + + public StringFormatGroup.FormatResult format(String service, String endpointName) { + Map endpointNameLookup = directLookup.get(service); + if (endpointNameLookup != null && endpointNameLookup.get(endpointName) != null) { + return new StringFormatGroup.FormatResult(true, endpointNameLookup.get(endpointName), endpointName); + } + + Map rules = groupedRules.get(service); + if (rules != null) { + final StringFormatGroup stringFormatGroup = rules.get(getGroupedRulesKey(endpointName)); + if (stringFormatGroup != null) { + return stringFormatGroup.format(endpointName); + } + } + + return new StringFormatGroup.FormatResult(false, endpointName, endpointName); + } + + void sortRulesAll() { + groupedRules.entrySet().forEach(rules -> { + sortRulesByService(rules.getKey()); + }); + } + + void sortRulesByService(String serviceName) { + Map rules = groupedRules.get(serviceName); + if (rules != null) { + rules.entrySet().forEach(stringFormatGroup -> { + stringFormatGroup.getValue() + .sortRules(new EndpointGroupingRule4Openapi.EndpointGroupingRulesComparator()); + }); + } + } + + String getGroupedRulesKey(String string) { + String[] ss = string.split("/"); + if (ss.length == 1) { //eg. POST:/ + return ss[0] + "/"; + } + if (ss.length > 1) { + return ss[0] + "/" + ss[1]; + } + return "/"; + } + + static class EndpointGroupingRulesComparator implements Comparator { + private static final String VAR_PATTERN = "\\(\\[\\^\\/\\]\\+\\)"; + + @Override + public int compare(final StringFormatGroup.PatternRule rule1, final StringFormatGroup.PatternRule rule2) { + + String pattern1 = rule1.getPattern().pattern(); + String pattern2 = rule2.getPattern().pattern(); + + if (getPatternVarsCount(pattern1) < getPatternVarsCount(pattern2)) { + return -1; + } else if (getPatternVarsCount(pattern1) > getPatternVarsCount(pattern2)) { + return 1; + } + + int length1 = getPatternLength(pattern1); + int length2 = getPatternLength(pattern2); + return length2 - length1; + } + + private int getPatternVarsCount(String pattern) { + return ",".concat(pattern).concat(",").split(VAR_PATTERN).length - 1; + } + + private int getPatternLength(String pattern) { + return pattern.replaceAll(VAR_PATTERN, "#").length(); + } + + } +} diff --git a/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRuleReader4Openapi.java b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRuleReader4Openapi.java new file mode 100644 index 0000000000..ab945e7aea --- /dev/null +++ b/oap-server/server-core/src/main/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRuleReader4Openapi.java @@ -0,0 +1,143 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.skywalking.oap.server.core.config.group.openapi; + +import java.io.File; +import java.io.FileNotFoundException; +import java.io.FileReader; +import java.io.Reader; +import java.util.HashMap; +import java.util.LinkedHashMap; +import java.util.List; +import java.util.Map; +import org.apache.skywalking.apm.util.StringUtil; +import org.apache.skywalking.oap.server.library.util.ResourceUtils; +import org.yaml.snakeyaml.Yaml; +import org.yaml.snakeyaml.constructor.SafeConstructor; + +public class EndpointGroupingRuleReader4Openapi { + + private final String openapiDefPath; + private final static String DEFAULT_ENDPOINT_NAME_FORMAT = "${METHOD}:${PATH}"; + private final static String DEFAULT_ENDPOINT_NAME_MATCH_RULE = "${METHOD}:${PATH}"; + private final Map requestMethodsMap = new HashMap() { + { + put("get", "GET"); + put("post", "POST"); + put("put", "PUT"); + put("delete", "DELETE"); + put("trace", "TRACE"); + put("options", "OPTIONS"); + put("head", "HEAD"); + put("patch", "PATCH"); + } + }; + + public EndpointGroupingRuleReader4Openapi(final String openapiDefPath) { + + this.openapiDefPath = openapiDefPath; + } + + public EndpointGroupingRule4Openapi read() throws FileNotFoundException { + EndpointGroupingRule4Openapi endpointGroupingRule = new EndpointGroupingRule4Openapi(); + + List fileList = ResourceUtils.getDirectoryFilesRecursive(openapiDefPath, 1); + for (File file : fileList) { + if (!file.getName().endsWith(".yaml")) { + continue; + } + Reader reader = new FileReader(file); + Yaml yaml = new Yaml(new SafeConstructor()); + Map openapiData = yaml.load(reader); + if (openapiData != null) { + String serviceName = getServiceName(openapiData, file); + LinkedHashMap> paths = + (LinkedHashMap>) openapiData.get("paths"); + + if (paths != null) { + paths.forEach((pathString, pathItem) -> { + pathItem.keySet().forEach(key -> { + String requestMethod = requestMethodsMap.get(key); + if (!StringUtil.isEmpty(requestMethod)) { + String endpointGroupName = formatEndPointName(pathString, requestMethod, openapiData); + String groupRegex = getGroupRegex(pathString, requestMethod, openapiData); + if (isTemplatePath(pathString)) { + endpointGroupingRule.addGroupedRule(serviceName, endpointGroupName, groupRegex); + } else { + endpointGroupingRule.addDirectLookup(serviceName, groupRegex, endpointGroupName); + } + } + }); + }); + } + } + } + endpointGroupingRule.sortRulesAll(); + return endpointGroupingRule; + } + + private String getServiceName(Map openapiData, File file) { + + String serviceName = (String) openapiData.get("x-sw-service-name"); + if (StringUtil.isEmpty(serviceName)) { + File directory = new File(file.getParent()); + if (openapiDefPath.equals(directory.getName())) { + throw new IllegalArgumentException( + "OpenAPI definition file: " + file.getAbsolutePath() + " found in root directory, but doesn't include x-sw-service-name extensive definition in the file."); + } + serviceName = directory.getName(); + } + + return serviceName; + } + + private boolean isTemplatePath(String pathString) { + return pathString.matches("(.*)\\{(.+?)}(.*)"); + } + + private String getGroupRegex(String pathString, String requstMathod, Map openapiData) { + String endPointNameMatchRuleTemplate = (String) openapiData.get("x-sw-endpoint-name-match-rule"); + String endPointNameMatchRule = replaceTemplateVars(DEFAULT_ENDPOINT_NAME_MATCH_RULE, pathString, requstMathod); + + if (!StringUtil.isEmpty(endPointNameMatchRuleTemplate)) { + endPointNameMatchRule = replaceTemplateVars(endPointNameMatchRuleTemplate, pathString, requstMathod); + } + + if (isTemplatePath(endPointNameMatchRule)) { + return endPointNameMatchRule.replaceAll("\\{(.+?)}", "([^/]+)"); + } + return endPointNameMatchRule; + } + + private String formatEndPointName(String pathString, String requstMethod, Map openapiData) { + String endPointNameFormat = (String) openapiData.get("x-sw-endpoint-name-format"); + + if (!StringUtil.isEmpty(endPointNameFormat)) { + return replaceTemplateVars(endPointNameFormat, pathString, requstMethod); + } + + return replaceTemplateVars(DEFAULT_ENDPOINT_NAME_FORMAT, pathString, requstMethod); + } + + private String replaceTemplateVars(String template, String pathString, String requstMathod) { + return template.replaceAll("\\$\\{METHOD}", requstMathod) + .replaceAll("\\$\\{PATH}", pathString); + } + +} diff --git a/oap-server/server-core/src/test/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingBenchmark4Openapi.java b/oap-server/server-core/src/test/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingBenchmark4Openapi.java new file mode 100644 index 0000000000..d2ffdf3dea --- /dev/null +++ b/oap-server/server-core/src/test/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingBenchmark4Openapi.java @@ -0,0 +1,172 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.skywalking.oap.server.core.config.group.openapi; + +import java.io.FileNotFoundException; +import lombok.SneakyThrows; +import org.openjdk.jmh.annotations.Benchmark; +import org.openjdk.jmh.annotations.BenchmarkMode; +import org.openjdk.jmh.annotations.Mode; +import org.openjdk.jmh.annotations.Scope; +import org.openjdk.jmh.annotations.State; +import org.openjdk.jmh.annotations.Threads; +import org.openjdk.jmh.profile.GCProfiler; +import org.openjdk.jmh.runner.Runner; +import org.openjdk.jmh.runner.RunnerException; +import org.openjdk.jmh.runner.options.Options; +import org.openjdk.jmh.runner.options.OptionsBuilder; + +@BenchmarkMode({Mode.Throughput}) +@Threads(4) +public class EndpointGroupingBenchmark4Openapi { + + @State(Scope.Benchmark) + public static class FormatClassPaths20 { + private EndpointGroupingRule4Openapi rule; + + @SneakyThrows + public FormatClassPaths20() { + rule = new EndpointGroupingRule4Openapi(); + for (int i = 0; i <= 3; i++) { + rule.addGroupedRule("serviceA", "GET:/products1/{id}/" + i, "GET:/products1/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "POST:/products1/{id}/" + i, "POST:/products1/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "GET:/products2/{id}/" + i, "GET:/products2/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "POST:/products3/{id}/" + i, "POST:/products3/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "GET:/products3/{id}/" + i, "GET:/products3/([^/]+)/" + i); + } + } + + public void format(String serviceName, String endpointName) { + rule.format(serviceName, endpointName); + } + } + + @State(Scope.Benchmark) + public static class FormatClassPaths50 { + private EndpointGroupingRule4Openapi rule; + + @SneakyThrows + public FormatClassPaths50() { + rule = new EndpointGroupingRule4Openapi(); + for (int i = 0; i <= 9; i++) { + rule.addGroupedRule("serviceA", "GET:/products1/{id}/" + i, "GET:/products1/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "POST:/products1/{id}/" + i, "POST:/products1/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "GET:/products2/{id}/" + i, "GET:/products2/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "POST:/products3/{id}/" + i, "POST:/products3/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "GET:/products3/{id}/" + i, "GET:/products3/([^/]+)/" + i); + } + } + + public void format(String serviceName, String endpointName) { + rule.format(serviceName, endpointName); + } + } + + @State(Scope.Benchmark) + public static class FormatClassPaths200 { + private EndpointGroupingRule4Openapi rule; + + @SneakyThrows + public FormatClassPaths200() { + rule = new EndpointGroupingRule4Openapi(); + for (int i = 0; i <= 39; i++) { + rule.addGroupedRule("serviceA", "GET:/products1/{id}/" + i, "GET:/products1/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "POST:/products1/{id}/" + i, "POST:/products1/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "GET:/products2/{id}/" + i, "GET:/products2/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "POST:/products3/{id}/" + i, "POST:/products3/([^/]+)/" + i); + rule.addGroupedRule("serviceA", "GET:/products3/{id}/" + i, "GET:/products3/([^/]+)/" + i); + } + } + + public void format(String serviceName, String endpointName) { + rule.format(serviceName, endpointName); + } + + } + + @Benchmark + public void formatEndpointNameMatchedPaths20(FormatClassPaths20 formatClass) { + formatClass.format("serviceA", "GET:/products1/123"); + } + + @Benchmark + public void formatEndpointNameMatchedPaths50(FormatClassPaths50 formatClass) { + formatClass.format("serviceA", "GET:/products1/123"); + } + + @Benchmark + public void formatEndpointNameMatchedPaths200(FormatClassPaths200 formatClass) { + formatClass.format("serviceA", "GET:/products1/123"); + } + + public static void main(String[] args) throws RunnerException, FileNotFoundException { + + Options opt = new OptionsBuilder() + .include(EndpointGroupingBenchmark4Openapi.class.getName()) + .addProfiler(GCProfiler.class) + .jvmArgsAppend("-Xmx512m", "-Xms512m") + .forks(1) + .build(); + + new Runner(opt).run(); + } +} + +/* +* The test is assumed each endpoint need to run all match within it's rules group. +* +# JMH version: 1.21 +# VM version: JDK 1.8.0_271, Java HotSpot(TM) 64-Bit Server VM, 25.271-b09 +# VM invoker: /Library/Java/JavaVirtualMachines/jdk1.8.0_271.jdk/Contents/Home/jre/bin/java +# VM options: -javaagent:/Applications/IntelliJ IDEA CE.app/Contents/lib/idea_rt.jar=51431:/Applications/IntelliJ IDEA CE.app/Contents/bin -Dfile.encoding=UTF-8 -Xmx512m -Xms512m +# Warmup: 5 iterations, 10 s each +# Measurement: 5 iterations, 10 s each +# Timeout: 10 min per iteration +# Threads: 4 threads, will synchronize iterations +# Benchmark mode: Throughput, ops/time + +Benchmark Mode Cnt Score Error Units +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths20 thrpt 5 4180207.544 ± 833644.395 ops/s +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths20:·gc.alloc.rate thrpt 5 4524.954 ± 903.291 MB/sec +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths20:·gc.alloc.rate.norm thrpt 5 1192.000 ± 0.001 B/op +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths20:·gc.churn.PS_Eden_Space thrpt 5 4550.511 ± 916.117 MB/sec +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths20:·gc.churn.PS_Eden_Space.norm thrpt 5 1198.713 ± 10.572 B/op +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths20:·gc.churn.PS_Survivor_Space thrpt 5 0.493 ± 0.118 MB/sec +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths20:·gc.churn.PS_Survivor_Space.norm thrpt 5 0.130 ± 0.039 B/op +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths20:·gc.count thrpt 5 1410.000 counts +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths20:·gc.time thrpt 5 783.000 ms +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths200 thrpt 5 600313.461 ± 58702.201 ops/s +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths200:·gc.alloc.rate thrpt 5 4260.484 ± 415.215 MB/sec +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths200:·gc.alloc.rate.norm thrpt 5 7816.000 ± 0.001 B/op +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths200:·gc.churn.PS_Eden_Space thrpt 5 4285.685 ± 407.822 MB/sec +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths200:·gc.churn.PS_Eden_Space.norm thrpt 5 7862.339 ± 46.737 B/op +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths200:·gc.churn.PS_Survivor_Space thrpt 5 0.444 ± 0.061 MB/sec +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths200:·gc.churn.PS_Survivor_Space.norm thrpt 5 0.815 ± 0.062 B/op +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths200:·gc.count thrpt 5 1328.000 counts +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths200:·gc.time thrpt 5 729.000 ms +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths50 thrpt 5 2001647.224 ± 139386.146 ops/s +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths50:·gc.alloc.rate thrpt 5 4173.062 ± 291.166 MB/sec +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths50:·gc.alloc.rate.norm thrpt 5 2296.000 ± 0.001 B/op +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths50:·gc.churn.PS_Eden_Space thrpt 5 4198.202 ± 271.551 MB/sec +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths50:·gc.churn.PS_Eden_Space.norm thrpt 5 2309.878 ± 14.994 B/op +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths50:·gc.churn.PS_Survivor_Space thrpt 5 0.393 ± 0.171 MB/sec +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths50:·gc.churn.PS_Survivor_Space.norm thrpt 5 0.216 ± 0.086 B/op +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths50:·gc.count thrpt 5 1301.000 counts +EndpointGroupingBenchmark4Openapi.formatEndpointNameMatchedPaths50:·gc.time thrpt 5 715.000 ms + */ \ No newline at end of file diff --git a/oap-server/server-core/src/test/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRuleReader4OpenapiTest.java b/oap-server/server-core/src/test/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRuleReader4OpenapiTest.java new file mode 100644 index 0000000000..b66df94c8c --- /dev/null +++ b/oap-server/server-core/src/test/java/org/apache/skywalking/oap/server/core/config/group/openapi/EndpointGroupingRuleReader4OpenapiTest.java @@ -0,0 +1,97 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.skywalking.oap.server.core.config.group.openapi; + +import java.io.IOException; +import org.apache.skywalking.oap.server.core.config.group.EndpointNameGrouping; +import org.junit.Assert; +import org.junit.Test; + +public class EndpointGroupingRuleReader4OpenapiTest { + + @Test + public void testReadingRule() throws IOException { + + EndpointGroupingRuleReader4Openapi reader = new EndpointGroupingRuleReader4Openapi("openapi-definitions"); + EndpointGroupingRule4Openapi rule = reader.read(); + EndpointNameGrouping nameGrouping = new EndpointNameGrouping(); + nameGrouping.setEndpointGroupingRule4Openapi(rule); + + //default x-sw-service-name x-sw-endpoint-name-match-rule and x-sw-endpoint-name-format + // test direct lookup + String endpointName = nameGrouping.format("serviceA", "GET:/products"); + Assert.assertEquals("GET:/products", endpointName); + + endpointName = nameGrouping.format("serviceA", "GET:/products/123"); + Assert.assertEquals("GET:/products/{id}", endpointName); + + endpointName = nameGrouping.format("serviceA", "GET:/products/123/abc/ef"); + Assert.assertEquals("GET:/products/123/abc/ef", endpointName); + + endpointName = nameGrouping.format("serviceA", "GET:/products/123/relatedProducts"); + Assert.assertEquals("GET:/products/{id}/relatedProducts", endpointName); + + endpointName = nameGrouping.format("serviceA", "GET:/products/1/relatedProducts"); + Assert.assertEquals("GET:/products/{id}/relatedProducts", endpointName); + + //test custom x-sw-service-name same x-sw-endpoint-name-match-rule and x-sw-endpoint-name-format + endpointName = nameGrouping.format("serviceA-1", "POST:/customer"); + Assert.assertEquals("POST:/customer", endpointName); + + endpointName = nameGrouping.format("serviceA-1", ":/customers/1"); + Assert.assertEquals(":/customers/{id}", endpointName); + + //test different x-sw-endpoint-name-match-rule and x-sw-endpoint-name-format + endpointName = nameGrouping.format("serviceB", "GET:/products"); + Assert.assertEquals("/products:", endpointName); + + endpointName = nameGrouping.format("serviceB", "GET:/products/asia/cn"); + Assert.assertEquals("/products/{region}/{country}:", endpointName); + + //test match priority, not match /products/{region}/{country}: + endpointName = nameGrouping.format("serviceB", "GET:/products/12/relatedProducts"); + Assert.assertEquals("/products/{id}/relatedProducts:", endpointName); + + //test not match, return the origin + endpointName = nameGrouping.format("serviceA", "GET:/products/"); + Assert.assertNotEquals("GET:/products", endpointName); + + endpointName = nameGrouping.format("serviceA", "GET:/products/123/"); + Assert.assertEquals("GET:/products/123/", endpointName); + + endpointName = nameGrouping.format("serviceC", "GET:/products/123"); + Assert.assertEquals("GET:/products/123", endpointName); + + endpointName = nameGrouping.format("serviceA", "GET:/products/1/ratings/123"); + Assert.assertEquals("GET:/products/1/ratings/123", endpointName); + + endpointName = nameGrouping.format("serviceA-1", ":/customers/1/123"); + Assert.assertEquals(":/customers/1/123", endpointName); + + endpointName = nameGrouping.format("serviceB", "/products/:"); + Assert.assertEquals("/products/:", endpointName); + + endpointName = nameGrouping.format("serviceB", "{GET}:/products"); + Assert.assertEquals("{GET}:/products", endpointName); + + endpointName = nameGrouping.format("serviceB", "/products/1/2/3:"); + Assert.assertEquals("/products/1/2/3:", endpointName); + + } +} diff --git a/oap-server/server-core/src/test/resources/openapi-definitions/serviceA/customerAPI-v1.yaml b/oap-server/server-core/src/test/resources/openapi-definitions/serviceA/customerAPI-v1.yaml new file mode 100644 index 0000000000..20dc3f0db1 --- /dev/null +++ b/oap-server/server-core/src/test/resources/openapi-definitions/serviceA/customerAPI-v1.yaml @@ -0,0 +1,172 @@ +# Licensed to the Apache Software Foundation (ASF) under one or more +# contributor license agreements. See the NOTICE file distributed with +# this work for additional information regarding copyright ownership. +# The ASF licenses this file to You 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. + +openapi: 3.0.0 +x-sw-service-name: serviceA-1 +x-sw-endpoint-name-match-rule: "<${METHOD}>:${PATH}" +x-sw-endpoint-name-format: "<${METHOD}>:${PATH}" +info: + description: OpenAPI definition for SkyWalking test. + version: v1 + title: Customer API + +tags: + - name: customer + description: customer + +paths: + /customers: + get: + tags: + - customer + summary: Get all customers list + description: Get all customers list. + operationId: getCustomers + responses: + "200": + description: Success + content: + application/json: + schema: + type: array + items: + $ref: "#/components/schemas/Customer" + /customers/{id}: + get: + tags: + - customer + summary: Get customer details + description: Get customer details with the given id. + operationId: getCustomer + parameters: + - name: id + in: path + description: Customer id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/CustomerDetails" + "400": + description: Invalid customer id + post: + tags: + - customer + summary: Update customer details + description: Update customer details with the given id. + operationId: updateCustomer + parameters: + - name: id + in: path + description: Customer id + required: true + schema: + type: integer + format: int64 + - name: name + in: query + description: Customer name + required: true + schema: + type: string + responses: + "200": + description: successful operation + delete: + tags: + - customer + summary: Delete customer details + description: Delete customer details with the given id. + operationId: deleteCustomer + parameters: + - name: id + in: path + description: Customer id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + + /customer/{region}/{country}: + get: + tags: + - customer + summary: Get customers regional + description: Get customers regional with the given id. + operationId: getCustomersRegional + parameters: + - name: region + in: path + description: Customers region + required: true + schema: + type: string + - name: country + in: path + description: Customers country + required: true + schema: + type: string + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/Customer" + "400": + description: Invalid parameters supplied +components: + schemas: + Customer: + type: object + description: Customer id and name + properties: + id: + type: integer + format: int64 + description: Customer id + name: + type: string + description: Customer name + required: + - id + - name + CustomerDetails: + type: object + description: Customer details + properties: + id: + type: integer + format: int64 + description: Customer id + name: + type: string + description: Customer name + description: + type: string + description: Customer description + required: + - id + - name diff --git a/oap-server/server-core/src/test/resources/openapi-definitions/serviceA/productAPI-v1.yaml b/oap-server/server-core/src/test/resources/openapi-definitions/serviceA/productAPI-v1.yaml new file mode 100644 index 0000000000..2a6d5722ad --- /dev/null +++ b/oap-server/server-core/src/test/resources/openapi-definitions/serviceA/productAPI-v1.yaml @@ -0,0 +1,180 @@ +# Licensed to the Apache Software Foundation (ASF) under one or more +# contributor license agreements. See the NOTICE file distributed with +# this work for additional information regarding copyright ownership. +# The ASF licenses this file to You 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. + +openapi: 3.0.0 + +info: + description: OpenAPI definition for SkyWalking test. + version: v1 + title: Product API + +tags: + - name: product + description: product + - name: relatedProducts + description: Related Products + +paths: + /products: + get: + tags: + - product + summary: Get all products list + description: Get all products list. + operationId: getProducts + responses: + "200": + description: Success + content: + application/json: + schema: + type: array + items: + $ref: "#/components/schemas/Product" + /products/{id}: + get: + tags: + - product + summary: Get product details + description: Get product details with the given id. + operationId: getProduct + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/ProductDetails" + "400": + description: Invalid product id + post: + tags: + - product + summary: Update product details + description: Update product details with the given id. + operationId: updateProduct + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + - name: name + in: query + description: Product name + required: true + schema: + type: string + responses: + "200": + description: successful operation + delete: + tags: + - product + summary: Delete product details + description: Delete product details with the given id. + operationId: deleteProduct + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + /products/{id}/relatedProducts: + get: + tags: + - relatedProducts + summary: Get related products + description: Get related products with the given product id. + operationId: getRelatedProducts + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/RelatedProducts" + "400": + description: Invalid product id + +components: + schemas: + Product: + type: object + description: Product id and name + properties: + id: + type: integer + format: int64 + description: Product id + name: + type: string + description: Product name + required: + - id + - name + ProductDetails: + type: object + description: Product details + properties: + id: + type: integer + format: int64 + description: Product id + name: + type: string + description: Product name + description: + type: string + description: Product description + required: + - id + - name + RelatedProducts: + type: object + description: Related Products + properties: + id: + type: integer + format: int32 + description: Product id + relatedProducts: + type: array + description: List of related products + items: + $ref: "#/components/schemas/Product" diff --git a/oap-server/server-core/src/test/resources/openapi-definitions/serviceA/serviceA-1/customerAPI-a-1-v1.yaml b/oap-server/server-core/src/test/resources/openapi-definitions/serviceA/serviceA-1/customerAPI-a-1-v1.yaml new file mode 100644 index 0000000000..d4a60a532d --- /dev/null +++ b/oap-server/server-core/src/test/resources/openapi-definitions/serviceA/serviceA-1/customerAPI-a-1-v1.yaml @@ -0,0 +1,171 @@ +# Licensed to the Apache Software Foundation (ASF) under one or more +# contributor license agreements. See the NOTICE file distributed with +# this work for additional information regarding copyright ownership. +# The ASF licenses this file to You 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. + +openapi: 3.0.0 +x-sw-service-name: serviceA + +info: + description: OpenAPI definition for SkyWalking test. + version: v1 + title: Customer API + +tags: + - name: customer + description: customer + +paths: + /customers: + get: + tags: + - customer + summary: Get all customers list + description: Get all customers list. + operationId: getCustomers + responses: + "200": + description: Success + content: + application/json: + schema: + type: array + items: + $ref: "#/components/schemas/Customer" + /customers/{id}: + get: + tags: + - customer + summary: Get customer details + description: Get customer details with the given id. + operationId: getCustomer + parameters: + - name: id + in: path + description: Customer id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/CustomerDetails" + "400": + description: Invalid customer id + post: + tags: + - customer + summary: Update customer details + description: Update customer details with the given id. + operationId: updateCustomer + parameters: + - name: id + in: path + description: Customer id + required: true + schema: + type: integer + format: int64 + - name: name + in: query + description: Customer name + required: true + schema: + type: string + responses: + "200": + description: successful operation + delete: + tags: + - customer + summary: Delete customer details + description: Delete customer details with the given id. + operationId: deleteCustomer + parameters: + - name: id + in: path + description: Customer id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + + /customer/{region}/{country}: + get: + tags: + - customer + summary: Get customers regional + description: Get customers regional with the given id. + operationId: getCustomersRegional + parameters: + - name: region + in: path + description: Customers region + required: true + schema: + type: string + - name: country + in: path + description: Customers country + required: true + schema: + type: string + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/Customer" + "400": + description: Invalid parameters supplied +components: + schemas: + Customer: + type: object + description: Customer id and name + properties: + id: + type: integer + format: int64 + description: Customer id + name: + type: string + description: Customer name + required: + - id + - name + CustomerDetails: + type: object + description: Customer details + properties: + id: + type: integer + format: int64 + description: Customer id + name: + type: string + description: Customer name + description: + type: string + description: Customer description + required: + - id + - name diff --git a/oap-server/server-core/src/test/resources/openapi-definitions/serviceB/productAPI-v2.yaml b/oap-server/server-core/src/test/resources/openapi-definitions/serviceB/productAPI-v2.yaml new file mode 100644 index 0000000000..b30a8b3665 --- /dev/null +++ b/oap-server/server-core/src/test/resources/openapi-definitions/serviceB/productAPI-v2.yaml @@ -0,0 +1,210 @@ +# Licensed to the Apache Software Foundation (ASF) under one or more +# contributor license agreements. See the NOTICE file distributed with +# this work for additional information regarding copyright ownership. +# The ASF licenses this file to You 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. + +openapi: 3.0.0 +x-sw-endpoint-name-format: "${PATH}:<${METHOD}>" + +info: + description: OpenAPI definition for SkyWalking test. + version: v2 + title: Product API + +tags: + - name: product + description: product + - name: relatedProducts + description: Related Products + +paths: + /products: + get: + tags: + - product + summary: Get all products list + description: Get all products list. + operationId: getProducts + responses: + "200": + description: Success + content: + application/json: + schema: + type: array + items: + $ref: "#/components/schemas/Product" + /products/{region}/{country}: + get: + tags: + - product + summary: Get products regional + description: Get products regional with the given id. + operationId: getProductRegional + parameters: + - name: region + in: path + description: Products region + required: true + schema: + type: string + - name: country + in: path + description: Products country + required: true + schema: + type: string + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/Product" + "400": + description: Invalid parameters supplied + /products/{id}: + get: + tags: + - product + summary: Get product details + description: Get product details with the given id. + operationId: getProduct + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/ProductDetails" + "400": + description: Invalid product id + post: + tags: + - product + summary: Update product details + description: Update product details with the given id. + operationId: updateProduct + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + - name: name + in: query + description: Product name + required: true + schema: + type: string + responses: + "200": + description: successful operation + delete: + tags: + - product + summary: Delete product details + description: Delete product details with the given id. + operationId: deleteProduct + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + /products/{id}/relatedProducts: + get: + tags: + - relatedProducts + summary: Get related products + description: Get related products with the given product id. + operationId: getRelatedProducts + parameters: + - name: id + in: path + description: Product id + required: true + schema: + type: integer + format: int64 + responses: + "200": + description: successful operation + content: + application/json: + schema: + $ref: "#/components/schemas/RelatedProducts" + "400": + description: Invalid product id + +components: + schemas: + Product: + type: object + description: Product id and name + properties: + id: + type: integer + format: int64 + description: Product id + name: + type: string + description: Product name + required: + - id + - name + ProductDetails: + type: object + description: Product details + properties: + id: + type: integer + format: int64 + description: Product id + name: + type: string + description: Product name + description: + type: string + description: Product description + required: + - id + - name + RelatedProducts: + type: object + description: Related Products + properties: + id: + type: integer + format: int32 + description: Product id + relatedProducts: + type: array + description: List of related products + items: + $ref: "#/components/schemas/Product" diff --git a/oap-server/server-library/library-util/src/main/java/org/apache/skywalking/oap/server/library/util/ResourceUtils.java b/oap-server/server-library/library-util/src/main/java/org/apache/skywalking/oap/server/library/util/ResourceUtils.java index 1381090a21..93b26207a7 100644 --- a/oap-server/server-library/library-util/src/main/java/org/apache/skywalking/oap/server/library/util/ResourceUtils.java +++ b/oap-server/server-library/library-util/src/main/java/org/apache/skywalking/oap/server/library/util/ResourceUtils.java @@ -24,8 +24,10 @@ import java.io.InputStream; import java.io.InputStreamReader; import java.io.Reader; import java.net.URL; +import java.util.ArrayList; import java.util.Arrays; import java.util.HashSet; +import java.util.List; import java.util.Objects; import java.util.Set; @@ -58,12 +60,51 @@ public class ResourceUtils { throw new FileNotFoundException("path not found: " + parentPath); } final Set nameSet = new HashSet<>(Arrays.asList(fileNames)); - final File[] listFiles = Objects.requireNonNull(new File(url.getPath()) - .listFiles((dir, name) -> nameSet.contains(name)), "No files in " + parentPath); + final File[] listFiles = Objects.requireNonNull( + new File(url.getPath()) + .listFiles((dir, name) -> nameSet.contains(name)), "No files in " + parentPath); if (listFiles.length == 0) { throw new FileNotFoundException("files not found:" + nameSet); } return listFiles; } + + /** + * @param directoryPath the directory path + * @param maxDepth the max directory depth to get the files, the given directory is 0 as the tree root + * @return all normal files which in this directory and subDirectory according to the maxDepth + * @throws FileNotFoundException the directory not exist in the given path + */ + public static List getDirectoryFilesRecursive(String directoryPath, + int maxDepth) throws FileNotFoundException { + + URL url = ResourceUtils.class.getClassLoader().getResource(directoryPath); + if (url == null) { + throw new FileNotFoundException("path not found: " + directoryPath); + } + List fileList = new ArrayList<>(); + return getDirectoryFilesRecursive(url.getPath(), fileList, maxDepth); + } + + private static List getDirectoryFilesRecursive(String directoryPath, List fileList, int maxDepth) { + if (maxDepth < 0) { + return fileList; + } + maxDepth--; + File file = new File(directoryPath); + if (file.isDirectory()) { + File[] subFiles = file.listFiles(); + if (subFiles != null) { + for (File subFile : subFiles) { + if (subFile.isDirectory()) { + getDirectoryFilesRecursive(subFile.getPath(), fileList, maxDepth); + } else { + fileList.add(subFile); + } + } + } + } + return fileList; + } } -- GitLab