[ { "path": ".editorconfig", "content": "root = true\n\n[*]\ncharset = utf-8\nindent_style = space\nindent_size = 4\ntrim_trailing_whitespace = true\ninsert_final_newline = true\nend_of_line = lf\n\n[*.md]\ntrim_trailing_whitespace = false\n" }, { "path": ".github/ISSUE_TEMPLATE/bug_report.md", "content": "---\nname: Bug report\nabout: Create a report to help us improve\ntitle: ''\nlabels: ''\nassignees: ''\n\n---\n\n---\nname: 🐞 Bug report\nabout: Create a report to help us improve\ntitle: ''\nlabels: ''\nassignees: ''\n\n---\n\n**Describe the bug**\nA clear and concise description of what the bug is.\n\n**To Reproduce**\nSteps to reproduce the behavior:\n1. Execute '...'\n2. Send a request to '....'\n3. See error\n\n**Expected behavior**\nA clear and concise description of what you expected to happen.\n\n**Version**\nThe version number of Easegress.\n\n**Configuration**\n* EaseAgent Configuration\n```\n```\n\n**Logs**\n```\nEaseAgent logs, if applicable.\n```\n\n**OS and Hardware**\n - OS: [e.g. Ubuntu 20.04]\n - CPU:[e.g. Intel(R) Core(TM) i5-8265U]\n - Memory: [e.g. 16GB]\n\n**Additional context**\nAdd any other context about the problem here.\n\n\n---\nThanks for contributing 🎉!\n" }, { "path": ".github/ISSUE_TEMPLATE/feature_request.md", "content": "---\nname: Feature request\nabout: Suggest an idea for this project\ntitle: ''\nlabels: ''\nassignees: ''\n\n---\n\n---\nname: 🚀 Feature request\nabout: Suggest an idea for this project\ntitle: ''\nlabels: ''\nassignees: ''\n\n---\n\n**Is your feature request related to a problem? Please describe.**\nA clear and concise description of what the problem is. Ex. I'm always frustrated when [...]\n\n**Describe the solution you'd like**\nA clear and concise description of what you want to happen.\n\n**Describe alternatives you've considered**\nA clear and concise description of any alternative solutions or features you've considered.\n\n**Additional context**\nAdd any other context or screenshots about the feature request here.\n\n\n---\nThanks for contributing 🎉!\n" }, { "path": ".github/workflows/build.yml", "content": "# This is a basic workflow to help you get started with Actions\n\nname: Build & Test\n\n# Controls when the workflow will run\non:\n # Triggers the workflow on push or pull request events but only for the master branch\n push:\n branches: [ master ]\n paths-ignore:\n - 'doc/**'\n - 'resources/**'\n - '**.md'\n pull_request:\n branches: [ master ]\n paths-ignore:\n - 'doc/**'\n - 'resources/**'\n - '**.md'\n\n # Allows you to run this workflow manually from the Actions tab\n workflow_dispatch:\n\n# A workflow run is made up of one or more jobs that can run sequentially or in parallel\njobs:\n # This workflow contains a single job called \"build\"\n build:\n # The type of runner that the job will run on\n runs-on: ${{ matrix.os }}\n\n strategy:\n fail-fast: false\n matrix:\n os: [ ubuntu-latest, windows-latest ]\n java-version: [ 8, 11, 16, 17 ]\n java-distribution: [ adopt, adopt-openj9 ]\n\n # Steps represent a sequence of tasks that will be executed as part of the job\n steps:\n\n\n # Checks-out your repository under $GITHUB_WORKSPACE, so your job can access it\n - name: Checkout Codebase\n uses: actions/checkout@v2\n\n - name: Setup Java Version\n uses: actions/setup-java@v3.14.1\n with:\n java-version: ${{ matrix.java-version }}\n distribution: ${{ matrix.java-distribution }}\n architecture: x64\n cache: 'maven'\n\n # Runs a single command using the runners shell\n - name: Build with Maven\n run: mvn clean package\n" }, { "path": ".github/workflows/license-checker.yml", "content": "name: License checker\n\non:\n push:\n branches:\n - maste\n paths-ignore:\n - 'doc/**'\n - 'resources/**'\n - '**.md'\n pull_request:\n branches:\n - master\n paths-ignore:\n - 'doc/**'\n - 'resources/**'\n - '**.md'\n\njobs:\n check-license:\n runs-on: ubuntu-latest\n\n steps:\n - uses: actions/checkout@v2\n - name: Check License Header\n uses: apache/skywalking-eyes@main\n env:\n GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}\n with:\n log: info\n" }, { "path": ".gitignore", "content": ".idea/\n*.iml\ntarget/\ndependency-reduced-pom.xml\n*.class\n*.swp\n**/*.versionsBackup\n\n.patch\n.classpath\n.factorypath\n.project\n.settings\n.DS_Store\n.vscode\nlogs/\ntmp/\n" }, { "path": ".licenserc.yaml", "content": "header:\n license:\n content: |\n Copyright (c) 2022, MegaEase\n All rights reserved.\n \n Licensed under the Apache License, Version 2.0 (the \"License\");\n you may not use this file except in compliance with the License.\n You may obtain a copy of the License at\n \n http://www.apache.org/licenses/LICENSE-2.0\n \n Unless required by applicable law or agreed to in writing, software\n distributed under the License is distributed on an \"AS IS\" BASIS,\n WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n See the License for the specific language governing permissions and\n limitations under the License.\n paths:\n - './'\n\n paths-ignore:\n - '.editorconfig'\n - 'install'\n - 'release'\n - '**/target/**'\n - '**/*.xml'\n - '**/*.yaml'\n - '**/*.properties'\n - '**/*.md'\n - '**/*.iml'\n - 'LICENSE'\n - '.github/'\n - '.git/'\n - 'doc/'\n - '**/resources/**'\n - 'CHANGELOG.md'\n - '.gitignore'\n - 'README.md'\n\n comment: on-failure\n\ndependency:\n files:\n - go.mod\n" }, { "path": "AOSP-Checkstyles.xml", "content": "\n\n\n\n\n\n \n\n \n\n \n \n \n \n \n \n \n \n \n \n \n\n \n \n \n \n \n\n \n \n \n \n \n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "CODE_OF_CONDUCT.md", "content": "# Contributor Covenant Code of Conduct\n\n## Our Pledge\n\nWe as members, contributors, and leaders pledge to make participation in our\ncommunity a harassment-free experience for everyone, regardless of age, body\nsize, visible or invisible disability, ethnicity, sex characteristics, gender\nidentity and expression, level of experience, education, socio-economic status,\nnationality, personal appearance, race, religion, or sexual identity\nand orientation.\n\nWe pledge to act and interact in ways that contribute to an open, welcoming,\ndiverse, inclusive, and healthy community.\n\n## Our Standards\n\nExamples of behavior that contributes to a positive environment for our\ncommunity include:\n\n* Demonstrating empathy and kindness toward other people\n* Being respectful of differing opinions, viewpoints, and experiences\n* Giving and gracefully accepting constructive feedback\n* Accepting responsibility and apologizing to those affected by our mistakes,\n and learning from the experience\n* Focusing on what is best not just for us as individuals, but for the\n overall community\n\nExamples of unacceptable behavior include:\n\n* The use of sexualized language or imagery, and sexual attention or\n advances of any kind\n* Trolling, insulting or derogatory comments, and personal or political attacks\n* Public or private harassment\n* Publishing others' private information, such as a physical or email\n address, without their explicit permission\n* Other conduct which could reasonably be considered inappropriate in a\n professional setting\n\n## Enforcement Responsibilities\n\nCommunity leaders are responsible for clarifying and enforcing our standards of\nacceptable behavior and will take appropriate and fair corrective action in\nresponse to any behavior that they deem inappropriate, threatening, offensive,\nor harmful.\n\nCommunity leaders have the right and responsibility to remove, edit, or reject\ncomments, commits, code, wiki edits, issues, and other contributions that are\nnot aligned to this Code of Conduct, and will communicate reasons for moderation\ndecisions when appropriate.\n\n## Scope\n\nThis Code of Conduct applies within all community spaces, and also applies when\nan individual is officially representing the community in public spaces.\nExamples of representing our community include using an official e-mail address,\nposting via an official social media account, or acting as an appointed\nrepresentative at an online or offline event.\n\n## Enforcement\n\nInstances of abusive, harassing, or otherwise unacceptable behavior may be\nreported to the community leaders responsible for enforcement at\nservice@megaease.com.\nAll complaints will be reviewed and investigated promptly and fairly.\n\nAll community leaders are obligated to respect the privacy and security of the\nreporter of any incident.\n\n## Enforcement Guidelines\n\nCommunity leaders will follow these Community Impact Guidelines in determining\nthe consequences for any action they deem in violation of this Code of Conduct:\n\n### 1. Correction\n\n**Community Impact**: Use of inappropriate language or other behavior deemed\nunprofessional or unwelcome in the community.\n\n**Consequence**: A private, written warning from community leaders, providing\nclarity around the nature of the violation and an explanation of why the\nbehavior was inappropriate. A public apology may be requested.\n\n### 2. Warning\n\n**Community Impact**: A violation through a single incident or series\nof actions.\n\n**Consequence**: A warning with consequences for continued behavior. No\ninteraction with the people involved, including unsolicited interaction with\nthose enforcing the Code of Conduct, for a specified period of time. This\nincludes avoiding interactions in community spaces as well as external channels\nlike social media. Violating these terms may lead to a temporary or\npermanent ban.\n\n### 3. Temporary Ban\n\n**Community Impact**: A serious violation of community standards, including\nsustained inappropriate behavior.\n\n**Consequence**: A temporary ban from any sort of interaction or public\ncommunication with the community for a specified period of time. No public or\nprivate interaction with the people involved, including unsolicited interaction\nwith those enforcing the Code of Conduct, is allowed during this period.\nViolating these terms may lead to a permanent ban.\n\n### 4. Permanent Ban\n\n**Community Impact**: Demonstrating a pattern of violation of community\nstandards, including sustained inappropriate behavior, harassment of an\nindividual, or aggression toward or disparagement of classes of individuals.\n\n**Consequence**: A permanent ban from any sort of public interaction within\nthe community.\n\n## Attribution\n\nThis Code of Conduct is adapted from the [Contributor Covenant][homepage],\nversion 2.0, available at\nhttps://www.contributor-covenant.org/version/2/0/code_of_conduct.html.\n\nCommunity Impact Guidelines were inspired by [Mozilla's code of conduct\nenforcement ladder](https://github.com/mozilla/diversity).\n\n[homepage]: https://www.contributor-covenant.org\n\nFor answers to common questions about this code of conduct, see the FAQ at\nhttps://www.contributor-covenant.org/faq. Translations are available at\nhttps://www.contributor-covenant.org/translations.\n" }, { "path": "LICENSE", "content": " Apache License\n Version 2.0, January 2004\n http://www.apache.org/licenses/\n\n TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION\n\n 1. Definitions.\n\n \"License\" shall mean the terms and conditions for use, reproduction,\n and distribution as defined by Sections 1 through 9 of this document.\n\n \"Licensor\" shall mean the copyright owner or entity authorized by\n the copyright owner that is granting the License.\n\n \"Legal Entity\" shall mean the union of the acting entity and all\n other entities that control, are controlled by, or are under common\n control with that entity. For the purposes of this definition,\n \"control\" means (i) the power, direct or indirect, to cause the\n direction or management of such entity, whether by contract or\n otherwise, or (ii) ownership of fifty percent (50%) or more of the\n outstanding shares, or (iii) beneficial ownership of such entity.\n\n \"You\" (or \"Your\") shall mean an individual or Legal Entity\n exercising permissions granted by this License.\n\n \"Source\" form shall mean the preferred form for making modifications,\n including but not limited to software source code, documentation\n source, and configuration files.\n\n \"Object\" form shall mean any form resulting from mechanical\n transformation or translation of a Source form, including but\n not limited to compiled object code, generated documentation,\n and conversions to other media types.\n\n \"Work\" shall mean the work of authorship, whether in Source or\n Object form, made available under the License, as indicated by a\n copyright notice that is included in or attached to the work\n (an example is provided in the Appendix below).\n\n \"Derivative Works\" shall mean any work, whether in Source or Object\n form, that is based on (or derived from) the Work and for which the\n editorial revisions, annotations, elaborations, or other modifications\n represent, as a whole, an original work of authorship. For the purposes\n of this License, Derivative Works shall not include works that remain\n separable from, or merely link (or bind by name) to the interfaces of,\n the Work and Derivative Works thereof.\n\n \"Contribution\" shall mean any work of authorship, including\n the original version of the Work and any modifications or additions\n to that Work or Derivative Works thereof, that is intentionally\n submitted to Licensor for inclusion in the Work by the copyright owner\n or by an individual or Legal Entity authorized to submit on behalf of\n the copyright owner. For the purposes of this definition, \"submitted\"\n means any form of electronic, verbal, or written communication sent\n to the Licensor or its representatives, including but not limited to\n communication on electronic mailing lists, source code control systems,\n and issue tracking systems that are managed by, or on behalf of, the\n Licensor for the purpose of discussing and improving the Work, but\n excluding communication that is conspicuously marked or otherwise\n designated in writing by the copyright owner as \"Not a Contribution.\"\n\n \"Contributor\" shall mean Licensor and any individual or Legal Entity\n on behalf of whom a Contribution has been received by Licensor and\n subsequently incorporated within the Work.\n\n 2. Grant of Copyright License. Subject to the terms and conditions of\n this License, each Contributor hereby grants to You a perpetual,\n worldwide, non-exclusive, no-charge, royalty-free, irrevocable\n copyright license to reproduce, prepare Derivative Works of,\n publicly display, publicly perform, sublicense, and distribute the\n Work and such Derivative Works in Source or Object form.\n\n 3. Grant of Patent License. Subject to the terms and conditions of\n this License, each Contributor hereby grants to You a perpetual,\n worldwide, non-exclusive, no-charge, royalty-free, irrevocable\n (except as stated in this section) patent license to make, have made,\n use, offer to sell, sell, import, and otherwise transfer the Work,\n where such license applies only to those patent claims licensable\n by such Contributor that are necessarily infringed by their\n Contribution(s) alone or by combination of their Contribution(s)\n with the Work to which such Contribution(s) was submitted. If You\n institute patent litigation against any entity (including a\n cross-claim or counterclaim in a lawsuit) alleging that the Work\n or a Contribution incorporated within the Work constitutes direct\n or contributory patent infringement, then any patent licenses\n granted to You under this License for that Work shall terminate\n as of the date such litigation is filed.\n\n 4. Redistribution. You may reproduce and distribute copies of the\n Work or Derivative Works thereof in any medium, with or without\n modifications, and in Source or Object form, provided that You\n meet the following conditions:\n\n (a) You must give any other recipients of the Work or\n Derivative Works a copy of this License; and\n\n (b) You must cause any modified files to carry prominent notices\n stating that You changed the files; and\n\n (c) You must retain, in the Source form of any Derivative Works\n that You distribute, all copyright, patent, trademark, and\n attribution notices from the Source form of the Work,\n excluding those notices that do not pertain to any part of\n the Derivative Works; and\n\n (d) If the Work includes a \"NOTICE\" text file as part of its\n distribution, then any Derivative Works that You distribute must\n include a readable copy of the attribution notices contained\n within such NOTICE file, excluding those notices that do not\n pertain to any part of the Derivative Works, in at least one\n of the following places: within a NOTICE text file distributed\n as part of the Derivative Works; within the Source form or\n documentation, if provided along with the Derivative Works; or,\n within a display generated by the Derivative Works, if and\n wherever such third-party notices normally appear. The contents\n of the NOTICE file are for informational purposes only and\n do not modify the License. You may add Your own attribution\n notices within Derivative Works that You distribute, alongside\n or as an addendum to the NOTICE text from the Work, provided\n that such additional attribution notices cannot be construed\n as modifying the License.\n\n You may add Your own copyright statement to Your modifications and\n may provide additional or different license terms and conditions\n for use, reproduction, or distribution of Your modifications, or\n for any such Derivative Works as a whole, provided Your use,\n reproduction, and distribution of the Work otherwise complies with\n the conditions stated in this License.\n\n 5. Submission of Contributions. Unless You explicitly state otherwise,\n any Contribution intentionally submitted for inclusion in the Work\n by You to the Licensor shall be under the terms and conditions of\n this License, without any additional terms or conditions.\n Notwithstanding the above, nothing herein shall supersede or modify\n the terms of any separate license agreement you may have executed\n with Licensor regarding such Contributions.\n\n 6. Trademarks. This License does not grant permission to use the trade\n names, trademarks, service marks, or product names of the Licensor,\n except as required for reasonable and customary use in describing the\n origin of the Work and reproducing the content of the NOTICE file.\n\n 7. Disclaimer of Warranty. Unless required by applicable law or\n agreed to in writing, Licensor provides the Work (and each\n Contributor provides its Contributions) on an \"AS IS\" BASIS,\n WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or\n implied, including, without limitation, any warranties or conditions\n of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A\n PARTICULAR PURPOSE. You are solely responsible for determining the\n appropriateness of using or redistributing the Work and assume any\n risks associated with Your exercise of permissions under this License.\n\n 8. Limitation of Liability. In no event and under no legal theory,\n whether in tort (including negligence), contract, or otherwise,\n unless required by applicable law (such as deliberate and grossly\n negligent acts) or agreed to in writing, shall any Contributor be\n liable to You for damages, including any direct, indirect, special,\n incidental, or consequential damages of any character arising as a\n result of this License or out of the use or inability to use the\n Work (including but not limited to damages for loss of goodwill,\n work stoppage, computer failure or malfunction, or any and all\n other commercial damages or losses), even if such Contributor\n has been advised of the possibility of such damages.\n\n 9. Accepting Warranty or Additional Liability. While redistributing\n the Work or Derivative Works thereof, You may choose to offer,\n and charge a fee for, acceptance of support, warranty, indemnity,\n or other liability obligations and/or rights consistent with this\n License. However, in accepting such obligations, You may act only\n on Your own behalf and on Your sole responsibility, not on behalf\n of any other Contributor, and only if You agree to indemnify,\n defend, and hold each Contributor harmless for any liability\n incurred by, or claims asserted against, such Contributor by reason\n of your accepting any such warranty or additional liability.\n\n END OF TERMS AND CONDITIONS\n\n APPENDIX: How to apply the Apache License to your work.\n\n To apply the Apache License to your work, attach the following\n boilerplate notice, with the fields enclosed by brackets \"[]\"\n replaced with your own identifying information. (Don't include\n the brackets!) The text should be enclosed in the appropriate\n comment syntax for the file format. We also recommend that a\n file or class name and description of purpose be included on the\n same \"printed page\" as the copyright notice for easier\n identification within third-party archives.\n\n Copyright [yyyy] [name of copyright owner]\n\n Licensed under the Apache License, Version 2.0 (the \"License\");\n you may not use this file except in compliance with the License.\n You may obtain a copy of the License at\n\n http://www.apache.org/licenses/LICENSE-2.0\n\n Unless required by applicable law or agreed to in writing, software\n distributed under the License is distributed on an \"AS IS\" BASIS,\n WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n See the License for the specific language governing permissions and\n limitations under the License.\n" }, { "path": "README.md", "content": "# EaseAgent\n\nA lightweight & opening Java Agent for Cloud-Native and APM system\n\n- [EaseAgent](#easeagent)\n - [Overview](#overview)\n - [Purpose](#purpose)\n - [Principles](#principles)\n - [Features](#features)\n - [Architecture Diagram](#architecture-diagram)\n - [Description](#description)\n - [QuickStart](#quickstart)\n - [Get And Set Environment Variable](#get-and-set-environment-variable)\n - [Setup Environment Variable](#setup-environment-variable)\n - [Download](#download)\n - [Build From the Source](#build-from-the-source)\n - [Get Configuration file](#get-configuration-file)\n - [Monitor Spring Petclinic](#monitor-spring-petclinic)\n - [Prerequisites](#prerequisites)\n - [Initialize and Start the project](#initialize-and-start-the-project)\n - [Metric](#metric)\n - [Tracing](#tracing)\n - [Build Spring Petclinic](#build-spring-petclinic)\n - [Add an Enhancement Plugin](#add-an-enhancement-plugin)\n - [User Manual](#user-manual)\n - [Enhancement Plugin Development Guide](#enhancement-plugin-development-guide)\n - [Report Plugin Development Guide](#report-plugin-development-guide)\n - [Community](#community)\n - [Licenses](#licenses)\n\n## Overview\n- EaseAgent is the underlying component that provides non-intrusive extensions to applications of the Java ecosystem. \n- EaseAgent can collect distributed application tracing, metrics, and logs, which could be used in the APM system and improve the observability of a distributed system. for the tracing, EaseAgent follows the [Google Dapper](https://research.google/pubs/pub36356/) paper.\n- EaseAgent also can work with Cloud-Native architecture. For example, it can help Service Mesh (especially for [EaseMesh](https://github.com/megaease/easemesh/) ) to do some control panel work.\n- EaseAgent supports plugins mechanism development, which is easy to extend or add new functionality.\n\n### Purpose\n- EaseAgent can be a Java agent for APM(Application Performance Management) system.\n- EaseAgent collects the basic metrics and the service tracing logs, which is very helpful for performance analysis and troubleshooting.\n- EaseAgent is compatible with mainstream monitoring ecosystems, such as Kafka, ElasticSearch, Prometheus, Zipkin, etc.\n- EaseAgent majorly focuses on the Spring Boot development environments, but users can support any Java ecosystem applications through plugins.\n- EaseAgent can support scenario-specific business requirements through the plugin mechanism, such as traffic redirection, traffic coloring, etc.\n\n### Principles\n- Safe to Java application/service.\n- Instrumenting a Java application in a non-intrusive way.\n- Lightweight and very low CPU, memory, and I/O resource usage.\n- Highly extensible, users can easily do extensions through a simple and clear plugin interface.\n- Design for Micro-Service architecture, collecting the data from a service perspective.\n\n## Features\n* Easy to use. It is right out of the box for Metrics, Tracing and Logs collecting.\n * Collecting Metric & Tracing Logs.\n * `JDBC 4.0`\n * `HTTP Servlet`、`HTTP Filter`\n * `Spring Boot 2.2.x`: `WebClient` 、 `RestTemplate`、`FeignClient`\n * `RabbitMQ Client 5.x`、 `Kafka Client 2.4.x`\n * `Jedis 3.5.x`、 `Lettuce 5.3.x (sync、async)`\n * `ElasticSearch Client >= 7.x (sync、async)`\n * `Mongodb Client >=4.0.x (sync、async)`\n * `Motan`\n * `Dubbo`\n * `SofaRpc >= 5.3.0`\n * Collecting Access Logs.\n * `HTTP Servlet`、`HTTP Filter`\n * `Spring Cloud Gateway`\n * Instrumenting the `traceId` and `spanId` into user application logging automatically\n * Supplying the `health check` endpoint\n * Supplying the `readiness check` endpoint for `SpringBoot2.2.x`\n * Supplying the `agent info` endpoint\n \n* Data Reports\n * Console Reporter.\n * Prometheus Exports.\n * Http Reporter.\n * Kafka Reporter.\n * Custom Reporter.\n\n* Easy to Extend\n * Simple and clear Plugin Interface, creating a plugin as few as three classes.\n * Extremely cleanly packaged `Tracing` and `Metric` API, with a small amount of code to achieve business support.\n\n* Standardization\n * The tracing data format is fully compatible with the Zipkin data format.\n * Metric data format fully supports integration with `Prometheus`.\n * The application log format is fully compatible with the `Opentelemetry` data format.\n\n\n## Architecture Diagram\n![image](./doc/images/EaseAgent-Architecture-Base-v2.0.png)\n\n#### Description\n**Plugin Framework** in `core` module is base on [Byte buddy](https://github.com/raphw/byte-buddy) technology.\n\n1. Easeagent's plugin defines where (which classes and methods) to make enhancements by implementing the `Points` and what to do at the point of enhancement by implementing the `Interceptor`.\n2. When the program invokes the enhanced method of class defined by Points, the `unique index`(uid) owned by the method will be used as a parameter to call the common interface of `Agent Common Method Advice`, which finds the `Agent Interceptor Chain` by the `Unique Index` and calls the `before` method of each Interceptor in the chain in order of priority.\n3. Normally, both the `Metric Interceptor` and the `Tracing Interceptor` are in the agent interceptor chain and are called sequentially.\n4. According to call the `Metric API` and `Tracing API` in interceptors, the `Metric` and `Tracing` information will be stored in `MetricRegistry` and `Tracing`.\n5. The `Reporter` module will get information from `MetricRegistry` and `Tracing` and send it to `Kafka`.\n6. The `after` method of each interceptor in the `Agent Interceptor Chain` will be invoked in the reverse order of the `before` invoked at last.\n7. The `tracing` data can be sent to `kafka` server or `zipkin` server, the `metric` data can be sent to `kafka` server and pull by `Prometheus` server.\n\n## QuickStart\n\n### Get And Set Environment Variable\n\nSetup Environment Variable and then download the latest release of `easeagent.jar` or build it from the source.\n\n#### Setup Environment Variable\n```\n$ cd ~/easeagent #[Replace with agent path]\n$ export EASE_AGENT_PATH=`pwd` # export EASE_AGENT_PATH=[Replace with agent path]\n$ mkdir plugins\n```\n\n#### Download\nDownload `easeagent.jar` from releases [releases](https://github.com/megaease/easeagent/releases).\n\n```\n$ curl -Lk https://github.com/megaease/easeagent/releases/latest/download/easeagent.jar -O\n```\n\n#### Build From the Source\n\nYou need Java 1.8+ and git:\n\nDownload EaseAgent with `git clone https://github.com/megaease/easeagent.git`.\n```\n$ cd easeagent\n$ mvn clean package -Dmaven.test.skip\n$ cp ./build/target/easeagent-dep.jar $EASE_AGENT_PATH/easeagent.jar\n```\nThe `./build/target/easeagent-dep.jar` is the agent jar with all the dependencies.\n\n> For the Windows platform, please make sure git `core.autocrlf` is set to false before git clone.\n> You can use `git config --global core.autocrlf false` to modify `core.autocrlf`.\n\n[How to use easeagent.jar on host?](doc/how-to-use/use-on-host.md)\n\n[How to use easeagent.jar in docker?](doc/how-to-use/use-in-docker.md)\n\n#### Get Configuration file\nExtracting the default configuration file.\n```\n$ cd $EASE_AGENT_PATH\n$ jar xf easeagent.jar agent.properties easeagent-log4j2.xml\n```\n\nBy default, there is an agent.properties configuration file, which is configured to print all output data to the console.\n\n### Monitor Spring Petclinic\n#### Prerequisites\n- Make sure you have installed the docker, docker-compose in your environment.\n- Make sure your docker version is higher than v19.+.\n- Make sure your docker-compose version is higher than v2.+.\n\n[Project Details](https://github.com/megaease/easeagent-spring-petclinic)\n\n#### Initialize and Start the project\n```\n$ git clone https://github.com/megaease/easeagent-spring-petclinic.git\n$ cd easeagent-spring-petclinic\n$ git submodule update --init\n$ ./spring-petclinic.sh start\n```\n\n> The script will download the latest release of EaseAgent. \n> If you want to use your own built EaseAgent, copy it to the directory: `easeagent/downloaded`\n>> ```$ cp $EASE_AGENT_PATH/easeagent.jar easeagent/downloaded/easeagent-latest.jar``` \n\nIt requires `Docker` to pull images from the docker hub, be patient. \n\nOpen Browser to visit grafana UI: [http://localhost:3000](http://localhost:3000).\n\n#### Metric\nClick the `search dashboards`, the first icon in the left menu bar. Choose the `spring-petclinic-easeagent` to open the dashboard we prepare for you.\n\nPrometheus Metric Schedule: [Prometheus Metric](doc/prometheus-metric-schedule.md)\n\n![metric](doc/images/grafana-metric.png)\n\n#### Tracing\n\nIf you want to check the tracing-data, you could click the explore in the left menu bar. Click the Search - beta to switch search mode. Click search query button in the right up corner, there is a list containing many tracing. Choose one to click.\n\n![tracing](doc/images/grafana-tracing.png)\n\n#### Build Spring Petclinic\n\n[Spring Petclinic Demo](doc/spring-petclinic-demo.md)\n\n### Add an Enhancement Plugin\n[Add a Demo Plugin to EaseAgent](doc/add-plugin-demo.md)\n\n## User Manual\nFor more information, please refer to the [User Manual](./doc/user-manual.md).\n\n## Enhancement Plugin Development Guide\nRefer to [Plugin Development Guide](./doc/development-guide.md).\n\n## Report Plugin Development Guide\nReport plugin enables user report tracing/metric data to different kinds of the backend in a different format.\n\nRefer to [Report Plugin Development Guide](./doc/report-development-guide.md)\n\n## Community\n\n* [Github Issues](https://github.com/megaease/easeagent/issues)\n* [Join Slack Workspace](https://join.slack.com/t/openmegaease/shared_invite/zt-upo7v306-lYPHvVwKnvwlqR0Zl2vveA) for requirement, issue and development.\n* [MegaEase on Twitter](https://twitter.com/megaease)\n\nIf you have any questions, welcome to discuss them in our community. Welcome to join!\n\n\n## Licenses\nEaseAgent is licensed under the Apache License, Version 2.0. See [LICENSE](./LICENSE) for the full license text.\n" }, { "path": "build/pom.xml", "content": "\n\n\n \n easeagent\n com.megaease.easeagent\n 2.3.0\n \n 4.0.0\n\n build\n\n \n \n org.apache.kafka\n kafka-clients\n provided\n \n \n javax.servlet\n javax.servlet-api\n provided\n \n \n com.megaease.easeagent\n metrics\n ${project.version}\n \n \n com.megaease.easeagent\n zipkin\n ${project.version}\n \n \n com.megaease.easeagent\n log4j2-api\n ${project.version}\n \n \n com.megaease.easeagent\n core\n ${project.version}\n \n \n com.megaease.easeagent\n loader\n ${project.version}\n \n \n\n \n easeagent\n \n \n org.apache.maven.plugins\n maven-compiler-plugin\n 3.6.1\n \n ${version.java}\n ${version.java}\n ${encoding.file}\n \n -Xlint:unchecked\n \n \n \n \n maven-antrun-plugin\n 3.0.0\n \n \n prepare-package\n \n \n \n \n \n \n\n \n\n \n \n \n \n \n \n \n run\n \n \n \n \n \n org.apache.maven.plugins\n maven-shade-plugin\n ${version.maven-shade-plugin}\n \n true\n \n \n \n \n \n \n \n org.apache.logging.log4j:*\n \n **\n \n \n \n \n \n \n package\n \n shade\n \n \n \n \n \n com.github.edwgiz\n maven-shade-plugin.log4j2-cachefile-transformer\n 2.14.0\n \n \n \n \n pl.project13.maven\n git-commit-id-plugin\n 4.9.10\n \n \n \n revision\n \n \n \n \n true\n \n false\n true\n \n yyyy-MM-dd'T'HH:mm:ssZ\n true\n ${project.build.outputDirectory}/git.properties\n \n \n git.commit.*\n \n \n \n \n org.apache.maven.plugins\n maven-assembly-plugin\n 3.0.0\n \n \n src/assembly/src.xml\n \n \n \n true\n true\n com.megaease.easeagent.Main\n com.megaease.easeagent.StartBootstrap\n log4j.configurationFile\n ${version}\n \n \n \n false\n \n \n \n \n package\n \n single\n \n \n \n \n \n \n\n\n\n" }, { "path": "build/src/assembly/src.xml", "content": "\n dep\n /\n \n jar\n \n \n \n ${project.build.outputDirectory}\n \n \n git.properties\n agent.yaml\n agent.properties\n user-minimal-cfg.properties\n easeagent-log4j2.xml\n \n \n\n \n\n \n ${project.basedir}/target/inner-plugins/\n plugins\n \n original-*.jar\n \n \n *.jar\n \n \n \n ${project.basedir}/target/log4j2/\n log4j2\n \n *.jar\n \n \n\n \n ${project.basedir}/target/boot/\n boot\n \n *.jar\n \n \n \n \n \n boot\n \n com.megaease.easeagent:plugin-api\n \n false\n \n \n lib\n \n **/com/megaease/easeagent/plugin/*\n **/com/megaease/easeagent/boot/*\n **/com/megaease/easeagent/log4j2/*\n \n \n com.megaease.easeagent:build\n \n \n \n \n \n \n com.megaease.easeagent:loader\n \n true\n \n \n\n" }, { "path": "build/src/main/java/com/megaease/easeagent/StartBootstrap.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent;\n\nimport com.megaease.easeagent.core.Bootstrap;\n\nimport java.lang.instrument.Instrumentation;\n\npublic class StartBootstrap {\n private StartBootstrap() {}\n\n public static void premain(String args, Instrumentation inst, String javaAgentJarPath) {\n Bootstrap.start(args, inst, javaAgentJarPath);\n }\n}\n" }, { "path": "build/src/main/resources/agent-to-cloud_1.0.properties", "content": "# for v2.0 agent send to megacloud-v1.0 kafka topic setting\n\nplugin.observability.global.metric.topic=application-meter\nplugin.observability.access.metric.topic=application-log\nplugin.observability.redis.metric.topic=platform-meter\nplugin.observability.rabbit.metric.topic=platform-meter\nplugin.observability.kafka.metric.topic=platform-meter\nplugin.observability.elasticsearch.metric.topic=platform-meter\n\nplugin.observability.jvmGc.metric.topic=platform-meter\nplugin.observability.jvmMemory.metric.topic=platform-meter\n\n\n\n" }, { "path": "build/src/main/resources/agent.properties", "content": "name=demo-service\nsystem=demo-system\n### http server\n# When the enabled value = false, agent will not start the http server\n# You can use -Deaseagent.server.enabled=[true | false] to override.\neaseagent.server.enabled=true\n# http server port. You can use -Deaseagent.server.port=[port] to override.\neaseagent.server.port=9900\n# Enable health/readiness\neaseagent.health.readiness.enabled=true\nplugin.integrability.global.healthReady.enabled=true\n# forwarded headers page\n# Pass-through headers from the root process all the way to the end\n# format: easeagent.progress.forwarded.headers={headerName}\n#easeagent.progress.forwarded.headers=X-Forwarded-For\n#easeagent.progress.forwarded.headers=X-Location,X-Mesh-Service-Canary,X-Phone-Os\n###\n### default tracings configuration\n###\n# sampledType:\n## counting: percentage sampling, sampled limit 0.01 to 1, 1 is always sample, 0 is never sample, 0.1 is ten samples per hundred\n## rate_limiting: traces per second, sampled >= 0, 0 is never sample, 10 is max 10 traces per second\n## boundary: percentage sampling by traceId, sampled limit 0.0001 to 1, 1 is always sample, 0 is never sample\n## if sampled=0.001, when (traceId^random)%10000<=(0.001*10000) sampled\n## sampledType must be used with sampled, otherwise the default value is used Sampler.ALWAYS_SAMPLE\nobservability.tracings.sampledType=\nobservability.tracings.sampled=1\n# get header from response headers then tag to tracing span\n# format: observability.tracings.tag.response.headers.{key}={value}\n# support ease mesh\n# X-EG-Circuit-Breaker\n# X-EG-Retryer\n# X-EG-Rate-Limiter\n# X-EG-Time-Limiter\nobservability.tracings.tag.response.headers.eg.0=X-EG-Circuit-Breaker\nobservability.tracings.tag.response.headers.eg.1=X-EG-Retryer\nobservability.tracings.tag.response.headers.eg.2=X-EG-Rate-Limiter\nobservability.tracings.tag.response.headers.eg.3=X-EG-Time-Limiter\n# -------------------- plugin global config ---------------------\nplugin.observability.global.init.enabled=true\nplugin.observability.global.tracing.enabled=true\nplugin.observability.global.metric.enabled=true\nplugin.observability.global.metric.interval=30\nplugin.observability.global.metric.topic=application-metrics\nplugin.observability.global.metric.url=/application-metrics\n##\n# if different with reporter.outputServer.appendType,\n# following options can be used in user config file to override\n# the default or global one\n#\n## when it's scrape by prometheus, noop can be used\n# plugin.observability.global.metric.appendType=noop\n## for debug, console can be used\n# plugin.observability.global.metric.appendType=console\n# plugin.observability.global.metric.appendType=http\n#\n# add service name to header enabled by name for easemesh\nplugin.integrability.global.addServiceNameHead.enabled=true\n# redirect the middleware address when env has address, see: com.megaease.easeagent.plugin.api.middleware.RedirectProcessor\n# about redirect: jdbc, kafka, rabbitmq, redis,\nplugin.integrability.global.redirect.enabled=true\n# forwarded headers enabled.\n# headers see config: easeagent.progress.forwarded.headers.???=???\nplugin.integrability.global.forwarded.enabled=true\nplugin.hook.global.foundation.enabled=true\nplugin.observability.global.log.enabled=true\nplugin.observability.global.log.topic=application-log\nplugin.observability.global.log.url=/application-log\n#plugin.observability.global.log.appendType=console\nplugin.observability.global.log.level=INFO\nplugin.observability.global.log.encoder=LogDataJsonEncoder\n#plugin.observability.global.log.encoder.collectMdcKeys=\n# support pattern:\n# \"logLevel\": \"%-5level\",\n# \"threadId\": \"%thread\",\n# \"location\": \"%logger{36}\",\n# \"message\": \"%msg%n\",\nplugin.observability.global.log.encoder.timestamp=%d{UNIX_MILLIS}\nplugin.observability.global.log.encoder.logLevel=%-5level\nplugin.observability.global.log.encoder.threadId=%thread\nplugin.observability.global.log.encoder.location=%logger{36}\nplugin.observability.global.log.encoder.message=%msg%n%xEx{3}\n#\n# -------------------- access ---------------------\n## access: servlet and spring gateway\nplugin.observability.access.log.encoder=AccessLogJsonEncoder\n# plugin.observability.access.metric.appendType=kafka\n#plugin.observability.logback.log.enabled=false\n#plugin.observability.log4j2.log.enabled=false\n# ----------------------------------------------\n# if the plugin configuration is consistent with the global namespace,\n# do not add configuration items not commented out in this default configuration file.\n# otherwise, they can not be overridden by Global configuration in user's configuration file.\n#\n# -------------------- jvm ---------------------\n# plugin.observability.jvmGc.metric.enabled=true\n# plugin.observability.jvmGc.metric.interval=30\nplugin.observability.jvmGc.metric.topic=platform-metrics\nplugin.observability.jvmGc.metric.url=/platform-metrics\n# plugin.observability.jvmGc.metric.appendType=kafka\n# plugin.observability.jvmMemory.metric.enabled=true\n# plugin.observability.jvmMemory.metric.interval=30\nplugin.observability.jvmMemory.metric.topic=platform-metrics\nplugin.observability.jvmMemory.metric.url=/platform-metrics\n# plugin.observability.jvmMemory.metric.appendType=kafka\n#\n# -------------------- async ---------------------\n# plugin.observability.async.tracing.enabled=true\n#\n# -------------------- elasticsearch redirect ---------------------\n# plugin.integrability.elasticsearch.redirect.enabled=true\n# plugin.observability.elasticsearch.tracing.enabled=true\n# elasticsearch metric\n# plugin.observability.elasticsearch.metric.enabled=true\n# plugin.observability.elasticsearch.metric.interval=30\nplugin.observability.elasticsearch.metric.topic=platform-metrics\nplugin.observability.elasticsearch.metric.url=/platform-metrics\n# plugin.observability.elasticsearch.metric.appendType=kafka\n#\n# -------------------- httpServlet ---------------------\n# plugin.observability.httpServlet.tracing.enabled=true\n# plugin.observability.httpServlet.metric.enabled=true\n# plugin.observability.httpServlet.metric.interval=30\n# plugin.observability.httpServlet.metric.topic=application-metrics\n# plugin.observability.httpServlet.metric.url=/application-metrics\n# plugin.observability.httpServlet.metric.appendType=kafka\n#\n#\n# -------------------- tomcat ---------------------\n# plugin.observability.tomcat.tracing.enabled=true\n# plugin.observability.tomcat.metric.enabled=true\n# plugin.observability.tomcat.metric.interval=30\n# plugin.observability.tomcat.metric.topic=application-metrics\n# plugin.observability.tomcat.metric.url=/application-metrics\n# plugin.observability.tomcat.metric.appendType=kafka\n#\n# -------------------- jdbc ---------------------\n## jdbc tracing\n# plugin.observability.jdbc.tracing.enabled=true\n# jdbcStatement metric\n# plugin.observability.jdbcStatement.metric.enabled=true\n# plugin.observability.jdbcStatement.metric.interval=30\n# plugin.observability.jdbcStatement.metric.topic=application-metrics\n# plugin.observability.jdbcStatement.metric.url=/application-metrics\n# plugin.observability.jdbcStatement.metric.appendType=kafka\n## jdbcConnection metric\n# plugin.observability.jdbcConnection.metric.enabled=true\n# plugin.observability.jdbcConnection.metric.interval=30\n# plugin.observability.jdbcConnection.metric.topic=application-metrics\n# plugin.observability.jdbcConnection.metric.url=/application-metrics\n# plugin.observability.jdbcConnection.metric.appendType=kafka\n## sql compress\n## compress.enabled=true, can use md5Dictionary to compress\n## compress.enabled=false, use original sql\nplugin.observability.jdbc.sql.compress.enabled=true\n## md5Dictionary metric\n# plugin.observability.md5Dictionary.metric.enabled=true\n# plugin.observability.md5Dictionary.metric.interval=30\n# plugin.observability.md5Dictionary.metric.topic=application-metrics\n# plugin.observability.md5Dictionary.metric.url=/application-metrics\n# plugin.observability.md5Dictionary.metric.appendType=kafka\n## jdbc redirect\n# plugin.integrability.jdbc.redirect.enabled=true\n#\n# -------------------- kafka ---------------------\n# kafka tracing\n# plugin.observability.kafka.tracing.enabled=true\n# kafka metric\n# plugin.observability.kafka.metric.enabled=true\n# plugin.observability.kafka.metric.interval=30\nplugin.observability.kafka.metric.topic=platform-metrics\nplugin.observability.kafka.metric.url=/platform-metrics\n# plugin.observability.kafka.metric.appendType=kafka\n# kafka redirect\n# plugin.integrability.kafka.redirect.enabled=true\n#\n# -------------------- rabbitmq ---------------------\n# rabbitmq tracing\n# plugin.observability.rabbitmq.tracing.enabled=true\n# rabbitmq metric\n# plugin.observability.rabbitmq.metric.enabled=true\n# plugin.observability.rabbitmq.metric.interval=30\nplugin.observability.rabbitmq.metric.topic=platform-metrics\nplugin.observability.rabbitmq.metric.url=/platform-metrics\n# plugin.observability.rabbitmq.metric.appendType=kafka\n# rabbitmq redirect\n# plugin.integrability.rabbitmq.redirect.enabled=true\n#\n# -------------------- redis ---------------------\n# redis tracing\n# plugin.observability.redis.tracing.enabled=true\n# redis metric\n# plugin.observability.redis.metric.enabled=true\n# plugin.observability.redis.metric.interval=30\nplugin.observability.redis.metric.topic=platform-metrics\nplugin.observability.redis.metric.url=/platform-metrics\n# plugin.observability.redis.metric.appendType=kafka\n# redis redirect\n# plugin.integrability.redis.redirect.enabled=true\n#\n# -------------------- springGateway ---------------------\n# springGateway tracing\n# plugin.observability.springGateway.tracing.enabled=true\n# springGateway metric\n# plugin.observability.springGateway.metric.enabled=true\n# plugin.observability.springGateway.metric.interval=30\n# plugin.observability.springGateway.metric.topic=application-metrics\n# plugin.observability.springGateway.metric.url=/application-metrics\n# plugin.observability.springGateway.metric.appendType=kafka\n#\n# -------------------- request ---------------------\n## httpclient tracing\\uFF1Ahttpclient and httpclient5\n# plugin.observability.httpclient.tracing.enabled=true\n## okHttp tracing\n# plugin.observability.okHttp.tracing.enabled=true\n## webclient tracing\n# plugin.observability.webclient.tracing.enabled=true\n## feignClient tracing\n# plugin.observability.feignClient.tracing.enabled=true\n## restTemplate tracing\n# plugin.observability.restTemplate.tracing.enabled=true\n## httpURLConnection tracing\n# plugin.observability.httpURLConnection.tracing.enabled=true\n# -------------------- service name ---------------------\n## add service name to header by name for easemesh. default name: X-Mesh-RPC-Service\n# plugin.integrability.serviceName.addServiceNameHead.propagate.head=X-Mesh-RPC-Service\n#\n# -------------------- mongodb ---------------------\n## mongodb tracing\n# plugin.observability.mongodb.tracing.enabled=true\n## mongodb metric\n# plugin.observability.mongodb.metric.enabled=true\n# plugin.observability.mongodb.metric.interval=30\nplugin.observability.mongodb.metric.topic=platform-metrics\nplugin.observability.mongodb.metric.url=/platform-metrics\n# plugin.observability.mongodb.metric.appendType=kafka\n## mongodb redirect\n# plugin.integrability.mongodb.redirect.enabled=true\n## mongodb foundation\n# plugin.hook.mongodb.foundation.enabled=true\n#\n# -------------------- motan ---------------------\n# motan tracing\n# plugin.observability.motan.tracing.enabled=true\n## motan args collect switch\n# plugin.observability.motan.tracing.args.collect.enabled=false\n## motan result collect switch\n# plugin.observability.motan.tracing.result.collect.enabled=false\n# motan metric\n# plugin.observability.motan.metric.enabled=true\n# plugin.observability.motan.metric.interval=30\nplugin.observability.motan.metric.topic=platform-metrics\nplugin.observability.motan.metric.url=/platform-metrics\n# plugin.observability.motan.metric.appendType=kafka\n#\n# -------------------- dubbo ---------------------\n## dubbo tracing\n#plugin.observability.dubbo.tracing.enabled=true\n## dubbo arguments collect switch\n#plugin.observability.dubbo.tracing.args.collect.enabled=false\n## dubbo return result collect switch\n#plugin.observability.dubbo.tracing.result.collect.enabled=false\n## dubbo metric\n#plugin.observability.dubbo.metric.enabled=true\n#plugin.observability.dubbo.metric.interval=30\n#plugin.observability.dubbo.metric.topic=platform-metrics\n#plugin.observability.dubbo.metric.url=/platform-metrics\n# plugin.observability.dubbo.metric.appendType=kafka\n# -------------------- sofarpc ---------------------\n# sofarpc tracing\n# plugin.observability.sofarpc.tracing.enabled=true\n## sofarpc args collect switch\n# plugin.observability.sofarpc.tracing.args.collect.enabled=false\n## sofarpc result collect switch\n# plugin.observability.sofarpc.tracing.result.collect.enabled=false\n# sofarpc metric\n# plugin.observability.sofarpc.metric.enabled=true\n# plugin.observability.sofarpc.metric.interval=30\nplugin.observability.sofarpc.metric.topic=platform-metrics\nplugin.observability.sofarpc.metric.url=/platform-metrics\n# plugin.observability.sofarpc.metric.appendType=kafka\n# -------------- output ------------------\n## http/kafka/zipkin server host and port for tracing and metric\n###### example ######\n## http: [http|https]://127.0.0.1:8080/report\n## kafka: 192.168.1.2:9092, 192.168.1.3:9092, 192.168.1.3:9092\n## zipkin: [http|https]://127.0.0.1:8080/zipkin\nreporter.outputServer.bootstrapServer=127.0.0.1:9092\nreporter.outputServer.appendType=console\nreporter.outputServer.timeout=1000\n## enabled=false: disable output tracing and metric\n## enabled=true: output tracing and metric\nreporter.outputServer.enabled=true\n## username and password for http basic auth\nreporter.outputServer.username=\nreporter.outputServer.password=\n## enable=false: disable mtls\n## enable=true: enable tls\n## key, cert, ca_cert is enabled when tls.enable=true\nreporter.outputServer.tls.enable=false\nreporter.outputServer.tls.key=\nreporter.outputServer.tls.cert=\nreporter.outputServer.tls.ca_cert=\n# --- redefine to output properties\nreporter.log.output.messageMaxBytes=999900\nreporter.log.output.reportThread=1\nreporter.log.output.queuedMaxSpans=1000\nreporter.log.output.queuedMaxSize=1000000\nreporter.log.output.messageTimeout=1000\n## sender.appendType config\n## [http] send to http server\n## [kafka] send to kafka\n## [console] send to console\n## reporter.log.sender.appendType=console\n## enabled=true:\n# reporter.log.sender.enabled=true\n# reporter.log.sender.url=/application-log\n## sender.appendType config\n## [http] send to http server\n## [kafka] send to kafka\n## [console] send to console\n# reporter.tracing.sender.appendType=http\n# reporter.tracing.sender.appendType=console\n## enabled=true:\nreporter.tracing.sender.enabled=true\n## url is only used in http\n## append to outputServer.bootstrapServer\n###### example ######\n## reporter.outputServer.bootstrapServer=http://127.0.0.1:8080/report\n## reporter.tracing.sender.url=/tracing\n## final output url: http://127.0.0.1:8080/report/tracing\n## if url is start with [http|https], url override reporter.outputServer.bootstrapServer\n###### example ######\n## reporter.outputServer.bootstrapServer=http://127.0.0.1:8080/report\n## reporter.tracing.sender.url=http://127.0.0.10:9090/tracing\n## final output url: http://127.0.0.10:9090/tracing\nreporter.tracing.sender.url=/application-tracing-log\n## topic for kafka use\nreporter.tracing.sender.topic=application-tracing-log\nreporter.tracing.encoder=SpanJsonEncoder\n# --- redefine to output properties\nreporter.tracing.output.messageMaxBytes=999900\nreporter.tracing.output.reportThread=1\nreporter.tracing.output.queuedMaxSpans=1000\nreporter.tracing.output.queuedMaxSize=1000000\nreporter.tracing.output.messageTimeout=1000\n## sender.appendType config\n## [http] send to http server\n## [metricKafka] send to kafka\n## [console] send to console\n#reporter.metric.sender.appendType=http\n#reporter.metric.sender.appendType=console\n## url is only used in http\n## append to outputServer.bootstrapServer\n###### example ######\n## reporter.outputServer.bootstrapServer=http://127.0.0.1:8080/report\n## reporter.metric.sender.url=/metric\n## final output url: http://127.0.0.1:8080/report/metric\n## if url is start with [http|https], url override reporter.outputServer.bootstrapServer\n###### example ######\n## reporter.outputServer.bootstrapServer=http://127.0.0.1:8080/report\n## reporter.metric.sender.url=http://127.0.0.10:9090/metric\n## final output url: http://127.0.0.10:9090/metric\n#reporter.metric.sender.url=/metrics\n\n## support spring boot 3.5.3: jdk17+3.5.3\n#runtime.code.version.points.jdk=jdk17\n#runtime.code.version.points.spring-boot=3.x.x\n" }, { "path": "build/src/main/resources/easeagent-log4j2.xml", "content": "\n\n \n \n \n %d [%10.10t] %5p %10.10c{1} - %msg%n\n \n \n \n \n \n \n \n \n \n\n \n \n \n" }, { "path": "build/src/main/resources/user-minimal-cfg.properties", "content": "## This a minimal configuration required to start the EaseAgent.\n## In most cases, this is all the configuration items that the normal user needs to be concerned about.\n## \n## When the user specifies a user configration file as this one, the items in user config file will override the default \n## configuration in agent.properties packaged in easeagent.jar\n##\n## -Deaseagent.config.path=/path/to/user-cfg-file\n\nname=demo-springweb\nsystem=demo-system\n\n###\n### report configuration \n###\nreporter.outputServer.bootstrapServer=http://127.0.0.1:9411\nreporter.outputServer.appendType=console\n\n##\n## Global metric configuration\n## the appendType is same as outputServer, so comment out\n# plugin.observability.global.metric.appendType=console\n\n##\n## tracing sender\n\n## [http] send to http server\n## [kafka] send to kafka\n## [console] send to console\n#\nreporter.tracing.sender.appendType=\n# reporter.tracing.sender.url=http://tempo:9411/api/v2/spans\nreporter.tracing.sender.url=http://localhost:9411/api/v2/spans\n\n## access log sender\n## [http] send to http server\n## [kafka] send to kafka\n## [console] send to console\n## the appendType is same as outputServer, so comment out\n## reporter.log.sender.appendType=console\n\n\n" }, { "path": "config/pom.xml", "content": "\n\n\n\n \n easeagent\n com.megaease.easeagent\n 2.3.0\n \n 4.0.0\n\n config\n\n \n \n com.megaease.easeagent\n plugin-api\n \n\n \n com.fasterxml.jackson.core\n jackson-databind\n \n\n \n com.megaease.easeagent\n log4j2-api\n \n \n org.slf4j\n slf4j-api\n \n \n com.megaease.easeagent\n log4j2-mock\n ${project.version}\n test\n \n \n org.yaml\n snakeyaml\n \n \n com.google.guava\n guava\n \n \n com.github.stefanbirkner\n system-rules\n 1.19.0\n test\n \n \n\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/AutoRefreshConfigItem.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.api.config.Config;\n\nimport java.util.function.BiFunction;\n\npublic class AutoRefreshConfigItem {\n private volatile T value;\n\n public AutoRefreshConfigItem(Config config, String name, BiFunction func) {\n ConfigUtils.bindProp(name, config, func, v -> this.value = v);\n }\n\n public T getValue() {\n return value;\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/CompatibilityConversion.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.ProgressFields;\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\n\nimport javax.annotation.Nonnull;\nimport java.util.*;\nimport java.util.function.BiFunction;\n\npublic class CompatibilityConversion {\n private static final Logger LOGGER = LoggerFactory.getLogger(CompatibilityConversion.class);\n protected static final String[] REQUEST_NAMESPACE = new String[]{\n ConfigConst.Namespace.HTTPCLIENT,\n ConfigConst.Namespace.OK_HTTP,\n ConfigConst.Namespace.WEB_CLIENT,\n ConfigConst.Namespace.FEIGN_CLIENT,\n ConfigConst.Namespace.REST_TEMPLATE,\n };\n private static final Map>> KEY_TO_NAMESPACE;\n private static final Set METRIC_SKIP;\n private static final Set TRACING_SKIP;\n\n static {\n Map>> map = new HashMap<>();\n map.put(ConfigConst.Observability.KEY_METRICS_ACCESS, SingleBuilder.observability(ConfigConst.Namespace.ACCESS));\n\n map.put(ConfigConst.Observability.KEY_METRICS_REQUEST, MultipleBuilder.observability(Arrays.asList(REQUEST_NAMESPACE)));\n map.put(ConfigConst.Observability.KEY_METRICS_JDBC_STATEMENT, SingleBuilder.observability(ConfigConst.Namespace.JDBC_STATEMENT));\n map.put(ConfigConst.Observability.KEY_METRICS_JDBC_CONNECTION, SingleBuilder.observability(ConfigConst.Namespace.JDBC_CONNECTION));\n map.put(ConfigConst.Observability.KEY_METRICS_MD5_DICTIONARY, SingleBuilder.observability(ConfigConst.Namespace.MD5_DICTIONARY));\n map.put(ConfigConst.Observability.KEY_METRICS_RABBIT, SingleBuilder.observability(ConfigConst.Namespace.RABBITMQ));\n map.put(ConfigConst.Observability.KEY_METRICS_KAFKA, SingleBuilder.observability(ConfigConst.Namespace.KAFKA));\n map.put(ConfigConst.Observability.KEY_METRICS_CACHE, SingleBuilder.observability(ConfigConst.Namespace.REDIS));\n map.put(ConfigConst.Observability.KEY_METRICS_JVM_GC, null);\n map.put(ConfigConst.Observability.KEY_METRICS_JVM_MEMORY, null);\n\n map.put(ConfigConst.Observability.KEY_TRACE_REQUEST, MultipleBuilder.observability(Arrays.asList(REQUEST_NAMESPACE)));\n map.put(ConfigConst.Observability.KEY_TRACE_REMOTE_INVOKE, SingleBuilder.observability(ConfigConst.Namespace.WEB_CLIENT));\n map.put(ConfigConst.Observability.KEY_TRACE_KAFKA, SingleBuilder.observability(ConfigConst.Namespace.KAFKA));\n map.put(ConfigConst.Observability.KEY_TRACE_JDBC, SingleBuilder.observability(ConfigConst.Namespace.JDBC));\n map.put(ConfigConst.Observability.KEY_TRACE_CACHE, SingleBuilder.observability(ConfigConst.Namespace.REDIS));\n map.put(ConfigConst.Observability.KEY_TRACE_RABBIT, SingleBuilder.observability(ConfigConst.Namespace.RABBITMQ));\n\n KEY_TO_NAMESPACE = map;\n\n TRACING_SKIP = new HashSet<>();\n TRACING_SKIP.add(ConfigConst.Observability.KEY_COMM_ENABLED);\n TRACING_SKIP.add(ConfigConst.Observability.KEY_COMM_SAMPLED_TYPE);\n TRACING_SKIP.add(ConfigConst.Observability.KEY_COMM_SAMPLED);\n TRACING_SKIP.add(ConfigConst.Observability.KEY_COMM_OUTPUT);\n TRACING_SKIP.add(ConfigConst.Observability.KEY_COMM_TAG);\n\n METRIC_SKIP = new HashSet<>();\n METRIC_SKIP.add(ConfigConst.Observability.KEY_METRICS_JVM_GC);\n METRIC_SKIP.add(ConfigConst.Observability.KEY_METRICS_JVM_MEMORY);\n }\n\n public static Map transform(Map oldConfigs) {\n Map changedKeys = new HashMap<>();\n Map newConfigs = new HashMap<>();\n for (Map.Entry entry : oldConfigs.entrySet()) {\n Conversion conversion = transformConversion(entry.getKey());\n Object changed = conversion.transform(newConfigs, entry.getValue());\n if (conversion.isChange()) {\n changedKeys.put(entry.getKey(), changed);\n }\n }\n if (changedKeys.isEmpty()) {\n return oldConfigs;\n }\n if (LOGGER.isInfoEnabled()) {\n LOGGER.info(\"config key has transform: \");\n for (Map.Entry entry : changedKeys.entrySet()) {\n LOGGER.info(\"{} to {}\", entry.getKey(), entry.getValue());\n }\n }\n return newConfigs;\n }\n\n private static Conversion transformConversion(String key) {\n if (key.startsWith(\"observability.metrics.\")) {\n return metricConversion(key);\n } else if (key.startsWith(\"observability.tracings.\")) {\n return tracingConversion(key);\n } else if (key.startsWith(ConfigConst.GlobalCanaryLabels.SERVICE_HEADERS + \".\")) {\n// return forwardedHeadersConversion(key);\n }\n return new FinalConversion(key, false);\n }\n\n private static Conversion metricConversion(String key) {\n if (key.equals(ConfigConst.Observability.METRICS_ENABLED)) {\n return new MultipleFinalConversion(Arrays.asList(new FinalConversion(ConfigConst.Observability.METRICS_ENABLED, true),\n new FinalConversion(ConfigConst.Plugin.OBSERVABILITY_GLOBAL_METRIC_ENABLED, true)), true);\n }\n return conversion(key, METRIC_SKIP, ConfigConst.PluginID.METRIC);\n }\n\n\n private static Conversion tracingConversion(String key) {\n if (key.equals(ConfigConst.Observability.TRACE_ENABLED)) {\n return new FinalConversion(ConfigConst.Plugin.OBSERVABILITY_GLOBAL_TRACING_ENABLED, true);\n }\n return conversion(key, TRACING_SKIP, ConfigConst.PluginID.TRACING);\n }\n\n// private static Conversion forwardedHeadersConversion(String key) {\n// return new FinalConversion(key.replace(ConfigConst.GlobalCanaryLabels.SERVICE_HEADERS + \".\", ProgressFields.EASEAGENT_PROGRESS_FORWARDED_HEADERS_CONFIG + \".\"), true);\n// }\n\n private static Conversion conversion(String key, Set skipSet, String pluginId) {\n String[] keys = ConfigConst.split(key);\n if (keys.length < 4) {\n return new FinalConversion(key, false);\n }\n String key2 = keys[2];\n if (skipSet.contains(key2)) {\n return new FinalConversion(key, false);\n }\n BiFunction> builder = KEY_TO_NAMESPACE.get(key2);\n if (builder == null) {\n builder = SingleBuilder.observability(key2);\n }\n String[] properties = new String[keys.length - 3];\n int index = 0;\n for (int i = 3; i < keys.length; i++) {\n properties[index++] = keys[i];\n }\n return builder.apply(pluginId, ConfigConst.join(properties));\n }\n\n interface Conversion {\n K transform(Map configs, String value);\n\n boolean isChange();\n }\n\n static class FinalConversion implements Conversion {\n private final String key;\n private final boolean change;\n\n public FinalConversion(String key, boolean change) {\n this.key = key;\n this.change = change;\n }\n\n @Override\n public String transform(Map configs, String value) {\n configs.put(key, value);\n return key;\n }\n\n public boolean isChange() {\n return change;\n }\n }\n\n static class MultipleFinalConversion implements Conversion> {\n private final List conversions;\n private final boolean change;\n\n MultipleFinalConversion(@Nonnull List conversions, boolean change) {\n this.conversions = conversions;\n this.change = change;\n }\n\n @Override\n public List transform(Map configs, String value) {\n List result = new ArrayList<>();\n for (FinalConversion conversion : conversions) {\n result.add(conversion.transform(configs, value));\n }\n return result;\n }\n\n @Override\n public boolean isChange() {\n return change;\n }\n }\n\n static class SingleConversion implements Conversion {\n private final String domain;\n private final String namespace;\n private final String id;\n private final String properties;\n\n public SingleConversion(String domain, String namespace, String id, String properties) {\n this.domain = domain;\n this.namespace = namespace;\n this.id = id;\n this.properties = properties;\n }\n\n @Override\n public String transform(Map configs, String value) {\n String key = ConfigUtils.buildPluginProperty(domain, namespace, id, properties);\n configs.put(key, value);\n return key;\n }\n\n @Override\n public boolean isChange() {\n return true;\n }\n }\n\n static class MultipleConversion implements Conversion> {\n private final String domain;\n private final List namespaces;\n private final String id;\n private final String properties;\n\n public MultipleConversion(String domain, List namespaces, String id, String properties) {\n this.domain = domain;\n this.namespaces = namespaces;\n this.id = id;\n this.properties = properties;\n }\n\n @Override\n public List transform(Map configs, String value) {\n List keys = new ArrayList<>();\n for (String namespace : namespaces) {\n String key = ConfigUtils.buildPluginProperty(domain, namespace, id, properties);\n keys.add(key);\n configs.put(key, value);\n }\n return keys;\n }\n\n @Override\n public boolean isChange() {\n return true;\n }\n }\n\n static class SingleBuilder implements BiFunction> {\n private final String domain;\n private final String namespace;\n\n public SingleBuilder(String domain, String namespace) {\n this.domain = domain;\n this.namespace = namespace;\n }\n\n @Override\n public Conversion apply(String id, String properties) {\n return new SingleConversion(domain, namespace, id, properties);\n }\n\n static SingleBuilder observability(String namespace) {\n return new SingleBuilder(ConfigConst.OBSERVABILITY, namespace);\n }\n }\n\n static class MultipleBuilder implements BiFunction> {\n private final String domain;\n private final List namespaces;\n\n public MultipleBuilder(String domain, List namespaces) {\n this.domain = domain;\n this.namespaces = namespaces;\n }\n\n @Override\n public Conversion apply(String id, String properties) {\n return new MultipleConversion(domain, namespaces, id, properties);\n }\n\n static MultipleBuilder observability(List namespaces) {\n return new MultipleBuilder(ConfigConst.OBSERVABILITY, namespaces);\n }\n }\n\n\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/ConfigAware.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.api.config.Config;\n\npublic interface ConfigAware {\n void setConfig(Config config);\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/ConfigFactory.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.utils.ImmutableMap;\nimport com.megaease.easeagent.plugin.utils.SystemEnv;\nimport com.megaease.easeagent.plugin.utils.common.JsonUtil;\nimport com.megaease.easeagent.plugin.utils.common.StringUtils;\n\nimport java.io.File;\nimport java.util.HashMap;\nimport java.util.Map;\nimport java.util.TreeMap;\n\npublic class ConfigFactory {\n private static final Logger LOGGER = LoggerFactory.getLogger(ConfigFactory.class);\n private static final String CONFIG_PROP_FILE = \"agent.properties\";\n private static final String CONFIG_YAML_FILE = \"agent.yaml\";\n\n public static final String AGENT_CONFIG_PATH_PROP_KEY = \"easeagent.config.path\";\n\n public static final String AGENT_SERVICE = \"name\";\n public static final String AGENT_SYSTEM = \"system\";\n\n public static final String AGENT_SERVER_PORT = \"easeagent.server.port\";\n public static final String AGENT_SERVER_ENABLED = \"easeagent.server.enabled\";\n\n public static final String EASEAGENT_ENV_CONFIG = \"EASEAGENT_ENV_CONFIG\";\n\n private static final Map AGENT_CONFIG_KEYS_TO_PROPS =\n ImmutableMap.builder()\n .put(\"easeagent.name\", AGENT_SERVICE)\n .put(\"easeagent.system\", AGENT_SYSTEM)\n .put(\"easeagent.server.port\", AGENT_SERVER_PORT)\n .put(\"easeagent.server.enabled\", AGENT_SERVER_ENABLED)\n .build();\n\n // OTEL_SERVICE_NAME=xxx\n private static final Map AGENT_ENV_KEY_TO_PROPS = new HashMap<>();\n\n\n static {\n for (Map.Entry entry : AGENT_CONFIG_KEYS_TO_PROPS.entrySet()) {\n // dot.case -> UPPER_UNDERSCORE\n AGENT_ENV_KEY_TO_PROPS.put(\n ConfigPropertiesUtils.toEnvVarName(entry.getKey()),\n entry.getValue()\n );\n }\n }\n\n /**\n * update config value from environment variables and java properties\n *

\n * java properties > environment variables > env:EASEAGENT_ENV_CONFIG={} > default\n */\n static Map updateEnvCfg() {\n Map envCfg = new TreeMap<>();\n\n String configEnv = SystemEnv.get(EASEAGENT_ENV_CONFIG);\n if (StringUtils.isNotEmpty(configEnv)) {\n Map map = JsonUtil.toMap(configEnv);\n Map strMap = new HashMap<>();\n if (!map.isEmpty()) {\n for (Map.Entry entry : map.entrySet()) {\n strMap.put(entry.getKey(), entry.getValue().toString());\n }\n }\n envCfg.putAll(strMap);\n }\n\n // override by environment variables, eg: export EASEAGENT_NAME=xxx\n for (Map.Entry entry : AGENT_ENV_KEY_TO_PROPS.entrySet()) {\n String value = SystemEnv.get(entry.getKey());\n if (!StringUtils.isEmpty(value)) {\n envCfg.put(entry.getValue(), value);\n }\n }\n\n // override by java properties; eg: java -Deaseagent.name=xxx\n for (Map.Entry entry : AGENT_CONFIG_KEYS_TO_PROPS.entrySet()) {\n String value = System.getProperty(entry.getKey());\n if (!StringUtils.isEmpty(value)) {\n envCfg.put(entry.getValue(), value);\n }\n }\n\n return envCfg;\n }\n\n private ConfigFactory() {\n }\n\n /**\n * Get config file path from system properties or environment variables\n */\n public static String getConfigPath() {\n // get config path from -Deaseagent.config.path=/easeagent/agent.properties || export EASEAGENT_CONFIG_PATH=/easeagent/agent.properties\n String path = ConfigPropertiesUtils.getString(AGENT_CONFIG_PATH_PROP_KEY);\n\n if (StringUtils.isEmpty(path)) {\n // eg: -Dotel.javaagent.configuration-file=/easeagent/agent.properties || export OTEL_JAVAAGENT_CONFIGURATION_FILE=/easeagent/agent.properties\n path = OtelSdkConfigs.getConfigPath();\n }\n return path;\n }\n\n public static GlobalConfigs loadConfigs(String pathname, ClassLoader loader) {\n // load property configuration file if exist\n GlobalConfigs configs = loadDefaultConfigs(loader, CONFIG_PROP_FILE);\n\n // load yaml configuration file if exist\n GlobalConfigs yConfigs = loadDefaultConfigs(loader, CONFIG_YAML_FILE);\n configs.mergeConfigs(yConfigs);\n\n // override by user special config file\n if (StringUtils.isNotEmpty(pathname)) {\n GlobalConfigs configsFromOuterFile = ConfigLoader.loadFromFile(new File(pathname));\n LOGGER.info(\"Loaded user special config file: {}\", pathname);\n configs.mergeConfigs(configsFromOuterFile);\n }\n\n // override by opentelemetry sdk env config\n configs.updateConfigsNotNotify(OtelSdkConfigs.updateEnvCfg());\n\n // check environment cfg override\n configs.updateConfigsNotNotify(updateEnvCfg());\n\n if (LOGGER.isDebugEnabled()) {\n final String display = configs.toPrettyDisplay();\n LOGGER.debug(\"Loaded conf:\\n{}\", display);\n }\n return configs;\n }\n\n private static GlobalConfigs loadDefaultConfigs(ClassLoader loader, String file) {\n GlobalConfigs globalConfigs = JarFileConfigLoader.load(file);\n if (globalConfigs != null) {\n return globalConfigs;\n }\n return ConfigLoader.loadFromClasspath(loader, file);\n }\n\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/ConfigLoader.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.config.yaml.YamlReader;\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport org.yaml.snakeyaml.parser.ParserException;\n\nimport java.io.File;\nimport java.io.FileInputStream;\nimport java.io.IOException;\nimport java.io.InputStream;\nimport java.util.Collections;\nimport java.util.HashMap;\nimport java.util.Map;\nimport java.util.Properties;\n\npublic class ConfigLoader {\n private static final Logger LOGGER = LoggerFactory.getLogger(ConfigLoader.class);\n\n private static boolean checkYaml(String filename) {\n return filename.endsWith(\".yaml\") || filename.endsWith(\".yml\");\n }\n\n static GlobalConfigs loadFromFile(File file) {\n try (FileInputStream in = new FileInputStream(file)) {\n return ConfigLoader.loadFromStream(in, file.getAbsolutePath());\n } catch (IOException e) {\n LOGGER.warn(\"Load config file failure: {}\", file.getAbsolutePath());\n }\n return new GlobalConfigs(Collections.emptyMap());\n }\n\n static GlobalConfigs loadFromStream(InputStream in, String filename) throws IOException {\n if (in != null) {\n Map map;\n if (checkYaml(filename)) {\n try {\n map = new YamlReader().load(in).compress();\n } catch (ParserException e) {\n LOGGER.warn(\"Wrong Yaml format, load config file failure: {}\", filename);\n map = Collections.emptyMap();\n }\n } else {\n map = extractPropsMap(in);\n }\n return new GlobalConfigs(map);\n } else {\n return new GlobalConfigs(Collections.emptyMap());\n }\n }\n\n private static HashMap extractPropsMap(InputStream in) throws IOException {\n Properties properties = new Properties();\n properties.load(in);\n HashMap map = new HashMap<>();\n for (String one : properties.stringPropertyNames()) {\n map.put(one, properties.getProperty(one));\n }\n return map;\n }\n\n static GlobalConfigs loadFromClasspath(ClassLoader classLoader, String file) {\n try (InputStream in = classLoader.getResourceAsStream(file)) {\n return ConfigLoader.loadFromStream(in, file);\n } catch (IOException e) {\n LOGGER.warn(\"Load config file:{} by classloader:{} failure: {}\", file, classLoader.toString(), e);\n }\n\n return new GlobalConfigs(Collections.emptyMap());\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/ConfigManagerMXBean.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport java.io.IOException;\nimport java.util.List;\nimport java.util.Map;\n\npublic interface ConfigManagerMXBean {\n void updateConfigs(Map configs);\n\n void updateService(String json, String version) throws IOException;\n\n void updateCanary(String json, String version) throws IOException;\n\n void updateService2(Map configs, String version);\n\n void updateCanary2(Map configs, String version);\n\n Map getConfigs();\n\n List availableConfigNames();\n\n default void healthz() {\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/ConfigNotifier.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.config.ChangeItem;\nimport com.megaease.easeagent.plugin.api.config.ConfigChangeListener;\n\nimport java.util.List;\nimport java.util.concurrent.CopyOnWriteArrayList;\nimport java.util.stream.Collectors;\n\npublic class ConfigNotifier {\n private static final Logger LOGGER = LoggerFactory.getLogger(ConfigNotifier.class);\n private final CopyOnWriteArrayList listeners = new CopyOnWriteArrayList<>();\n private final String prefix;\n\n public ConfigNotifier(String prefix) {\n this.prefix = prefix;\n }\n\n public Runnable addChangeListener(ConfigChangeListener listener) {\n final boolean add = listeners.add(listener);\n return () -> {\n if (add) {\n listeners.remove(listener);\n }\n };\n }\n\n public void handleChanges(List list) {\n final List changes = this.prefix.isEmpty() ? list : filterChanges(list);\n if (changes.isEmpty()) {\n return;\n }\n listeners.forEach(one -> {\n try {\n one.onChange(changes);\n } catch (Exception e) {\n LOGGER.warn(\"Notify config changes to listener failure: {}\", e);\n }\n });\n }\n\n private List filterChanges(List list) {\n return list.stream().filter(one -> one.getFullName().startsWith(prefix))\n .map(e -> new ChangeItem(e.getFullName().substring(prefix.length()), e.getFullName(), e.getOldValue(), e.getNewValue()))\n .collect(Collectors.toList());\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/ConfigPropertiesUtils.java", "content": "/*\n * Copyright (c) 2022, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.utils.SystemEnv;\n\nimport javax.annotation.Nullable;\nimport java.util.Locale;\n\n/**\n * Get config from system properties or environment variables.\n */\nfinal class ConfigPropertiesUtils {\n\n public static boolean getBoolean(String propertyName, boolean defaultValue) {\n String strValue = getString(propertyName);\n return strValue == null ? defaultValue : Boolean.parseBoolean(strValue);\n }\n\n public static int getInt(String propertyName, int defaultValue) {\n String strValue = getString(propertyName);\n if (strValue == null) {\n return defaultValue;\n }\n try {\n return Integer.parseInt(strValue);\n } catch (NumberFormatException ignored) {\n return defaultValue;\n }\n }\n\n @Nullable\n public static String getString(String propertyName) {\n String value = System.getProperty(propertyName);\n if (value != null) {\n return value;\n }\n return SystemEnv.get(toEnvVarName(propertyName));\n }\n\n /**\n * dot.case -> UPPER_UNDERSCORE\n */\n public static String toEnvVarName(String propertyName) {\n return propertyName.toUpperCase(Locale.ROOT).replace('-', '_').replace('.', '_');\n }\n\n private ConfigPropertiesUtils() {\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/ConfigUtils.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.fasterxml.jackson.databind.JsonNode;\nimport com.fasterxml.jackson.databind.ObjectMapper;\nimport com.megaease.easeagent.plugin.api.config.ChangeItem;\nimport com.megaease.easeagent.plugin.api.config.Config;\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\n\nimport java.io.IOException;\nimport java.util.*;\nimport java.util.function.BiFunction;\nimport java.util.function.Consumer;\nimport java.util.stream.Collectors;\n\nimport static com.megaease.easeagent.plugin.api.config.ConfigConst.*;\n\npublic class ConfigUtils {\n private ConfigUtils() {\n }\n\n public static void bindProp(String name, Config configs, BiFunction func, Consumer consumer, R def) {\n Runnable process = () -> {\n R result = func.apply(configs, name);\n result = firstNotNull(result, def);\n if (result != null) {\n consumer.accept(result);\n }\n };\n process.run();\n configs.addChangeListener(list -> {\n boolean hasChange = list.stream().map(ChangeItem::getFullName).anyMatch(fn -> fn.equals(name));\n if (hasChange) {\n process.run();\n }\n });\n }\n\n @SafeVarargs\n private static R firstNotNull(R... ars) {\n for (R one : ars) {\n if (one != null) {\n return one;\n }\n }\n return null;\n }\n\n public static void bindProp(String name, Config configs, BiFunction func, Consumer consumer) {\n bindProp(name, configs, func, consumer, null);\n }\n\n public static Map json2KVMap(String json) throws IOException {\n ObjectMapper mapper = new ObjectMapper();\n JsonNode node = mapper.readTree(json);\n List> list = extractKVs(null, node);\n return list.stream().collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));\n }\n\n public static List> extractKVs(String prefix, JsonNode node) {\n List> rst = new LinkedList<>();\n if (node.isObject()) {\n Iterator names = node.fieldNames();\n while (names.hasNext()) {\n String current = names.next();\n rst.addAll(extractKVs(join(prefix, current), node.path(current)));\n }\n } else if (node.isArray()) {\n int len = node.size();\n for (int i = 0; i < len; i++) {\n rst.addAll(extractKVs(join(prefix, i + \"\"), node.path(i)));\n }\n } else {\n rst.add(new AbstractMap.SimpleEntry<>(prefix, node.asText(\"\")));\n }\n return rst;\n }\n\n private static String join(String prefix, String current) {\n return prefix == null ? current : ConfigConst.join(prefix, current);\n }\n\n public static boolean isGlobal(String namespace) {\n return PLUGIN_GLOBAL.equals(namespace);\n }\n\n public static boolean isPluginConfig(String key) {\n return key != null && key.startsWith(PLUGIN_PREFIX);\n }\n\n public static boolean isPluginConfig(String key, String domain, String namespace, String id) {\n return key != null && key.startsWith(ConfigConst.join(PLUGIN, domain, namespace, id));\n }\n\n public static PluginProperty pluginProperty(String path) {\n String[] configs = path.split(\"\\\\\" + DELIMITER);\n if (configs.length < 5) {\n throw new ValidateUtils.ValidException(String.format(\"Property[%s] must be format: %s\", path, ConfigConst.join(PLUGIN, \"\", \"\", \"\", \"\")));\n }\n\n for (int idOffsetEnd = 3; idOffsetEnd < configs.length - 1; idOffsetEnd++) {\n new PluginProperty(configs[1], configs[2],\n ConfigConst.join(Arrays.copyOfRange(configs, 3, idOffsetEnd)),\n ConfigConst.join(Arrays.copyOfRange(configs, idOffsetEnd + 1, configs.length)));\n }\n\n return new PluginProperty(configs[1], configs[2], configs[3], ConfigConst.join(Arrays.copyOfRange(configs, 4, configs.length)));\n }\n\n\n public static String requireNonEmpty(String obj, String message) {\n if (obj == null || obj.trim().isEmpty()) {\n throw new ValidateUtils.ValidException(message);\n }\n return obj.trim();\n }\n\n public static String buildPluginProperty(String domain, String namespace, String id, String property) {\n return String.format(PLUGIN_FORMAT, domain, namespace, id, property);\n }\n\n public static String buildCodeVersionKey(String key) {\n return RUNTIME_CODE_VERSION_POINTS_PREFIX + key;\n }\n\n /**\n * extract config item with a fromPrefix to and convert the prefix to 'toPrefix' for configuration Compatibility\n *\n * @param cfg config source map\n * @param fromPrefix from\n * @param toPrefix to\n * @return Extracted and converted KV map\n */\n public static Map extractAndConvertPrefix(Map cfg, String fromPrefix, String toPrefix) {\n Map convert = new HashMap<>();\n\n Set keys = new HashSet<>();\n cfg.forEach((key, value) -> {\n if (key.startsWith(fromPrefix)) {\n keys.add(key);\n key = toPrefix + key.substring(fromPrefix.length());\n convert.put(key, value);\n }\n });\n\n // override, new configuration KV override previous KV\n convert.putAll(extractByPrefix(cfg, toPrefix));\n\n return convert;\n }\n\n /**\n * Extract config items from config by prefix\n *\n * @param config config\n * @param prefix prefix\n * @return Extracted KV\n */\n public static Map extractByPrefix(Config config, String prefix) {\n return extractByPrefix(config.getConfigs(), prefix);\n }\n\n public static Map extractByPrefix(Map cfg, String prefix) {\n Map extract = new TreeMap<>();\n\n // override, new configuration KV override previous KV\n cfg.forEach((key, value) -> {\n if (key.startsWith(prefix)) {\n extract.put(key, value);\n }\n });\n\n return extract;\n }\n\n public static int isChanged(String name, Map map, String check) {\n if (map.get(name) == null || map.get(name).equals(check)) {\n return 0;\n }\n return 1;\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/Configs.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.config.ChangeItem;\nimport com.megaease.easeagent.plugin.api.config.Config;\nimport com.megaease.easeagent.plugin.api.config.ConfigChangeListener;\n\nimport java.util.*;\nimport java.util.stream.Collectors;\n\npublic class Configs implements Config {\n private static final Logger LOGGER = LoggerFactory.getLogger(Configs.class);\n protected Map source;\n protected ConfigNotifier notifier;\n\n protected Configs() {\n }\n\n public Configs(Map source) {\n this.source = new TreeMap<>(source);\n notifier = new ConfigNotifier(\"\");\n }\n\n public void updateConfigsNotNotify(Map changes) {\n this.source.putAll(changes);\n }\n\n public void updateConfigs(Map changes) {\n Map dump = new TreeMap<>(this.source);\n List items = new LinkedList<>();\n changes.forEach((name, value) -> {\n String old = dump.get(name);\n if (!Objects.equals(old, value)) {\n dump.put(name, value);\n items.add(new ChangeItem(name, name, old, value));\n }\n });\n if (!items.isEmpty()) {\n LOGGER.info(\"change items: {}\", items);\n this.source = dump;\n this.notifier.handleChanges(items);\n }\n }\n\n protected boolean hasText(String text) {\n return text != null && text.trim().length() > 0;\n }\n\n @Override\n public Map getConfigs() {\n return new TreeMap<>(this.source);\n }\n\n public String toPrettyDisplay() {\n return this.source.toString();\n }\n\n public boolean hasPath(String path) {\n return this.source.containsKey(path);\n }\n\n\n public String getString(String name) {\n return this.source.get(name);\n }\n\n public String getString(String name, String defVal) {\n String val = this.source.get(name);\n\n return val == null ? defVal : val;\n }\n\n public Integer getInt(String name) {\n String value = this.source.get(name);\n if (value == null) {\n return null;\n }\n try {\n return Integer.parseInt(value);\n } catch (Exception e) {\n return null;\n }\n }\n\n @Override\n public Integer getInt(String name, int defValue) {\n Integer anInt = getInt(name);\n if (anInt == null) {\n return defValue;\n }\n return anInt;\n }\n\n public Boolean getBooleanNullForUnset(String name) {\n String value = this.source.get(name);\n if (value == null) {\n return null;\n }\n return value.equalsIgnoreCase(\"yes\") || value.equalsIgnoreCase(\"true\");\n }\n\n public Boolean getBoolean(String name) {\n String value = this.source.get(name);\n if (value == null) {\n return false;\n }\n return value.equalsIgnoreCase(\"yes\") || value.equalsIgnoreCase(\"true\");\n }\n\n @Override\n public Boolean getBoolean(String name, boolean defValue) {\n Boolean aBoolean = getBooleanNullForUnset(name);\n if (aBoolean == null) {\n return defValue;\n }\n return aBoolean;\n }\n\n public Double getDouble(String name) {\n String value = this.source.get(name);\n if (value == null) {\n return null;\n }\n try {\n return Double.parseDouble(value);\n } catch (Exception e) {\n return null;\n }\n }\n\n @Override\n public Double getDouble(String name, double defValue) {\n Double aDouble = getDouble(name);\n if (aDouble == null) {\n return defValue;\n }\n return aDouble;\n }\n\n public Long getLong(String name) {\n String value = this.source.get(name);\n if (value == null) {\n return null;\n }\n try {\n return Long.parseLong(value);\n } catch (Exception e) {\n return null;\n }\n }\n\n @Override\n public Long getLong(String name, long defValue) {\n Long aLong = getLong(name);\n if (aLong == null) {\n return defValue;\n }\n return aLong;\n }\n\n public List getStringList(String name) {\n String value = this.source.get(name);\n if (value == null) {\n return Collections.emptyList();\n }\n return Arrays.stream(value.split(\",\")).filter(Objects::nonNull).collect(Collectors.toList());\n }\n\n @Override\n public Runnable addChangeListener(ConfigChangeListener listener) {\n return notifier.addChangeListener(listener);\n }\n\n @Override\n public Set keySet() {\n return this.source.keySet();\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/GlobalConfigs.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.config.report.ReportConfigAdapter;\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\n\nimport java.io.IOException;\nimport java.util.HashMap;\nimport java.util.List;\nimport java.util.Map;\nimport java.util.TreeMap;\n\npublic class GlobalConfigs extends Configs implements ConfigManagerMXBean {\n Configs originalConfig;\n\n public GlobalConfigs(Map source) {\n super();\n this.originalConfig = new Configs(source);\n // reporter adapter\n Map map = new TreeMap<>(source);\n ReportConfigAdapter.convertConfig(map);\n // check environment config\n this.source = new TreeMap<>(map);\n this.notifier = new ConfigNotifier(\"\");\n }\n\n public Configs getOriginalConfig() {\n return this.originalConfig;\n }\n\n @Override\n public void updateConfigsNotNotify(Map changes) {\n // update original config\n Map newGlobalCfg = new TreeMap<>(this.originalConfig.getConfigs());\n newGlobalCfg.putAll(changes);\n this.originalConfig.updateConfigsNotNotify(changes);\n\n // report adapter\n ReportConfigAdapter.convertConfig(newGlobalCfg);\n\n super.updateConfigsNotNotify(newGlobalCfg);\n }\n\n @Override\n public void updateConfigs(Map changes) {\n // update original config\n Map newGlobalCfg = new TreeMap<>(this.originalConfig.getConfigs());\n newGlobalCfg.putAll(changes);\n this.originalConfig.updateConfigsNotNotify(changes);\n\n // report adapter\n ReportConfigAdapter.convertConfig(newGlobalCfg);\n\n super.updateConfigs(newGlobalCfg);\n }\n\n public void mergeConfigs(GlobalConfigs configs) {\n Map merged = configs.getOriginalConfig().getConfigs();\n if (merged.isEmpty()) {\n return;\n }\n this.updateConfigsNotNotify(merged);\n return;\n }\n\n @Override\n public List availableConfigNames() {\n throw new UnsupportedOperationException();\n }\n\n @Override\n public void updateService(String json, String version) throws IOException {\n this.updateConfigs(ConfigUtils.json2KVMap(json));\n }\n\n @Override\n public void updateCanary(String json, String version) throws IOException {\n Map originals = ConfigUtils.json2KVMap(json);\n HashMap rst = new HashMap<>();\n originals.forEach((k, v) -> rst.put(ConfigConst.join(ConfigConst.GLOBAL_CANARY_LABELS, k), v));\n this.updateConfigs(rst);\n }\n\n @Override\n public void updateService2(Map configs, String version) {\n this.updateConfigs(configs);\n }\n\n @Override\n public void updateCanary2(Map configs, String version) {\n HashMap rst = new HashMap<>();\n for (Map.Entry entry : configs.entrySet()) {\n String k = entry.getKey();\n String v = entry.getValue();\n rst.put(ConfigConst.join(ConfigConst.GLOBAL_CANARY_LABELS, k), v);\n }\n this.updateConfigs(CompatibilityConversion.transform(rst));\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/JarFileConfigLoader.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\n\nimport java.io.File;\nimport java.io.IOException;\nimport java.io.InputStream;\nimport java.util.jar.JarFile;\nimport java.util.zip.ZipEntry;\n\npublic class JarFileConfigLoader {\n private static final Logger LOGGER = LoggerFactory.getLogger(JarFileConfigLoader.class);\n\n static GlobalConfigs load(String file) {\n String agentJarPath = System.getProperty(ConfigConst.AGENT_JAR_PATH);\n if (agentJarPath == null) {\n return null;\n }\n try {\n JarFile jarFile = new JarFile(new File(agentJarPath));\n ZipEntry zipEntry = jarFile.getEntry(file);\n if (zipEntry == null) {\n return null;\n }\n try (InputStream in = jarFile.getInputStream(zipEntry)) {\n return ConfigLoader.loadFromStream(in, file);\n } catch (IOException e) {\n LOGGER.debug(\"Load config file:{} failure: {}\", file, e);\n }\n } catch (IOException e) {\n LOGGER.debug(\"create JarFile:{} failure: {}\", agentJarPath, e);\n }\n return null;\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/OtelSdkConfigs.java", "content": "/*\n * Copyright (c) 2022, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.google.common.base.Splitter;\nimport com.megaease.easeagent.plugin.utils.ImmutableMap;\nimport com.megaease.easeagent.plugin.utils.SystemEnv;\nimport com.megaease.easeagent.plugin.utils.common.StringUtils;\n\nimport java.util.HashMap;\nimport java.util.Map;\nimport java.util.TreeMap;\n\n\n/**\n * Compatible with opentelemetry-java.\n *

\n * {@see https://github.com/open-telemetry/opentelemetry-java/blob/main/sdk-extensions/autoconfigure/README.md#disabling-opentelemetrysdk}\n */\npublic class OtelSdkConfigs {\n private static final String OTEL_RESOURCE_ATTRIBUTES_KEY = \"otel.resource.attributes\";\n\n private static final String CONFIG_PATH_PROP_KEY = \"otel.javaagent.configuration-file\";\n\n private static final Splitter.MapSplitter OTEL_RESOURCE_ATTRIBUTES_SPLITTER\n = Splitter.on(\",\")\n .omitEmptyStrings()\n .withKeyValueSeparator(\"=\");\n\n private static final Map SDK_ATTRIBUTES_TO_EASE_AGENT_PROPS =\n ImmutableMap.builder()\n .put(\"sdk.disabled\", \"easeagent.server.enabled\")\n .put(\"service.name\", \"name\") //\"easeagent.name\"\n .put(\"service.namespace\", \"system\") //\"easeagent.system\"\n .build();\n\n // -Dotel.service.name=xxx\n private static final Map OTEL_SDK_PROPS_TO_EASE_AGENT_PROPS = new HashMap<>();\n\n // OTEL_SERVICE_NAME=xxx\n private static final Map OTEL_SDK_ENV_VAR_TO_EASE_AGENT_PROPS = new HashMap<>();\n\n static {\n for (Map.Entry entry : SDK_ATTRIBUTES_TO_EASE_AGENT_PROPS.entrySet()) {\n // lower.hyphen -> UPPER_UNDERSCORE\n OTEL_SDK_PROPS_TO_EASE_AGENT_PROPS.put(\n \"otel.\" + entry.getKey(),\n entry.getValue()\n );\n }\n\n for (Map.Entry entry : OTEL_SDK_PROPS_TO_EASE_AGENT_PROPS.entrySet()) {\n // dot.case -> UPPER_UNDERSCORE\n OTEL_SDK_ENV_VAR_TO_EASE_AGENT_PROPS.put(\n ConfigPropertiesUtils.toEnvVarName(entry.getKey()),\n entry.getValue()\n );\n }\n }\n\n /**\n * Get config path from java properties or environment variables\n */\n static String getConfigPath() {\n return ConfigPropertiesUtils.getString(CONFIG_PATH_PROP_KEY);\n }\n\n\n /**\n * update config value from environment variables and java properties\n *

\n * java properties > environment variables > OTEL_RESOURCE_ATTRIBUTES\n */\n static Map updateEnvCfg() {\n Map envCfg = new TreeMap<>();\n\n String configEnv = ConfigPropertiesUtils.getString(OTEL_RESOURCE_ATTRIBUTES_KEY);\n if (StringUtils.isNotEmpty(configEnv)) {\n Map map = OTEL_RESOURCE_ATTRIBUTES_SPLITTER.split(configEnv);\n if (!map.isEmpty()) {\n for (Map.Entry entry : SDK_ATTRIBUTES_TO_EASE_AGENT_PROPS.entrySet()) {\n String value = map.get(entry.getKey());\n if (!StringUtils.isEmpty(value)) {\n envCfg.put(entry.getValue(), value);\n }\n }\n }\n }\n\n // override by environment variables, eg: export OTEL_SERVICE_NAME=xxx\n for (Map.Entry entry : OTEL_SDK_ENV_VAR_TO_EASE_AGENT_PROPS.entrySet()) {\n String value = SystemEnv.get(entry.getKey());\n if (!StringUtils.isEmpty(value)) {\n envCfg.put(entry.getValue(), value);\n }\n }\n\n // override by java properties; eg: java -Dotel.service.name=xxx\n for (Map.Entry entry : OTEL_SDK_PROPS_TO_EASE_AGENT_PROPS.entrySet()) {\n String value = System.getProperty(entry.getKey());\n if (!StringUtils.isEmpty(value)) {\n envCfg.put(entry.getValue(), value);\n }\n }\n\n return envCfg;\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/PluginConfig.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.config.Const;\nimport com.megaease.easeagent.plugin.api.config.IPluginConfig;\nimport com.megaease.easeagent.plugin.api.config.PluginConfigChangeListener;\nimport com.megaease.easeagent.plugin.utils.common.StringUtils;\n\nimport javax.annotation.Nonnull;\nimport java.util.*;\nimport java.util.function.Consumer;\nimport java.util.stream.Collectors;\n\npublic class PluginConfig implements IPluginConfig {\n private static final Logger LOGGER = LoggerFactory.getLogger(PluginConfig.class);\n private final Set listeners;\n private final String domain;\n private final String namespace;\n private final String id;\n private final Map global;\n private final Map cover;\n private final boolean enabled;\n\n protected PluginConfig(@Nonnull String domain, @Nonnull String id, @Nonnull Map global, @Nonnull String namespace, @Nonnull Map cover, @Nonnull Set listeners) {\n this.domain = domain;\n this.namespace = namespace;\n this.id = id;\n this.global = global;\n this.cover = cover;\n this.listeners = listeners;\n Boolean b = getBoolean(Const.ENABLED_CONFIG);\n if (b == null) {\n enabled = false;\n } else {\n enabled = b;\n }\n }\n\n public static PluginConfig build(@Nonnull String domain, @Nonnull String id,\n @Nonnull Map global, @Nonnull String namespace,\n @Nonnull Map cover, PluginConfig oldConfig) {\n Set listeners;\n if (oldConfig == null) {\n listeners = new HashSet<>();\n } else {\n listeners = oldConfig.listeners;\n }\n return new PluginConfig(domain, id, global, namespace, cover, listeners);\n }\n\n @Override\n public String domain() {\n return domain;\n }\n\n @Override\n public String namespace() {\n return namespace;\n }\n\n @Override\n public String id() {\n return id;\n }\n\n\n @Override\n public boolean hasProperty(String property) {\n return global.containsKey(property) || cover.containsKey(property);\n }\n\n @Override\n public String getString(String property) {\n String value = cover.get(property);\n if (value != null) {\n return value;\n }\n return global.get(property);\n }\n\n\n @Override\n public Integer getInt(String property) {\n String value = this.getString(property);\n if (value == null) {\n return null;\n }\n try {\n return Integer.parseInt(value);\n } catch (Exception e) {\n return null;\n }\n }\n\n private boolean isTrue(String value) {\n return value.equalsIgnoreCase(\"yes\") || value.equalsIgnoreCase(\"true\");\n }\n\n @Override\n public Boolean getBoolean(String property) {\n String value = cover.get(property);\n boolean implB = true;\n if (value != null) {\n implB = isTrue(value);\n }\n value = global.get(property);\n boolean globalB = false;\n if (value != null) {\n globalB = isTrue(value);\n }\n return implB && globalB;\n }\n\n @Override\n public boolean enabled() {\n return enabled;\n }\n\n @Override\n public Double getDouble(String property) {\n String value = this.getString(property);\n if (value == null) {\n return null;\n }\n try {\n return Double.parseDouble(value);\n } catch (Exception e) {\n return null;\n }\n }\n\n @Override\n public Long getLong(String property) {\n String value = this.getString(property);\n if (value == null) {\n return null;\n }\n try {\n return Long.parseLong(value);\n } catch (Exception e) {\n return null;\n }\n }\n\n @Override\n public List getStringList(String property) {\n String value = this.getString(property);\n if (StringUtils.isEmpty(value)) {\n return Collections.emptyList();\n }\n return Arrays.stream(value.split(\",\"))\n .map(String::trim)\n .collect(Collectors.toList());\n }\n\n @Override\n public IPluginConfig getGlobal() {\n return new Global(domain, id, global, namespace);\n }\n\n @Override\n public Set keySet() {\n Set keys = new HashSet<>(global.keySet());\n keys.addAll(cover.keySet());\n return keys;\n }\n\n @Override\n public void addChangeListener(PluginConfigChangeListener listener) {\n synchronized (listeners) {\n listeners.add(listener);\n }\n }\n\n public void foreachConfigChangeListener(Consumer action) {\n Set oldListeners;\n synchronized (listeners) {\n oldListeners = new HashSet<>(listeners);\n }\n for (PluginConfigChangeListener oldListener : oldListeners) {\n try {\n action.accept(oldListener);\n } catch (Exception e) {\n LOGGER.error(\"PluginConfigChangeListener<{}> change plugin config fail : {}\", oldListener.getClass(), e.getMessage());\n }\n }\n }\n\n public class Global extends PluginConfig implements IPluginConfig {\n\n public Global(String domain, String id, Map global, String namespace) {\n super(domain, id, global, namespace, Collections.emptyMap(), Collections.emptySet());\n }\n\n @Override\n public void addChangeListener(PluginConfigChangeListener listener) {\n PluginConfig.this.addChangeListener(listener);\n }\n\n @Override\n public void foreachConfigChangeListener(Consumer action) {\n PluginConfig.this.foreachConfigChangeListener(action);\n }\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/PluginConfigManager.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.config.ChangeItem;\nimport com.megaease.easeagent.plugin.api.config.Config;\nimport com.megaease.easeagent.plugin.api.config.ConfigChangeListener;\nimport com.megaease.easeagent.plugin.api.config.IConfigFactory;\n\nimport java.util.*;\n\nimport static com.megaease.easeagent.plugin.api.config.ConfigConst.PLUGIN_GLOBAL;\n\npublic class PluginConfigManager implements IConfigFactory {\n private static final Logger LOGGER = LoggerFactory.getLogger(PluginConfigManager.class);\n private Runnable shutdownRunnable;\n private final Configs configs;\n private final Map pluginSourceConfigs;\n private final Map pluginConfigs;\n\n private PluginConfigManager(Configs configs, Map pluginSourceConfigs, Map pluginConfigs) {\n this.configs = Objects.requireNonNull(configs, \"configs must not be null.\");\n this.pluginSourceConfigs = Objects.requireNonNull(pluginSourceConfigs, \"pluginSourceConfigs must not be null.\");\n this.pluginConfigs = Objects.requireNonNull(pluginConfigs, \"pluginConfigs must not be null.\");\n }\n\n public static PluginConfigManager.Builder builder(Configs configs) {\n PluginConfigManager pluginConfigManager = new PluginConfigManager(configs, new HashMap<>(), new HashMap<>());\n return pluginConfigManager.new Builder();\n }\n\n @Override\n public Config getConfig() {\n return this.configs;\n }\n\n @Override\n public String getConfig(String property) {\n return configs.getString(property);\n }\n\n @Override\n public String getConfig(String property, String defaultValue) {\n return configs.getString(property, defaultValue);\n }\n\n public PluginConfig getConfig(String domain, String namespace, String id) {\n return getConfig(domain, namespace, id, null);\n }\n\n public synchronized PluginConfig getConfig(String domain, String namespace, String id, PluginConfig oldConfig) {\n Key key = new Key(domain, namespace, id);\n PluginConfig pluginConfig = pluginConfigs.get(key);\n if (pluginConfig != null) {\n return pluginConfig;\n }\n Map globalConfig = getGlobalConfig(domain, id);\n Map coverConfig = getCoverConfig(domain, namespace, id);\n PluginConfig newPluginConfig = PluginConfig.build(domain, id, globalConfig, namespace, coverConfig, oldConfig);\n pluginConfigs.put(key, newPluginConfig);\n return newPluginConfig;\n }\n\n private Map getGlobalConfig(String domain, String id) {\n return getConfigSource(domain, PLUGIN_GLOBAL, id);\n }\n\n private Map getCoverConfig(String domain, String namespace, String id) {\n return getConfigSource(domain, namespace, id);\n }\n\n private Map getConfigSource(String domain, String namespace, String id) {\n PluginSourceConfig sourceConfig = pluginSourceConfigs.get(new Key(domain, namespace, id));\n if (sourceConfig == null) {\n return Collections.emptyMap();\n }\n return sourceConfig.getProperties();\n }\n\n\n private Set keys(Set keys) {\n Set propertyKeys = new HashSet<>();\n for (String k : keys) {\n if (!ConfigUtils.isPluginConfig(k)) {\n continue;\n }\n PluginProperty property = ConfigUtils.pluginProperty(k);\n Key key = new Key(property.getDomain(), property.getNamespace(), property.getId());\n propertyKeys.add(key);\n }\n return propertyKeys;\n }\n\n\n public void shutdown() {\n shutdownRunnable.run();\n }\n\n protected synchronized void onChange(Map sources) {\n Set sourceKeys = keys(sources.keySet());\n Map newSources = buildNewSources(sourceKeys, sources);\n for (Key sourceKey : sourceKeys) {\n pluginSourceConfigs.put(sourceKey, PluginSourceConfig.build(sourceKey.getDomain(), sourceKey.getNamespace(), sourceKey.getId(), newSources));\n }\n Set changeKeys = buildChangeKeys(sourceKeys);\n\n for (Key changeKey : changeKeys) {\n final PluginConfig oldConfig = pluginConfigs.remove(changeKey);\n final PluginConfig newConfig = getConfig(changeKey.getDomain(), changeKey.getNamespace(), changeKey.id, oldConfig);\n if (oldConfig == null) {\n continue;\n }\n try {\n oldConfig.foreachConfigChangeListener(listener -> listener.onChange(oldConfig, newConfig));\n } catch (Exception e) {\n LOGGER.warn(\"change config<{}> fail: {}\", changeKey.toString(), e.getMessage());\n }\n }\n }\n\n private Map buildNewSources(Set sourceKeys, Map sources) {\n Map newSources = new HashMap<>();\n for (Key sourceKey : sourceKeys) {\n PluginSourceConfig pluginSourceConfig = pluginSourceConfigs.get(sourceKey);\n if (pluginSourceConfig == null) {\n continue;\n }\n newSources.putAll(pluginSourceConfig.getSource());\n }\n newSources.putAll(sources);\n return newSources;\n }\n\n private Set buildChangeKeys(Set sourceKeys) {\n Set changeKeys = new HashSet<>(sourceKeys);\n for (Key key : sourceKeys) {\n if (!ConfigUtils.isGlobal(key.getNamespace())) {\n continue;\n }\n for (Key oldKey : pluginConfigs.keySet()) {\n if (!key.id.equals(oldKey.id)) {\n continue;\n }\n changeKeys.add(oldKey);\n }\n }\n return changeKeys;\n }\n\n class Key {\n private final String domain;\n private final String namespace;\n private final String id;\n\n public Key(String domain, String namespace, String id) {\n this.domain = domain;\n this.namespace = namespace;\n this.id = id;\n }\n\n public String getDomain() {\n return domain;\n }\n\n public String getNamespace() {\n return namespace;\n }\n\n public String getId() {\n return id;\n }\n\n @Override\n public boolean equals(Object o) {\n if (this == o) return true;\n if (o == null || getClass() != o.getClass()) return false;\n Key key = (Key) o;\n return Objects.equals(domain, key.domain) &&\n Objects.equals(namespace, key.namespace) &&\n Objects.equals(id, key.id);\n }\n\n @Override\n public int hashCode() {\n\n return Objects.hash(domain, namespace, id);\n }\n\n @Override\n public String toString() {\n return \"Key{\" +\n \"domain='\" + domain + '\\'' +\n \", namespace='\" + namespace + '\\'' +\n \", id='\" + id + '\\'' +\n '}';\n }\n }\n\n public class Builder {\n public PluginConfigManager build() {\n synchronized (PluginConfigManager.this) {\n Map sources = configs.getConfigs();\n Set sourceKeys = keys(sources.keySet());\n for (Key sourceKey : sourceKeys) {\n pluginSourceConfigs.put(sourceKey, PluginSourceConfig.build(sourceKey.getDomain(), sourceKey.getNamespace(), sourceKey.getId(), sources));\n }\n for (Key key : pluginSourceConfigs.keySet()) {\n getConfig(key.getDomain(), key.getNamespace(), key.getId());\n }\n shutdownRunnable = configs.addChangeListener(new ChangeListener());\n }\n return PluginConfigManager.this;\n }\n }\n\n class ChangeListener implements ConfigChangeListener {\n\n @Override\n public void onChange(List list) {\n Map sources = new HashMap<>();\n for (ChangeItem changeItem : list) {\n sources.put(changeItem.getFullName(), changeItem.getNewValue());\n }\n PluginConfigManager.this.onChange(sources);\n }\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/PluginProperty.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport java.util.Objects;\n\npublic class PluginProperty {\n private final String domain;\n private final String namespace;\n private final String id;\n private final String property;\n\n public PluginProperty(String domain, String namespace, String id, String property) {\n this.domain = domain;\n this.namespace = namespace;\n this.id = id;\n this.property = property;\n }\n\n public String getDomain() {\n return domain;\n }\n\n public String getNamespace() {\n return namespace;\n }\n\n public String getId() {\n return id;\n }\n\n public String getProperty() {\n return property;\n }\n\n @Override\n public boolean equals(Object o) {\n if (this == o) return true;\n if (o == null || getClass() != o.getClass()) return false;\n PluginProperty that = (PluginProperty) o;\n return Objects.equals(domain, that.domain) &&\n Objects.equals(namespace, that.namespace) &&\n Objects.equals(id, that.id) &&\n Objects.equals(property, that.property);\n }\n\n @Override\n public int hashCode() {\n\n return Objects.hash(domain, namespace, id, property);\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/PluginSourceConfig.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport java.util.HashMap;\nimport java.util.Map;\nimport java.util.Objects;\n\npublic class PluginSourceConfig {\n private final String domain;\n private final String namespace;\n private final String id;\n private final Map source;\n private final Map properties;\n\n public PluginSourceConfig(String domain, String namespace, String id, Map source, Map properties) {\n this.domain = Objects.requireNonNull(domain, \"domain must not be null.\");\n this.namespace = Objects.requireNonNull(namespace, \"namespace must not be null.\");\n this.id = Objects.requireNonNull(id, \"id must not be null.\");\n this.source = Objects.requireNonNull(source, \"source must not be null.\");\n this.properties = Objects.requireNonNull(properties, \"properties must not be null.\");\n }\n\n public static PluginSourceConfig build(String domain, String namespace, String id, Map source) {\n Map pluginSource = new HashMap<>();\n Map properties = new HashMap<>();\n for (Map.Entry sourceEntry : source.entrySet()) {\n String key = sourceEntry.getKey();\n if (!ConfigUtils.isPluginConfig(key, domain, namespace, id)) {\n continue;\n }\n pluginSource.put(key, sourceEntry.getValue());\n PluginProperty property = ConfigUtils.pluginProperty(key);\n properties.put(property, sourceEntry.getValue());\n }\n return new PluginSourceConfig(domain, namespace, id, pluginSource, properties);\n }\n\n public Map getSource() {\n return source;\n }\n\n public String getDomain() {\n return domain;\n }\n\n public String getNamespace() {\n return namespace;\n }\n\n public String getId() {\n return id;\n }\n\n public Map getProperties() {\n Map result = new HashMap<>();\n for (Map.Entry propertyEntry : properties.entrySet()) {\n result.put(propertyEntry.getKey().getProperty(), propertyEntry.getValue());\n }\n return result;\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/ValidateUtils.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\npublic class ValidateUtils {\n public static class ValidException extends RuntimeException {\n public ValidException(String message) {\n super(message);\n }\n }\n\n public interface Validator {\n void validate(String name, String value);\n }\n\n public static void validate(Configs configs, String name, Validator... vs){\n String value = configs.getString(name);\n for (Validator one : vs) {\n one.validate(name, value);\n }\n }\n\n public static final Validator HasText = (name, value) -> {\n if (value == null || value.trim().length() == 0) {\n throw new ValidException(String.format(\"Property[%s] has no non-empty value\", name));\n }\n };\n\n public static final Validator Bool = (name, value) -> {\n String upper = value.toUpperCase();\n if (upper.equals(\"TRUE\") || upper.equals(\"FALSE\")) {\n return;\n }\n throw new ValidException(String.format(\"Property[%s] has no boolean value\", name));\n };\n\n public static final Validator NumberInt = (name, value) -> {\n try {\n Integer.parseInt(value.trim());\n } catch (Exception e) {\n throw new ValidException(String.format(\"Property[%s] has no integer value\", name));\n }\n };\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/WrappedConfigManager.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.async.ThreadUtils;\n\nimport java.io.IOException;\nimport java.util.List;\nimport java.util.Map;\n\npublic class WrappedConfigManager implements ConfigManagerMXBean {\n private final ClassLoader customClassLoader;\n private final ConfigManagerMXBean conf;\n\n public WrappedConfigManager(ClassLoader customClassLoader, ConfigManagerMXBean config) {\n this.customClassLoader = customClassLoader;\n this.conf = config;\n }\n\n @Override\n public void updateConfigs(Map configs) {\n ThreadUtils.callWithClassLoader(customClassLoader, () -> {\n conf.updateConfigs(configs);\n return null;\n });\n }\n\n @Override\n public void updateService(String json, String version) throws IOException {\n try {\n ThreadUtils.callWithClassLoader(customClassLoader, () -> {\n try {\n conf.updateService(json, version);\n } catch (IOException e) {\n throw new RuntimeException(e);\n }\n return null;\n });\n } catch (RuntimeException e) {\n if (e.getCause() instanceof IOException) {\n throw (IOException) e.getCause();\n }\n throw e;\n }\n }\n\n @Override\n public void updateCanary(String json, String version) throws IOException {\n try {\n ThreadUtils.callWithClassLoader(customClassLoader, () -> {\n try {\n conf.updateCanary(json, version);\n } catch (IOException e) {\n throw new RuntimeException(e);\n }\n return null;\n });\n } catch (RuntimeException e) {\n if (e.getCause() instanceof IOException) {\n throw (IOException) e.getCause();\n }\n throw e;\n }\n }\n\n @Override\n public void updateService2(Map configs, String version) {\n ThreadUtils.callWithClassLoader(customClassLoader, () -> {\n conf.updateService2(configs, version);\n return null;\n });\n }\n\n @Override\n public void updateCanary2(Map configs, String version) {\n ThreadUtils.callWithClassLoader(customClassLoader, () -> {\n conf.updateCanary2(configs, version);\n return null;\n });\n }\n\n @Override\n public Map getConfigs() {\n return ThreadUtils.callWithClassLoader(customClassLoader, conf::getConfigs);\n }\n\n @Override\n public List availableConfigNames() {\n return ThreadUtils.callWithClassLoader(customClassLoader, conf::availableConfigNames);\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/report/ReportConfigAdapter.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.config.report;\n\nimport com.megaease.easeagent.plugin.api.config.Config;\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\nimport com.megaease.easeagent.plugin.api.config.Const;\nimport com.megaease.easeagent.plugin.utils.NoNull;\nimport com.megaease.easeagent.plugin.utils.common.StringUtils;\nimport lombok.extern.slf4j.Slf4j;\n\nimport java.util.HashMap;\nimport java.util.HashSet;\nimport java.util.Map;\nimport java.util.TreeMap;\n\nimport static com.megaease.easeagent.config.ConfigUtils.extractAndConvertPrefix;\nimport static com.megaease.easeagent.config.ConfigUtils.extractByPrefix;\nimport static com.megaease.easeagent.config.report.ReportConfigConst.*;\n\n@Slf4j\npublic class ReportConfigAdapter {\n private ReportConfigAdapter() {}\n\n public static void convertConfig(Map config) {\n Map cfg = extractAndConvertReporterConfig(config);\n config.putAll(cfg);\n }\n\n public static Map extractReporterConfig(Config configs) {\n Map cfg = extractByPrefix(configs.getConfigs(), REPORT);\n\n // default config\n cfg.put(TRACE_ENCODER, NoNull.of(cfg.get(TRACE_ENCODER), SPAN_JSON_ENCODER_NAME));\n cfg.put(METRIC_ENCODER, NoNull.of(cfg.get(METRIC_ENCODER), METRIC_JSON_ENCODER_NAME));\n cfg.put(LOG_ENCODER, NoNull.of(cfg.get(LOG_ENCODER), LOG_DATA_JSON_ENCODER_NAME));\n cfg.put(LOG_ACCESS_ENCODER, NoNull.of(cfg.get(LOG_ACCESS_ENCODER), ACCESS_LOG_JSON_ENCODER_NAME));\n\n cfg.put(TRACE_SENDER_NAME, NoNull.of(cfg.get(TRACE_SENDER_NAME), getDefaultAppender(cfg)));\n cfg.put(METRIC_SENDER_NAME, NoNull.of(cfg.get(METRIC_SENDER_NAME), getDefaultAppender(cfg)));\n cfg.put(LOG_ACCESS_SENDER_NAME, NoNull.of(cfg.get(LOG_ACCESS_SENDER_NAME), getDefaultAppender(cfg)));\n cfg.put(LOG_SENDER_NAME, NoNull.of(cfg.get(LOG_SENDER_NAME), getDefaultAppender(cfg)));\n\n return cfg;\n }\n\n public static String getDefaultAppender(Map cfg) {\n String outputAppender = cfg.get(join(OUTPUT_SERVER_V2, APPEND_TYPE_KEY));\n\n if (StringUtils.isEmpty(outputAppender)) {\n return Const.DEFAULT_APPEND_TYPE;\n }\n\n return outputAppender;\n }\n\n private static Map extractAndConvertReporterConfig(Map srcConfig) {\n Map extract = extractTracingConfig(srcConfig);\n Map outputCfg = new TreeMap<>(extract);\n\n // metric config\n extract = extractMetricPluginConfig(srcConfig);\n outputCfg.putAll(extract);\n\n // log config\n extract = extractLogPluginConfig(srcConfig);\n outputCfg.putAll(extract);\n\n // if there are access log in metric\n updateAccessLogCfg(outputCfg);\n\n // all extract configuration will be overridden by config items start with \"report\" in srcConfig\n extract = extractByPrefix(srcConfig, REPORT);\n outputCfg.putAll(extract);\n\n return outputCfg;\n }\n\n\n /**\n * this can be deleted if there is not any v1 configuration needed to compatible with\n * convert v1 tracing config to v2\n */\n private static Map extractTracingConfig(Map srcCfg) {\n // outputServer config\n Map extract = extractAndConvertPrefix(srcCfg, OUTPUT_SERVER_V1, OUTPUT_SERVER_V2);\n Map outputCfg = new TreeMap<>(extract);\n\n // async output config\n extract = extractAndConvertPrefix(srcCfg, TRACE_OUTPUT_V1, TRACE_ASYNC);\n\n String target = srcCfg.get(join(TRACE_OUTPUT_V1, \"target\"));\n extract.remove(join(TRACE_ASYNC, \"target\"));\n\n if (!StringUtils.isEmpty(outputCfg.get(TRACE_SENDER_NAME))) {\n log.info(\"Reporter V2 config trace sender as: {}\", outputCfg.get(TRACE_SENDER_NAME));\n } else if (\"system\".equals(target)) {\n // check output servers\n if (StringUtils.hasText(outputCfg.get(BOOTSTRAP_SERVERS))) {\n outputCfg.put(TRACE_SENDER_NAME, KAFKA_SENDER_NAME);\n outputCfg.put(TRACE_SENDER_TOPIC_V2, extract.remove(join(TRACE_ASYNC, TOPIC_KEY)));\n } else {\n outputCfg.put(TRACE_SENDER_NAME, CONSOLE_SENDER_NAME);\n }\n } else if (\"zipkin\".equals(target)) {\n outputCfg.put(TRACE_SENDER_NAME, ZIPKIN_SENDER_NAME);\n String url = extract.remove(join(TRACE_ASYNC, \"target.zipkinUrl\"));\n if (StringUtils.isEmpty(url)) {\n outputCfg.put(TRACE_SENDER_NAME, CONSOLE_SENDER_NAME);\n } else {\n outputCfg.put(join(TRACE_SENDER, \"url\"), url);\n }\n } else if (!StringUtils.isEmpty(target)) {\n outputCfg.put(TRACE_SENDER_NAME, CONSOLE_SENDER_NAME);\n log.info(\"Unsupported output configuration item:{}={}\", TRACE_OUTPUT_TARGET_V1, target);\n }\n outputCfg.putAll(extract);\n\n return outputCfg;\n }\n\n /**\n * For Compatibility, call after metric config adapter\n *\n * extract 'reporter.metric.access.*' to 'reporter.log.access.*'\n */\n private static void updateAccessLogCfg(Map outputCfg) {\n // reporter.metric.access.*\n String prefix = join(METRIC_V2, ConfigConst.Namespace.ACCESS);\n Map metricAccess = extractByPrefix(outputCfg, prefix);\n Map accessLog = extractAndConvertPrefix(metricAccess, prefix, LOG_ACCESS);\n\n // access log use `kafka` sender\n if (METRIC_KAFKA_SENDER_NAME.equals(accessLog.get(LOG_ACCESS_SENDER_NAME))) {\n accessLog.put(LOG_ACCESS_SENDER_NAME, KAFKA_SENDER_NAME);\n }\n\n outputCfg.putAll(accessLog);\n }\n\n /**\n * metric report configuration\n *\n * extract `plugin.observability.global.metric.*` config items to reporter.metric.sender.*`\n *\n * extract `plugin.observability.[namespace].metric.*` config items\n * to reporter.metric.[namespace].sender.*`\n *\n * @param srcCfg source configuration map\n * @return metric reporter config start with 'reporter.metric.[namespace].sender'\n */\n private static Map extractMetricPluginConfig(Map srcCfg) {\n final String globalKey = \".\" + ConfigConst.PLUGIN_GLOBAL + \".\";\n final String prefix = join(ConfigConst.PLUGIN, ConfigConst.OBSERVABILITY);\n int metricKeyLength = ConfigConst.METRIC_SERVICE_ID.length();\n\n Map global = extractGlobalMetricConfig(srcCfg);\n HashSet namespaces = new HashSet<>();\n Map metricConfigs = new HashMap<>(global);\n\n for (Map.Entry e : srcCfg.entrySet()) {\n String key = e.getKey();\n if (!key.startsWith(prefix)) {\n continue;\n }\n int idx = key.indexOf(ConfigConst.METRIC_SERVICE_ID, prefix.length());\n if (idx < 0) {\n continue;\n }\n String namespaceWithSeparator = key.substring(prefix.length(), idx);\n String suffix = key.substring(idx + metricKeyLength + 1);\n String newKey;\n\n if (namespaceWithSeparator.equals(globalKey)) {\n continue;\n } else {\n if (!namespaces.contains(namespaceWithSeparator)) {\n namespaces.add(namespaceWithSeparator);\n Map d = extractAndConvertPrefix(global,\n METRIC_V2 + \".\", METRIC_V2 + namespaceWithSeparator);\n metricConfigs.putAll(d);\n }\n }\n\n if (suffix.startsWith(ENCODER_KEY) || suffix.startsWith(ASYNC_KEY)) {\n newKey = METRIC_V2 + namespaceWithSeparator + suffix;\n } else if (suffix.equals(INTERVAL_KEY)) {\n newKey = METRIC_V2 + namespaceWithSeparator + join(ASYNC_KEY, suffix);\n } else {\n newKey = METRIC_V2 + namespaceWithSeparator + join(SENDER_KEY, suffix);\n }\n\n if (newKey.endsWith(APPEND_TYPE_KEY) && e.getValue().equals(\"kafka\")) {\n metricConfigs.put(newKey, METRIC_KAFKA_SENDER_NAME);\n } else {\n metricConfigs.put(newKey, e.getValue());\n }\n }\n\n return metricConfigs;\n }\n\n private static Map extractGlobalMetricConfig(Map srcCfg) {\n final String prefix = join(ConfigConst.PLUGIN, ConfigConst.OBSERVABILITY,\n ConfigConst.PLUGIN_GLOBAL,\n ConfigConst.PluginID.METRIC);\n Map global = new TreeMap<>();\n Map extract = extractAndConvertPrefix(srcCfg, prefix, METRIC_SENDER);\n\n for (Map.Entry e : extract.entrySet()) {\n if (e.getKey().startsWith(ENCODER_KEY, METRIC_SENDER.length() + 1)) {\n global.put(join(METRIC_V2, e.getKey().substring(METRIC_SENDER.length() + 1)), e.getValue());\n } else if (e.getKey().endsWith(INTERVAL_KEY)) {\n global.put(join(METRIC_ASYNC, INTERVAL_KEY), e.getValue());\n } else if (e.getKey().endsWith(APPEND_TYPE_KEY) && e.getValue().equals(\"kafka\")) {\n global.put(e.getKey(), METRIC_KAFKA_SENDER_NAME);\n } else {\n global.put(e.getKey(), e.getValue());\n }\n }\n\n\n // global log level (async)\n global.putAll(extractByPrefix(srcCfg, METRIC_SENDER));\n global.putAll(extractByPrefix(srcCfg, METRIC_ASYNC));\n global.putAll(extractByPrefix(srcCfg, METRIC_ENCODER));\n\n return global;\n }\n\n /**\n * metric report configuration\n *\n * extract `plugin.observability.global.metric.*` config items to reporter.metric.sender.*`\n *\n * extract `plugin.observability.[namespace].metric.*` config items\n * to reporter.metric.[namespace].sender.*`\n *\n * @param srcCfg source configuration map\n * @return metric reporter config start with 'reporter.metric.[namespace].sender'\n */\n private static Map extractLogPluginConfig(Map srcCfg) {\n final String globalKey = \".\" + ConfigConst.PLUGIN_GLOBAL + \".\";\n final String prefix = join(ConfigConst.PLUGIN, ConfigConst.OBSERVABILITY);\n\n String typeKey = join(\"\", ConfigConst.PluginID.LOG, \"\");\n int typeKeyLength = ConfigConst.PluginID.LOG.length();\n\n final String reporterPrefix = LOGS;\n\n Map global = extractGlobalLogConfig(srcCfg);\n HashSet namespaces = new HashSet<>();\n Map outputConfigs = new TreeMap<>(global);\n\n for (Map.Entry e : srcCfg.entrySet()) {\n String key = e.getKey();\n if (!key.startsWith(prefix)) {\n continue;\n }\n int idx = key.indexOf(typeKey, prefix.length());\n if (idx < 0) {\n continue;\n } else {\n idx += 1;\n }\n String namespaceWithSeparator = key.substring(prefix.length(), idx);\n String suffix = key.substring(idx + typeKeyLength + 1);\n String newKey;\n\n if (namespaceWithSeparator.equals(globalKey)) {\n continue;\n } else {\n if (!namespaces.contains(namespaceWithSeparator)) {\n namespaces.add(namespaceWithSeparator);\n Map d = extractAndConvertPrefix(global,\n reporterPrefix + \".\", reporterPrefix + namespaceWithSeparator);\n outputConfigs.putAll(d);\n }\n }\n\n if (suffix.startsWith(ENCODER_KEY) || suffix.startsWith(ASYNC_KEY)) {\n newKey = reporterPrefix + namespaceWithSeparator + suffix;\n } else {\n newKey = reporterPrefix + namespaceWithSeparator + join(SENDER_KEY, suffix);\n }\n\n outputConfigs.put(newKey, e.getValue());\n }\n\n return outputConfigs;\n }\n\n /**\n * extract `plugin.observability.global.log.*` config items to `reporter.log.sender.*`\n * extract `plugin.observability.global.log.output.*` config items to `reporter.log.output.*`\n *\n * @param srcCfg source config map\n * @return reporter log config\n */\n private static Map extractGlobalLogConfig(Map srcCfg) {\n final String prefix = join(ConfigConst.PLUGIN, ConfigConst.OBSERVABILITY,\n ConfigConst.PLUGIN_GLOBAL,\n ConfigConst.PluginID.LOG);\n Map global = new TreeMap<>();\n Map extract = extractAndConvertPrefix(srcCfg, prefix, LOG_SENDER);\n\n for (Map.Entry e : extract.entrySet()) {\n String key = e.getKey();\n if (key.startsWith(ENCODER_KEY, LOG_SENDER.length() + 1)) {\n global.put(join(LOGS, key.substring(LOG_SENDER.length() + 1)), e.getValue());\n } else if (key.startsWith(ASYNC_KEY, LOG_SENDER.length() + 1)) {\n global.put(join(LOGS, key.substring(LOG_SENDER.length() + 1)), e.getValue());\n } else {\n global.put(e.getKey(), e.getValue());\n }\n }\n\n // global log level (async)\n global.putAll(extractByPrefix(srcCfg, LOG_SENDER));\n global.putAll(extractByPrefix(srcCfg, LOG_ASYNC));\n global.putAll(extractByPrefix(srcCfg, LOG_ENCODER));\n\n return global;\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/report/ReportConfigConst.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.config.report;\n\n@SuppressWarnings(\"unused\")\npublic class ReportConfigConst {\n private ReportConfigConst() {}\n\n public static final String KAFKA_SENDER_NAME = \"kafka\";\n public static final String METRIC_KAFKA_SENDER_NAME = \"metricKafka\";\n public static final String CONSOLE_SENDER_NAME = \"console\";\n public static final String ZIPKIN_SENDER_NAME = \"http\";\n\n public static final String NOOP_SENDER_NAME = \"noop\";\n\n public static final String SPAN_JSON_ENCODER_NAME = \"SpanJsonEncoder\";\n public static final String METRIC_JSON_ENCODER_NAME = \"MetricJsonEncoder\";\n public static final String LOG_DATA_JSON_ENCODER_NAME = \"LogDataJsonEncoder\";\n public static final String ACCESS_LOG_JSON_ENCODER_NAME = \"AccessLogJsonEncoder\";\n\n public static final String HTTP_SPAN_JSON_ENCODER_NAME = \"HttpSpanJsonEncoder\";\n\n public static final String LOG_ENCODER_NAME = \"StringEncoder\";\n\n static final String DELIMITER = \".\";\n public static final String TOPIC_KEY = \"topic\";\n public static final String LOG_APPENDER_KEY = \"appenderName\";\n\n public static final String ENABLED_KEY = \"enabled\";\n public static final String SENDER_KEY = \"sender\";\n public static final String ENCODER_KEY = \"encoder\";\n public static final String ASYNC_KEY = \"output\";\n public static final String APPEND_TYPE_KEY = \"appendType\";\n public static final String INTERVAL_KEY = \"interval\";\n\n public static final String ASYNC_THREAD_KEY = \"reportThread\";\n public static final String ASYNC_MSG_MAX_BYTES_KEY = \"messageMaxBytes\";\n public static final String ASYNC_MSG_TIMEOUT_KEY = \"messageTimeout\";\n public static final String ASYNC_QUEUE_MAX_SIZE_KEY = \"queuedMaxSize\";\n public static final String ASYNC_QUEUE_MAX_LOGS_KEY = \"queuedMaxLogs\";\n public static final String ASYNC_QUEUE_MAX_ITEMS_KEY = \"queuedMaxItems\";\n\n /**\n * Reporter v2 configuration\n */\n // -- lv1 --\n public static final String REPORT = \"reporter\";\n // ---- lv2 ----\n public static final String OUTPUT_SERVER_V2 = join(REPORT, \"outputServer\");\n public static final String TRACE_V2 = join(REPORT, \"tracing\");\n public static final String LOGS = join(REPORT, \"log\");\n public static final String METRIC_V2 = join(REPORT, \"metric\");\n public static final String GENERAL = join(REPORT, \"general\");\n // ------ lv3 ------\n public static final String BOOTSTRAP_SERVERS = join(OUTPUT_SERVER_V2, \"bootstrapServer\");\n public static final String OUTPUT_SERVERS_ENABLE = join(OUTPUT_SERVER_V2, ENABLED_KEY);\n public static final String OUTPUT_SERVERS_TIMEOUT = join(OUTPUT_SERVER_V2, \"timeout\");\n\n public static final String OUTPUT_SECURITY_PROTOCOL_V2 = join(OUTPUT_SERVER_V2, \"security.protocol\");\n public static final String OUTPUT_SERVERS_SSL = join(OUTPUT_SERVER_V2, \"ssl\");\n\n public static final String LOG_ASYNC = join(LOGS, ASYNC_KEY);\n\n public static final String LOG_SENDER = join(LOGS, SENDER_KEY);\n public static final String LOG_ENCODER = join(LOGS, ENCODER_KEY);\n\n public static final String LOG_ACCESS = join(LOGS, \"access\");\n public static final String LOG_ACCESS_SENDER = join(LOG_ACCESS, SENDER_KEY);\n public static final String LOG_ACCESS_ENCODER = join(LOG_ACCESS, ENCODER_KEY);\n\n public static final String TRACE_SENDER = join(TRACE_V2, SENDER_KEY);\n public static final String TRACE_ENCODER = join(TRACE_V2, ENCODER_KEY);\n public static final String TRACE_ASYNC = join(TRACE_V2, ASYNC_KEY);\n\n public static final String METRIC_SENDER = join(METRIC_V2, SENDER_KEY);\n public static final String METRIC_ENCODER = join(METRIC_V2, ENCODER_KEY);\n public static final String METRIC_ASYNC = join(METRIC_V2, ASYNC_KEY);\n\n // -------- lv4 --------\n public static final String LOG_SENDER_TOPIC = join(LOG_SENDER, TOPIC_KEY);\n public static final String LOG_SENDER_NAME = join(LOG_SENDER, APPEND_TYPE_KEY);\n\n public static final String LOG_ACCESS_SENDER_NAME = join(LOG_ACCESS_SENDER, APPEND_TYPE_KEY);\n public static final String LOG_ACCESS_SENDER_ENABLED = join(LOG_ACCESS_SENDER, ENABLED_KEY);\n public static final String LOG_ACCESS_SENDER_TOPIC = join(LOG_ACCESS_SENDER, TOPIC_KEY);\n\n public static final String LOG_ASYNC_MESSAGE_MAX_BYTES = join(LOG_ASYNC, ASYNC_MSG_MAX_BYTES_KEY);\n public static final String LOG_ASYNC_REPORT_THREAD = join(LOG_ASYNC, ASYNC_THREAD_KEY);\n public static final String LOG_ASYNC_MESSAGE_TIMEOUT = join(LOG_ASYNC, ASYNC_MSG_TIMEOUT_KEY);\n public static final String LOG_ASYNC_QUEUED_MAX_LOGS = join(LOG_ASYNC, ASYNC_QUEUE_MAX_LOGS_KEY);\n public static final String LOG_ASYNC_QUEUED_MAX_SIZE = join(LOG_ASYNC, ASYNC_QUEUE_MAX_SIZE_KEY);\n\n public static final String TRACE_SENDER_NAME = join(TRACE_SENDER, APPEND_TYPE_KEY);\n public static final String TRACE_SENDER_ENABLED_V2 = join(TRACE_SENDER, ENABLED_KEY);\n public static final String TRACE_SENDER_TOPIC_V2 = join(TRACE_SENDER, TOPIC_KEY);\n\n public static final String TRACE_ASYNC_MESSAGE_MAX_BYTES_V2 = join(TRACE_ASYNC, ASYNC_MSG_MAX_BYTES_KEY);\n public static final String TRACE_ASYNC_REPORT_THREAD_V2 = join(TRACE_ASYNC, ASYNC_THREAD_KEY);\n public static final String TRACE_ASYNC_MESSAGE_TIMEOUT_V2 = join(TRACE_ASYNC, ASYNC_MSG_TIMEOUT_KEY);\n public static final String TRACE_ASYNC_QUEUED_MAX_SPANS_V2 = join(TRACE_ASYNC, \"queuedMaxSpans\");\n public static final String TRACE_ASYNC_QUEUED_MAX_SIZE_V2 = join(TRACE_ASYNC, ASYNC_QUEUE_MAX_SIZE_KEY);\n\n public static final String METRIC_SENDER_NAME = join(METRIC_SENDER, APPEND_TYPE_KEY);\n public static final String METRIC_SENDER_ENABLED = join(METRIC_SENDER, ENABLED_KEY);\n public static final String METRIC_SENDER_TOPIC = join(METRIC_SENDER, TOPIC_KEY);\n public static final String METRIC_SENDER_APPENDER = join(METRIC_SENDER, LOG_APPENDER_KEY);\n\n public static final String METRIC_ASYNC_INTERVAL = join(METRIC_ASYNC, INTERVAL_KEY);\n public static final String METRIC_ASYNC_QUEUED_MAX_ITEMS = join(METRIC_ASYNC, ASYNC_QUEUE_MAX_ITEMS_KEY);\n public static final String METRIC_ASYNC_MESSAGE_MAX_BYTES = join(METRIC_ASYNC, ASYNC_MSG_MAX_BYTES_KEY);\n\n public static final String OUTPUT_SSL_KEYSTORE_TYPE_V2 = join(OUTPUT_SERVERS_SSL, \"keystore.type\");\n public static final String OUTPUT_KEY_V2 = join(OUTPUT_SERVERS_SSL, \"keystore.key\");\n public static final String OUTPUT_CERT_V2 = join(OUTPUT_SERVERS_SSL, \"keystore.certificate.chain\");\n public static final String OUTPUT_TRUST_CERT_V2 = join(OUTPUT_SERVERS_SSL, \"truststore.certificates\");\n public static final String OUTPUT_TRUST_CERT_TYPE_V2 = join(OUTPUT_SERVERS_SSL, \"truststore.type\");\n public static final String OUTPUT_ENDPOINT_IDENTIFICATION_ALGORITHM_V2 = join(OUTPUT_SERVERS_SSL, \"endpoint.identification.algorithm\");\n\n /**\n * Reporter v1 configuration\n */\n public static final String OBSERVABILITY = \"observability\";\n // ---- lv2 ----\n public static final String TRACING = join(OBSERVABILITY, \"tracings\");\n public static final String OUTPUT_SERVER_V1 = join(OBSERVABILITY, \"outputServer\");\n // ------ lv3 ------\n public static final String TRACE_OUTPUT_V1 = join(TRACING, ASYNC_KEY);\n public static final String BOOTSTRAP_SERVERS_V1 = join(OUTPUT_SERVER_V1, \"bootstrapServer\");\n\n // --------- lv4 ---------\n public static final String TRACE_OUTPUT_ENABLED_V1 = join(TRACE_OUTPUT_V1, ENABLED_KEY);\n public static final String TRACE_OUTPUT_TOPIC_V1 = join(TRACE_OUTPUT_V1, TOPIC_KEY);\n public static final String TRACE_OUTPUT_TARGET_V1 = join(TRACE_OUTPUT_V1, \"target\");\n public static final String TRACE_OUTPUT_TARGET_ZIPKIN_URL = join(TRACE_OUTPUT_V1, \"target.zipkinUrl\");\n\n public static final String TRACE_OUTPUT_REPORT_THREAD_V1 = join(TRACE_OUTPUT_V1, ASYNC_THREAD_KEY);\n public static final String TRACE_OUTPUT_MESSAGE_TIMEOUT_V1 = join(TRACE_OUTPUT_V1, ASYNC_MSG_TIMEOUT_KEY);\n public static final String TRACE_OUTPUT_QUEUED_MAX_SPANS_V1 = join(TRACE_OUTPUT_V1, \"queuedMaxSpans\");\n public static final String TRACE_OUTPUT_QUEUED_MAX_SIZE_V1 = join(TRACE_OUTPUT_V1, ASYNC_QUEUE_MAX_SIZE_KEY);\n\n public static final String GLOBAL_METRIC = \"plugin.observability.global.metric\";\n public static final String GLOBAL_METRIC_ENABLED = join(GLOBAL_METRIC, ENABLED_KEY);\n public static final String GLOBAL_METRIC_TOPIC = join(GLOBAL_METRIC, TOPIC_KEY);\n public static final String GLOBAL_METRIC_APPENDER = join(GLOBAL_METRIC, APPEND_TYPE_KEY);\n\n public static String join(String... texts) {\n return String.join(DELIMITER, texts);\n }\n}\n" }, { "path": "config/src/main/java/com/megaease/easeagent/config/yaml/YamlReader.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config.yaml;\n\nimport org.yaml.snakeyaml.DumperOptions;\nimport org.yaml.snakeyaml.Yaml;\n\nimport java.io.InputStream;\nimport java.util.Collections;\nimport java.util.Deque;\nimport java.util.HashMap;\nimport java.util.LinkedHashMap;\nimport java.util.LinkedList;\nimport java.util.List;\nimport java.util.Map;\nimport java.util.Objects;\nimport java.util.stream.Collectors;\n\npublic class YamlReader {\n\n private Map yaml;\n\n private static final DumperOptions DUMPER_OPTIONS;\n\n static {\n DUMPER_OPTIONS = new DumperOptions();\n DUMPER_OPTIONS.setLineBreak(DumperOptions.LineBreak.getPlatformLineBreak());\n }\n\n public YamlReader() {\n // ignored\n }\n\n public YamlReader load(InputStream in) {\n if (in != null) {\n yaml = new Yaml(DUMPER_OPTIONS).load(in);\n }\n return this;\n }\n\n public Map getYaml() {\n return yaml;\n }\n\n public static YamlReader merge(YamlReader target, YamlReader source) {\n Map targetMap = target.yaml;\n Map sourceMap = source.yaml;\n\n merge(targetMap, sourceMap);\n\n YamlReader result = new YamlReader();\n result.yaml = new HashMap<>(targetMap);\n return result;\n }\n\n @SuppressWarnings(\"unchecked\")\n private static void merge(Map target, Map source) {\n source.forEach((key, value) -> {\n Object existing = target.get(key);\n if (value instanceof Map && existing instanceof Map) {\n Map result = new LinkedHashMap<>((Map) existing);\n merge(result, (Map) value);\n target.put(key, result);\n } else {\n target.put(key, value);\n }\n });\n }\n\n public Map compress() {\n if (Objects.isNull(yaml) || yaml.size() == 0) {\n return Collections.emptyMap();\n }\n\n final Deque keyStack = new LinkedList<>();\n final Map resultMap = new HashMap<>();\n\n compress(yaml, keyStack, resultMap);\n\n return resultMap;\n }\n\n @SuppressWarnings(\"unchecked\")\n private void compress(Map result, Deque keyStack, Map resultMap) {\n result.forEach((k, v) -> {\n keyStack.addLast(String.valueOf(k));\n\n if (v instanceof Map) {\n compress((Map) v, keyStack, resultMap);\n keyStack.removeLast();\n return;\n }\n\n if (v instanceof List) {\n String value = ((List) v).stream()\n .map(String::valueOf)\n .collect(Collectors.joining(\",\"));\n\n resultMap.put(String.join(\".\", keyStack), value);\n keyStack.removeLast();\n return;\n }\n\n resultMap.put(String.join(\".\", keyStack), String.valueOf(v));\n keyStack.removeLast();\n });\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/CompatibilityConversionTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.api.ProgressFields;\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\nimport org.junit.Test;\n\nimport java.util.Collections;\nimport java.util.Map;\n\nimport static com.megaease.easeagent.config.CompatibilityConversion.REQUEST_NAMESPACE;\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertTrue;\n\npublic class CompatibilityConversionTest {\n\n @Test\n public void transform() {\n Map newMap = CompatibilityConversion.transform(Collections.singletonMap(ConfigConst.Observability.METRICS_ENABLED, \"false\"));\n assertEquals(2, newMap.size());\n assertTrue(newMap.containsKey(ConfigConst.Plugin.OBSERVABILITY_GLOBAL_METRIC_ENABLED));\n assertTrue(newMap.containsKey(ConfigConst.Observability.METRICS_ENABLED));\n newMap = CompatibilityConversion.transform(Collections.singletonMap(ConfigConst.Observability.TRACE_ENABLED, \"false\"));\n assertEquals(1, newMap.size());\n assertTrue(newMap.containsKey(ConfigConst.Plugin.OBSERVABILITY_GLOBAL_TRACING_ENABLED));\n// newMap = CompatibilityConversion.transform(Collections.singletonMap(\"globalCanaryHeaders.serviceHeaders.mesh-app-backend.0\", \"X-canary\"));\n// assertEquals(1, newMap.size());\n// assertTrue(newMap.containsKey(ProgressFields.EASEAGENT_PROGRESS_FORWARDED_HEADERS_CONFIG + \".mesh-app-backend.0\"));\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.outputServer.bootstrapServer\", \"tstatssta\"));\n assertEquals(1, newMap.size());\n assertTrue(newMap.containsKey(\"observability.outputServer.bootstrapServer\"));\n\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.metrics.access.enabled\", \"true\"));\n assertEquals(1, newMap.size());\n assertEquals(\"true\", newMap.get(\"plugin.observability.access.metric.enabled\"));\n\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.metrics.access.interval\", \"30\"));\n assertEquals(1, newMap.size());\n assertEquals(\"30\", newMap.get(\"plugin.observability.access.metric.interval\"));\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.tracings.remoteInvoke.enabled\", \"true\"));\n assertEquals(1, newMap.size());\n assertEquals(\"true\", newMap.get(\"plugin.observability.webclient.tracing.enabled\"));\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.tracings.request.enabled\", \"true\"));\n assertEquals(REQUEST_NAMESPACE.length, newMap.size());\n for (String s : REQUEST_NAMESPACE) {\n assertEquals(\"true\", newMap.get(\"plugin.observability.\" + s + \".tracing.enabled\"));\n }\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.metrics.jvmGc.enabled\", \"true\"));\n assertEquals(1, newMap.size());\n assertEquals(\"true\", newMap.get(\"observability.metrics.jvmGc.enabled\"));\n\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.tracings.sampledType\", \"counting\"));\n assertEquals(1, newMap.size());\n assertEquals(\"counting\", newMap.get(\"observability.tracings.sampledType\"));\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.tracings.sampled\", \"100\"));\n assertEquals(1, newMap.size());\n assertEquals(\"100\", newMap.get(\"observability.tracings.sampled\"));\n\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.tracings.output.enabled\", \"true\"));\n assertEquals(1, newMap.size());\n assertEquals(\"true\", newMap.get(\"observability.tracings.output.enabled\"));\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.metrics.jdbcConnection.interval\", \"30\"));\n assertEquals(1, newMap.size());\n assertEquals(\"30\", newMap.get(\"plugin.observability.jdbcConnection.metric.interval\"));\n\n newMap = CompatibilityConversion.transform(Collections.singletonMap(\"observability.metrics.aaaaaaaaaa.interval\", \"30\"));\n assertEquals(1, newMap.size());\n assertEquals(\"30\", newMap.get(\"plugin.observability.aaaaaaaaaa.metric.interval\"));\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/ConfigFactoryTest.java", "content": "/*\n * Copyright (c) 2022, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.utils.SystemEnv;\nimport org.junit.Test;\nimport org.mockito.MockedStatic;\nimport org.mockito.Mockito;\n\nimport java.io.File;\nimport java.net.URISyntaxException;\n\nimport static com.megaease.easeagent.config.ConfigFactory.AGENT_SERVICE;\nimport static com.megaease.easeagent.config.ConfigFactory.AGENT_SYSTEM;\nimport static org.junit.Assert.assertEquals;\n\npublic class ConfigFactoryTest {\n\n @Test\n public void test_yaml() {\n Configs config = ConfigFactory.loadConfigs(null, this.getClass().getClassLoader());\n assertEquals(\"test-service\", config.getString(AGENT_SERVICE));\n assertEquals(\"demo-system\", config.getString(AGENT_SYSTEM));\n }\n\n @Test\n public void test_env() {\n try (MockedStatic mock = Mockito.mockStatic(SystemEnv.class)) {\n mock.when(() -> SystemEnv.get(ConfigFactory.EASEAGENT_ENV_CONFIG)).thenReturn(\"{\\\"name\\\":\\\"env-service\\\"}\");\n\n Configs config = ConfigFactory.loadConfigs(null, this.getClass().getClassLoader());\n assertEquals(\"env-service\", config.getString(AGENT_SERVICE));\n assertEquals(\"demo-system\", config.getString(AGENT_SYSTEM));\n }\n }\n\n\n @Test\n public void test_loadConfigs() {\n try (MockedStatic mockSystemEnv = Mockito.mockStatic(SystemEnv.class)) {\n mockSystemEnv.when(() -> SystemEnv.get(\"EASEAGENT_NAME\")).thenReturn(\"service1\");\n mockSystemEnv.when(() -> SystemEnv.get(\"EASEAGENT_SYSTEM\")).thenReturn(\"system1\");\n\n Configs config = ConfigFactory.loadConfigs(null, this.getClass().getClassLoader());\n assertEquals(\"service1\", config.getString(AGENT_SERVICE));\n assertEquals(\"system1\", config.getString(AGENT_SYSTEM));\n\n System.setProperty(\"easeagent.name\", \"service2\");\n System.setProperty(\"easeagent.system\", \"system2\");\n config = ConfigFactory.loadConfigs(null, this.getClass().getClassLoader());\n assertEquals(\"service2\", config.getString(AGENT_SERVICE));\n assertEquals(\"system2\", config.getString(AGENT_SYSTEM));\n }\n\n }\n\n @Test\n public void test_loadConfigsFromUserSpec() throws URISyntaxException {\n String userSpec = new File(this.getClass().getClassLoader().getResource(\"user-spec.properties\").toURI()).getPath();\n\n try (MockedStatic mockSystemEnv = Mockito.mockStatic(SystemEnv.class)) {\n mockSystemEnv.when(() -> SystemEnv.get(\"EASEAGENT_CONFIG_PATH\")).thenReturn(userSpec);\n String path = ConfigFactory.getConfigPath();\n assertEquals(userSpec, path);\n }\n }\n\n\n @Test\n public void test_loadConfigsFromOtelUserSpec() throws URISyntaxException {\n String userSpec = new File(this.getClass().getClassLoader().getResource(\"user-spec.properties\").toURI()).getPath();\n\n try (MockedStatic mockSystemEnv = Mockito.mockStatic(SystemEnv.class)) {\n mockSystemEnv.when(() -> SystemEnv.get(\"OTEL_JAVAAGENT_CONFIGURATION_FILE\")).thenReturn(userSpec);\n String path = ConfigFactory.getConfigPath();\n assertEquals(userSpec, path);\n }\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/ConfigPropertiesUtilsTest.java", "content": "/*\n * Copyright (c) 2022, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport org.junit.*;\nimport org.junit.contrib.java.lang.system.EnvironmentVariables;\nimport org.junit.contrib.java.lang.system.RestoreSystemProperties;\n\npublic class ConfigPropertiesUtilsTest {\n\n @Rule\n public final EnvironmentVariables environmentVariables\n = new EnvironmentVariables();\n\n @Rule\n public final RestoreSystemProperties restoreSystemProperties\n = new RestoreSystemProperties();\n\n @BeforeClass\n public static void beforeClass() {\n // EnvironmentVariables and restoreSystemProperties does not work with Java 16\n Assume.assumeTrue(\n System.getProperty(\"java.version\").startsWith(\"1.8\")\n || System.getProperty(\"java.version\").startsWith(\"11\")\n );\n }\n\n @Test\n public void getString_systemProperty() {\n environmentVariables.set(\"TEST_PROPERTY_STRING\", \"env\");\n System.setProperty(\"test.property.string\", \"sys\");\n Assert.assertEquals(\"sys\", ConfigPropertiesUtils.getString(\"test.property.string\"));\n }\n\n @Test\n public void getString_environmentVariable() {\n environmentVariables.set(\"TEST_PROPERTY_STRING\", \"env\");\n Assert.assertEquals(\"env\", ConfigPropertiesUtils.getString(\"test.property.string\"));\n }\n\n @Test\n public void getString_none() {\n Assert.assertNull(ConfigPropertiesUtils.getString(\"test.property.string.none\"));\n }\n\n @Test\n public void getInt_systemProperty() {\n environmentVariables.set(\"TEST_PROPERTY_INT\", \"12\");\n System.setProperty(\"test.property.int\", \"42\");\n Assert.assertEquals(42, ConfigPropertiesUtils.getInt(\"test.property.int\", -1));\n }\n\n @Test\n public void getInt_environmentVariable() {\n environmentVariables.set(\"TEST_PROPERTY_INT\", \"12\");\n Assert.assertEquals(12, ConfigPropertiesUtils.getInt(\"test.property.int\", -1));\n }\n\n @Test\n public void getInt_none() {\n Assert.assertEquals(-1, ConfigPropertiesUtils.getInt(\"test.property.int\", -1));\n }\n\n @Test\n public void getInt_invalidNumber() {\n System.setProperty(\"test.property.int\", \"not a number\");\n Assert.assertEquals(-1, ConfigPropertiesUtils.getInt(\"test.property.int\", -1));\n }\n\n @Test\n public void getBoolean_systemProperty() {\n environmentVariables.set(\"TEST_PROPERTY_BOOLEAN\", \"false\");\n System.setProperty(\"test.property.boolean\", \"true\");\n Assert.assertTrue(ConfigPropertiesUtils.getBoolean(\"test.property.boolean\", false));\n }\n\n @Test\n public void getBoolean_environmentVariable() {\n environmentVariables.set(\"TEST_PROPERTY_BOOLEAN\", \"true\");\n Assert.assertTrue(ConfigPropertiesUtils.getBoolean(\"test.property.boolean\", false));\n }\n\n @Test\n public void getBoolean_none() {\n Assert.assertFalse(ConfigPropertiesUtils.getBoolean(\"test.property.boolean\", false));\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/ConfigUtilsTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.api.config.Config;\nimport org.junit.Assert;\nimport org.junit.Test;\n\nimport java.io.IOException;\nimport java.util.Collections;\nimport java.util.Map;\n\nimport static org.junit.Assert.assertTrue;\n\npublic class ConfigUtilsTest {\n @Test\n public void test_bindProp() throws Exception {\n Configs configs = new Configs(Collections.singletonMap(\"hello\", \"world\"));\n String[] rst = new String[]{null};\n ConfigUtils.bindProp(\"hello\", configs, Config::getString, v -> rst[0] = v);\n Assert.assertEquals(\"world\", rst[0]);\n configs.updateConfigs(Collections.singletonMap(\"hello\", \"test\"));\n Assert.assertEquals(\"test\", rst[0]);\n configs.updateConfigs(Collections.singletonMap(\"hello\", \"one\"));\n Assert.assertEquals(\"one\", rst[0]);\n }\n\n @Test\n public void test_json2KVMap() throws Exception {\n Map map = ConfigUtils.json2KVMap(\"{\\n\" +\n \" \\\"output\\\": {\\n\" +\n \" \\\"servers\\\": \\\"127.0.0.1\\\",\\n\" +\n \" \\\"timeout\\\": 1000,\\n\" +\n \" \\\"enabled\\\": true,\\n\" +\n \" \\\"arr\\\": [\\\"x\\\", { \\\"test\\\": 0 }]\\n\" +\n \" },\\n\" +\n \" \\\"hello\\\":null,\\n\" +\n \" \\\"metrics\\\": {\\n\" +\n \" \\\"obj\\\": {\\n\" +\n \" \\\"a\\\": 1,\\n\" +\n \" \\\"b\\\": \\\"2\\\",\\n\" +\n \" \\\"c\\\": false\\n\" +\n \" },\\n\" +\n \" \\\"request\\\": {\\n\" +\n \" \\\"topic\\\": \\\"hello\\\",\\n\" +\n \" \\\"enabled\\\": false\\n\" +\n \" }\\n\" +\n \" }\\n\" +\n \"}\");\n Assert.assertEquals(\"127.0.0.1\", map.get(\"output.servers\"));\n Assert.assertEquals(\"1000\", map.get(\"output.timeout\"));\n Assert.assertEquals(\"true\", map.get(\"output.enabled\"));\n Assert.assertEquals(\"x\", map.get(\"output.arr.0\"));\n Assert.assertEquals(\"0\", map.get(\"output.arr.1.test\"));\n Assert.assertEquals(\"\", map.get(\"hello\"));\n Assert.assertEquals(\"1\", map.get(\"metrics.obj.a\"));\n Assert.assertEquals(\"2\", map.get(\"metrics.obj.b\"));\n Assert.assertEquals(\"false\", map.get(\"metrics.obj.c\"));\n Assert.assertEquals(\"hello\", map.get(\"metrics.request.topic\"));\n Assert.assertEquals(\"false\", map.get(\"metrics.request.enabled\"));\n }\n\n @Test\n public void test_json2KVMap_2() throws IOException {\n Map map = ConfigUtils.json2KVMap(\"{\\\"serviceHeaders\\\":{\\\"mesh-app-backend\\\":[\\\"X-canary\\\"]}}\");\n Assert.assertEquals(\"X-canary\", map.get(\"serviceHeaders.mesh-app-backend.0\"));\n }\n\n @Test\n public void isGlobal() {\n Assert.assertTrue(ConfigUtils.isGlobal(\"global\"));\n Assert.assertFalse(ConfigUtils.isGlobal(\"globaldf\"));\n }\n\n @Test\n public void isPluginConfig() {\n Assert.assertTrue(ConfigUtils.isPluginConfig(\"plugin.\"));\n Assert.assertFalse(ConfigUtils.isPluginConfig(\"plugin\"));\n Assert.assertFalse(ConfigUtils.isPluginConfig(\"plugins.\"));\n Assert.assertTrue(ConfigUtils.isPluginConfig(\"plugin.observability.kafka.kafka-trace\", \"observability\", \"kafka\", \"kafka-trace\"));\n Assert.assertFalse(ConfigUtils.isPluginConfig(\"plugin.observability.kafka.kafka-trace\", \"observabilitys\", \"kafka\", \"kafka-trace\"));\n Assert.assertFalse(ConfigUtils.isPluginConfig(\"plugin.observability.kafka.kafka-trace\", \"observability\", \"kafkas\", \"kafka-trace\"));\n Assert.assertFalse(ConfigUtils.isPluginConfig(\"plugin.observability.kafka.kafka-trace\", \"observability\", \"kafka\", \"kafka-traces\"));\n }\n\n @Test\n public void pluginProperty() {\n PluginProperty pluginProperty = ConfigUtils.pluginProperty(\"plugin.observability.kafka.self.enabled\");\n Assert.assertEquals(\"observability\", pluginProperty.getDomain());\n Assert.assertEquals(\"kafka\", pluginProperty.getNamespace());\n Assert.assertEquals(\"self\", pluginProperty.getId());\n Assert.assertEquals(\"enabled\", pluginProperty.getProperty());\n pluginProperty = ConfigUtils.pluginProperty(\"plugin.observability.kafka.self.tcp.enabled\");\n Assert.assertEquals(\"observability\", pluginProperty.getDomain());\n Assert.assertEquals(\"kafka\", pluginProperty.getNamespace());\n Assert.assertEquals(\"self\", pluginProperty.getId());\n Assert.assertEquals(\"tcp.enabled\", pluginProperty.getProperty());\n try {\n ConfigUtils.pluginProperty(\"plugin.observability.kafka.self\");\n assertTrue(\"must be error\", false);\n } catch (Exception e) {\n Assert.assertNotNull(e);\n }\n\n\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/ConfigsTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.fasterxml.jackson.databind.ObjectMapper;\nimport com.megaease.easeagent.plugin.api.config.ChangeItem;\nimport com.megaease.easeagent.plugin.api.config.Config;\nimport org.junit.Assert;\nimport org.junit.Test;\n\nimport java.util.Collections;\nimport java.util.LinkedList;\nimport java.util.List;\n\npublic class ConfigsTest {\n\n @Test\n public void test_check_change_count1() throws Exception {\n final ObjectMapper json = new ObjectMapper();\n GlobalConfigs configs = new GlobalConfigs(Collections.singletonMap(\"hello\", \"world\"));\n List rst = addListener(configs);\n\n configs.updateService(\"{}\", null);\n Assert.assertEquals(0, rst.size());\n\n configs.updateService(json.writeValueAsString(Collections.singletonMap(\"hello\", \"world\")), null);\n Assert.assertEquals(0, rst.size());\n\n configs.updateService(json.writeValueAsString(Collections.singletonMap(\"hello\", \"world2\")), null);\n Assert.assertEquals(1, rst.size());\n }\n\n @Test\n public void test_check_change_count() {\n Configs configs = new Configs(Collections.singletonMap(\"hello\", \"world\"));\n List rst = addListener(configs);\n\n configs.updateConfigs(Collections.emptyMap());\n Assert.assertEquals(0, rst.size());\n\n configs.updateConfigs(Collections.singletonMap(\"hello\", \"world\"));\n Assert.assertEquals(0, rst.size());\n\n configs.updateConfigs(Collections.singletonMap(\"hello\", \"world2\"));\n Assert.assertEquals(1, rst.size());\n }\n\n @Test\n public void test_check_old() {\n Configs configs = new Configs(Collections.singletonMap(\"hello\", \"world\"));\n List rst = addListener(configs);\n configs.updateConfigs(Collections.singletonMap(\"hello\", \"test\"));\n ChangeItem first = rst.get(0);\n Assert.assertEquals(\"hello\", first.getFullName());\n Assert.assertEquals(\"hello\", first.getName());\n Assert.assertEquals(\"world\", first.getOldValue());\n Assert.assertEquals(\"test\", first.getNewValue());\n }\n\n\n @Test\n public void test_check_new() {\n Configs configs = new Configs(Collections.singletonMap(\"hello\", \"world\"));\n List rst = addListener(configs);\n configs.updateConfigs(Collections.singletonMap(\"name\", \"666\"));\n ChangeItem first = rst.get(0);\n Assert.assertEquals(\"name\", first.getFullName());\n Assert.assertEquals(\"name\", first.getName());\n Assert.assertNull(first.getOldValue());\n Assert.assertEquals(\"666\", first.getNewValue());\n }\n\n\n private List addListener(Config config) {\n List rst = new LinkedList<>();\n config.addChangeListener(rst::addAll);\n return rst;\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/IPluginConfigConstTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\nimport org.junit.Test;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertNull;\n\npublic class IPluginConfigConstTest {\n @Test\n public void testExtractHeaderName() {\n String prefix = \"globalCanaryHeaders.serviceHeaders\";\n assertEquals(ConfigConst.GlobalCanaryLabels.SERVICE_HEADERS, prefix);\n assertEquals(\"hello\", ConfigConst.GlobalCanaryLabels.extractHeaderName(prefix + \".test-aaa.0.hello\"));\n assertEquals(\"world\", ConfigConst.GlobalCanaryLabels.extractHeaderName(prefix + \".test-aaa.1.world\"));\n assertNull(ConfigConst.GlobalCanaryLabels.extractHeaderName(prefix + \".test-bbb\"));\n assertNull(ConfigConst.GlobalCanaryLabels.extractHeaderName(prefix + \"test-bbb\"));\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/JarFileConfigLoaderTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\nimport org.junit.After;\nimport org.junit.Test;\n\nimport java.io.File;\nimport java.net.URISyntaxException;\n\nimport static org.junit.Assert.*;\n\npublic class JarFileConfigLoaderTest {\n\n @After\n public void after() {\n System.clearProperty(ConfigConst.AGENT_JAR_PATH);\n }\n\n\n @Test\n public void load() throws URISyntaxException {\n String fileName = \"agent.properties\";\n assertNull(JarFileConfigLoader.load(null));\n assertNull(JarFileConfigLoader.load(fileName));\n String jarPath = new File(this.getClass().getClassLoader().getResource(\"easeagent_config.jar\").toURI()).getPath();\n System.setProperty(ConfigConst.AGENT_JAR_PATH, jarPath);\n\n GlobalConfigs config = JarFileConfigLoader.load(fileName);\n assertNotNull(config);\n assertEquals(\"demo-jar-service\", config.getString(\"name\"));\n assertEquals(\"demo-system\", config.getString(\"system\"));\n\n config = JarFileConfigLoader.load(\"agent.yaml\");\n assertNotNull(config);\n assertEquals(\"test-jar-service\", config.getString(\"name\"));\n assertEquals(\"demo-system\", config.getString(\"system\"));\n\n config = JarFileConfigLoader.load(\"agentaaaa.yaml\");\n assertNull(config);\n\n\n String nullJarPath = new File(\"easeagent_config_null.jar\").getPath();\n System.setProperty(ConfigConst.AGENT_JAR_PATH, nullJarPath);\n\n config = JarFileConfigLoader.load(\"agent.yaml\");\n assertNull(config);\n\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/OtelSdkConfigsTest.java", "content": "/*\n * Copyright (c) 2022, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.utils.SystemEnv;\nimport org.junit.After;\nimport org.junit.Assert;\nimport org.junit.Test;\n\nimport java.lang.reflect.Field;\nimport java.util.Map;\n\n\npublic class OtelSdkConfigsTest {\n\n @After\n public void after() throws NoSuchFieldException, IllegalAccessException {\n Field field = SystemEnv.class.getDeclaredField(\"ENVIRONMENTS\");\n field.setAccessible(true);\n Object env = field.get(null);\n field.setAccessible(false);\n Map envMap = (Map) env;\n envMap.remove(\"OTEL_RESOURCE_ATTRIBUTES\");\n envMap.remove(\"OTEL_SERVICE_NAME\");\n envMap.remove(\"OTEL_SERVICE_NAMESPACE\");\n System.clearProperty(\"otel.service.name\");\n System.clearProperty(\"otel.service.namespace\");\n }\n\n @Test\n public void updateEnvCfg() {\n //value from system env \"OTEL_RESOURCE_ATTRIBUTES\n String attributes = \"service.name=service1,service.namespace=namespace1\";\n SystemEnv.set(\"OTEL_RESOURCE_ATTRIBUTES\", attributes);\n Map envCfg = OtelSdkConfigs.updateEnvCfg();\n Assert.assertEquals(\"service1\", envCfg.get(\"name\"));\n Assert.assertEquals(\"namespace1\", envCfg.get(\"system\"));\n\n // override by system env\n SystemEnv.set(\"OTEL_SERVICE_NAME\", \"service2\");\n SystemEnv.set(\"OTEL_SERVICE_NAMESPACE\", \"namespace2\");\n envCfg = OtelSdkConfigs.updateEnvCfg();\n Assert.assertEquals(\"service2\", envCfg.get(\"name\"));\n Assert.assertEquals(\"namespace2\", envCfg.get(\"system\"));\n\n // override by system property\n System.setProperty(\"otel.service.name\", \"service3\");\n System.setProperty(\"otel.service.namespace\", \"namespace3\");\n envCfg = OtelSdkConfigs.updateEnvCfg();\n Assert.assertEquals(\"service3\", envCfg.get(\"name\"));\n Assert.assertEquals(\"namespace3\", envCfg.get(\"system\"));\n\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/PluginConfigManagerTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.api.config.IPluginConfig;\nimport com.megaease.easeagent.plugin.api.config.PluginConfigChangeListener;\nimport org.junit.Test;\n\nimport java.util.Collections;\nimport java.util.HashMap;\nimport java.util.Map;\nimport java.util.concurrent.atomic.AtomicReference;\n\nimport static org.junit.Assert.*;\n\npublic class PluginConfigManagerTest {\n public static String DOMAIN = \"testDomain\";\n public static String NAMESPACE = \"testNamespace\";\n public static String TEST_TRACE_ID = \"test-trace\";\n public static String GLOBAL_ID = TEST_TRACE_ID;\n public static String TEST_METRIC_ID = \"test-metric\";\n public static String TEST_AAA_ID = \"test-AAA\";\n\n\n PluginConfigManager build() {\n return build(PluginSourceConfigTest.buildSource());\n }\n\n PluginConfigManager build(Map source) {\n Configs configs = new Configs(source);\n return PluginConfigManager.builder(configs).build();\n }\n\n @Test\n public void testBuild() {\n build();\n assertTrue(true);\n }\n\n @Test\n public void getConfig() {\n PluginConfigManager pluginConfigManager = build();\n for (Map.Entry entry : PluginSourceConfigTest.buildSource().entrySet()) {\n assertEquals(pluginConfigManager.getConfig(entry.getKey()), entry.getValue());\n }\n }\n\n private void checkPluginConfigString(PluginConfig pluginConfig, Map source) {\n for (Map.Entry entry : source.entrySet()) {\n assertEquals(pluginConfig.getString(entry.getKey()), entry.getValue());\n }\n }\n\n public static Map buildSource() {\n Map source = getSource(\"global\", GLOBAL_ID, PluginConfigTest.globalSource());\n source.putAll(getSource(\"global\", TEST_AAA_ID, PluginConfigTest.globalSource()));\n source.putAll(getSource(NAMESPACE, TEST_TRACE_ID, PluginConfigTest.coverSource()));\n source.put(\"plugin.testDomain.testssss.self.lll\", \"aaa\");\n source.put(\"plugin.testDomain.testssss.kafka.lll\", \"aaa\");\n source.put(\"plugin.testDomain.testssss.kafka.lll\", \"aaa\");\n source.put(\"plugin.testDomain.testssss.kafka.lll\", \"aaa\");\n return source;\n }\n\n\n public static Map getSource(String namespace, String id, Map properties) {\n Map s = new HashMap<>();\n for (Map.Entry pEntry : properties.entrySet()) {\n s.put(\"plugin.\" + DOMAIN + \".\" + namespace + \".\" + id + \".\" + pEntry.getKey(), pEntry.getValue());\n }\n return s;\n }\n\n @Test\n public void getConfig1() throws InterruptedException {\n PluginConfigManager pluginConfigManager = build();\n PluginConfig pluginConfig = pluginConfigManager.getConfig(PluginSourceConfigTest.DOMAIN, \"global\", PluginSourceConfigTest.GLOBAL_ID);\n checkPluginConfigString(pluginConfig, PluginConfigTest.globalSource());\n pluginConfig = pluginConfigManager.getConfig(PluginSourceConfigTest.DOMAIN, PluginSourceConfigTest.NAMESPACE, PluginSourceConfigTest.TEST_TRACE_ID);\n checkPluginConfigString(pluginConfig, PluginConfigTest.globalSource());\n pluginConfig = pluginConfigManager.getConfig(PluginSourceConfigTest.DOMAIN, PluginSourceConfigTest.NAMESPACE, PluginSourceConfigTest.TEST_METRIC_ID);\n checkPluginConfigString(pluginConfig, PluginConfigTest.globalSource());\n\n Map source = buildSource();\n source.putAll(getSource(\"trace\", TEST_TRACE_ID, PluginConfigTest.coverSource()));\n source.putAll(getSource(\"trace\", GLOBAL_ID, PluginConfigTest.coverSource()));\n build(source);\n\n Configs configs = new Configs(buildSource());\n pluginConfigManager = PluginConfigManager.builder(configs).build();\n final PluginConfig pluginConfig1 = pluginConfigManager.getConfig(DOMAIN, NAMESPACE, TEST_TRACE_ID);\n final AtomicReference oldPluginConfig = new AtomicReference<>();\n final AtomicReference newPluginConfig = new AtomicReference<>();\n PluginConfigTest.checkAllType(pluginConfig1);\n pluginConfig1.addChangeListener((oldConfig, newConfig) -> {\n oldPluginConfig.set(oldConfig);\n newPluginConfig.set(newConfig);\n });\n\n configs.updateConfigs(Collections.singletonMap(String.format(\"plugin.%s.%s.%s.enabled\", DOMAIN, NAMESPACE, TEST_TRACE_ID), \"false\"));\n assertNotNull(oldPluginConfig.get());\n assertNotNull(newPluginConfig.get());\n assertTrue(oldPluginConfig.get() == pluginConfig1);\n final AtomicReference oldPluginConfigListener = new AtomicReference<>();\n final AtomicReference newPluginConfigListener = new AtomicReference<>();\n ((PluginConfig) oldPluginConfig.get()).foreachConfigChangeListener(listener -> oldPluginConfigListener.set(listener));\n ((PluginConfig) newPluginConfig.get()).foreachConfigChangeListener(listener -> newPluginConfigListener.set(listener));\n assertTrue(oldPluginConfigListener.get() == newPluginConfigListener.get());\n assertTrue(oldPluginConfig.get().getBoolean(\"enabled\"));\n assertFalse(newPluginConfig.get().getBoolean(\"enabled\"));\n PluginConfigTest.checkAllType((PluginConfig) oldPluginConfig.get());\n\n configs.updateConfigs(Collections.singletonMap(String.format(\"plugin.%s.%s.%s.enabled\", DOMAIN, NAMESPACE, TEST_TRACE_ID), \"true\"));\n assertFalse(oldPluginConfig.get().getBoolean(\"enabled\"));\n assertTrue(newPluginConfig.get().getBoolean(\"enabled\"));\n\n\n configs.updateConfigs(Collections.singletonMap(String.format(\"plugin.%s.global.%s.enabled\", DOMAIN, TEST_TRACE_ID), \"false\"));\n assertTrue(oldPluginConfig.get().getBoolean(\"enabled\"));\n assertFalse(newPluginConfig.get().getBoolean(\"enabled\"));\n\n\n IPluginConfig newConfig = newPluginConfig.get();\n configs.updateConfigs(Collections.singletonMap(String.format(\"plugin.%s.global.%s.enabled\", DOMAIN, TEST_AAA_ID), \"false\"));\n assertTrue(newPluginConfig.get() == newConfig);\n\n configs.updateConfigs(Collections.singletonMap(String.format(\"ssss.%s.global.%s.enabled\", DOMAIN, TEST_AAA_ID), \"false\"));\n }\n\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/PluginConfigTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport com.megaease.easeagent.plugin.api.config.PluginConfigChangeListener;\nimport org.junit.Test;\n\nimport java.util.HashMap;\nimport java.util.List;\nimport java.util.Map;\nimport java.util.Set;\nimport java.util.concurrent.atomic.AtomicInteger;\nimport java.util.function.Consumer;\n\nimport static org.junit.Assert.*;\n\npublic class PluginConfigTest {\n\n public static Map globalSource() {\n Map global = new HashMap<>();\n global.put(\"enabled\", \"true\");\n global.put(\"tcp.enabled\", \"true\");\n global.put(\"host\", \"127.0.0.1\");\n global.put(\"count\", \"127\");\n global.put(\"double\", \"127.1\");\n global.put(\"double_1\", \"127.2\");\n global.put(\"list\", \"a,b,c\");\n return global;\n }\n\n public static Map coverSource() {\n Map cover = new HashMap<>();\n cover.put(\"tcp.enabled\", \"false\");\n cover.put(\"http.enabled\", \"true\");\n cover.put(\"host\", \"127.0.0.3\");\n cover.put(\"count\", \"127\");\n cover.put(\"double\", \"127.3\");\n cover.put(\"list\", \"a,b,c\");\n return cover;\n }\n\n PluginConfig build() {\n String domain = \"testdomain\";\n String id = \"testid\";\n String namespace = \"NAMESPACE\";\n Map global = globalSource();\n Map cover = coverSource();\n return PluginConfig.build(domain, id, global, namespace, cover, null);\n }\n\n @Test\n public void domain() {\n assertEquals(\"testdomain\", build().domain());\n }\n\n @Test\n public void namespace() {\n assertEquals(\"NAMESPACE\", build().namespace());\n }\n\n @Test\n public void id() {\n assertEquals(\"testid\", build().id());\n }\n\n @Test\n public void hasProperty() {\n checkHasProperty(build());\n\n }\n\n public static void checkHasProperty(PluginConfig config) {\n assertTrue(config.hasProperty(\"enabled\"));\n assertTrue(config.hasProperty(\"tcp.enabled\"));\n assertTrue(config.hasProperty(\"http.enabled\"));\n assertFalse(config.hasProperty(\"http.enabled.cccc\"));\n }\n\n @Test\n public void getString() {\n checkString(build());\n }\n\n public static void checkString(PluginConfig config) {\n assertEquals(\"true\", config.getString(\"enabled\"));\n assertEquals(\"false\", config.getString(\"tcp.enabled\"));\n assertEquals(\"127\", config.getString(\"count\"));\n assertEquals(\"127.0.0.3\", config.getString(\"host\"));\n assertEquals(\"true\", config.getString(\"http.enabled\"));\n assertNull(config.getString(\"http.enabled.sss\"));\n }\n\n @Test\n public void getInt() {\n checkInt(build());\n }\n\n public static void checkInt(PluginConfig config) {\n assertEquals(127, (int) config.getInt(\"count\"));\n assertNull(config.getInt(\"enabled\"));\n assertNull(config.getInt(\"cccccccccccccc\"));\n }\n\n\n @Test\n public void getBoolean() {\n checkBoolean(build());\n }\n\n public static void checkBoolean(PluginConfig config) {\n assertTrue(config.getBoolean(\"enabled\"));\n assertFalse(config.getBoolean(\"tcp.enabled\"));\n assertFalse(config.getBoolean(\"http.enabled\"));\n assertFalse(config.getBoolean(\"http.enabled.ssss\"));\n }\n\n @Test\n public void getDouble() {\n checkDouble(build());\n }\n\n\n public static void checkDouble(PluginConfig config) {\n assertTrue(Math.abs(config.getDouble(\"double\") - 127.3) < 0.0001);\n assertTrue(Math.abs(config.getDouble(\"double_1\") - 127.2) < 0.0001);\n assertNull(config.getDouble(\"enabled\"));\n }\n\n @Test\n public void getLong() {\n checkLong(build());\n }\n\n public static void checkLong(PluginConfig config) {\n assertEquals(127l, (long) config.getLong(\"count\"));\n assertNull(config.getLong(\"enabled\"));\n assertNull(config.getLong(\"cccccccccccccc\"));\n }\n\n @Test\n public void getStringList() {\n checkStringList(build());\n }\n\n\n public static void checkStringList(PluginConfig config) {\n List list = config.getStringList(\"list\");\n assertEquals(3, list.size());\n assertEquals(\"a\", list.get(0));\n assertEquals(\"b\", list.get(1));\n assertEquals(\"c\", list.get(2));\n }\n\n\n @Test\n public void addChangeListener() {\n PluginConfig config = build();\n config.addChangeListener((oldConfig, newConfig) -> {\n });\n AtomicInteger count = new AtomicInteger(0);\n config.foreachConfigChangeListener(new Consumer() {\n @Override\n public void accept(PluginConfigChangeListener listener) {\n count.incrementAndGet();\n }\n });\n assertEquals(1, count.get());\n }\n\n @Test\n public void getConfigChangeListener() {\n addChangeListener();\n }\n\n @Test\n public void keySet() {\n checkKeySet(build());\n }\n\n public static void checkKeySet(PluginConfig config) {\n Set set = config.keySet();\n Map source = globalSource();\n source.putAll(coverSource());\n assertEquals(set.size(), source.size());\n for (String s : set) {\n assertTrue(source.containsKey(s));\n }\n }\n\n\n public static void checkAllType(PluginConfig config) {\n checkHasProperty(config);\n checkString(config);\n checkBoolean(config);\n checkInt(config);\n checkLong(config);\n checkDouble(config);\n checkStringList(config);\n checkKeySet(config);\n }\n\n\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/PluginPropertyTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport org.junit.Test;\n\nimport java.util.HashMap;\nimport java.util.Map;\n\nimport static org.junit.Assert.*;\n\npublic class PluginPropertyTest {\n\n public PluginProperty build() {\n return new PluginProperty(\"testdomain\", \"testnamespace\", \"testId\", \"testproperty\");\n }\n\n @Test\n public void getDomain() {\n assertEquals(\"testdomain\", build().getDomain());\n }\n\n @Test\n public void getNamespace() {\n assertEquals(\"testnamespace\", build().getNamespace());\n }\n\n @Test\n public void getId() {\n assertEquals(\"testId\", build().getId());\n }\n\n @Test\n public void getProperty() {\n assertEquals(\"testproperty\", build().getProperty());\n }\n\n @Test\n public void testEquals() {\n assertTrue(build().equals(build()));\n PluginProperty property = new PluginProperty(\"testdomain\", \"testnamespace\", \"testId\", \"testproperty1\");\n assertFalse(build().equals(property));\n Map pluginPropertyStringMap = new HashMap<>();\n pluginPropertyStringMap.put(build(), \"testValue\");\n assertTrue(pluginPropertyStringMap.containsKey(build()));\n assertEquals(\"testValue\", pluginPropertyStringMap.get(build()));\n assertFalse(pluginPropertyStringMap.containsKey(property));\n }\n\n @Test\n public void testHashCode() {\n assertEquals(build().hashCode(), build().hashCode());\n PluginProperty property = new PluginProperty(\"testdomain\", \"testnamespace\", \"testId\", \"testproperty1\");\n assertNotEquals(build().hashCode(), property.hashCode());\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/PluginSourceConfigTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport org.junit.Test;\n\nimport java.util.Collections;\nimport java.util.HashMap;\nimport java.util.Map;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertNotNull;\n\npublic class PluginSourceConfigTest {\n\n public static String DOMAIN = \"testDomain\";\n public static String NAMESPACE = \"testNamespace\";\n public static String TEST_TRACE_ID = \"test-trace\";\n public static String GLOBAL_ID = TEST_TRACE_ID;\n public static String TEST_METRIC_ID = \"test-metric\";\n public static String TEST_AAA_ID = \"test-AAA\";\n\n\n public static Map getSource(String namespace, String id) {\n Map properties = PluginConfigTest.globalSource();\n Map s = new HashMap<>();\n for (Map.Entry pEntry : properties.entrySet()) {\n s.put(\"plugin.\" + DOMAIN + \".\" + namespace + \".\" + id + \".\" + pEntry.getKey(), pEntry.getValue());\n }\n return s;\n }\n\n public static Map getSource(String id) {\n return getSource(NAMESPACE, id);\n }\n\n\n public static Map getGlobal() {\n return getSource(\"global\", GLOBAL_ID);\n }\n\n public static Map buildSource() {\n Map source = getGlobal();\n source.putAll(getSource(TEST_METRIC_ID));\n source.putAll(getSource(TEST_TRACE_ID));\n source.put(\"plugin.testDomain.testssss.self.lll\", \"aaa\");\n source.put(\"plugin.testDomain.testssss.kafka.lll\", \"aaa\");\n source.put(\"plugin.testDomain.testssss.kafka.lll\", \"aaa\");\n source.put(\"plugin.testDomain.testssss.kafka.lll\", \"aaa\");\n return source;\n }\n\n PluginSourceConfig buildImpl(String namespace, String id) {\n\n Map source = buildSource();\n return PluginSourceConfig.build(DOMAIN, namespace, id, source);\n }\n\n @Test\n public void build() {\n PluginSourceConfig config = buildImpl(NAMESPACE, TEST_TRACE_ID);\n assertNotNull(config);\n\n }\n\n @Test\n public void getSource() {\n assertEquals(buildImpl(\"global\", GLOBAL_ID).getSource(), getGlobal());\n assertEquals(Collections.emptyMap(), buildImpl(NAMESPACE, TEST_AAA_ID).getSource());\n assertEquals(getSource(TEST_TRACE_ID), buildImpl(NAMESPACE, TEST_TRACE_ID).getSource());\n assertEquals(getSource(TEST_METRIC_ID), buildImpl(NAMESPACE, TEST_METRIC_ID).getSource());\n }\n\n @Test\n public void getDomain() {\n assertEquals(buildImpl(NAMESPACE, TEST_AAA_ID).getDomain(), DOMAIN);\n assertEquals(buildImpl(NAMESPACE, TEST_TRACE_ID).getDomain(), DOMAIN);\n assertEquals(buildImpl(NAMESPACE, TEST_METRIC_ID).getDomain(), DOMAIN);\n }\n\n @Test\n public void getNamespace() {\n assertEquals(buildImpl(NAMESPACE, TEST_AAA_ID).getNamespace(), NAMESPACE);\n assertEquals(buildImpl(NAMESPACE, TEST_TRACE_ID).getNamespace(), NAMESPACE);\n assertEquals(buildImpl(NAMESPACE, TEST_METRIC_ID).getNamespace(), NAMESPACE);\n }\n\n\n @Test\n public void getId() {\n assertEquals(buildImpl(NAMESPACE, TEST_AAA_ID).getId(), TEST_AAA_ID);\n assertEquals(buildImpl(NAMESPACE, TEST_TRACE_ID).getId(), TEST_TRACE_ID);\n assertEquals(buildImpl(NAMESPACE, TEST_METRIC_ID).getId(), TEST_METRIC_ID);\n }\n\n @Test\n public void getProperties() {\n\n assertEquals(buildImpl(\"global\", GLOBAL_ID).getProperties(), PluginConfigTest.globalSource());\n assertEquals(buildImpl(NAMESPACE, TEST_TRACE_ID).getProperties(), PluginConfigTest.globalSource());\n assertEquals(buildImpl(NAMESPACE, TEST_METRIC_ID).getProperties(), PluginConfigTest.globalSource());\n assertEquals(buildImpl(NAMESPACE, TEST_AAA_ID).getProperties(), new HashMap<>());\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/ValidateUtilsTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config;\n\nimport org.junit.Assert;\nimport org.junit.Test;\n\nimport java.util.Collections;\n\nimport static com.megaease.easeagent.config.ValidateUtils.*;\n\npublic class ValidateUtilsTest {\n @Test\n public void test_hasText() throws Exception {\n Configs configs = new Configs(Collections.emptyMap());\n try {\n ValidateUtils.validate(configs, \"hello\", HasText);\n Assert.fail(\"Never get here.\");\n } catch (ValidateUtils.ValidException e) {\n Assert.assertTrue(e.getMessage().contains(\"has no non-empty value\"));\n }\n }\n\n @Test\n public void test_numberInt() throws Exception {\n Configs configs = new Configs(Collections.singletonMap(\"hello\", \"test\"));\n try {\n ValidateUtils.validate(configs, \"hello\", HasText, NumberInt);\n Assert.fail(\"Never get here.\");\n } catch (ValidateUtils.ValidException e) {\n Assert.assertTrue(e.getMessage().contains(\"has no integer value\"));\n }\n }\n\n @Test\n public void test_numberInt2() throws Exception {\n Configs configs = new Configs(Collections.singletonMap(\"hello\", \"100\"));\n try {\n ValidateUtils.validate(configs, \"hello\", HasText, NumberInt);\n } catch (ValidateUtils.ValidException e) {\n Assert.fail(\"Never get here.\");\n }\n }\n\n @Test\n public void test_bool() throws Exception {\n Configs configs = new Configs(Collections.singletonMap(\"hello\", \"test\"));\n try {\n ValidateUtils.validate(configs, \"hello\", HasText, Bool);\n Assert.fail(\"Never get here.\");\n } catch (ValidateUtils.ValidException e) {\n Assert.assertTrue(e.getMessage().contains(\"has no boolean value\"));\n }\n }\n\n @Test\n public void test_bool2() throws Exception {\n Configs configs = new Configs(Collections.singletonMap(\"hello\", \"true\"));\n try {\n ValidateUtils.validate(configs, \"hello\", HasText, Bool);\n } catch (ValidateUtils.ValidException e) {\n Assert.fail(\"Never get here.\");\n }\n }\n\n}" }, { "path": "config/src/test/java/com/megaease/easeagent/config/report/ReportConfigAdapterTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.config.report;\n\nimport com.megaease.easeagent.config.GlobalConfigs;\nimport com.megaease.easeagent.plugin.utils.NoNull;\nimport org.junit.Assert;\nimport org.junit.Test;\n\nimport java.util.HashMap;\n\nimport static com.megaease.easeagent.config.report.ReportConfigConst.*;\n\npublic class ReportConfigAdapterTest {\n @Test\n public void test_zipkin_target() {\n HashMap cfg = new HashMap<>();\n String url = \"http://localhost:9411/api/v2/spans\";\n cfg.put(\"observability.tracings.output.target\", \"zipkin\");\n cfg.put(\"observability.tracings.output.target.zipkinUrl\", url);\n GlobalConfigs configs = new GlobalConfigs(cfg);\n\n String senderName = configs.getString(TRACE_SENDER_NAME);\n Assert.assertEquals(ZIPKIN_SENDER_NAME, senderName);\n Assert.assertEquals(url, configs.getString(join(TRACE_SENDER, \"url\")));\n }\n\n @Test\n public void test_console_target() {\n HashMap cfg = new HashMap<>();\n String url = \"\";\n cfg.put(\"observability.tracings.output.target\", \"zipkin\");\n cfg.put(\"observability.tracings.output.target.zipkinUrl\", url);\n GlobalConfigs configs = new GlobalConfigs(cfg);\n\n Assert.assertEquals(CONSOLE_SENDER_NAME, configs.getString(TRACE_SENDER_NAME));\n\n cfg.clear();\n cfg.put(\"observability.tracings.output.target\", \"system\");\n cfg.put(\"observability.tracings.output.topic\", \"log-tracing\");\n configs = new GlobalConfigs(cfg);\n\n Assert.assertEquals(CONSOLE_SENDER_NAME, configs.getString(TRACE_SENDER_NAME));\n }\n\n @Test\n public void test_kafka_target() {\n HashMap cfg = new HashMap<>();\n cfg.put(\"observability.tracings.output.target\", \"system\");\n cfg.put(\"observability.tracings.output.topic\", \"log-tracing\");\n cfg.put(\"observability.outputServer.bootstrapServer\", \"127.0.0.1:9092\");\n\n GlobalConfigs configs = new GlobalConfigs(cfg);\n\n Assert.assertEquals(KAFKA_SENDER_NAME, configs.getString(TRACE_SENDER_NAME));\n }\n\n @Test\n public void test_trace_output() {\n /*\n * observability.tracings.output.messageMaxBytes=999900\n * observability.tracings.output.reportThread=1\n * observability.tracings.output.queuedMaxSpans=1000\n * observability.tracings.output.queuedMaxSize=1000000\n * observability.tracings.output.messageTimeout=1000\n */\n HashMap cfg = new HashMap<>();\n cfg.put(\"observability.tracings.output.messageMaxBytes\", \"123\");\n cfg.put(\"observability.tracings.output.reportThread\", \"2\");\n\n GlobalConfigs configs = new GlobalConfigs(cfg);\n long m = configs.getLong(TRACE_ASYNC_MESSAGE_MAX_BYTES_V2);\n Assert.assertEquals(123L, m);\n m = configs.getLong(TRACE_ASYNC_REPORT_THREAD_V2);\n Assert.assertEquals(2L, m);\n\n // override by v2\n cfg.clear();\n cfg.put(\"observability.tracings.output.queuedMaxSpans\", \"123\");\n cfg.put(TRACE_ASYNC_QUEUED_MAX_SPANS_V2, \"1000\");\n\n configs = new GlobalConfigs(cfg);\n m = configs.getLong(TRACE_ASYNC_QUEUED_MAX_SPANS_V2);\n Assert.assertEquals(1000L, m);\n }\n\n @Test\n public void test_metric_global_v1() {\n /*\n * plugin.observability.global.metric.interval=30\n * plugin.observability.global.metric.topic=application-meter\n * plugin.observability.global.metric.appendType=kafka\n */\n HashMap cfg = new HashMap<>();\n cfg.put(\"plugin.observability.global.metric.interval\", \"30\");\n cfg.put(\"plugin.observability.global.metric.topic\", \"application-meter\");\n cfg.put(\"plugin.observability.global.metric.appendType\", \"kafka\");\n\n GlobalConfigs configs = new GlobalConfigs(cfg);\n Assert.assertEquals(30L, (long)NoNull.of(configs.getLong(METRIC_ASYNC_INTERVAL), 0L));\n Assert.assertEquals(\"application-meter\", configs.getString(METRIC_SENDER_TOPIC));\n Assert.assertEquals(METRIC_KAFKA_SENDER_NAME, configs.getString(METRIC_SENDER_NAME));\n }\n\n @Test\n public void test_metric_async_update() {\n HashMap cfg = new HashMap<>();\n cfg.put(\"plugin.observability.global.metric.output.messageMaxBytes\", \"100\");\n cfg.put(\"plugin.observability.access.metric.output.interval\", \"100\");\n cfg.put(METRIC_ASYNC_MESSAGE_MAX_BYTES, \"1000\");\n cfg.put(METRIC_ASYNC_INTERVAL, \"200\");\n\n GlobalConfigs configs = new GlobalConfigs(cfg);\n Assert.assertEquals(\"1000\", configs.getString(METRIC_ASYNC_MESSAGE_MAX_BYTES));\n Assert.assertEquals(\"1000\", configs.getString(join(METRIC_V2, \"access\", ASYNC_KEY, ASYNC_MSG_MAX_BYTES_KEY)));\n Assert.assertEquals(\"100\", configs.getString(join(METRIC_V2, \"access\", ASYNC_KEY, INTERVAL_KEY)));\n\n HashMap changes = new HashMap<>();\n changes.put(METRIC_ASYNC_MESSAGE_MAX_BYTES, \"2000\");\n changes.put(METRIC_ASYNC_INTERVAL, \"150\");\n configs.updateConfigs(changes);\n Assert.assertEquals(\"2000\", configs.getString(join(METRIC_V2, \"access\", ASYNC_KEY, ASYNC_MSG_MAX_BYTES_KEY)));\n Assert.assertEquals(\"2000\", configs.getString(METRIC_ASYNC_MESSAGE_MAX_BYTES));\n Assert.assertEquals(\"100\", configs.getString(join(METRIC_V2, \"access\", ASYNC_KEY, INTERVAL_KEY)));\n }\n\n @Test\n public void test_log_global() {\n HashMap cfg = new HashMap<>();\n cfg.put(\"plugin.observability.global.log.topic\", \"application-log\");\n cfg.put(\"plugin.observability.global.log.url\", \"/application-log\");\n cfg.put(\"plugin.observability.global.log.appendType\", \"kafka\");\n cfg.put(\"plugin.observability.global.log.output.messageMaxBytes\", \"100\");\n\n GlobalConfigs configs = new GlobalConfigs(cfg);\n Assert.assertEquals(\"application-log\", configs.getString(LOG_SENDER_TOPIC));\n Assert.assertEquals(KAFKA_SENDER_NAME, configs.getString(LOG_SENDER_NAME));\n Assert.assertEquals(\"100\", configs.getString(LOG_ASYNC_MESSAGE_MAX_BYTES));\n }\n\n @Test\n public void test_access_in_metric() {\n HashMap cfg = new HashMap<>();\n cfg.put(\"plugin.observability.global.log.topic\", \"application-log\");\n cfg.put(\"plugin.observability.global.log.url\", \"/application-log\");\n cfg.put(\"plugin.observability.global.log.appendType\", \"kafka\");\n cfg.put(\"plugin.observability.global.log.encoder\", \"APP_JSON\");\n cfg.put(\"plugin.observability.global.log.output.messageMaxBytes\", \"100\");\n cfg.put(\"plugin.observability.access.log.topic\", \"access-log\");\n cfg.put(\"plugin.observability.access.log.encoder\", \"LOG_ACCESS_JSON\");\n cfg.put(\"plugin.observability.access.metric.encoder\", \"ACCESS_JSON\");\n\n GlobalConfigs configs = new GlobalConfigs(cfg);\n Assert.assertEquals(\"application-log\", configs.getString(LOG_SENDER_TOPIC));\n Assert.assertEquals(\"access-log\", configs.getString(LOG_ACCESS_SENDER_TOPIC));\n Assert.assertEquals(\"ACCESS_JSON\", configs.getString(LOG_ACCESS_ENCODER));\n Assert.assertEquals(KAFKA_SENDER_NAME, configs.getString(LOG_ACCESS_SENDER_NAME));\n Assert.assertEquals(\"100\", configs.getString(LOG_ASYNC_MESSAGE_MAX_BYTES));\n }\n\n @Test\n public void test_access_log_update() {\n HashMap cfg = new HashMap<>();\n cfg.put(\"plugin.observability.global.log.topic\", \"application-log\");\n cfg.put(\"plugin.observability.global.log.url\", \"/application-log\");\n cfg.put(\"plugin.observability.global.log.appendType\", \"kafka\");\n cfg.put(\"plugin.observability.global.log.encoder\", \"APP_JSON\");\n cfg.put(\"plugin.observability.global.log.output.messageMaxBytes\", \"100\");\n cfg.put(\"plugin.observability.access.log.topic\", \"access-log\");\n cfg.put(\"plugin.observability.access.log.encoder\", \"LOG_ACCESS_JSON\");\n\n GlobalConfigs configs = new GlobalConfigs(cfg);\n Assert.assertEquals(\"application-log\", configs.getString(LOG_SENDER_TOPIC));\n Assert.assertEquals(\"access-log\", configs.getString(LOG_ACCESS_SENDER_TOPIC));\n Assert.assertEquals(\"LOG_ACCESS_JSON\", configs.getString(LOG_ACCESS_ENCODER));\n Assert.assertEquals(KAFKA_SENDER_NAME, configs.getString(LOG_ACCESS_SENDER_NAME));\n Assert.assertEquals(\"100\", configs.getString(LOG_ASYNC_MESSAGE_MAX_BYTES));\n\n // update\n HashMap changes = new HashMap<>();\n changes.put(\"plugin.observability.access.log.appendType\", CONSOLE_SENDER_NAME);\n configs.updateConfigs(changes);\n Assert.assertEquals(CONSOLE_SENDER_NAME, configs.getString(LOG_ACCESS_SENDER_NAME));\n\n changes.put(\"plugin.observability.access.metric.appendType\", KAFKA_SENDER_NAME);\n changes.put(\"plugin.observability.access.log.appendType\", CONSOLE_SENDER_NAME);\n configs.updateConfigs(changes);\n Assert.assertEquals(KAFKA_SENDER_NAME, configs.getString(LOG_ACCESS_SENDER_NAME));\n }\n\n @Test\n public void test_log_global_async() {\n HashMap cfg = new HashMap<>();\n cfg.put(\"plugin.observability.global.log.topic\", \"application-log\");\n cfg.put(\"plugin.observability.global.log.url\", \"/application-log\");\n cfg.put(\"plugin.observability.global.log.appendType\", \"kafka\");\n cfg.put(\"plugin.observability.global.log.output.messageMaxBytes\", \"100\");\n cfg.put(\"plugin.observability.access.log.topic\", \"access-log\");\n cfg.put(\"plugin.observability.access.log.encoder\", \"LOG_ACCESS_JSON\");\n\n cfg.put(LOG_ASYNC_MESSAGE_MAX_BYTES, \"1000\");\n cfg.put(LOG_ASYNC_QUEUED_MAX_LOGS, \"200\");\n\n GlobalConfigs configs = new GlobalConfigs(cfg);\n Assert.assertEquals(\"application-log\", configs.getString(LOG_SENDER_TOPIC));\n Assert.assertEquals(\"access-log\", configs.getString(LOG_ACCESS_SENDER_TOPIC));\n Assert.assertEquals(\"LOG_ACCESS_JSON\", configs.getString(LOG_ACCESS_ENCODER));\n Assert.assertEquals(KAFKA_SENDER_NAME, configs.getString(LOG_ACCESS_SENDER_NAME));\n\n Assert.assertEquals(\"1000\", configs.getString(LOG_ASYNC_MESSAGE_MAX_BYTES));\n Assert.assertEquals(\"1000\", configs.getString(join(LOG_ACCESS, ASYNC_KEY, ASYNC_MSG_MAX_BYTES_KEY)));\n }\n\n @Test\n public void test_log_async_update() {\n HashMap cfg = new HashMap<>();\n cfg.put(\"plugin.observability.global.log.output.messageMaxBytes\", \"100\");\n cfg.put(\"plugin.observability.access.log.output.queuedMaxLogs\", \"100\");\n cfg.put(LOG_ASYNC_MESSAGE_MAX_BYTES, \"1000\");\n cfg.put(LOG_ASYNC_QUEUED_MAX_LOGS, \"200\");\n\n GlobalConfigs configs = new GlobalConfigs(cfg);\n Assert.assertEquals(\"1000\", configs.getString(LOG_ASYNC_MESSAGE_MAX_BYTES));\n Assert.assertEquals(\"1000\", configs.getString(join(LOG_ACCESS, ASYNC_KEY, ASYNC_MSG_MAX_BYTES_KEY)));\n Assert.assertEquals(\"100\", configs.getString(join(LOG_ACCESS, ASYNC_KEY, ASYNC_QUEUE_MAX_LOGS_KEY)));\n\n HashMap changes = new HashMap<>();\n changes.put(LOG_ASYNC_MESSAGE_MAX_BYTES, \"2000\");\n changes.put(LOG_ASYNC_QUEUED_MAX_LOGS, \"150\");\n configs.updateConfigs(changes);\n Assert.assertEquals(\"2000\", configs.getString(join(LOG_ACCESS, ASYNC_KEY, ASYNC_MSG_MAX_BYTES_KEY)));\n Assert.assertEquals(\"2000\", configs.getString(LOG_ASYNC_MESSAGE_MAX_BYTES));\n Assert.assertEquals(\"100\", configs.getString(join(LOG_ACCESS, ASYNC_KEY, ASYNC_QUEUE_MAX_LOGS_KEY)));\n }\n}\n" }, { "path": "config/src/test/java/com/megaease/easeagent/config/yaml/YamlReaderTest.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.config.yaml;\n\nimport org.junit.Assert;\nimport org.junit.Test;\n\nimport java.io.IOException;\nimport java.io.InputStream;\nimport java.util.Collections;\nimport java.util.HashMap;\nimport java.util.Map;\nimport java.util.Properties;\n\npublic class YamlReaderTest {\n\n private static final String CONFIG_FILE_YAML = \"agent.yaml\";\n private static final String CONFIG_FILE_PROPERTIES = \"agent.properties\";\n\n @Test\n public void testCompress() {\n InputStream inputYaml = Thread.currentThread().getContextClassLoader().getResourceAsStream(CONFIG_FILE_YAML);\n Map yamlMap = new YamlReader().load(inputYaml).compress();\n\n InputStream inputProperties = YamlReaderTest.class.getClassLoader().getResourceAsStream(CONFIG_FILE_PROPERTIES);\n Map yamlProperties = extractPropsMap(inputProperties);\n\n yamlMap.forEach((k, v) -> {\n Assert.assertTrue(yamlProperties.containsKey(k));\n if (!k.equals(\"name\")) {\n Assert.assertEquals(v, yamlProperties.get(k));\n }\n });\n\n yamlProperties.forEach((k, v) -> {\n Assert.assertTrue(yamlMap.containsKey(k));\n if (!k.equals(\"name\")) {\n Assert.assertEquals(v, yamlMap.get(k));\n }\n });\n }\n\n private static Map extractPropsMap(InputStream in) {\n Properties properties = new Properties();\n try {\n properties.load(in);\n HashMap map = new HashMap<>();\n for (String one : properties.stringPropertyNames()) {\n map.put(one, properties.getProperty(one));\n }\n return map;\n } catch (IOException e) {\n e.printStackTrace();\n }\n return Collections.emptyMap();\n }\n\n}\n" }, { "path": "config/src/test/resources/agent.properties", "content": "name=demo-service\nsystem=demo-system\n\nuser.list=a,b\n\n### http server\n# When the enabled value = false, agent will not start the http server\n# You can use -Deaseagent.server.enabled=[true | false] to override.\neaseagent.server.enabled=true\n# http server port. You can use -Deaseagent.server.port=[port] to override.\neaseagent.server.port=9900\n# Enable health/readiness\neaseagent.health.readiness.enabled=true\n# forwarded headers page\n# Pass-through headers from the root process all the way to the end\n# format: easeagent.progress.forwarded.headers={headerName}\n#easeagent.progress.forwarded.headers=X-Forwarded-For\n#easeagent.progress.forwarded.headers=X-Location,X-Mesh-Service-Canary,X-Phone-Os\n###\n### default tracings reporter configuration\n###\n# sampledType:\n## counting: percentage sampling, sampled limit 0.01 to 1, 1 is always sample, 0 is never sample, 0.1 is ten samples per hundred\n## rate_limiting: traces per second, sampled >= 0, 0 is never sample, 10 is max 10 traces per second\n## boundary: percentage sampling by traceId, sampled limit 0.0001 to 1, 1 is always sample, 0 is never sample\n## if sampled=0.001, when (traceId^random)%10000<=(0.001*10000) sampled\n## sampledType must be used with sampled, otherwise the default value is used Sampler.ALWAYS_SAMPLE\nobservability.tracings.sampledType=\nobservability.tracings.sampled=1\n# get header from response headers then tag to tracing span\n# format: observability.tracings.tag.response.headers.{key}={value}\n# support ease mesh\n# X-EG-Circuit-Breaker\n# X-EG-Retryer\n# X-EG-Rate-Limiter\n# X-EG-Time-Limiter\nobservability.tracings.tag.response.headers.eg.0=X-EG-Circuit-Breaker\nobservability.tracings.tag.response.headers.eg.1=X-EG-Retryer\nobservability.tracings.tag.response.headers.eg.2=X-EG-Rate-Limiter\nobservability.tracings.tag.response.headers.eg.3=X-EG-Time-Limiter\n\n# -------------------- plugin global config ---------------------\nplugin.observability.global.tracing.enabled=true\nplugin.observability.global.metric.enabled=true\nplugin.observability.global.metric.interval=30\nplugin.observability.global.metric.topic=application-meter\n# plugin.observability.global.metric.appendType=console\n## output by http\n#plugin.observability.global.metric.appendType=http\n# add service name to header enabled by name for easemesh\nplugin.integrability.global.addServiceNameHead.enabled=true\n# redirect the middleware address when env has address, see: com.megaease.easeagent.plugin.api.middleware.RedirectProcessor\n# about redirect: jdbc, kafka, rabbitmq, redis,\nplugin.integrability.global.redirect.enabled=true\n# forwarded headers enabled.\n# headers see config: easeagent.progress.forwarded.headers.???=???\nplugin.integrability.global.forwarded.enabled=true\nplugin.hook.global.foundation.enabled=true\n\n\n# ----------------------------------------------\n# if the plugin configuration is consistent with the global namespace,\n# do not add configuration items not commented out in this default configuration file.\n# otherwise, they can not be overridden by Global configuration in user's configuration file.\n\n#\n# -------------------- jvm ---------------------\n# plugin.observability.jvmGc.metric.enabled=true\n# plugin.observability.jvmGc.metric.interval=30\nplugin.observability.jvmGc.metric.topic=platform-metrics\nplugin.observability.jvmGc.metric.url=/platform-metrics\n# plugin.observability.jvmGc.metric.appendType=kafka\n# plugin.observability.jvmMemory.metric.enabled=true\n# plugin.observability.jvmMemory.metric.interval=30\nplugin.observability.jvmMemory.metric.topic=platform-metrics\nplugin.observability.jvmMemory.metric.url=/platform-metrics\n# plugin.observability.jvmMemory.metric.appendType=kafka\n#\n# -------------------- async ---------------------\n# plugin.observability.async.tracing.enabled=true\n#\n# -------------------- elasticsearch redirect ---------------------\n# plugin.integrability.elasticsearch.redirect.enabled=true\n# plugin.observability.elasticsearch.tracing.enabled=true\n# elasticsearch metric\n# plugin.observability.elasticsearch.metric.enabled=true\n# plugin.observability.elasticsearch.metric.interval=30\nplugin.observability.elasticsearch.metric.topic=platform-metrics\nplugin.observability.elasticsearch.metric.url=/platform-metrics\n# plugin.observability.elasticsearch.metric.appendType=kafka\n#\n# -------------------- httpServlet ---------------------\n# plugin.observability.httpServlet.tracing.enabled=true\n# plugin.observability.httpServlet.metric.enabled=true\n# plugin.observability.httpServlet.metric.interval=30\nplugin.observability.httpServlet.metric.topic=application-metrics\nplugin.observability.httpServlet.metric.url=/application-metrics\n# plugin.observability.httpServlet.metric.appendType=kafka\n#\n# -------------------- jdbc ---------------------\n## jdbc tracing\n# plugin.observability.jdbc.tracing.enabled=true\n# jdbcStatement metric\n# plugin.observability.jdbcStatement.metric.enabled=true\n# plugin.observability.jdbcStatement.metric.interval=30\nplugin.observability.jdbcStatement.metric.topic=application-metrics\nplugin.observability.jdbcStatement.metric.url=/application-metrics\n# plugin.observability.jdbcStatement.metric.appendType=kafka\n## jdbcConnection metric\n# plugin.observability.jdbcConnection.metric.enabled=true\n# plugin.observability.jdbcConnection.metric.interval=30\nplugin.observability.jdbcConnection.metric.topic=application-metrics\nplugin.observability.jdbcConnection.metric.url=/application-metrics\n# plugin.observability.jdbcConnection.metric.appendType=kafka\n## sql compress\n## compress.enabled=true, can use md5Dictionary to compress\n## compress.enabled=false, use original sql\nplugin.observability.jdbc.sql.compress.enabled=true\n\n## md5Dictionary metric\n# plugin.observability.md5Dictionary.metric.enabled=true\n# plugin.observability.md5Dictionary.metric.interval=30\nplugin.observability.md5Dictionary.metric.topic=application-metrics\nplugin.observability.md5Dictionary.metric.url=/application-metrics\n# plugin.observability.md5Dictionary.metric.appendType=kafka\n## jdbc redirect\n# plugin.integrability.jdbc.redirect.enabled=true\n#\n# -------------------- kafka ---------------------\n# kafka tracing\n# plugin.observability.kafka.tracing.enabled=true\n# kafka metric\n# plugin.observability.kafka.metric.enabled=true\n# plugin.observability.kafka.metric.interval=30\nplugin.observability.kafka.metric.topic=platform-metrics\nplugin.observability.kafka.metric.url=/platform-metrics\n# plugin.observability.kafka.metric.appendType=kafka\n# kafka redirect\n# plugin.integrability.kafka.redirect.enabled=true\n#\n# -------------------- rabbitmq ---------------------\n# rabbitmq tracing\n# plugin.observability.rabbitmq.tracing.enabled=true\n# rabbitmq metric\n# plugin.observability.rabbitmq.metric.enabled=true\n# plugin.observability.rabbitmq.metric.interval=30\nplugin.observability.rabbitmq.metric.topic=platform-metrics\nplugin.observability.rabbitmq.metric.url=/platform-metrics\n# plugin.observability.rabbitmq.metric.appendType=kafka\n# rabbitmq redirect\n# plugin.integrability.rabbitmq.redirect.enabled=true\n#\n# -------------------- redis ---------------------\n# redis tracing\n# plugin.observability.redis.tracing.enabled=true\n# redis metric\n# plugin.observability.redis.metric.enabled=true\n# plugin.observability.redis.metric.interval=30\nplugin.observability.redis.metric.topic=platform-metrics\nplugin.observability.redis.metric.url=/platform-metrics\n# plugin.observability.redis.metric.appendType=kafka\n# redis redirect\n# plugin.integrability.redis.redirect.enabled=true\n#\n# -------------------- springGateway ---------------------\n# springGateway tracing\n# plugin.observability.springGateway.tracing.enabled=true\n# springGateway metric\n# plugin.observability.springGateway.metric.enabled=true\n# plugin.observability.springGateway.metric.interval=30\nplugin.observability.springGateway.metric.topic=application-metrics\nplugin.observability.springGateway.metric.url=/application-metrics\n# plugin.observability.springGateway.metric.appendType=kafka\n#\n# -------------------- request ---------------------\n## httpclient tracing：httpclient and httpclient5\n# plugin.observability.httpclient.tracing.enabled=true\n## okHttp tracing\n# plugin.observability.okHttp.tracing.enabled=true\n## webclient tracing\n# plugin.observability.webclient.tracing.enabled=true\n## feignClient tracing\n# plugin.observability.feignClient.tracing.enabled=true\n## restTemplate tracing\n# plugin.observability.restTemplate.tracing.enabled=true\n#\n# -------------------- access ---------------------\n## access: servlet and spring gateway\n# plugin.observability.access.metric.enabled=true\n# plugin.observability.access.metric.interval=30\nplugin.observability.access.metric.topic=application-log\nplugin.observability.access.metric.url=/application-log\n# plugin.observability.access.metric.appendType=kafka\n#\n# -------------------- service name ---------------------\n## add service name to header by name for easemesh. default name: X-Mesh-RPC-Service\n# plugin.integrability.serviceName.addServiceNameHead.propagate.head=X-Mesh-RPC-Service\n#\n# -------------------- mongodb ---------------------\n## mongodb tracing\n# plugin.observability.mongodb.tracing.enabled=true\n## mongodb metric\n# plugin.observability.mongodb.metric.enabled=true\n# plugin.observability.mongodb.metric.interval=30\nplugin.observability.mongodb.metric.topic=platform-metrics\nplugin.observability.mongodb.metric.url=/platform-metrics\n# plugin.observability.mongodb.metric.appendType=kafka\n## mongodb redirect\n# plugin.integrability.mongodb.redirect.enabled=true\n## mongodb foundation\n# plugin.hook.mongodb.foundation.enabled=true\n\n# -------------- output ------------------\n## http/kafka/zipkin server host and port for tracing and metric\n###### example ######\n## http: [http|https]://127.0.0.1:8080/report\n## kafka: 192.168.1.2:9092, 192.168.1.3:9092, 192.168.1.3:9092\n## zipkin: [http|https]://127.0.0.1:8080/zipkin\n\nreporter.outputServer.bootstrapServer=127.0.0.1:9092\nreporter.outputServer.appendType=console\nreporter.outputServer.timeout=1000\n\n## enabled=false: disable output tracing and metric\n## enabled=true: output tracing and metric\nreporter.outputServer.enabled=true\n\n## username and password for http basic auth\nreporter.outputServer.username=\nreporter.outputServer.password=\n## enable=false: disable mtls\n## enable=true: enable tls\n## key, cert, ca_cert is enabled when tls.enable=true\nreporter.outputServer.tls.enable=false\nreporter.outputServer.tls.key=\nreporter.outputServer.tls.cert=\nreporter.outputServer.tls.ca_cert=\n\n\n# --- redefine to output properties\nreporter.log.output.messageMaxBytes=999900\nreporter.log.output.reportThread=1\nreporter.log.output.queuedMaxSpans=1000\nreporter.log.output.queuedMaxSize=1000000\nreporter.log.output.messageTimeout=1000\n\n## sender.appendType config\n## [http] send to http server\n## [kafka] send to kafka\n## [console] send to console\n## reporter.log.sender.appendType=console\n## enabled=true:\n# reporter.log.sender.enabled=true\n# reporter.log.sender.url=/application-log\n\n\n## sender.appendType config\n## [http] send to http server\n## [kafka] send to kafka\n## [console] send to console\n# reporter.tracing.sender.appendType=http\n# reporter.tracing.sender.appendType=console\n\n## enabled=true:\nreporter.tracing.sender.enabled=true\n\n## url is only used in http\n## append to outputServer.bootstrapServer\n###### example ######\n## reporter.outputServer.bootstrapServer=http://127.0.0.1:8080/report\n## reporter.tracing.sender.url=/tracing\n## final output url: http://127.0.0.1:8080/report/tracing\n## if url is start with [http|https], url override reporter.outputServer.bootstrapServer\n###### example ######\n## reporter.outputServer.bootstrapServer=http://127.0.0.1:8080/report\n## reporter.tracing.sender.url=http://127.0.0.10:9090/tracing\n## final output url: http://127.0.0.10:9090/tracing\nreporter.tracing.sender.url=/application-tracing-log\n\n## topic for kafka use\nreporter.tracing.sender.topic=log-tracing\n\nreporter.tracing.encoder=SpanJsonEncoder\n\n# --- redefine to output properties\nreporter.tracing.output.messageMaxBytes=999900\nreporter.tracing.output.reportThread=1\nreporter.tracing.output.queuedMaxSpans=1000\nreporter.tracing.output.queuedMaxSize=1000000\nreporter.tracing.output.messageTimeout=1000\n\n## sender.appendType config\n## [http] send to http server\n## [metricKafka] send to kafka\n## [console] send to console\n#reporter.metric.sender.appendType=http\n#reporter.metric.sender.appendType=console\n\n## url is only used in http\n## append to outputServer.bootstrapServer\n###### example ######\n## reporter.outputServer.bootstrapServer=http://127.0.0.1:8080/report\n## reporter.metric.sender.url=/metric\n## final output url: http://127.0.0.1:8080/report/metric\n## if url is start with [http|https], url override reporter.outputServer.bootstrapServer\n###### example ######\n## reporter.outputServer.bootstrapServer=http://127.0.0.1:8080/report\n## reporter.metric.sender.url=http://127.0.0.10:9090/metric\n## final output url: http://127.0.0.10:9090/metric\n#reporter.metric.sender.url=/metrics\n\n## topic for kafka use\nreporter.metric.sender.topic=application-meter\n\n#reporter.metric.encoder=MetricJsonEncoder\nreporter.metric.output.interval=30\n" }, { "path": "config/src/test/resources/agent.yaml", "content": "name: test-service\nsystem: demo-system\n\nuser:\n list:\n - \"a\"\n - \"b\"\n\n### http server\neaseagent:\n server:\n # When the enabled value = false, agent will not start the http server\n # You can use -Deaseagent.server.enabled=[true | false] to override.\n enabled: true\n # http server port. You can use -Deaseagent.server.port=[port] to override.\n port: 9900\n health:\n readiness:\n # Enable health/readiness\n enabled: true\n# progress:\n# forwarded:\n# headers: X-Forwarded-For\n# headers: X-Location,X-Mesh-Service-Canary,X-Phone-Os\n\n###\n### default tracings reporter configuration\n###\n# sampledType:\n## counting: percentage sampling, sampled limit 0.01 to 1, 1 is always sample, 0 is never sample, 0.1 is ten samples per hundred\n## rate_limiting: traces per second, sampled >= 0, 0 is never sample, 10 is max 10 traces per second\n## boundary: percentage sampling by traceId, sampled limit 0.0001 to 1, 1 is always sample, 0 is never sample\n## if sampled=0.001, when (traceId^random)%10000<=(0.001*10000) sampled\n## sampledType must be used with sampled, otherwise the default value is used Sampler.ALWAYS_SAMPLE\n#\n# get header from response headers then tag to tracing span\n# format: observability.tracings.tag.response.headers.{key}={value}\n# support ease mesh\n# X-EG-Circuit-Breaker\n# X-EG-Retryer\n# X-EG-Rate-Limiter\n# X-EG-Time-Limiter\nobservability:\n tracings:\n sampledType: \"\"\n sampled: 1\n tag:\n response:\n headers:\n eg:\n 0: X-EG-Circuit-Breaker\n 1: X-EG-Retryer\n 2: X-EG-Rate-Limiter\n 3: X-EG-Time-Limiter\n\nplugin:\n hook:\n global:\n foundation:\n enabled: true\n integrability:\n global:\n # add service name to header enabled by name for easemesh\n addServiceNameHead:\n enabled: true\n # forwarded headers enabled.\n # headers see config: easeagent.progress.forwarded.headers.???=???\n forwarded:\n enabled: true\n # redirect the middleware address when env has address, see: com.megaease.easeagent.plugin.api.middleware.RedirectProcessor\n # about redirect: jdbc, kafka, rabbitmq, redis,\n redirect:\n enabled: true\n # -------------------- service name ---------------------\n ## add service name to header by name for easemesh. default name: X-Mesh-RPC-Service\n # serviceName:\n # addServiceNameHead:\n # propagate:\n # head: X-Mesh-RPC-Service\n observability:\n # -------------------- async ---------------------\n # async:\n # tracing:\n # enabled: true\n # -------------------- access ---------------------\n access:\n metric:\n # enabled: true\n # interval: 30\n topic: application-log\n url: /application-log\n # appendType: kafka\n # -------------------- elasticsearch redirect ---------------------\n elasticsearch:\n metric:\n # enabled: true\n # interval: 30\n topic: platform-metrics\n url: /platform-metrics\n # appendType: kafka\n # -------------------- plugin global config ---------------------\n global:\n metric:\n enabled: true\n interval: 30\n topic: application-meter\n ## output by http\n # appendType: console\n # appendType: http\n tracing:\n enabled: true\n # -------------------- httpServlet ---------------------\n httpServlet:\n metric:\n # enabled: true\n # interval: 30\n topic: application-metrics\n url: /application-metrics\n # appendType: kafka\n # tracing:\n # enabled: true\n # -------------------- request ---------------------\n ## httpclient tracing, httpclient and httpclient5\n # httpclient:\n # tracing:\n # enabled: true\n ## okHttp tracing\n # okHttp:\n # tracing:\n # enabled: true\n ## webclient tracing\n # webclient:\n # tracing:\n # enabled: true\n ## feignClient tracing\n # feignClient:\n # tracing:\n # enabled: true\n ## restTemplate tracing\n # restTemplate:\n # tracing:\n # enabled: true\n # -------------------- jdbc ---------------------\n jdbc:\n # tracing:\n # enabled: true\n ## sql compress\n ## compress.enabled=true, can use md5Dictionary to compress\n ## compress.enabled=false, use original sql\n sql:\n compress:\n enabled: true\n ## jdbc redirect\n # redirect:\n # enabled: true\n jdbcConnection:\n metric:\n # enabled: true\n # interval: 30\n topic: application-metrics\n url: /application-metrics\n # appendType: kafka\n jdbcStatement:\n metric:\n # enabled: true\n # interval: 30\n topic: application-metrics\n url: /application-metrics\n # appendType: kafka\n # ----------------------------------------------\n # if the plugin configuration is consistent with the global namespace,\n # do not add configuration items not commented out in this default configuration file.\n # otherwise, they can not be overridden by Global configuration in user's configuration file.\n # -------------------- jvm ---------------------\n jvmGc:\n metric:\n # enabled: true\n # interval: 30\n topic: platform-metrics\n url: /platform-metrics\n # appendType: kafka\n jvmMemory:\n metric:\n # enabled: true\n # interval: 30\n topic: platform-metrics\n url: /platform-metrics\n # appendType: kafka\n # -------------------- kafka ---------------------\n kafka:\n # kafka tracing\n # tracing:\n # enabled: true\n # kafka metric\n metric:\n # enabled: true\n # interval: 30\n topic: platform-metrics\n url: /platform-metrics\n # appendType: kafka\n # kafka redirect\n # redirect:\n # enabled: true\n ## md5Dictionary metric\n md5Dictionary:\n metric:\n # enabled: true\n # interval: 30\n topic: application-metrics\n url: /application-metrics\n # appendType: kafka\n # -------------------- mongodb ---------------------\n mongodb:\n ## mongodb tracing\n # tracing:\n # enabled: true\n ## mongodb metric\n metric:\n # enabled: true\n # interval: 30\n topic: platform-metrics\n url: /platform-metrics\n # appendType: kafka\n ## mongodb redirect\n # redirect:\n # enabled: true\n ## mongodb foundation\n # foundation:\n # enabled: true\n # -------------------- rabbitmq ---------------------\n rabbitmq:\n # rabbitmq tracing\n # tracing:\n # enabled: true\n # rabbitmq metric\n metric:\n # enabled: true\n # interval: 30\n topic: platform-metrics\n url: /platform-metrics\n # appendType: kafka\n # rabbitmq redirect\n # redirect:\n # enabled: true\n # -------------------- redis ---------------------\n redis:\n # redis tracing\n # tracing:\n # enabled: true\n # redis metric\n metric:\n # enabled: true\n # interval: 30\n topic: platform-metrics\n url: /platform-metrics\n # redis redirect\n # redirect:\n # enabled: true\n # -------------------- springGateway ---------------------\n springGateway:\n # springGateway tracing\n # tracing:\n # enabled: true\n metric:\n # enabled: true\n # interval: 30\n topic: application-metrics\n url: /application-metrics\n# appendType: kafka\n\n# -------------- output ------------------\n## http/kafka/zipkin server host and port for tracing and metric\n###### example ######\n## http: [http|https]://127.0.0.1:8080/report\n## kafka: 192.168.1.2:9092, 192.168.1.3:9092, 192.168.1.3:9092\n## zipkin: [http|https]://127.0.0.1:8080/zipkin\n\nreporter:\n outputServer:\n appendType: console\n bootstrapServer: 127.0.0.1:9092\n ## enabled=false: disable output tracing and metric\n ## enabled=true: output tracing and metric\n enabled: true\n ## username and password for http basic auth\n username: ''\n password: ''\n timeout: 1000\n ## enable=false: disable mtls\n ## enable=true: enable tls\n ## key, cert, ca_cert is enabled when tls.enable=true\n tls:\n enable: false\n key: ''\n cert: ''\n ca_cert: ''\n # --- redefine to output properties\n log:\n output:\n messageMaxBytes: 999900\n messageTimeout: 1000\n queuedMaxSize: 1000000\n queuedMaxSpans: 1000\n reportThread: 1\n ## sender.appendType config\n ## [http] send to http server\n ## [kafka] send to kafka\n ## [console] send to console\n # sender:\n # enabled: true\n # url: /application-log\n # appendType: console\n metric:\n ## reporter.metric.encoder=MetricJsonEncoder\n output:\n interval: 30\n ## topic for kafka use\n sender:\n topic: application-meter\n tracing:\n encoder: SpanJsonEncoder\n # --- redefine to output properties\n output:\n messageMaxBytes: 999900\n messageTimeout: 1000\n queuedMaxSize: 1000000\n queuedMaxSpans: 1000\n reportThread: 1\n ## sender.appendType config\n ## [http] send to http server\n ## [kafka] send to kafka\n ## [console] send to console\n sender:\n enabled: true\n ## topic for kafka use\n topic: log-tracing\n url: /application-tracing-log\n# appendType: http\n# appendType: console\n" }, { "path": "config/src/test/resources/user-spec.properties", "content": "name=user-spec\nsystem=system-spec\n" }, { "path": "config/src/test/resources/user-spec2.properties", "content": "name=user-spec2\nsystem=system-spec2\n" }, { "path": "config/src/test/resources/user.properties", "content": "name=demo-user\nsystem=demo-system\n\n## topic for kafka use\nreporter.metric.sender.topic=application-meter\n\n#reporter.metric.encoder=MetricJsonEncoder\nreporter.metric.output.interval=30\n" }, { "path": "context/pom.xml", "content": "\n\n\n\n \n easeagent\n com.megaease.easeagent\n 2.3.0\n \n 4.0.0\n\n context\n\n\n \n \n com.megaease.easeagent\n plugin-api\n ${project.version}\n \n \n com.megaease.easeagent\n config\n ${project.version}\n \n \n com.megaease.easeagent\n log4j2-api\n \n \n com.megaease.easeagent\n log4j2-mock\n ${project.version}\n test\n \n \n com.megaease.easeagent\n config-mock\n ${project.version}\n test\n \n \n\n" }, { "path": "context/src/main/java/com/megaease/easeagent/context/AsyncContextImpl.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.plugin.api.Cleaner;\nimport com.megaease.easeagent.plugin.api.InitializeContext;\nimport com.megaease.easeagent.plugin.api.context.AsyncContext;\nimport com.megaease.easeagent.plugin.api.trace.SpanContext;\n\nimport java.util.HashMap;\nimport java.util.Map;\nimport java.util.Objects;\nimport java.util.function.Supplier;\n\npublic class AsyncContextImpl implements AsyncContext {\n private final SpanContext spanContext;\n private final Map context;\n private final Supplier supplier;\n\n private AsyncContextImpl(SpanContext spanContext, Map context, Supplier supplier) {\n this.spanContext = Objects.requireNonNull(spanContext, \"spanContext must not be null\");\n this.context = Objects.requireNonNull(context, \"context must not be null\");\n this.supplier = Objects.requireNonNull(supplier, \"supplier must not be null\");\n }\n\n public static AsyncContextImpl build(SpanContext spanContext,\n Supplier supplier,\n Map context) {\n Map contextMap = context == null ? new HashMap<>() : new HashMap<>(context);\n return new AsyncContextImpl(spanContext, contextMap, supplier);\n }\n\n @Override\n public boolean isNoop() {\n return false;\n }\n\n @Override\n public SpanContext getSpanContext() {\n return spanContext;\n }\n\n @Override\n public Cleaner importToCurrent() {\n return supplier.get().importAsync(this);\n }\n\n @Override\n public Map getAll() {\n return context;\n }\n\n @Override\n @SuppressWarnings(\"unchecked\")\n public T get(Object o) {\n return (T) this.context.get(o);\n }\n\n @Override\n @SuppressWarnings(\"unchecked\")\n public V put(Object key, V value) {\n return (V) this.context.put(key, value);\n }\n}\n" }, { "path": "context/src/main/java/com/megaease/easeagent/context/ContextManager.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.config.Configs;\nimport com.megaease.easeagent.config.PluginConfigManager;\nimport com.megaease.easeagent.context.log.LoggerFactoryImpl;\nimport com.megaease.easeagent.context.log.LoggerMdc;\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.InitializeContext;\nimport com.megaease.easeagent.plugin.api.Reporter;\nimport com.megaease.easeagent.plugin.api.config.IPluginConfig;\nimport com.megaease.easeagent.plugin.api.context.IContextManager;\nimport com.megaease.easeagent.plugin.api.logging.ILoggerFactory;\nimport com.megaease.easeagent.plugin.api.logging.Mdc;\nimport com.megaease.easeagent.plugin.api.metric.MetricProvider;\nimport com.megaease.easeagent.plugin.api.metric.MetricRegistry;\nimport com.megaease.easeagent.plugin.api.metric.MetricRegistrySupplier;\nimport com.megaease.easeagent.plugin.api.metric.name.NameFactory;\nimport com.megaease.easeagent.plugin.api.metric.name.Tags;\nimport com.megaease.easeagent.plugin.api.trace.ITracing;\nimport com.megaease.easeagent.plugin.api.trace.TracingProvider;\nimport com.megaease.easeagent.plugin.api.trace.TracingSupplier;\nimport com.megaease.easeagent.plugin.bridge.*;\nimport com.megaease.easeagent.plugin.utils.NoNull;\n\nimport javax.annotation.Nonnull;\nimport java.util.function.Supplier;\n\npublic class ContextManager implements IContextManager {\n private static final Logger LOGGER = LoggerFactory.getLogger(ContextManager.class.getName());\n private static final ThreadLocal LOCAL_SESSION_CONTEXT = ThreadLocal.withInitial(SessionContext::new);\n private final PluginConfigManager pluginConfigManager;\n private final Supplier sessionContextSupplier;\n private final GlobalContext globalContext;\n private volatile TracingSupplier tracingSupplier = (supplier) -> null;\n private volatile MetricRegistrySupplier metric = NoOpMetrics.NO_OP_METRIC_SUPPLIER;\n\n\n private ContextManager(@Nonnull Configs conf, @Nonnull PluginConfigManager pluginConfigManager, @Nonnull ILoggerFactory loggerFactory, @Nonnull Mdc mdc) {\n this.pluginConfigManager = pluginConfigManager;\n this.sessionContextSupplier = new SessionContextSupplier();\n this.globalContext = new GlobalContext(conf, new MetricRegistrySupplierImpl(), loggerFactory, mdc);\n }\n\n public static ContextManager build(Configs conf) {\n LOGGER.info(\"build context manager.\");\n ProgressFieldsManager.init(conf);\n PluginConfigManager pluginConfigManager = PluginConfigManager.builder(conf).build();\n LoggerFactoryImpl loggerFactory = LoggerFactoryImpl.build();\n ILoggerFactory iLoggerFactory = NoOpLoggerFactory.INSTANCE;\n Mdc mdc = NoOpLoggerFactory.NO_OP_MDC_INSTANCE;\n if (loggerFactory != null) {\n iLoggerFactory = loggerFactory;\n mdc = new LoggerMdc(loggerFactory.factory().mdc());\n }\n ContextManager contextManager = new ContextManager(conf, pluginConfigManager, iLoggerFactory, mdc);\n EaseAgent.loggerFactory = contextManager.globalContext.getLoggerFactory();\n EaseAgent.loggerMdc = contextManager.globalContext.getMdc();\n EaseAgent.initializeContextSupplier = contextManager;\n EaseAgent.metricRegistrySupplier = contextManager.globalContext.getMetric();\n EaseAgent.configFactory = contextManager.pluginConfigManager;\n return contextManager;\n }\n\n @Override\n public InitializeContext getContext() {\n return this.sessionContextSupplier.get();\n }\n\n public void setTracing(@Nonnull TracingProvider tracing) {\n LOGGER.info(\"set tracing supplier function.\");\n this.tracingSupplier = tracing.tracingSupplier();\n }\n\n public void setMetric(@Nonnull MetricProvider metricProvider) {\n LOGGER.info(\"set metric supplier function.\");\n this.metric = metricProvider.metricSupplier();\n }\n\n private class SessionContextSupplier implements Supplier {\n @Override\n public InitializeContext get() {\n SessionContext context = LOCAL_SESSION_CONTEXT.get();\n ITracing tracing = context.getTracing();\n if (tracing == null || tracing.isNoop()) {\n context.setCurrentTracing(NoNull.of(tracingSupplier.get(this), NoOpTracer.NO_OP_TRACING));\n }\n if (context.getSupplier() == null) {\n context.setSupplier(this);\n }\n return context;\n }\n }\n\n public class MetricRegistrySupplierImpl implements MetricRegistrySupplier {\n\n @Override\n public MetricRegistry newMetricRegistry(IPluginConfig config, NameFactory nameFactory, Tags tags) {\n return NoNull.of(metric.newMetricRegistry(config, nameFactory, tags), NoOpMetrics.NO_OP_METRIC);\n }\n\n @Override\n public Reporter reporter(IPluginConfig config) {\n return NoNull.of(metric.reporter(config), NoOpReporter.NO_OP_REPORTER);\n }\n }\n}\n" }, { "path": "context/src/main/java/com/megaease/easeagent/context/GlobalContext.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.config.Configs;\nimport com.megaease.easeagent.plugin.api.logging.ILoggerFactory;\nimport com.megaease.easeagent.plugin.api.logging.Mdc;\nimport com.megaease.easeagent.plugin.api.metric.MetricRegistrySupplier;\n\nimport javax.annotation.Nonnull;\n\npublic class GlobalContext {\n private final Configs conf;\n private final MetricRegistrySupplier metric;\n private final ILoggerFactory loggerFactory;\n private final Mdc mdc;\n\n public GlobalContext(@Nonnull Configs conf, @Nonnull MetricRegistrySupplier metric, @Nonnull ILoggerFactory loggerFactory, @Nonnull Mdc mdc) {\n this.conf = conf;\n this.metric = metric;\n this.loggerFactory = loggerFactory;\n this.mdc = mdc;\n }\n\n\n public Configs getConf() {\n return conf;\n }\n\n public Mdc getMdc() {\n return mdc;\n }\n\n\n public ILoggerFactory getLoggerFactory() {\n return loggerFactory;\n }\n\n public MetricRegistrySupplier getMetric() {\n return metric;\n }\n}\n" }, { "path": "context/src/main/java/com/megaease/easeagent/context/ProgressFieldsManager.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.config.Configs;\nimport com.megaease.easeagent.plugin.api.ProgressFields;\nimport com.megaease.easeagent.plugin.api.config.ChangeItem;\n\nimport java.util.HashMap;\nimport java.util.Map;\nimport java.util.function.Consumer;\n\npublic class ProgressFieldsManager {\n\n private ProgressFieldsManager() {\n }\n\n public static void init(Configs configs) {\n Consumer> changeListener = ProgressFields.changeListener();\n changeListener.accept(configs.getConfigs());\n configs.addChangeListener(list -> {\n Map map = new HashMap<>();\n for (ChangeItem changeItem : list) {\n String key = changeItem.getFullName();\n if (ProgressFields.isProgressFields(key)) {\n map.put(key, changeItem.getNewValue());\n }\n changeListener.accept(map);\n }\n });\n }\n}\n" }, { "path": "context/src/main/java/com/megaease/easeagent/context/RetBound.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport java.util.HashMap;\nimport java.util.Map;\n\npublic class RetBound {\n int size;\n Map local;\n\n RetBound(int size) {\n this.size = size;\n }\n\n public int size() {\n return this.size;\n }\n\n public Object get(String key) {\n if (local == null) {\n return null;\n }\n return local.get(key);\n }\n\n public void put(String key, Object value) {\n if (local == null) {\n this.local = new HashMap<>();\n }\n this.local.put(key, value);\n }\n}\n" }, { "path": "context/src/main/java/com/megaease/easeagent/context/SessionContext.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.Cleaner;\nimport com.megaease.easeagent.plugin.api.InitializeContext;\nimport com.megaease.easeagent.plugin.api.ProgressFields;\nimport com.megaease.easeagent.plugin.api.config.IPluginConfig;\nimport com.megaease.easeagent.plugin.api.context.AsyncContext;\nimport com.megaease.easeagent.plugin.api.context.RequestContext;\nimport com.megaease.easeagent.plugin.api.trace.*;\nimport com.megaease.easeagent.plugin.bridge.NoOpCleaner;\nimport com.megaease.easeagent.plugin.bridge.NoOpIPluginConfig;\nimport com.megaease.easeagent.plugin.bridge.NoOpTracer;\nimport com.megaease.easeagent.plugin.field.NullObject;\nimport com.megaease.easeagent.plugin.utils.NoNull;\n\nimport java.util.*;\nimport java.util.function.Supplier;\n\n@SuppressWarnings(\"unused, unchecked\")\npublic class SessionContext implements InitializeContext {\n private static final Logger LOGGER = LoggerFactory.getLogger(SessionContext.class);\n private static final Setter NOOP_SETTER = (name, value) -> {\n };\n private ITracing tracing = NoOpTracer.NO_OP_TRACING;\n\n private Supplier supplier;\n private final Deque configs = new ArrayDeque<>();\n private final Deque retStack = new ArrayDeque<>();\n private final Deque retBound = new ArrayDeque<>();\n\n private final Map context = new HashMap<>();\n private final Map entered = new HashMap<>();\n private boolean hasCleaner = false;\n\n @Override\n public boolean isNoop() {\n return false;\n }\n\n public Supplier getSupplier() {\n return supplier;\n }\n\n public void setSupplier(Supplier supplier) {\n this.supplier = supplier;\n }\n\n @Override\n public Tracing currentTracing() {\n return NoNull.of(tracing, NoOpTracer.NO_OP_TRACING);\n }\n\n @Override\n public V get(Object key) {\n return change(context.get(key));\n }\n\n @Override\n public V remove(Object key) {\n return change(context.remove(key));\n }\n\n @SuppressWarnings(\"unchecked\")\n private V change(Object o) {\n return o == null ? null : (V) o;\n }\n\n @Override\n public V put(Object key, V value) {\n context.put(key, value);\n return value;\n }\n\n @Override\n public V putLocal(String key, V value) {\n assert this.retBound.peek() != null;\n this.retBound.peek().put(key, value);\n return value;\n }\n\n @Override\n public V getLocal(String key) {\n assert this.retBound.peek() != null;\n return change(this.retBound.peek().get(key));\n }\n\n @Override\n public IPluginConfig getConfig() {\n if (configs.isEmpty()) {\n LOGGER.warn(\"context.configs was empty.\");\n return NoOpIPluginConfig.INSTANCE;\n }\n return configs.peek();\n }\n\n @Override\n public void pushConfig(IPluginConfig config) {\n configs.push(config);\n }\n\n @Override\n public IPluginConfig popConfig() {\n if (configs.isEmpty()) {\n LOGGER.warn(\"context.configs was empty.\");\n return NoOpIPluginConfig.INSTANCE;\n }\n return configs.pop();\n }\n\n @Override\n public int enter(Object key) {\n Integer count = entered.get(key);\n if (count == null) {\n count = 1;\n } else {\n count++;\n }\n entered.put(key, count);\n return count;\n }\n\n @Override\n public int exit(Object key) {\n Integer count = entered.get(key);\n if (count == null) {\n return 0;\n }\n entered.put(key, count - 1);\n return count;\n }\n\n @Override\n public AsyncContext exportAsync() {\n return AsyncContextImpl.build(tracing.exportAsync(), supplier, context);\n }\n\n @Override\n public Cleaner importAsync(AsyncContext snapshot) {\n Scope scope = tracing.importAsync(snapshot.getSpanContext());\n context.putAll(snapshot.getAll());\n if (hasCleaner) {\n return new AsyncCleaner(scope, false);\n } else {\n hasCleaner = true;\n return new AsyncCleaner(scope, true);\n }\n }\n\n @Override\n public RequestContext clientRequest(Request request) {\n return tracing.clientRequest(request);\n }\n\n @Override\n public RequestContext serverReceive(Request request) {\n return tracing.serverReceive(request);\n }\n\n @Override\n public Span consumerSpan(MessagingRequest request) {\n return tracing.consumerSpan(request);\n }\n\n @Override\n public Span producerSpan(MessagingRequest request) {\n return tracing.producerSpan(request);\n }\n\n @Override\n public Span nextSpan() {\n return tracing.nextSpan();\n }\n\n /**\n * called by framework to maintain stack\n */\n @Override\n @SuppressWarnings(\"ConstantConditions\")\n public void popToBound() {\n while (this.retStack.size() > this.retBound.peek().size()) {\n this.retStack.pop();\n }\n }\n\n /**\n * called by framework to maintain stack\n */\n public void pushRetBound() {\n this.retBound.push(new RetBound(this.retStack.size()));\n }\n\n /**\n * called by framework to maintain stack\n */\n public void popRetBound() {\n this.retBound.pop();\n }\n\n @Override\n public void push(T obj) {\n if (obj == null) {\n this.retStack.push(NullObject.NULL);\n } else {\n this.retStack.push(obj);\n }\n }\n\n @Override\n @SuppressWarnings(\"ConstantConditions\")\n public T pop() {\n if (this.retStack.size() <= this.retBound.peek().size()) {\n return null;\n }\n Object o = this.retStack.pop();\n if (o == NullObject.NULL) {\n return null;\n }\n return change(o);\n }\n\n @Override\n public T peek() {\n if (this.retStack.isEmpty()) {\n return null;\n }\n Object o = this.retStack.pop();\n if (o == NullObject.NULL) {\n return null;\n }\n return change(o);\n }\n\n @Override\n public Runnable wrap(Runnable task) {\n return new CurrentContextRunnable(exportAsync(), task);\n }\n\n @Override\n public boolean isWrapped(Runnable task) {\n return task instanceof CurrentContextRunnable;\n }\n\n @Override\n public boolean isNecessaryKeys(String key) {\n return tracing.propagationKeys().contains(key);\n }\n\n @Override\n public void consumerInject(Span span, MessagingRequest request) {\n Injector injector = tracing.messagingTracing().consumerInjector();\n injector.inject(span, request);\n }\n\n @Override\n public void producerInject(Span span, MessagingRequest request) {\n Injector injector = tracing.messagingTracing().producerInjector();\n injector.inject(span, request);\n }\n\n @Override\n public void injectForwardedHeaders(Setter setter) {\n Set fields = ProgressFields.getForwardedHeaders();\n if (fields.isEmpty()) {\n return;\n }\n for (String field : fields) {\n Object o = context.get(field);\n if ((o instanceof String)) {\n setter.setHeader(field, (String) o);\n }\n }\n }\n\n @Override\n public Cleaner importForwardedHeaders(Getter getter) {\n return importForwardedHeaders(getter, NOOP_SETTER);\n }\n\n private Cleaner importForwardedHeaders(Getter getter, Setter setter) {\n Set fields = ProgressFields.getForwardedHeaders();\n if (fields.isEmpty()) {\n return NoOpCleaner.INSTANCE;\n }\n List fieldArr = new ArrayList<>(fields.size());\n for (String field : fields) {\n String o = getter.header(field);\n if (o == null) {\n continue;\n }\n fieldArr.add(field);\n this.context.put(field, o);\n setter.setHeader(field, o);\n }\n if (fieldArr.isEmpty()) {\n return NoOpCleaner.INSTANCE;\n }\n return new FieldCleaner(fieldArr);\n }\n\n\n public ITracing getTracing() {\n return tracing;\n }\n\n @Override\n public void setCurrentTracing(ITracing tracing) {\n this.tracing = NoNull.of(tracing, NoOpTracer.NO_OP_TRACING);\n }\n\n @Override\n public void clear() {\n if (!this.configs.isEmpty()) {\n this.configs.clear();\n }\n if (!this.retStack.isEmpty()) {\n this.retStack.clear();\n }\n if (!this.retBound.isEmpty()) {\n this.retBound.clear();\n }\n if (!this.context.isEmpty()) {\n this.context.clear();\n }\n if (!this.entered.isEmpty()) {\n this.entered.clear();\n }\n this.hasCleaner = false;\n }\n\n public static class CurrentContextRunnable implements Runnable {\n private final AsyncContext asyncContext;\n private final Runnable task;\n\n public CurrentContextRunnable(AsyncContext asyncContext, Runnable task) {\n this.asyncContext = asyncContext;\n this.task = task;\n }\n\n @Override\n public void run() {\n try (Cleaner cleaner = asyncContext.importToCurrent()) {\n task.run();\n }\n }\n }\n\n private class FieldCleaner implements Cleaner {\n private final List fields;\n\n public FieldCleaner(List fields) {\n this.fields = fields;\n }\n\n @Override\n public void close() {\n for (String field : fields) {\n context.remove(field);\n }\n }\n }\n\n public class AsyncCleaner implements Cleaner {\n private final Scope scope;\n private final boolean clearContext;\n\n public AsyncCleaner(Scope scope, boolean clearContext) {\n this.scope = scope;\n this.clearContext = clearContext;\n }\n\n @Override\n public void close() {\n this.scope.close();\n if (clearContext) {\n SessionContext.this.clear();\n }\n }\n }\n}\n" }, { "path": "context/src/main/java/com/megaease/easeagent/context/log/LoggerFactoryImpl.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context.log;\n\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.log4j2.api.AgentLoggerFactory;\nimport com.megaease.easeagent.plugin.api.logging.ILoggerFactory;\nimport com.megaease.easeagent.plugin.api.logging.Logger;\n\nimport javax.annotation.Nonnull;\n\npublic class LoggerFactoryImpl implements ILoggerFactory {\n private final AgentLoggerFactory loggerFactory;\n\n private LoggerFactoryImpl(@Nonnull AgentLoggerFactory loggerFactory) {\n this.loggerFactory = loggerFactory;\n }\n\n @Override\n public Logger getLogger(String name) {\n return loggerFactory.getLogger(name);\n }\n\n public AgentLoggerFactory factory() {\n return loggerFactory;\n }\n\n public static LoggerFactoryImpl build() {\n AgentLoggerFactory loggerFactory = LoggerFactory.newFactory(LoggerImpl.LOGGER_SUPPLIER, LoggerImpl.class);\n if (loggerFactory == null) {\n return null;\n }\n\n return new LoggerFactoryImpl(loggerFactory);\n }\n}\n" }, { "path": "context/src/main/java/com/megaease/easeagent/context/log/LoggerImpl.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context.log;\n\nimport com.megaease.easeagent.log4j2.api.AgentLogger;\n\nimport java.util.function.Function;\nimport java.util.logging.Logger;\n\npublic class LoggerImpl extends AgentLogger implements com.megaease.easeagent.plugin.api.logging.Logger {\n public static final Function LOGGER_SUPPLIER = LoggerImpl::new;\n\n public LoggerImpl(Logger logger) {\n super(logger);\n }\n}\n" }, { "path": "context/src/main/java/com/megaease/easeagent/context/log/LoggerMdc.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context.log;\n\nimport com.megaease.easeagent.plugin.api.logging.Mdc;\n\npublic class LoggerMdc implements Mdc {\n private final com.megaease.easeagent.log4j2.api.Mdc mdc;\n\n public LoggerMdc(com.megaease.easeagent.log4j2.api.Mdc mdc) {\n this.mdc = mdc;\n }\n\n @Override\n public void put(String key, String value) {\n mdc.put(key, value);\n }\n\n @Override\n public void remove(String key) {\n mdc.remove(key);\n }\n\n @Override\n public String get(String key) {\n return mdc.get(key);\n }\n}\n" }, { "path": "context/src/test/java/com/megaease/easeagent/context/AsyncContextImplTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.plugin.api.Cleaner;\nimport com.megaease.easeagent.plugin.bridge.NoOpTracer;\nimport org.junit.Test;\n\nimport java.util.Collections;\nimport java.util.HashMap;\nimport java.util.Map;\n\nimport static org.junit.Assert.*;\n\npublic class AsyncContextImplTest {\n\n @Test\n public void build() {\n try {\n AsyncContextImpl.build(null, null, null);\n assertTrue(\"must be throw error\", false);\n } catch (Exception e) {\n assertNotNull(e);\n }\n try {\n AsyncContextImpl.build(NoOpTracer.NO_OP_SPAN_CONTEXT, null, null);\n assertTrue(\"must be throw error\", false);\n } catch (Exception e) {\n assertNotNull(e);\n }\n\n AsyncContextImpl asyncContext = AsyncContextImpl.build(NoOpTracer.NO_OP_SPAN_CONTEXT, () -> null, null);\n assertNotNull(asyncContext.getAll());\n }\n\n @Test\n public void isNoop() {\n AsyncContextImpl asyncContext = AsyncContextImpl.build(NoOpTracer.NO_OP_SPAN_CONTEXT, () -> null, null);\n assertFalse(asyncContext.isNoop());\n }\n\n @Test\n public void getSpanContext() {\n AsyncContextImpl asyncContext = AsyncContextImpl.build(NoOpTracer.NO_OP_SPAN_CONTEXT, () -> null, null);\n assertTrue(asyncContext.getSpanContext().isNoop());\n }\n\n @Test\n public void importToCurrent() {\n String name = \"test_name\";\n String value = \"test_value\";\n SessionContext sessionContext = new SessionContext();\n AsyncContextImpl asyncContext = AsyncContextImpl.build(NoOpTracer.NO_OP_SPAN_CONTEXT, () -> sessionContext, null);\n asyncContext.put(name, value);\n assertNull(sessionContext.get(name));\n try (Cleaner cleaner = asyncContext.importToCurrent()) {\n assertEquals(value, sessionContext.get(name));\n }\n assertNull(sessionContext.get(name));\n\n AsyncContextImpl asyncContext2 = AsyncContextImpl.build(NoOpTracer.NO_OP_SPAN_CONTEXT, () -> sessionContext, Collections.singletonMap(name, value));\n\n assertNull(sessionContext.get(name));\n try (Cleaner cleaner = asyncContext2.importToCurrent()) {\n assertEquals(value, sessionContext.get(name));\n }\n assertNull(sessionContext.get(name));\n\n }\n\n @Test\n public void getAll() {\n String name = \"test_name\";\n String value = \"test_value\";\n AsyncContextImpl asyncContext = AsyncContextImpl.build(NoOpTracer.NO_OP_SPAN_CONTEXT, () -> null, Collections.singletonMap(name, value));\n assertNotNull(asyncContext.getAll());\n assertEquals(value, asyncContext.getAll().get(name));\n }\n\n @Test\n public void get() {\n String name = \"test_name\";\n String value = \"test_value\";\n AsyncContextImpl asyncContext = AsyncContextImpl.build(NoOpTracer.NO_OP_SPAN_CONTEXT, () -> null, Collections.singletonMap(name, value));\n String v = asyncContext.get(name);\n assertNotNull(v);\n assertEquals(v, value);\n assertNull(asyncContext.get(name + \"test\"));\n\n }\n\n @Test\n public void put() {\n Map context = new HashMap<>();\n String name = \"test_name\";\n String value = \"test_value\";\n String name2 = name + \"2\";\n context.put(name, value);\n AsyncContextImpl asyncContext = AsyncContextImpl.build(NoOpTracer.NO_OP_SPAN_CONTEXT, () -> null, context);\n asyncContext.put(name2, value);\n assertNull(context.get(name2));\n assertEquals(value, asyncContext.get(name));\n assertEquals(value, asyncContext.get(name2));\n\n }\n}\n" }, { "path": "context/src/test/java/com/megaease/easeagent/context/ContextManagerTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.mock.config.MockConfig;\nimport com.megaease.easeagent.plugin.api.Reporter;\nimport com.megaease.easeagent.plugin.api.config.IPluginConfig;\nimport com.megaease.easeagent.plugin.api.metric.MetricRegistry;\nimport com.megaease.easeagent.plugin.api.metric.MetricRegistrySupplier;\nimport com.megaease.easeagent.plugin.api.metric.name.NameFactory;\nimport com.megaease.easeagent.plugin.api.metric.name.Tags;\nimport org.junit.Test;\n\nimport static org.junit.Assert.assertNotNull;\n\npublic class ContextManagerTest {\n\n @Test\n public void build() {\n ContextManager contextManager = ContextManager.build(MockConfig.getCONFIGS());\n assertNotNull(contextManager);\n contextManager.setTracing(() -> contextSupplier -> null);\n }\n\n @Test\n public void setTracing() {\n ContextManager contextManager = ContextManager.build(MockConfig.getCONFIGS());\n assertNotNull(contextManager);\n contextManager.setTracing(() -> contextSupplier -> null);\n }\n\n @Test\n public void setMetric() {\n ContextManager contextManager = ContextManager.build(MockConfig.getCONFIGS());\n assertNotNull(contextManager);\n contextManager.setMetric(() -> new MetricRegistrySupplier() {\n @Override\n public MetricRegistry newMetricRegistry(IPluginConfig config, NameFactory nameFactory, Tags tags) {\n return null;\n }\n\n @Override\n public Reporter reporter(IPluginConfig config) {\n return null;\n }\n });\n }\n}\n" }, { "path": "context/src/test/java/com/megaease/easeagent/context/GlobalContextTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.config.Configs;\nimport com.megaease.easeagent.context.log.LoggerFactoryImpl;\nimport com.megaease.easeagent.context.log.LoggerMdc;\nimport com.megaease.easeagent.plugin.bridge.NoOpMetrics;\nimport org.junit.Before;\nimport org.junit.Test;\n\nimport java.util.HashMap;\nimport java.util.Map;\n\nimport static org.junit.Assert.assertNotNull;\n\npublic class GlobalContextTest {\n GlobalContext globalContext;\n\n @Before\n public void before() {\n Map initConfigs = new HashMap<>();\n initConfigs.put(\"name\", \"demo-service\");\n initConfigs.put(\"system\", \"demo-system\");\n Configs configs = new Configs(initConfigs);\n LoggerFactoryImpl loggerFactory = LoggerFactoryImpl.build();\n globalContext = new GlobalContext(configs, NoOpMetrics.NO_OP_METRIC_SUPPLIER, loggerFactory, new LoggerMdc(loggerFactory.factory().mdc()));\n }\n\n\n @Test\n public void getConf() {\n assertNotNull(globalContext.getConf());\n }\n\n @Test\n public void getMdc() {\n assertNotNull(globalContext.getMdc());\n }\n\n @Test\n public void getLoggerFactory() {\n assertNotNull(globalContext.getLoggerFactory());\n }\n\n @Test\n public void getMetric() {\n assertNotNull(globalContext.getMetric());\n }\n}\n" }, { "path": "context/src/test/java/com/megaease/easeagent/context/ProgressFieldsManagerTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.config.Configs;\nimport com.megaease.easeagent.plugin.api.ProgressFields;\nimport org.junit.Test;\n\nimport java.util.Collections;\nimport java.util.HashMap;\nimport java.util.Set;\n\nimport static com.megaease.easeagent.plugin.api.ProgressFields.EASEAGENT_PROGRESS_FORWARDED_HEADERS_CONFIG;\nimport static org.junit.Assert.assertFalse;\nimport static org.junit.Assert.assertTrue;\n\npublic class ProgressFieldsManagerTest {\n\n @Test\n public void init() {\n HashMap source = new HashMap<>();\n source.put(\"plugin.observability.global.metrics.enabled\", \"true\");\n source.put(EASEAGENT_PROGRESS_FORWARDED_HEADERS_CONFIG, \"aaa,bbb,ccc\");\n Configs configs = new Configs(source);\n ProgressFieldsManager.init(configs);\n Set fields = ProgressFields.getForwardedHeaders();\n assertFalse(fields.isEmpty());\n }\n\n @Test\n public void isEmpty() {\n assertTrue(true);\n assertTrue(ProgressFields.isEmpty(new String[0]));\n assertFalse(ProgressFields.isEmpty(new String[1]));\n }\n\n @Test\n public void getFields() {\n HashMap source = new HashMap<>();\n source.put(\"plugin.observability.global.metrics.enabled\", \"true\");\n source.put(EASEAGENT_PROGRESS_FORWARDED_HEADERS_CONFIG, \"a,b,c\");\n Configs configs = new Configs(source);\n ProgressFieldsManager.init(configs);\n Set fields = ProgressFields.getForwardedHeaders();\n assertFalse(fields.isEmpty());\n assertTrue(fields.contains(\"a\"));\n assertTrue(fields.contains(\"b\"));\n assertTrue(fields.contains(\"c\"));\n\n configs.updateConfigs(Collections.singletonMap(EASEAGENT_PROGRESS_FORWARDED_HEADERS_CONFIG, \"a\"));\n configs.updateConfigs(Collections.singletonMap(EASEAGENT_PROGRESS_FORWARDED_HEADERS_CONFIG, \"c\"));\n fields = ProgressFields.getForwardedHeaders();\n assertFalse(fields.isEmpty());\n assertFalse(fields.contains(\"a\"));\n assertFalse(fields.contains(\"b\"));\n assertTrue(fields.contains(\"c\"));\n\n }\n}\n" }, { "path": "context/src/test/java/com/megaease/easeagent/context/RetBoundTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport org.junit.Test;\n\nimport static org.junit.Assert.*;\n\npublic class RetBoundTest {\n\n @Test\n public void size() {\n assertEquals(100, new RetBound(100).size());\n }\n\n @Test\n public void get() {\n RetBound retBound = new RetBound(1);\n Object o = new Object();\n Object o2 = new Object();\n retBound.put(\"a\", o);\n assertEquals(o, retBound.get(\"a\"));\n assertNotEquals(o2, retBound.get(\"a\"));\n }\n\n @Test\n public void put() {\n get();\n }\n}\n" }, { "path": "context/src/test/java/com/megaease/easeagent/context/SessionContextTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context;\n\nimport com.megaease.easeagent.config.PluginConfig;\nimport com.megaease.easeagent.config.PluginConfigManager;\nimport com.megaease.easeagent.mock.config.MockConfig;\nimport com.megaease.easeagent.plugin.api.Cleaner;\nimport com.megaease.easeagent.plugin.api.InitializeContext;\nimport com.megaease.easeagent.plugin.api.ProgressFields;\nimport com.megaease.easeagent.plugin.api.config.IPluginConfig;\nimport com.megaease.easeagent.plugin.api.context.AsyncContext;\nimport com.megaease.easeagent.plugin.api.context.RequestContext;\nimport com.megaease.easeagent.plugin.api.trace.*;\nimport com.megaease.easeagent.plugin.bridge.EaseAgent;\nimport com.megaease.easeagent.plugin.bridge.NoOpIPluginConfig;\nimport com.megaease.easeagent.plugin.bridge.NoOpTracer;\nimport org.junit.Before;\nimport org.junit.Test;\n\nimport java.util.Collections;\nimport java.util.HashMap;\nimport java.util.List;\nimport java.util.Map;\nimport java.util.concurrent.atomic.AtomicInteger;\nimport java.util.concurrent.atomic.AtomicReference;\nimport java.util.function.Supplier;\n\nimport static com.megaease.easeagent.plugin.api.ProgressFields.EASEAGENT_PROGRESS_FORWARDED_HEADERS_CONFIG;\nimport static org.junit.Assert.*;\n\npublic class SessionContextTest {\n\n @Before\n public void before() {\n ContextManager contextManager = ContextManager.build(MockConfig.getCONFIGS());\n assertNotNull(contextManager);\n }\n\n @Test\n public void isNoop() {\n SessionContext sessionContext = new SessionContext();\n assertFalse(sessionContext.isNoop());\n }\n\n @Test\n public void getSupplier() {\n SessionContext sessionContext = new SessionContext();\n assertNull(sessionContext.getSupplier());\n sessionContext.setSupplier(() -> null);\n assertNotNull(sessionContext.getSupplier());\n }\n\n @Test\n public void setSupplier() {\n getSupplier();\n }\n\n @Test\n public void currentTracing() {\n SessionContext sessionContext = new SessionContext();\n assertTrue(sessionContext.currentTracing().isNoop());\n sessionContext.setCurrentTracing(new MockITracing());\n assertNotNull(sessionContext.currentTracing());\n assertFalse(sessionContext.currentTracing().isNoop());\n }\n\n @Test\n public void get() {\n String name = \"test_name\";\n String value = \"test_value\";\n SessionContext sessionContext = new SessionContext();\n assertNull(sessionContext.get(name));\n sessionContext.put(name, value);\n assertEquals(value, sessionContext.get(name));\n sessionContext.remove(name);\n assertNull(sessionContext.get(name));\n }\n\n @Test\n public void remove() {\n get();\n }\n\n @Test\n public void put() {\n get();\n }\n\n @Test\n public void getLocal() {\n //Deprecated\n }\n\n @Test\n public void putLocal() {\n //Deprecated\n }\n\n\n @Test\n public void getConfig() {\n SessionContext sessionContext = new SessionContext();\n assertNotNull(sessionContext.getConfig());\n IPluginConfig iPluginConfig = sessionContext.getConfig();\n assertEquals(NoOpIPluginConfig.INSTANCE.domain(), iPluginConfig.domain());\n assertEquals(NoOpIPluginConfig.INSTANCE.namespace(), iPluginConfig.namespace());\n assertEquals(NoOpIPluginConfig.INSTANCE.id(), iPluginConfig.id());\n\n sessionContext.pushConfig(NoOpIPluginConfig.INSTANCE);\n assertNotNull(sessionContext.getConfig());\n PluginConfigManager pluginConfigManager = PluginConfigManager.builder(MockConfig.getCONFIGS()).build();\n String domain = \"observability\";\n String namespace = \"test_config\";\n String id = \"test\";\n PluginConfig config = pluginConfigManager.getConfig(domain, namespace, id);\n sessionContext.pushConfig(config);\n\n iPluginConfig = sessionContext.getConfig();\n assertEquals(domain, iPluginConfig.domain());\n assertEquals(namespace, iPluginConfig.namespace());\n assertEquals(id, iPluginConfig.id());\n\n IPluginConfig oppConfig = sessionContext.popConfig();\n assertEquals(domain, oppConfig.domain());\n assertEquals(namespace, oppConfig.namespace());\n assertEquals(id, oppConfig.id());\n\n iPluginConfig = sessionContext.getConfig();\n assertEquals(NoOpIPluginConfig.INSTANCE.domain(), iPluginConfig.domain());\n assertEquals(NoOpIPluginConfig.INSTANCE.namespace(), iPluginConfig.namespace());\n assertEquals(NoOpIPluginConfig.INSTANCE.id(), iPluginConfig.id());\n\n }\n\n @Test\n public void pushConfig() {\n getConfig();\n }\n\n @Test\n public void popConfig() {\n getConfig();\n }\n\n @Test\n public void enter() {\n SessionContext sessionContext = new SessionContext();\n Object key1 = new Object();\n Object key2 = new Object();\n assertEquals(1, sessionContext.enter(key1));\n assertEquals(1, sessionContext.enter(key2));\n assertEquals(2, sessionContext.enter(key1));\n assertEquals(2, sessionContext.enter(key2));\n assertEquals(2, sessionContext.exit(key1));\n assertEquals(2, sessionContext.exit(key2));\n assertEquals(1, sessionContext.exit(key1));\n assertEquals(1, sessionContext.exit(key2));\n\n\n assertTrue(sessionContext.enter(key1, 1));\n assertTrue(sessionContext.enter(key2, 1));\n assertTrue(sessionContext.enter(key1, 2));\n assertTrue(sessionContext.enter(key2, 2));\n assertTrue(sessionContext.exit(key1, 2));\n assertTrue(sessionContext.exit(key2, 2));\n assertTrue(sessionContext.exit(key1, 1));\n assertTrue(sessionContext.exit(key2, 1));\n\n }\n\n @Test\n public void exit() {\n enter();\n }\n\n @Test\n public void exportAsync() {\n SessionContext sessionContext = new SessionContext();\n sessionContext.setSupplier(() -> EaseAgent.initializeContextSupplier.getContext());\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n String name = \"test_name\";\n String value = \"test_value\";\n assertNull(sessionContext.get(name));\n sessionContext.put(name, value);\n assertEquals(value, sessionContext.get(name));\n AsyncContext asyncContext = sessionContext.exportAsync();\n assertTrue(asyncContext.getSpanContext().isNoop());\n assertEquals(1, iTracing.exportAsyncCount.get());\n\n assertEquals(value, asyncContext.get(name));\n assertFalse(asyncContext.isNoop());\n SessionContext sessionContext2 = new SessionContext();\n sessionContext2.setCurrentTracing(iTracing);\n assertNull(sessionContext2.get(name));\n try (Cleaner ignored = sessionContext2.importAsync(asyncContext)) {\n assertEquals(value, sessionContext2.get(name));\n assertEquals(1, iTracing.importAsyncCount.get());\n try (Cleaner ignored1 = sessionContext2.importAsync(asyncContext)) {\n assertEquals(2, iTracing.importAsyncCount.get());\n }\n assertEquals(value, sessionContext2.get(name));\n }\n\n }\n\n @Test\n public void importAsync() {\n exportAsync();\n }\n\n @Test\n public void clientRequest() {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n sessionContext.clientRequest(new EmptyRequest());\n assertEquals(1, iTracing.clientRequestCount.get());\n }\n\n @Test\n public void serverReceive() {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n sessionContext.serverReceive(new EmptyRequest());\n assertEquals(1, iTracing.serverReceiveCount.get());\n\n }\n\n @Test\n public void consumerSpan() {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n sessionContext.consumerSpan(new EmptyRequest());\n assertEquals(1, iTracing.consumerSpanCount.get());\n\n }\n\n @Test\n public void producerSpan() {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n sessionContext.producerSpan(new EmptyRequest());\n assertEquals(1, iTracing.producerSpanCount.get());\n }\n\n @Test\n public void nextSpan() {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n sessionContext.nextSpan();\n assertEquals(1, iTracing.nextSpanCount.get());\n\n }\n\n @Test\n public void popToBound() {\n //Deprecated\n }\n\n @Test\n public void pushRetBound() {\n //Deprecated\n }\n\n @Test\n public void popRetBound() {\n //Deprecated\n }\n\n @Test\n public void push() {\n //Deprecated\n }\n\n @Test\n public void pop() {\n //Deprecated\n }\n\n @Test\n public void peek() {\n //Deprecated\n }\n\n @Test\n public void wrap() throws InterruptedException {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n String name = \"test_name\";\n String value = \"test_value\";\n sessionContext.put(name, value);\n\n AtomicReference lastSessionContext = new AtomicReference<>();\n Supplier newContextSupplier = () -> {\n SessionContext sessionContext1 = new SessionContext();\n sessionContext1.setCurrentTracing(iTracing);\n lastSessionContext.set(sessionContext1);\n return sessionContext1;\n };\n sessionContext.setSupplier(newContextSupplier);\n Runnable runnable = () -> {\n assertNotNull(lastSessionContext.get());\n assertEquals(value, lastSessionContext.get().get(name));\n\n };\n runnable = sessionContext.wrap(runnable);\n assertTrue(sessionContext.isWrapped(runnable));\n Thread thread = new Thread(runnable);\n thread.start();\n thread.join();\n assertNull(lastSessionContext.get().get(name));\n assertEquals(1, iTracing.exportAsyncCount.get());\n assertEquals(1, iTracing.importAsyncCount.get());\n }\n\n @Test\n public void isWrapped() throws InterruptedException {\n wrap();\n }\n\n @Test\n public void isNecessaryKeys() {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n iTracing.setPropagationKeys(Collections.singletonList(\"b3\"));\n sessionContext.setCurrentTracing(iTracing);\n assertTrue(sessionContext.isNecessaryKeys(\"b3\"));\n assertFalse(sessionContext.isNecessaryKeys(\"aaa\"));\n\n }\n\n @Test\n public void consumerInject() {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n sessionContext.consumerInject(NoOpTracer.NO_OP_SPAN, new EmptyRequest());\n assertEquals(1, iTracing.mockMessagingTracing.consumerInjectorCount.get());\n\n }\n\n @Test\n public void producerInject() {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n sessionContext.producerInject(NoOpTracer.NO_OP_SPAN, new EmptyRequest());\n assertEquals(1, iTracing.mockMessagingTracing.producerInjectorCount.get());\n\n }\n\n @Test\n public void injectForwardedHeaders() {\n String forwarded = \"test_forwarded_value\";\n String forwardedKey = EASEAGENT_PROGRESS_FORWARDED_HEADERS_CONFIG;\n String headerValue = \"test_value\";\n ProgressFields.changeListener().accept(Collections.singletonMap(forwardedKey, forwarded));\n ProgressFields.changeListener().accept(Collections.singletonMap(forwardedKey + \"2\", forwarded + \"2\"));\n SessionContext sessionContext = new SessionContext();\n sessionContext.importForwardedHeaders(name -> {\n if (name.equals(forwarded)) {\n return headerValue;\n }\n return null;\n });\n\n\n Map values = new HashMap<>();\n sessionContext.injectForwardedHeaders(values::put);\n assertEquals(1, values.size());\n assertEquals(headerValue, values.get(forwarded));\n\n ProgressFields.changeListener().accept(Collections.singletonMap(forwardedKey, \"\"));\n ProgressFields.changeListener().accept(Collections.singletonMap(forwardedKey + \"2\", \"\"));\n }\n\n @Test\n public void getTracing() {\n SessionContext sessionContext = new SessionContext();\n MockITracing iTracing = new MockITracing();\n sessionContext.setCurrentTracing(iTracing);\n assertNotNull(sessionContext.getTracing());\n assertTrue(sessionContext.getTracing() instanceof MockITracing);\n }\n\n @Test\n public void setCurrentTracing() {\n getTracing();\n }\n\n @Test\n public void clear() {\n String name = \"test_name\";\n String value = \"test_value\";\n SessionContext sessionContext = new SessionContext();\n sessionContext.put(name, value);\n assertEquals(1, sessionContext.enter(\"test_key\"));\n sessionContext.clear();\n assertNull(sessionContext.get(name));\n assertEquals(1, sessionContext.enter(\"test_key\"));\n sessionContext.clear();\n }\n\n\n public static class EmptyRequest implements MessagingRequest {\n\n @Override\n public String operation() {\n return null;\n }\n\n @Override\n public String channelKind() {\n return null;\n }\n\n @Override\n public String channelName() {\n return null;\n }\n\n @Override\n public Object unwrap() {\n return null;\n }\n\n @Override\n public Span.Kind kind() {\n return null;\n }\n\n @Override\n public String header(String name) {\n return null;\n }\n\n @Override\n public String name() {\n return null;\n }\n\n @Override\n public boolean cacheScope() {\n return false;\n }\n\n @Override\n public void setHeader(String name, String value) {\n\n }\n }\n\n\n public static class MockITracing extends NoOpTracer.NoopTracing {\n public AtomicInteger exportAsyncCount = new AtomicInteger();\n public AtomicInteger importAsyncCount = new AtomicInteger();\n public AtomicInteger clientRequestCount = new AtomicInteger();\n public AtomicInteger serverReceiveCount = new AtomicInteger();\n public AtomicInteger consumerSpanCount = new AtomicInteger();\n public AtomicInteger producerSpanCount = new AtomicInteger();\n public AtomicInteger nextSpanCount = new AtomicInteger();\n public MockMessagingTracing mockMessagingTracing = new MockMessagingTracing();\n private List propagationKeys = Collections.emptyList();\n\n public MockITracing setPropagationKeys(List propagationKeys) {\n this.propagationKeys = propagationKeys;\n return this;\n }\n\n @Override\n public SpanContext exportAsync() {\n exportAsyncCount.incrementAndGet();\n return super.exportAsync();\n }\n\n @Override\n public Scope importAsync(SpanContext snapshot) {\n importAsyncCount.incrementAndGet();\n return super.importAsync(snapshot);\n }\n\n @Override\n public RequestContext clientRequest(Request request) {\n clientRequestCount.incrementAndGet();\n return super.clientRequest(request);\n }\n\n @Override\n public RequestContext serverReceive(Request request) {\n serverReceiveCount.incrementAndGet();\n return super.serverReceive(request);\n }\n\n @Override\n public Span consumerSpan(MessagingRequest request) {\n consumerSpanCount.incrementAndGet();\n return super.consumerSpan(request);\n }\n\n @Override\n public Span producerSpan(MessagingRequest request) {\n producerSpanCount.incrementAndGet();\n return super.producerSpan(request);\n }\n\n @Override\n public Span nextSpan() {\n nextSpanCount.incrementAndGet();\n return super.nextSpan();\n }\n\n @Override\n public List propagationKeys() {\n return propagationKeys;\n }\n\n @Override\n public MessagingTracing messagingTracing() {\n return mockMessagingTracing;\n }\n\n @Override\n public boolean isNoop() {\n return false;\n }\n }\n\n\n public static class MockMessagingTracing extends NoOpTracer.EmptyMessagingTracing {\n public AtomicInteger producerInjectorCount = new AtomicInteger();\n public AtomicInteger consumerInjectorCount = new AtomicInteger();\n\n @Override\n public Injector producerInjector() {\n return (span, request) -> producerInjectorCount.incrementAndGet();\n }\n\n @Override\n public Injector consumerInjector() {\n return (span, request) -> consumerInjectorCount.incrementAndGet();\n }\n }\n}\n" }, { "path": "context/src/test/java/com/megaease/easeagent/context/log/LoggerFactoryImplTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context.log;\n\nimport com.megaease.easeagent.log4j2.MDC;\nimport com.megaease.easeagent.log4j2.api.AgentLoggerFactory;\nimport com.megaease.easeagent.plugin.api.logging.Logger;\nimport org.junit.Test;\n\nimport static org.junit.Assert.assertNotNull;\n\npublic class LoggerFactoryImplTest {\n LoggerFactoryImpl loggerFactory = LoggerFactoryImpl.build();\n\n @Test\n public void getLogger() {\n Logger logger = loggerFactory.getLogger(LoggerFactoryImplTest.class.getName());\n logger.info(\"aaaa\");\n MDC.put(\"testMdc\", \"testMdc_value\");\n logger.info(\"bbbb\");\n assertNotNull(MDC.get(\"testMdc\"));\n }\n\n @Test\n public void factory() {\n AgentLoggerFactory agentLoggerFactory = loggerFactory.factory();\n assertNotNull(agentLoggerFactory);\n LoggerImpl logger = agentLoggerFactory.getLogger(LoggerFactoryImplTest.class.getName());\n logger.info(\"aaaa\");\n agentLoggerFactory.mdc().put(\"newFactory\", \"newFactory\");\n assertNotNull(MDC.get(\"newFactory\"));\n\n }\n\n @Test\n public void build() {\n LoggerFactoryImpl loggerFactory = LoggerFactoryImpl.build();\n assertNotNull(loggerFactory);\n }\n}\n" }, { "path": "context/src/test/java/com/megaease/easeagent/context/log/LoggerImplTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context.log;\n\nimport com.megaease.easeagent.plugin.api.logging.Logger;\nimport org.junit.Test;\n\nimport static org.junit.Assert.*;\n\npublic class LoggerImplTest {\n @Test\n public void test() {\n LoggerFactoryImpl loggerFactory = LoggerFactoryImpl.build();\n Logger logger = loggerFactory.getLogger(LoggerFactoryImplTest.class.getName());\n assertTrue(logger instanceof LoggerImpl);\n }\n}\n" }, { "path": "context/src/test/java/com/megaease/easeagent/context/log/LoggerMdcTest.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.context.log;\n\nimport org.junit.Test;\n\nimport static org.junit.Assert.*;\n\npublic class LoggerMdcTest {\n LoggerFactoryImpl loggerFactory = LoggerFactoryImpl.build();\n LoggerMdc mdc = new LoggerMdc(loggerFactory.factory().mdc());\n\n @Test\n public void put() {\n mdc.put(\"testA\", \"testB\");\n assertNull(org.slf4j.MDC.get(\"testA\"));\n assertNotNull(mdc.get(\"testA\"));\n }\n\n @Test\n public void remove() {\n mdc.put(\"testA\", \"testB\");\n assertNotNull(mdc.get(\"testA\"));\n mdc.remove(\"testA\");\n assertNull(mdc.get(\"testA\"));\n }\n\n @Test\n public void get() {\n mdc.put(\"testA\", \"testB\");\n assertNull(org.slf4j.MDC.get(\"testA\"));\n assertNotNull(mdc.get(\"testA\"));\n org.slf4j.MDC.put(\"testB\", \"testB\");\n assertNotNull(org.slf4j.MDC.get(\"testB\"));\n assertNull(mdc.get(\"testB\"));\n }\n}\n" }, { "path": "context/src/test/resources/log4j2.xml", "content": "\n\n \n \n \n \n \n \n \n \n \n \n \n\n" }, { "path": "core/pom.xml", "content": "\n\n\n \n easeagent\n com.megaease.easeagent\n 2.3.0\n \n 4.0.0\n\n core\n\n \n \n com.google.auto.service\n auto-service\n \n \n net.bytebuddy\n byte-buddy\n \n \n com.google.guava\n guava\n \n \n com.megaease.easeagent\n plugin-api\n ${project.version}\n \n \n com.megaease.easeagent\n context\n ${project.version}\n \n \n com.megaease.easeagent\n config\n ${project.version}\n \n \n com.megaease.easeagent\n report\n ${project.version}\n \n \n com.megaease.easeagent\n httpserver\n ${project.version}\n \n \n com.megaease.easeagent\n log4j2-api\n \n \n org.slf4j\n slf4j-api\n \n \n org.apache.commons\n commons-lang3\n \n \n commons-codec\n commons-codec\n \n \n com.fasterxml.jackson.core\n jackson-core\n \n \n com.fasterxml.jackson.core\n jackson-databind\n \n \n com.j256.simplejmx\n simplejmx\n 1.19\n test\n \n \n javax.servlet\n javax.servlet-api\n provided\n \n \n io.projectreactor\n reactor-core\n provided\n \n \n net.bytebuddy\n byte-buddy-agent\n test\n \n \n com.megaease.easeagent\n log4j2-mock\n ${project.version}\n test\n \n \n\n \n \n \n src/main/resources\n true\n \n \n \n \n org.codehaus.mojo\n cobertura-maven-plugin\n \n \n \n html\n \n \n \n **/AdviceTo*\n **/Configurable*\n \n \n \n \n \n \n\n\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/AppendBootstrapClassLoaderSearch.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core;\n\nimport com.google.common.base.Function;\nimport com.google.common.collect.Maps;\nimport com.google.common.io.CharStreams;\nimport com.google.common.io.Closeables;\nimport com.megaease.easeagent.plugin.AppendBootstrapLoader;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.dynamic.ClassFileLocator;\nimport net.bytebuddy.dynamic.loading.ClassInjector;\nimport net.bytebuddy.pool.TypePool;\n\nimport java.io.File;\nimport java.io.IOException;\nimport java.io.InputStream;\nimport java.io.InputStreamReader;\nimport java.lang.instrument.Instrumentation;\nimport java.net.URLConnection;\nimport java.nio.charset.StandardCharsets;\nimport java.security.AccessController;\nimport java.security.PrivilegedAction;\nimport java.util.Map;\nimport java.util.Set;\n\nimport static com.google.common.collect.FluentIterable.from;\nimport static com.google.common.collect.Maps.uniqueIndex;\nimport static java.util.Collections.list;\n\npublic final class AppendBootstrapClassLoaderSearch {\n private static final File TMP_FILE = new File(\n AccessController.doPrivileged(\n new PrivilegedAction() {\n @Override\n public String run() {\n return System.getProperty(\"java.io.tmpdir\");\n }\n })\n );\n\n static Set by(Instrumentation inst, ClassInjector.UsingInstrumentation.Target target) throws IOException {\n final Set names = findClassAnnotatedAutoService(AppendBootstrapLoader.class);\n ClassInjector.UsingInstrumentation.of(TMP_FILE, target, inst).inject(types(names));\n return names;\n }\n\n private static Map types(Set names) {\n final ClassLoader loader = AppendBootstrapClassLoaderSearch.class.getClassLoader();\n final ClassFileLocator locator = ClassFileLocator.ForClassLoader.of(loader);\n final TypePool pool = TypePool.Default.of(locator);\n\n return Maps.transformValues(\n uniqueIndex(names, input -> pool.describe(input).resolve()),\n input -> {\n try {\n return locator.locate(input).resolve();\n } catch (IOException e) {\n throw new IllegalStateException(e);\n }\n });\n }\n\n private static Set findClassAnnotatedAutoService(Class cls) throws IOException {\n final ClassLoader loader = AppendBootstrapClassLoaderSearch.class.getClassLoader();\n\n return from(list(loader.getResources(\"META-INF/services/\" + cls.getName())))\n .transform(input -> {\n try {\n final URLConnection connection = input.openConnection();\n final InputStream stream = connection.getInputStream();\n return new InputStreamReader(stream, StandardCharsets.UTF_8);\n } catch (IOException e) {\n throw new IllegalStateException(e);\n }\n })\n .transformAndConcat((Function>) input -> {\n try {\n return CharStreams.readLines(input);\n } catch (IOException e) {\n throw new IllegalStateException(e);\n } finally {\n Closeables.closeQuietly(input);\n }\n\n })\n .toSet();\n }\n\n private AppendBootstrapClassLoaderSearch() {\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/Bootstrap.java", "content": "/*\n * Copyright (c) 2017, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core;\n\nimport com.megaease.easeagent.config.*;\nimport com.megaease.easeagent.context.ContextManager;\nimport com.megaease.easeagent.core.config.*;\nimport com.megaease.easeagent.core.info.AgentInfoFactory;\nimport com.megaease.easeagent.core.plugin.BaseLoader;\nimport com.megaease.easeagent.core.plugin.BridgeDispatcher;\nimport com.megaease.easeagent.core.plugin.PluginLoader;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpHandlerProvider;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpServer;\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\nimport com.megaease.easeagent.plugin.api.metric.MetricProvider;\nimport com.megaease.easeagent.plugin.api.middleware.RedirectProcessor;\nimport com.megaease.easeagent.plugin.api.trace.TracingProvider;\nimport com.megaease.easeagent.plugin.bean.AgentInitializingBean;\nimport com.megaease.easeagent.plugin.bean.BeanProvider;\nimport com.megaease.easeagent.plugin.bridge.AgentInfo;\nimport com.megaease.easeagent.plugin.bridge.EaseAgent;\nimport com.megaease.easeagent.plugin.report.AgentReport;\nimport com.megaease.easeagent.plugin.utils.common.StringUtils;\nimport com.megaease.easeagent.report.AgentReportAware;\nimport com.megaease.easeagent.report.DefaultAgentReport;\nimport lombok.SneakyThrows;\nimport net.bytebuddy.agent.builder.AgentBuilder;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.dynamic.ClassFileLocator;\nimport net.bytebuddy.dynamic.DynamicType;\nimport net.bytebuddy.dynamic.loading.ClassInjector;\nimport net.bytebuddy.matcher.ElementMatcher;\nimport net.bytebuddy.utility.JavaModule;\n\nimport javax.management.MBeanServer;\nimport javax.management.ObjectName;\nimport java.lang.instrument.Instrumentation;\nimport java.lang.management.ManagementFactory;\nimport java.net.URLClassLoader;\nimport java.util.List;\nimport java.util.Set;\nimport java.util.concurrent.TimeUnit;\n\nimport static net.bytebuddy.matcher.ElementMatchers.*;\n\n@SuppressWarnings(\"unused\")\npublic class Bootstrap {\n private static final Logger LOGGER = LoggerFactory.getLogger(Bootstrap.class);\n\n private static final String AGENT_SERVER_PORT_KEY = ConfigFactory.AGENT_SERVER_PORT;\n private static final String AGENT_SERVER_ENABLED_KEY = ConfigFactory.AGENT_SERVER_ENABLED;\n\n private static final String AGENT_MIDDLEWARE_UPDATE = \"easeagent.middleware.update\";\n\n private static final int DEF_AGENT_SERVER_PORT = 9900;\n\n static final String MX_BEAN_OBJECT_NAME = \"com.megaease.easeagent:type=ConfigManager\";\n\n private static ContextManager contextManager;\n\n private Bootstrap() {\n }\n\n @SneakyThrows\n public static void start(String args, Instrumentation inst, String javaAgentJarPath) {\n long begin = System.nanoTime();\n System.setProperty(ConfigConst.AGENT_JAR_PATH, javaAgentJarPath);\n\n // add bootstrap classes\n Set bootstrapClassSet = AppendBootstrapClassLoaderSearch.by(inst, ClassInjector.UsingInstrumentation.Target.BOOTSTRAP);\n if (LOGGER.isDebugEnabled()) {\n LOGGER.debug(\"Injected class: {}\", bootstrapClassSet);\n }\n\n // initiate configuration\n String configPath = ConfigFactory.getConfigPath();\n if (StringUtils.isEmpty(configPath)) {\n configPath = args;\n }\n\n ClassLoader classLoader = Bootstrap.class.getClassLoader();\n final AgentInfo agentInfo = AgentInfoFactory.loadAgentInfo(classLoader);\n EaseAgent.agentInfo = agentInfo;\n final GlobalConfigs conf = ConfigFactory.loadConfigs(configPath, classLoader);\n wrapConfig(conf);\n\n // loader check\n GlobalAgentHolder.setAgentClassLoader((URLClassLoader) Bootstrap.class.getClassLoader());\n EaseAgent.agentClassloader = GlobalAgentHolder::getAgentClassLoader;\n\n // init Context/API\n contextManager = ContextManager.build(conf);\n EaseAgent.dispatcher = new BridgeDispatcher();\n\n // initInnerHttpServer\n initHttpServer(conf);\n\n // redirection\n RedirectProcessor.INSTANCE.init();\n\n // reporter\n final AgentReport agentReport = DefaultAgentReport.create(conf);\n GlobalAgentHolder.setAgentReport(agentReport);\n EaseAgent.agentReport = agentReport;\n\n // load plugins\n AgentBuilder builder = getAgentBuilder(conf, false);\n builder = PluginLoader.load(builder, conf);\n\n // provider & beans\n loadProvider(conf, agentReport);\n\n long installBegin = System.currentTimeMillis();\n builder.installOn(inst);\n LOGGER.info(\"installBegin use time: {}ms\", (System.currentTimeMillis() - installBegin));\n\n LOGGER.info(\"Initialization has took {}ns\", TimeUnit.NANOSECONDS.toMillis(System.nanoTime() - begin));\n }\n\n private static void initHttpServer(Configs conf) {\n // inner httpserver\n Integer port = conf.getInt(AGENT_SERVER_PORT_KEY);\n if (port == null) {\n port = DEF_AGENT_SERVER_PORT;\n }\n String portStr = System.getProperty(AGENT_SERVER_PORT_KEY, String.valueOf(port));\n port = Integer.parseInt(portStr);\n\n AgentHttpServer agentHttpServer = new AgentHttpServer(port);\n\n boolean httpServerEnabled = conf.getBoolean(AGENT_SERVER_ENABLED_KEY);\n if (httpServerEnabled) {\n agentHttpServer.startServer();\n LOGGER.info(\"start agent http server on port:{}\", port);\n }\n GlobalAgentHolder.setAgentHttpServer(agentHttpServer);\n\n // add httpHandler\n agentHttpServer.addHttpRoute(new ServiceUpdateAgentHttpHandler());\n agentHttpServer.addHttpRoute(new CanaryUpdateAgentHttpHandler());\n agentHttpServer.addHttpRoute(new CanaryListUpdateAgentHttpHandler());\n agentHttpServer.addHttpRoute(new PluginPropertyHttpHandler());\n agentHttpServer.addHttpRoute(new PluginPropertiesHttpHandler());\n }\n\n private static void loadProvider(final Configs conf, final AgentReport agentReport) {\n List providers = BaseLoader.loadOrdered(BeanProvider.class);\n providers.forEach(input -> provider(input, conf, agentReport));\n }\n\n private static void provider(final BeanProvider beanProvider, final Configs conf, final AgentReport agentReport) {\n if (beanProvider instanceof ConfigAware) {\n ((ConfigAware) beanProvider).setConfig(conf);\n }\n if (beanProvider instanceof AgentReportAware) {\n ((AgentReportAware) beanProvider).setAgentReport(agentReport);\n }\n\n if (beanProvider instanceof AgentHttpHandlerProvider) {\n GlobalAgentHolder.getAgentHttpServer()\n .addHttpRoutes(((AgentHttpHandlerProvider) beanProvider).getAgentHttpHandlers());\n }\n if (beanProvider instanceof AgentInitializingBean) {\n ((AgentInitializingBean) beanProvider).afterPropertiesSet();\n }\n if (beanProvider instanceof TracingProvider) {\n TracingProvider tracingProvider = (TracingProvider) beanProvider;\n contextManager.setTracing(tracingProvider);\n }\n if (beanProvider instanceof MetricProvider) {\n contextManager.setMetric((MetricProvider) beanProvider);\n }\n }\n\n public static AgentBuilder getAgentBuilder(Configs config, boolean test) {\n // config may use to add some classes to be ignored in future\n long buildBegin = System.currentTimeMillis();\n AgentBuilder builder = new AgentBuilder.Default()\n .with(LISTENER)\n .with(AgentBuilder.RedefinitionStrategy.RETRANSFORMATION)\n .with(AgentBuilder.InitializationStrategy.NoOp.INSTANCE)\n .with(AgentBuilder.TypeStrategy.Default.REDEFINE)\n .with(AgentBuilder.LocationStrategy.ForClassLoader.STRONG\n .withFallbackTo(ClassFileLocator.ForClassLoader.ofSystemLoader()));\n AgentBuilder.Ignored ignore = builder.ignore(isSynthetic())\n .or(nameStartsWith(\"sun.\").and(not(nameStartsWith(\"sun.net.www.protocol.http\")) ))\n .or(nameStartsWith(\"com.sun.\"))\n .or(nameStartsWith(\"brave.\"))\n .or(nameStartsWith(\"zipkin2.\"))\n .or(nameStartsWith(\"com.fasterxml\"))\n .or(nameStartsWith(\"org.apache.logging\")\n .and(not(hasSuperClass(named(\"org.apache.logging.log4j.spi.AbstractLogger\")))))\n .or(nameStartsWith(\"kotlin.\"))\n .or(nameStartsWith(\"javax.\"))\n .or(nameStartsWith(\"net.bytebuddy.\"))\n .or(nameStartsWith(\"com\\\\.sun\\\\.proxy\\\\.\\\\$Proxy.+\"))\n .or(nameStartsWith(\"java\\\\.lang\\\\.invoke\\\\.BoundMethodHandle\\\\$Species_L.+\"))\n .or(nameStartsWith(\"org.junit.\"))\n .or(nameStartsWith(\"junit.\"))\n .or(nameStartsWith(\"com.intellij.\"));\n\n // config used here to avoid warning of unused\n if (!test && config != null) {\n builder = ignore\n .or(nameStartsWith(\"com.megaease.easeagent.\"));\n } else {\n builder = ignore;\n }\n LOGGER.info(\"AgentBuilder use time: {}\", (System.currentTimeMillis() - buildBegin));\n return builder;\n }\n\n @SneakyThrows\n static void registerMBeans(ConfigManagerMXBean conf) {\n long begin = System.currentTimeMillis();\n MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();\n ObjectName mxBeanName = new ObjectName(MX_BEAN_OBJECT_NAME);\n ClassLoader customClassLoader = Thread.currentThread().getContextClassLoader();\n mbs.registerMBean(conf, mxBeanName);\n LOGGER.info(\"Register {} as MBean {}, use time: {}\",\n conf.getClass().getName(), mxBeanName, (System.currentTimeMillis() - begin));\n }\n\n private static ElementMatcher protectedLoaders() {\n return isBootstrapClassLoader().or(is(Bootstrap.class.getClassLoader()));\n }\n\n private static void wrapConfig(GlobalConfigs configs) {\n WrappedConfigManager wrappedConfigManager = new WrappedConfigManager(Bootstrap.class.getClassLoader(), configs);\n registerMBeans(wrappedConfigManager);\n GlobalAgentHolder.setWrappedConfigManager(wrappedConfigManager);\n }\n\n private static final AgentBuilder.Listener LISTENER = new AgentBuilder.Listener() {\n @Override\n public void onDiscovery(String typeName, ClassLoader classLoader, JavaModule module, boolean loaded) {\n // ignored\n }\n\n @Override\n public void onTransformation(TypeDescription td, ClassLoader ld, JavaModule m, boolean loaded, DynamicType dt) {\n LOGGER.debug(\"onTransformation: {} loaded: {} from classLoader {}\", td, loaded, ld);\n }\n\n @Override\n public void onIgnored(TypeDescription td, ClassLoader ld, JavaModule m, boolean loaded) {\n // ignored\n }\n\n @Override\n public void onError(String name, ClassLoader ld, JavaModule m, boolean loaded, Throwable error) {\n LOGGER.debug(\"Just for Debug-log, transform ends exceptionally, \" +\n \"which is sometimes normal and sometimes there is an error: {} error:{} loaded: {} from classLoader {}\",\n name, error, loaded, ld);\n }\n\n @Override\n public void onComplete(String name, ClassLoader ld, JavaModule m, boolean loaded) {\n // ignored\n }\n };\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/GlobalAgentHolder.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.core;\n\nimport com.megaease.easeagent.config.WrappedConfigManager;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpServer;\nimport com.megaease.easeagent.plugin.report.AgentReport;\n\nimport java.net.URLClassLoader;\n\npublic class GlobalAgentHolder {\n private static WrappedConfigManager wrappedConfigManager;\n private static AgentHttpServer agentHttpServer;\n private static AgentReport agentReport;\n private static URLClassLoader agentLoader;\n\n private GlobalAgentHolder() {}\n\n public static void setWrappedConfigManager(WrappedConfigManager config) {\n wrappedConfigManager = config;\n }\n\n public static WrappedConfigManager getWrappedConfigManager() {\n return wrappedConfigManager;\n }\n\n public static void setAgentHttpServer(AgentHttpServer server) {\n agentHttpServer = server;\n }\n\n public static AgentHttpServer getAgentHttpServer() {\n return agentHttpServer;\n }\n\n public static void setAgentReport(AgentReport report) {\n agentReport = report;\n }\n\n public static AgentReport getAgentReport() {\n return agentReport;\n }\n\n public static void setAgentClassLoader(URLClassLoader loader) {\n agentLoader = loader;\n }\n\n public static URLClassLoader getAgentClassLoader() {\n return agentLoader;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/config/CanaryListUpdateAgentHttpHandler.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.core.config;\n\nimport com.megaease.easeagent.core.GlobalAgentHolder;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpServer;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Response;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Status;\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\n\nimport java.util.HashMap;\nimport java.util.List;\nimport java.util.Map;\nimport java.util.concurrent.atomic.AtomicInteger;\n\npublic class CanaryListUpdateAgentHttpHandler extends ConfigsUpdateAgentHttpHandler {\n private static final Logger LOGGER = LoggerFactory.getLogger(CanaryListUpdateAgentHttpHandler.class);\n public static final AtomicInteger LAST_COUNT = new AtomicInteger(0);\n\n public CanaryListUpdateAgentHttpHandler() {\n this.mxBeanConfig = GlobalAgentHolder.getWrappedConfigManager();\n }\n\n @Override\n public String getPath() {\n return \"/config-global-transmission\";\n }\n\n @Override\n @SuppressWarnings(\"unchecked\")\n public Response processJsonConfig(Map map, Map urlParams) {\n LOGGER.info(\"call /config-global-transmission. configs: {}\", map);\n synchronized (LAST_COUNT) {\n List headers = (List) map.get(\"headers\");\n Map config = new HashMap<>();\n for (int i = 0; i < headers.size(); i++) {\n config.put(\"easeagent.progress.forwarded.headers.global.transmission.\" + i, headers.get(i));\n }\n int last = LAST_COUNT.get();\n if (headers.size() < last) {\n for (int i = headers.size(); i < last; i++) {\n config.put(\"easeagent.progress.forwarded.headers.global.transmission.\" + i, \"\");\n }\n }\n LAST_COUNT.set(headers.size());\n this.mxBeanConfig.updateConfigs(config);\n }\n return Response.newFixedLengthResponse(Status.OK, AgentHttpServer.JSON_TYPE, (String) null);\n }\n\n @Override\n public Response processConfig(Map config, Map urlParams, String version) {\n this.mxBeanConfig.updateCanary2(config, version);\n return null;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/config/CanaryUpdateAgentHttpHandler.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.core.config;\n\nimport com.megaease.easeagent.core.GlobalAgentHolder;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Response;\n\nimport java.util.Map;\n\npublic class CanaryUpdateAgentHttpHandler extends ConfigsUpdateAgentHttpHandler {\n public CanaryUpdateAgentHttpHandler() {\n this.mxBeanConfig = GlobalAgentHolder.getWrappedConfigManager();\n }\n\n @Override\n public String getPath() {\n return \"/config-canary\";\n }\n\n @Override\n public Response processConfig(Map config, Map urlParams, String version) {\n this.mxBeanConfig.updateCanary2(config, version);\n return null;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/config/ConfigsUpdateAgentHttpHandler.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.core.config;\n\nimport com.megaease.easeagent.config.ConfigManagerMXBean;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpHandler;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpServer;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.IHTTPSession;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Response;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Status;\nimport com.megaease.easeagent.httpserver.nanohttpd.router.RouterNanoHTTPD;\nimport com.megaease.easeagent.plugin.utils.common.JsonUtil;\nimport lombok.SneakyThrows;\nimport org.apache.commons.lang3.StringUtils;\n\nimport java.util.HashMap;\nimport java.util.Map;\n\npublic abstract class ConfigsUpdateAgentHttpHandler extends AgentHttpHandler {\n public abstract Response processConfig(Map config, Map urlParams, String version);\n\n protected ConfigManagerMXBean mxBeanConfig;\n\n static Map toConfigMap(Map map) {\n Map config = new HashMap<>(Math.max(map.size(), 8));\n map.forEach((s, o) -> config.put(s, o.toString()));\n return config;\n }\n\n @SneakyThrows\n @Override\n public Response process(RouterNanoHTTPD.UriResource uriResource, Map urlParams, IHTTPSession session) {\n String body = this.buildRequestBody(session);\n if (StringUtils.isEmpty(body)) {\n return Response.newFixedLengthResponse(Status.BAD_REQUEST, AgentHttpServer.JSON_TYPE, (String) null);\n }\n Map map = JsonUtil.toMap(body);\n if (map == null) {\n return Response.newFixedLengthResponse(Status.BAD_REQUEST, AgentHttpServer.JSON_TYPE, (String) null);\n }\n return processJsonConfig(map, urlParams);\n }\n\n public Response processJsonConfig(Map map, Map urlParams) {\n// String version = (String) map.remove(\"version\");\n// if (version == null) {\n// return Response.newFixedLengthResponse(Status.BAD_REQUEST, AgentHttpServer.JSON_TYPE, (String) null);\n// }\n Map config = toConfigMap(map);\n Response response = processConfig(config, urlParams, null);\n if (response != null) {\n return response;\n }\n return Response.newFixedLengthResponse(Status.OK, AgentHttpServer.JSON_TYPE, (String) null);\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/config/PluginPropertiesHttpHandler.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.core.config;\n\nimport com.megaease.easeagent.config.ConfigUtils;\nimport com.megaease.easeagent.core.GlobalAgentHolder;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpServer;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Response;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Status;\n\nimport java.util.HashMap;\nimport java.util.Map;\nimport java.util.Objects;\n\npublic class PluginPropertiesHttpHandler extends ConfigsUpdateAgentHttpHandler {\n public PluginPropertiesHttpHandler() {\n this.mxBeanConfig = GlobalAgentHolder.getWrappedConfigManager();\n }\n\n @Override\n public String getPath() {\n return \"/plugins/domains/:domain/namespaces/:namespace/:id/properties\";\n }\n\n @Override\n public Response processConfig(Map config, Map urlParams, String version) {\n try {\n String domain = ConfigUtils.requireNonEmpty(urlParams.get(\"domain\"), \"urlParams.domain must not be null and empty.\");\n String namespace = ConfigUtils.requireNonEmpty(urlParams.get(\"namespace\"), \"urlParams.namespace must not be null and empty.\");\n String id = ConfigUtils.requireNonEmpty(urlParams.get(\"id\"), \"urlParams.id must not be null and empty.\");\n Map changeConfig = new HashMap<>();\n for (Map.Entry propertyEntry : config.entrySet()) {\n String property = ConfigUtils.buildPluginProperty(domain,\n namespace,\n id,\n ConfigUtils.requireNonEmpty(propertyEntry.getKey(), \"body.key must not be null and empty.\"));\n String value = Objects.requireNonNull(propertyEntry.getValue(), String.format(\"body.%s must not be null.\", propertyEntry.getKey()));\n changeConfig.put(property, value);\n }\n this.mxBeanConfig.updateService2(changeConfig, version);\n return null;\n } catch (Exception e) {\n return Response.newFixedLengthResponse(Status.BAD_REQUEST, AgentHttpServer.JSON_TYPE, e.getMessage());\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/config/PluginPropertyHttpHandler.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.core.config;\n\nimport com.megaease.easeagent.config.ConfigManagerMXBean;\nimport com.megaease.easeagent.config.ConfigUtils;\nimport com.megaease.easeagent.core.GlobalAgentHolder;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpHandler;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpServer;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.IHTTPSession;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Response;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Status;\nimport com.megaease.easeagent.httpserver.nanohttpd.router.RouterNanoHTTPD;\nimport lombok.SneakyThrows;\n\nimport java.util.Collections;\nimport java.util.Map;\nimport java.util.Objects;\n\npublic class PluginPropertyHttpHandler extends AgentHttpHandler {\n ConfigManagerMXBean mxBeanConfig;\n\n public PluginPropertyHttpHandler() {\n this.mxBeanConfig = GlobalAgentHolder.getWrappedConfigManager();\n methods = Collections.singleton(\n com.megaease.easeagent.httpserver.nanohttpd.protocols.http.request.Method.GET);\n }\n\n @Override\n public String getPath() {\n return \"/plugins/domains/:domain/namespaces/:namespace/:id/properties/:property/:value/:version\";\n }\n\n @SneakyThrows\n @Override\n public Response process(RouterNanoHTTPD.UriResource uriResource, Map urlParams, IHTTPSession session) {\n String version = urlParams.get(\"version\");\n if (version == null) {\n return Response.newFixedLengthResponse(Status.BAD_REQUEST, AgentHttpServer.JSON_TYPE, (String) null);\n }\n try {\n String property = ConfigUtils.buildPluginProperty(\n ConfigUtils.requireNonEmpty(urlParams.get(\"domain\"), \"urlParams.domain must not be null and empty.\"),\n ConfigUtils.requireNonEmpty(urlParams.get(\"namespace\"), \"urlParams.namespace must not be null and empty.\"),\n ConfigUtils.requireNonEmpty(urlParams.get(\"id\"), \"urlParams.id must not be null and empty.\"),\n ConfigUtils.requireNonEmpty(urlParams.get(\"property\"), \"urlParams.property must not be null and empty.\"));\n String value = Objects.requireNonNull(urlParams.get(\"value\"), \"urlParams.value must not be null.\");\n this.mxBeanConfig.updateService2(Collections.singletonMap(property, value), version);\n return Response.newFixedLengthResponse(Status.OK, AgentHttpServer.JSON_TYPE, (String) null);\n } catch (Exception e) {\n return Response.newFixedLengthResponse(Status.BAD_REQUEST, AgentHttpServer.JSON_TYPE, e.getMessage());\n }\n\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/config/ServiceUpdateAgentHttpHandler.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.core.config;\n\nimport com.megaease.easeagent.config.CompatibilityConversion;\nimport com.megaease.easeagent.core.GlobalAgentHolder;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Response;\n\nimport java.util.Map;\n\npublic class ServiceUpdateAgentHttpHandler extends ConfigsUpdateAgentHttpHandler {\n public ServiceUpdateAgentHttpHandler() {\n this.mxBeanConfig = GlobalAgentHolder.getWrappedConfigManager();\n }\n\n @Override\n public String getPath() {\n return \"/config\";\n }\n\n @Override\n public Response processConfig(Map config, Map urlParams, String version) {\n this.mxBeanConfig.updateService2(CompatibilityConversion.transform(config), version);\n return null;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/health/HealthProvider.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.health;\n\nimport com.megaease.easeagent.plugin.api.config.Config;\nimport com.megaease.easeagent.config.ConfigAware;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpHandler;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpHandlerProvider;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpServer;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.IHTTPSession;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.IStatus;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Response;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Status;\nimport com.megaease.easeagent.httpserver.nanohttpd.router.RouterNanoHTTPD;\nimport com.megaease.easeagent.plugin.bean.BeanProvider;\nimport com.megaease.easeagent.plugin.bean.AgentInitializingBean;\nimport com.megaease.easeagent.plugin.api.health.AgentHealth;\n\nimport java.util.ArrayList;\nimport java.util.List;\nimport java.util.Map;\n\npublic class HealthProvider implements AgentHttpHandlerProvider, ConfigAware, AgentInitializingBean, BeanProvider {\n private static final String EASEAGENT_HEALTH_READINESS_ENABLED = \"easeagent.health.readiness.enabled\";\n\n private Config config;\n\n @Override\n public List getAgentHttpHandlers() {\n List list = new ArrayList<>();\n list.add(new HealthAgentHttpHandler());\n list.add(new LivenessAgentHttpHandler());\n list.add(new ReadinessAgentHttpHandler());\n return list;\n }\n\n @Override\n public void setConfig(Config config) {\n this.config = config;\n }\n\n @Override\n public void afterPropertiesSet() {\n AgentHealth.setReadinessEnabled(this.config.getBoolean(EASEAGENT_HEALTH_READINESS_ENABLED));\n }\n\n\n public static class HealthAgentHttpHandler extends AgentHttpHandler {\n\n @Override\n public String getPath() {\n return \"/health\";\n }\n\n @Override\n public Response process(RouterNanoHTTPD.UriResource uriResource, Map urlParams, IHTTPSession session) {\n return Response.newFixedLengthResponse(Status.OK, AgentHttpServer.JSON_TYPE, (String) null);\n }\n }\n\n public static class LivenessAgentHttpHandler extends HealthAgentHttpHandler {\n\n @Override\n public String getPath() {\n return \"/health/liveness\";\n }\n\n }\n\n public static class ReadinessAgentHttpHandler extends HealthAgentHttpHandler {\n\n @Override\n public String getPath() {\n return \"/health/readiness\";\n }\n\n @Override\n public Response process(RouterNanoHTTPD.UriResource uriResource, Map urlParams, IHTTPSession session) {\n if (AgentHealth.INSTANCE.isReadinessEnabled()) {\n if (AgentHealth.INSTANCE.isReady()) {\n return Response.newFixedLengthResponse(Status.OK, AgentHttpServer.JSON_TYPE, (String) null);\n }\n\n return Response.newFixedLengthResponse(HStatus.SERVICE_UNAVAILABLE,\n AgentHttpServer.JSON_TYPE, (String) null);\n }\n return super.process(uriResource, urlParams, session);\n }\n }\n\n enum HStatus implements IStatus {\n\n /**\n * service unavailable\n */\n SERVICE_UNAVAILABLE(503, \"Service Unavailable\"),\n ;\n private final int requestStatus;\n\n private final String description;\n\n HStatus(int requestStatus, String description) {\n this.requestStatus = requestStatus;\n this.description = description;\n }\n\n @Override\n public String getDescription() {\n return description;\n }\n\n @Override\n public int getRequestStatus() {\n return requestStatus;\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/info/AgentInfoFactory.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.info;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.bridge.AgentInfo;\n\nimport java.io.BufferedReader;\nimport java.io.IOException;\nimport java.io.InputStream;\nimport java.io.InputStreamReader;\n\npublic class AgentInfoFactory {\n private static final Logger LOGGER = LoggerFactory.getLogger(AgentInfoFactory.class);\n public static final String AGENT_TYPE = \"EaseAgent\";\n private static final String VERSION_FILE = \"version.txt\";\n\n\n public static AgentInfo loadAgentInfo(ClassLoader classLoader) {\n return new AgentInfo(AGENT_TYPE, loadVersion(classLoader, VERSION_FILE));\n }\n\n\n private static String loadVersion(ClassLoader classLoader, String file) {\n try (InputStream in = classLoader.getResourceAsStream(file)) {\n BufferedReader reader = new BufferedReader(new InputStreamReader(in));\n String version = reader.readLine();\n reader.close();\n return version;\n } catch (IOException e) {\n LOGGER.warn(\"Load config file:{} by classloader:{} failure: {}\", file, classLoader.toString(), e);\n }\n return \"\";\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/info/AgentInfoProvider.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.info;\n\nimport com.megaease.easeagent.core.utils.JsonUtil;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpHandler;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpHandlerProvider;\nimport com.megaease.easeagent.httpserver.nano.AgentHttpServer;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.IHTTPSession;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Response;\nimport com.megaease.easeagent.httpserver.nanohttpd.protocols.http.response.Status;\nimport com.megaease.easeagent.httpserver.nanohttpd.router.RouterNanoHTTPD;\nimport com.megaease.easeagent.plugin.bean.BeanProvider;\nimport com.megaease.easeagent.plugin.bridge.EaseAgent;\n\nimport java.util.ArrayList;\nimport java.util.List;\nimport java.util.Map;\n\npublic class AgentInfoProvider implements AgentHttpHandlerProvider, BeanProvider {\n\n @Override\n public List getAgentHttpHandlers() {\n List list = new ArrayList<>();\n list.add(new AgentInfoHttpHandler());\n return list;\n }\n\n public static class AgentInfoHttpHandler extends AgentHttpHandler {\n\n @Override\n public String getPath() {\n return \"/agent-info\";\n }\n\n @Override\n public Response process(RouterNanoHTTPD.UriResource uriResource, Map urlParams, IHTTPSession session) {\n return Response.newFixedLengthResponse(Status.OK, AgentHttpServer.JSON_TYPE, JsonUtil.toJson(EaseAgent.getAgentInfo()));\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/BaseLoader.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin;\n\nimport com.megaease.easeagent.plugin.Ordered;\nimport com.megaease.easeagent.plugin.api.logging.Logger;\nimport com.megaease.easeagent.plugin.bridge.EaseAgent;\n\nimport java.util.ArrayList;\nimport java.util.Comparator;\nimport java.util.Iterator;\nimport java.util.List;\nimport java.util.ServiceLoader;\n\npublic class BaseLoader {\n private static final Logger logger = EaseAgent.loggerFactory.getLogger(BaseLoader.class);\n\n public static List load(Class serviceClass) {\n List result = new ArrayList<>();\n java.util.ServiceLoader services = ServiceLoader.load(serviceClass);\n for (Iterator it = services.iterator(); it.hasNext(); ) {\n try {\n result.add(it.next());\n } catch (UnsupportedClassVersionError e) {\n logger.info(\"Unable to load class: {}\", e.getMessage());\n logger.info(\"Please check the plugin compile Java version configuration,\"\n + \" and it should not latter than current JVM runtime\");\n }\n }\n return result;\n }\n\n public static List loadOrdered(Class serviceClass) {\n List result = load(serviceClass);\n result.sort(Comparator.comparing(Ordered::order));\n return result;\n }\n\n private BaseLoader() {}\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/BridgeDispatcher.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin;\n\nimport com.megaease.easeagent.plugin.interceptor.MethodInfo;\nimport com.megaease.easeagent.plugin.api.Context;\nimport com.megaease.easeagent.plugin.api.InitializeContext;\nimport com.megaease.easeagent.plugin.api.dispatcher.IDispatcher;\nimport com.megaease.easeagent.plugin.bridge.EaseAgent;\n\npublic class BridgeDispatcher implements IDispatcher {\n @Override\n public void enter(int chainIndex, MethodInfo info) {\n InitializeContext context = EaseAgent.initializeContextSupplier\n .getContext();\n if (context.isNoop()) {\n return;\n }\n Dispatcher.enter(chainIndex, info, context);\n }\n\n @Override\n public Object exit(int chainIndex, MethodInfo methodInfo,\n Context context, Object result, Throwable e) {\n if (context.isNoop() || !(context instanceof InitializeContext)) {\n return result;\n }\n InitializeContext iContext = (InitializeContext)context;\n methodInfo.throwable(e);\n methodInfo.retValue(result);\n Dispatcher.exit(chainIndex, methodInfo, iContext);\n if (methodInfo.isChanged()) {\n result = methodInfo.getRetValue();\n }\n\n return result;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/CommonInlineAdvice.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin;\n\nimport com.megaease.easeagent.core.plugin.annotation.Index;\nimport com.megaease.easeagent.core.plugin.transformer.advice.AgentAdvice.NoExceptionHandler;\nimport com.megaease.easeagent.plugin.interceptor.MethodInfo;\nimport com.megaease.easeagent.plugin.api.InitializeContext;\nimport com.megaease.easeagent.plugin.bridge.EaseAgent;\nimport net.bytebuddy.asm.Advice;\nimport net.bytebuddy.implementation.bytecode.assign.Assigner;\n\n/**\n * uniform interceptor entrance\n * get interceptor chain thought index generated when transform\n */\n// suppress all warnings for the code at these warnings is intentionally written this way\n@SuppressWarnings(\"all\")\npublic class CommonInlineAdvice {\n private static final String CONTEXT = \"easeagent_context\";\n private static final String POS = \"easeagent_pos\";\n\n @Advice.OnMethodEnter(suppress = NoExceptionHandler.class)\n public static MethodInfo enter(@Index int index,\n @Advice.This(optional = true) Object invoker,\n @Advice.Origin(\"#t\") String type,\n @Advice.Origin(\"#m\") String method,\n @Advice.AllArguments(readOnly = false, typing = Assigner.Typing.DYNAMIC) Object[] args,\n @Advice.Local(CONTEXT) InitializeContext context) {\n context = EaseAgent.initializeContextSupplier.getContext();\n if (context.isNoop()) {\n return null;\n }\n\n MethodInfo methodInfo = MethodInfo.builder()\n .invoker(invoker)\n .type(type)\n .method(method)\n .args(args)\n .build();\n Dispatcher.enter(index, methodInfo, context);\n if (methodInfo.isChanged()) {\n args = methodInfo.getArgs();\n }\n\n return methodInfo;\n }\n\n @Advice.OnMethodExit(onThrowable = Exception.class, suppress = NoExceptionHandler.class)\n // @Advice.OnMethodExit(suppress = NoExceptionHandler.class)\n public static void exit(@Index int index,\n @Advice.Enter MethodInfo methodInfo,\n @Advice.Return(readOnly = false, typing = Assigner.Typing.DYNAMIC) Object result,\n @Advice.Thrown(readOnly = false, typing = Assigner.Typing.DYNAMIC) Throwable throwable,\n @Advice.Local(CONTEXT) InitializeContext context) {\n if (context.isNoop()) {\n return;\n }\n methodInfo.throwable(throwable);\n methodInfo.retValue(result);\n Dispatcher.exit(index, methodInfo, context);\n if (methodInfo.isChanged()) {\n result = methodInfo.getRetValue();\n }\n }\n\n @Advice.OnMethodExit(suppress = NoExceptionHandler.class)\n public static void exit(@Index int index,\n @Advice.This(optional = true) Object invoker,\n @Advice.Enter MethodInfo methodInfo,\n @Advice.Return(readOnly = false, typing = Assigner.Typing.DYNAMIC) Object result,\n @Advice.Local(CONTEXT) InitializeContext context) {\n if (context.isNoop()) {\n return;\n }\n methodInfo.setInvoker(invoker);\n methodInfo.retValue(result);\n Dispatcher.exit(index, methodInfo, context);\n if (methodInfo.isChanged()) {\n result = methodInfo.getRetValue();\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/Dispatcher.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin;\n\nimport com.google.auto.service.AutoService;\nimport com.megaease.easeagent.core.utils.AgentArray;\nimport com.megaease.easeagent.core.utils.ContextUtils;\nimport com.megaease.easeagent.plugin.AppendBootstrapLoader;\nimport com.megaease.easeagent.plugin.interceptor.MethodInfo;\nimport com.megaease.easeagent.plugin.api.InitializeContext;\nimport com.megaease.easeagent.plugin.interceptor.AgentInterceptorChain;\n\n@AutoService(AppendBootstrapLoader.class)\npublic final class Dispatcher {\n\n private Dispatcher() {\n }\n\n static AgentArray chains = new AgentArray<>();\n\n /**\n * for chains only modified during related class loading process,\n * so it doesn't need to consider updating process\n * otherwise, chain should store in context, avoiding changed during enter and exit\n */\n public static void enter(int index, MethodInfo info, InitializeContext ctx) {\n AgentInterceptorChain chain = chains.getUncheck(index);\n int pos = 0;\n ContextUtils.setBeginTime(ctx);\n chain.doBefore(info, pos, ctx);\n }\n\n public static Object exit(int index, MethodInfo info, InitializeContext ctx) {\n AgentInterceptorChain chain = chains.getUncheck(index);\n int pos = chain.size() - 1;\n ContextUtils.setEndTime(ctx);\n return chain.doAfter(info, pos, ctx);\n }\n\n public static AgentInterceptorChain register(int index, AgentInterceptorChain chain) {\n return chains.putIfAbsent(index, chain);\n }\n\n // for interceptor\n public static AgentInterceptorChain getChain(int index) {\n return chains.get(index);\n }\n\n public static boolean updateChain(int index, AgentInterceptorChain chain) {\n return chains.replace(index, chain) != null;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/PluginLoader.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin;\n\nimport com.megaease.easeagent.config.ConfigUtils;\nimport com.megaease.easeagent.config.Configs;\nimport com.megaease.easeagent.core.plugin.matcher.ClassTransformation;\nimport com.megaease.easeagent.core.plugin.matcher.MethodTransformation;\nimport com.megaease.easeagent.core.plugin.registry.PluginRegistry;\nimport com.megaease.easeagent.core.plugin.transformer.CompoundPluginTransformer;\nimport com.megaease.easeagent.core.plugin.transformer.DynamicFieldTransformer;\nimport com.megaease.easeagent.core.plugin.transformer.ForAdviceTransformer;\nimport com.megaease.easeagent.core.plugin.transformer.TypeFieldTransformer;\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.AgentPlugin;\nimport com.megaease.easeagent.plugin.CodeVersion;\nimport com.megaease.easeagent.plugin.Ordered;\nimport com.megaease.easeagent.plugin.Points;\nimport com.megaease.easeagent.plugin.field.AgentDynamicFieldAccessor;\nimport com.megaease.easeagent.plugin.interceptor.InterceptorProvider;\nimport com.megaease.easeagent.plugin.utils.common.StringUtils;\nimport net.bytebuddy.agent.builder.AgentBuilder;\n\nimport java.util.*;\nimport java.util.stream.Collectors;\nimport java.util.stream.StreamSupport;\n\npublic class PluginLoader {\n\n private PluginLoader() {\n }\n\n static Logger log = LoggerFactory.getLogger(PluginLoader.class);\n\n public static AgentBuilder load(AgentBuilder ab, Configs conf) {\n pluginLoad();\n pointsLoad(conf);\n providerLoad();\n Set sortedTransformations = classTransformationLoad();\n\n for (ClassTransformation transformation : sortedTransformations) {\n ab = ab.type(transformation.getClassMatcher(), transformation.getClassloaderMatcher())\n .transform(compound(transformation.isHasDynamicField(), transformation.getMethodTransformations(), transformation.getTypeFieldAccessor()));\n }\n return ab;\n }\n\n public static void providerLoad() {\n for (InterceptorProvider provider : BaseLoader.load(InterceptorProvider.class)) {\n String pointsClassName = PluginRegistry.getPointsClassName(provider.getAdviceTo());\n Points points = PluginRegistry.getPoints(pointsClassName);\n if (points == null) {\n log.debug(\"Unload provider:{}, can not found Points<{}>\", provider.getClass().getName(), pointsClassName);\n continue;\n } else {\n log.debug(\"Loading provider:{}\", provider.getClass().getName());\n }\n\n try {\n log.debug(\"provider for:{} at {}\",\n provider.getPluginClassName(), provider.getAdviceTo());\n PluginRegistry.register(provider);\n } catch (Exception | LinkageError e) {\n log.error(\n \"Unable to load provider in [class {}]\",\n provider.getClass().getName(),\n e);\n }\n }\n }\n\n public static Set classTransformationLoad() {\n Collection points = PluginRegistry.getPoints();\n return points.stream().map(point -> {\n try {\n return PluginRegistry.registerClassTransformation(point);\n } catch (Exception e) {\n log.error(\n \"Unable to load classTransformation in [class {}]\",\n point.getClass().getName(),\n e);\n return null;\n }\n }).filter(Objects::nonNull)\n .sorted(Comparator.comparing(Ordered::order))\n .collect(Collectors.toCollection(LinkedHashSet::new));\n }\n\n public static void pluginLoad() {\n for (AgentPlugin plugin : BaseLoader.loadOrdered(AgentPlugin.class)) {\n log.info(\n \"Loading plugin {}:{} [class {}]\",\n plugin.getDomain(),\n plugin.getNamespace(),\n plugin.getClass().getName());\n\n try {\n PluginRegistry.register(plugin);\n } catch (Exception | LinkageError e) {\n log.error(\n \"Unable to load extension {}:{} [class {}]\",\n plugin.getDomain(),\n plugin.getNamespace(),\n plugin.getClass().getName(),\n e);\n }\n }\n }\n\n public static void pointsLoad(Configs conf) {\n for (Points points : BaseLoader.load(Points.class)) {\n if (!isCodeVersion(points, conf)) {\n continue;\n } else {\n log.info(\"Loading points [class Points<{}>]\", points.getClass().getName());\n }\n\n try {\n PluginRegistry.register(points);\n } catch (Exception | LinkageError e) {\n log.error(\n \"Unable to load extension [class {}]\",\n points.getClass().getName(),\n e);\n }\n }\n }\n\n public static boolean isCodeVersion(Points points, Configs conf) {\n CodeVersion codeVersion = points.codeVersions();\n if (codeVersion.isEmpty()) {\n return true;\n }\n String versionKey = ConfigUtils.buildCodeVersionKey(codeVersion.getKey());\n Set versions = new HashSet<>(conf.getStringList(versionKey));\n if (versions.isEmpty()) {\n versions = Points.DEFAULT_VERSIONS;\n }\n Set pointVersions = codeVersion.getVersions();\n for (String version : versions) {\n if (pointVersions.contains(version)) {\n return true;\n }\n }\n log.info(\"Unload points [class Points<{}>], the config [{}={}] is not in Points.codeVersions()=[{}:{}]\",\n points.getClass().getCanonicalName(), versionKey, String.join(\",\", versions),\n codeVersion.getKey(), String.join(\",\", codeVersion.getVersions()));\n return false;\n }\n\n\n /**\n * @param methodTransformations method matchers under a special classMatcher\n * @return transform\n */\n public static AgentBuilder.Transformer compound(boolean hasDynamicField,\n Iterable methodTransformations, String typeFieldAccessor) {\n List agentTransformers = StreamSupport\n .stream(methodTransformations.spliterator(), false)\n .map(ForAdviceTransformer::new)\n .collect(Collectors.toList());\n\n if (hasDynamicField) {\n agentTransformers.add(new DynamicFieldTransformer(AgentDynamicFieldAccessor.DYNAMIC_FIELD_NAME));\n }\n\n if (StringUtils.hasText(typeFieldAccessor)) {\n agentTransformers.add(new TypeFieldTransformer(typeFieldAccessor));\n }\n\n return new CompoundPluginTransformer(agentTransformers);\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/annotation/EaseAgentInstrumented.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.annotation;\n\nimport java.lang.annotation.Retention;\nimport java.lang.annotation.RetentionPolicy;\n\n@Retention(RetentionPolicy.RUNTIME)\npublic @interface EaseAgentInstrumented {\n int value() default 0;\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/annotation/Index.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.annotation;\n\nimport java.lang.annotation.Retention;\nimport java.lang.annotation.RetentionPolicy;\n\n@Retention(RetentionPolicy.RUNTIME)\npublic @interface Index {\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/interceptor/InterceptorPluginDecorator.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.interceptor;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.AgentPlugin;\nimport com.megaease.easeagent.plugin.interceptor.Interceptor;\nimport com.megaease.easeagent.plugin.interceptor.MethodInfo;\nimport com.megaease.easeagent.plugin.api.Context;\nimport com.megaease.easeagent.plugin.api.InitializeContext;\nimport com.megaease.easeagent.plugin.api.config.AutoRefreshPluginConfigImpl;\nimport com.megaease.easeagent.plugin.api.config.AutoRefreshPluginConfigRegistry;\nimport com.megaease.easeagent.plugin.api.config.IPluginConfig;\nimport com.megaease.easeagent.plugin.bridge.NoOpIPluginConfig;\n\nimport java.util.function.Supplier;\n\npublic class InterceptorPluginDecorator implements Interceptor {\n private static final Logger LOGGER = LoggerFactory.getLogger(InterceptorPluginDecorator.class);\n private final Interceptor interceptor;\n private final AgentPlugin plugin;\n private final AutoRefreshPluginConfigImpl config;\n\n public InterceptorPluginDecorator(Interceptor interceptor, AgentPlugin plugin) {\n this.interceptor = interceptor;\n this.plugin = plugin;\n this.config = AutoRefreshPluginConfigRegistry.getOrCreate(plugin.getDomain(), plugin.getNamespace(), interceptor.getType());\n }\n\n public IPluginConfig getConfig() {\n return this.config.getConfig();\n }\n\n @Override\n public void before(MethodInfo methodInfo, Context context) {\n IPluginConfig cfg = this.config.getConfig();\n InitializeContext innerContext = (InitializeContext) context;\n innerContext.pushConfig(cfg);\n if (cfg == null || cfg.enabled() || cfg instanceof NoOpIPluginConfig) {\n innerContext.pushRetBound();\n this.interceptor.before(methodInfo, context);\n } else if (LOGGER.isDebugEnabled()) {\n LOGGER.debug(\"plugin.{}.{}.{} is not enabled\", config.domain(), config.namespace(), config.id());\n }\n }\n\n @Override\n public void after(MethodInfo methodInfo, Context context) {\n IPluginConfig cfg = context.getConfig();\n InitializeContext innerContext = (InitializeContext) context;\n try {\n if (cfg == null || cfg.enabled() || cfg instanceof NoOpIPluginConfig) {\n try {\n this.interceptor.after(methodInfo, context);\n } finally {\n innerContext.popToBound();\n innerContext.popRetBound();\n }\n }\n } finally {\n innerContext.popConfig();\n }\n }\n\n @Override\n public String getType() {\n return this.interceptor.getType();\n }\n\n @Override\n public void init(IPluginConfig config, String type, String method, String methodDescriptor) {\n this.interceptor.init(config, type, method, methodDescriptor);\n }\n\n @Override\n public void init(IPluginConfig config, int uniqueIndex) {\n this.interceptor.init(config, uniqueIndex);\n }\n\n @Override\n public int order() {\n int pluginOrder = this.plugin.order();\n int interceptorOrder = this.interceptor.order();\n return (interceptorOrder << 8) + pluginOrder;\n }\n\n public static Supplier getInterceptorSupplier(final AgentPlugin plugin, final Supplier supplier) {\n return () -> new InterceptorPluginDecorator(supplier.get(), plugin);\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/interceptor/ProviderChain.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.interceptor;\n\nimport com.megaease.easeagent.plugin.interceptor.InterceptorProvider;\nimport com.megaease.easeagent.plugin.interceptor.Interceptor;\n\nimport java.util.ArrayList;\nimport java.util.List;\nimport java.util.function.Supplier;\nimport java.util.stream.Collectors;\n\npublic class ProviderChain {\n private final List providers;\n\n ProviderChain(List providers) {\n this.providers = providers;\n }\n\n public List> getSupplierChain() {\n return this.providers.stream()\n .map(InterceptorProvider::getInterceptorProvider)\n .collect(Collectors.toCollection(ArrayList::new));\n }\n\n public static Builder builder() {\n return new Builder();\n }\n\n @SuppressWarnings(\"all\")\n public static class Builder {\n private List providers = new ArrayList<>();\n\n Builder() {\n }\n\n public Builder providers(List providers) {\n this.providers = providers;\n return this;\n }\n\n public Builder addProvider(InterceptorProvider supplier) {\n this.providers.add(supplier);\n return this;\n }\n\n public ProviderChain build() {\n return new ProviderChain(providers);\n }\n\n @Override\n public String toString() {\n return \"ProviderChain.Builder(providers=\" + this.providers + \")\";\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/interceptor/ProviderPluginDecorator.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.interceptor;\n\nimport com.megaease.easeagent.plugin.AgentPlugin;\nimport com.megaease.easeagent.plugin.interceptor.InterceptorProvider;\nimport com.megaease.easeagent.plugin.interceptor.Interceptor;\n\nimport java.util.function.Supplier;\n\npublic class ProviderPluginDecorator implements InterceptorProvider {\n private final AgentPlugin plugin;\n private final InterceptorProvider provider;\n\n public ProviderPluginDecorator(AgentPlugin plugin, InterceptorProvider provider) {\n this.plugin = plugin;\n this.provider = provider;\n }\n\n @Override\n public Supplier getInterceptorProvider() {\n return () -> {\n Supplier origin = ProviderPluginDecorator.this.provider.getInterceptorProvider();\n return new InterceptorPluginDecorator(origin.get(), this.plugin);\n };\n }\n\n @Override\n public String getAdviceTo() {\n return this.provider.getAdviceTo();\n }\n\n @Override\n public String getPluginClassName() {\n return this.provider.getPluginClassName();\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/matcher/ClassLoaderMatcherConvert.java", "content": "/*\n * Copyright (c) 2022, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n *\n */\npackage com.megaease.easeagent.core.plugin.matcher;\n\nimport com.megaease.easeagent.core.Bootstrap;\nimport com.megaease.easeagent.log4j2.FinalClassloaderSupplier;\nimport com.megaease.easeagent.plugin.matcher.loader.IClassLoaderMatcher;\nimport com.megaease.easeagent.plugin.matcher.loader.NegateClassLoaderMatcher;\nimport net.bytebuddy.matcher.ElementMatcher;\n\nimport static com.megaease.easeagent.plugin.matcher.loader.ClassLoaderMatcher.*;\nimport static net.bytebuddy.matcher.ElementMatchers.*;\n\npublic class ClassLoaderMatcherConvert implements Converter> {\n public static final ClassLoaderMatcherConvert INSTANCE = new ClassLoaderMatcherConvert();\n\n private static final ElementMatcher agentLoaderMatcher = is(Bootstrap.class.getClassLoader())\n .or(is(FinalClassloaderSupplier.CLASSLOADER));\n\n @Override\n public ElementMatcher convert(IClassLoaderMatcher source) {\n boolean negate;\n ElementMatcher matcher;\n if (source instanceof NegateClassLoaderMatcher) {\n negate = true;\n source = source.negate();\n } else {\n negate = false;\n }\n\n if (ALL.equals(source)) {\n matcher = any();\n } else {\n switch (source.getClassLoaderName()) {\n case BOOTSTRAP_NAME:\n matcher = isBootstrapClassLoader();\n break;\n case EXTERNAL_NAME:\n matcher = isExtensionClassLoader();\n break;\n case SYSTEM_NAME:\n matcher = isSystemClassLoader();\n break;\n case AGENT_NAME:\n matcher = agentLoaderMatcher;\n break;\n default:\n matcher = new NameMatcher(source.getClassLoaderName());\n break;\n }\n }\n\n if (negate) {\n return not(matcher);\n } else {\n return matcher;\n }\n }\n\n static class NameMatcher implements ElementMatcher {\n final String className;\n\n public NameMatcher(String name) {\n this.className = name;\n }\n\n @Override\n public boolean matches(ClassLoader target) {\n if (target == null) {\n return this.className == null;\n }\n return this.className.equals(target.getClass().getCanonicalName());\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/matcher/ClassMatcherConvert.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.matcher;\n\nimport com.megaease.easeagent.plugin.asm.Modifier;\nimport com.megaease.easeagent.plugin.matcher.ClassMatcher;\nimport com.megaease.easeagent.plugin.matcher.IClassMatcher;\nimport com.megaease.easeagent.plugin.matcher.operator.AndClassMatcher;\nimport com.megaease.easeagent.plugin.matcher.operator.NegateClassMatcher;\nimport com.megaease.easeagent.plugin.matcher.operator.OrClassMatcher;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.matcher.ElementMatcher.Junction;\nimport net.bytebuddy.matcher.NegatingMatcher;\n\nimport static net.bytebuddy.matcher.ElementMatchers.*;\n\npublic class ClassMatcherConvert\n implements Converter> {\n public static final ClassMatcherConvert INSTANCE = new ClassMatcherConvert();\n\n @Override\n public Junction convert(IClassMatcher source) {\n if (source == null) {\n return null;\n }\n\n if (source instanceof AndClassMatcher) {\n AndClassMatcher andMatcher = (AndClassMatcher) source;\n Junction leftMatcher = this.convert(andMatcher.getLeft());\n Junction rightMatcher = this.convert(andMatcher.getRight());\n return leftMatcher.and(rightMatcher);\n } else if (source instanceof OrClassMatcher) {\n OrClassMatcher andMatcher = (OrClassMatcher) source;\n Junction leftMatcher = this.convert(andMatcher.getLeft());\n Junction rightMatcher = this.convert(andMatcher.getRight());\n return leftMatcher.or(rightMatcher);\n } else if (source instanceof NegateClassMatcher) {\n NegateClassMatcher matcher = (NegateClassMatcher) source;\n Junction notMatcher = this.convert(matcher.getMatcher());\n return new NegatingMatcher<>(notMatcher);\n }\n\n if (!(source instanceof ClassMatcher)) {\n return null;\n }\n\n return this.convert((ClassMatcher) source);\n }\n\n private Junction convert(ClassMatcher matcher) {\n Junction c;\n switch (matcher.getMatchType()) {\n case NAMED:\n c = named(matcher.getName());\n break;\n case SUPER_CLASS:\n case INTERFACE:\n c = hasSuperType(named(matcher.getName()));\n break;\n case ANNOTATION:\n c = isAnnotatedWith(named(matcher.getName()));\n break;\n default:\n return null;\n }\n\n Junction mc = fromModifier(matcher.getModifier(), false);\n if (mc != null) {\n c = c.and(mc);\n }\n mc = fromModifier(matcher.getNotModifier(), true);\n if (mc != null) {\n c = c.and(mc);\n }\n\n // TODO: classloader matcher\n\n return c;\n }\n\n Junction fromModifier(int modifier, boolean not) {\n Junction mc = null;\n if ((modifier & ClassMatcher.MODIFIER_MASK) != 0) {\n if ((modifier & Modifier.ACC_ABSTRACT) != 0) {\n mc = isAbstract();\n }\n if ((modifier & Modifier.ACC_PUBLIC) != 0) {\n if (mc != null) {\n mc = not ? mc.or(isPublic()) : mc.and(isPublic());\n } else {\n mc = isPublic();\n }\n }\n if ((modifier & Modifier.ACC_PRIVATE) != 0) {\n if (mc != null) {\n mc = not ? mc.or(isPrivate()) : mc.and(isPrivate());\n } else {\n mc = isPrivate();\n }\n }\n\n if ((modifier & Modifier.ACC_INTERFACE) != 0) {\n if (mc != null) {\n mc = not ? mc.or(isInterface()) : mc.and(isInterface());\n } else {\n mc = isInterface();\n }\n }\n\n if ((modifier & Modifier.ACC_PROTECTED) != 0) {\n if (mc != null) {\n mc = not ? mc.or(isProtected()) : mc.and(isProtected());\n } else {\n mc = isProtected();\n }\n }\n if (not) {\n mc = new NegatingMatcher<>(mc);\n }\n }\n return mc;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/matcher/ClassTransformation.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.matcher;\n\nimport com.megaease.easeagent.plugin.Ordered;\nimport lombok.Data;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.matcher.ElementMatcher;\nimport net.bytebuddy.matcher.ElementMatcher.Junction;\n\nimport java.util.Set;\n\nimport static net.bytebuddy.matcher.ElementMatchers.any;\n\n@Data\npublic class ClassTransformation implements Ordered {\n private int order;\n private Junction classMatcher;\n private ElementMatcher classloaderMatcher;\n private Set methodTransformations;\n private boolean hasDynamicField;\n private String typeFieldAccessor;\n\n public ClassTransformation(int order,\n ElementMatcher classloaderMatcher,\n Junction classMatcher,\n Set methodTransformations,\n boolean hasDynamicField,\n String typeFieldAccessor) {\n this.order = order;\n if (classloaderMatcher == null) {\n this.classloaderMatcher = any();\n } else {\n this.classloaderMatcher = classloaderMatcher;\n }\n this.classMatcher = classMatcher;\n this.methodTransformations = methodTransformations;\n this.hasDynamicField = hasDynamicField;\n this.typeFieldAccessor = typeFieldAccessor;\n }\n\n public static Builder builder() {\n return new Builder();\n }\n\n @Override\n public int order() {\n return this.order;\n }\n\n public static class Builder {\n private int order;\n private Junction classMatcher;\n private ElementMatcher classloaderMatcher = null;\n private Set methodTransformations;\n private boolean hasDynamicField;\n\n private String typeFieldAccessor;\n\n Builder() {\n }\n\n public Builder order(int order) {\n this.order = order;\n return this;\n }\n\n public Builder classloaderMatcher(ElementMatcher clmMatcher) {\n this.classloaderMatcher = clmMatcher;\n return this;\n }\n\n public Builder classMatcher(Junction classMatcher) {\n this.classMatcher = classMatcher;\n return this;\n }\n\n public Builder methodTransformations(Set methodTransformations) {\n this.methodTransformations = methodTransformations;\n return this;\n }\n\n public Builder hasDynamicField(boolean hasDynamicField) {\n this.hasDynamicField = hasDynamicField;\n return this;\n }\n\n public Builder typeFieldAccessor(String typeFieldAccessor) {\n this.typeFieldAccessor = typeFieldAccessor;\n return this;\n }\n\n public ClassTransformation build() {\n return new ClassTransformation(order, classloaderMatcher, classMatcher,\n methodTransformations, hasDynamicField, typeFieldAccessor);\n }\n\n public String toString() {\n return \"ClassTransformation.Builder(order=\" + this.order + \", classMatcher=\" + this.classMatcher + \", methodTransformations=\" + this.methodTransformations + \", hasDynamicField=\" + this.hasDynamicField + \")\";\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/matcher/Converter.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.matcher;\n\npublic interface Converter {\n T convert(S source);\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/matcher/MethodMatcherConvert.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.matcher;\n\nimport com.megaease.easeagent.plugin.asm.Modifier;\nimport com.megaease.easeagent.plugin.matcher.ClassMatcher;\nimport com.megaease.easeagent.plugin.matcher.IMethodMatcher;\nimport com.megaease.easeagent.plugin.matcher.MethodMatcher;\nimport com.megaease.easeagent.plugin.matcher.operator.AndMethodMatcher;\nimport com.megaease.easeagent.plugin.matcher.operator.NegateMethodMatcher;\nimport com.megaease.easeagent.plugin.matcher.operator.OrMethodMatcher;\nimport net.bytebuddy.description.method.MethodDescription;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.matcher.ElementMatcher.Junction;\nimport net.bytebuddy.matcher.NegatingMatcher;\n\nimport static net.bytebuddy.matcher.ElementMatchers.*;\n\npublic class MethodMatcherConvert\n implements Converter> {\n\n public static final MethodMatcherConvert INSTANCE = new MethodMatcherConvert();\n\n @Override\n public Junction convert(IMethodMatcher source) {\n if (source == null) {\n return null;\n }\n\n if (source instanceof AndMethodMatcher) {\n AndMethodMatcher andMatcher = (AndMethodMatcher) source;\n Junction leftMatcher = this.convert(andMatcher.getLeft());\n Junction rightMatcher = this.convert(andMatcher.getRight());\n return leftMatcher.and(rightMatcher);\n } else if (source instanceof OrMethodMatcher) {\n OrMethodMatcher andMatcher = (OrMethodMatcher) source;\n Junction leftMatcher = this.convert(andMatcher.getLeft());\n Junction rightMatcher = this.convert(andMatcher.getRight());\n return leftMatcher.or(rightMatcher);\n } else if (source instanceof NegateMethodMatcher) {\n NegateMethodMatcher matcher = (NegateMethodMatcher) source;\n Junction notMatcher = this.convert(matcher.getMatcher());\n return new NegatingMatcher<>(notMatcher);\n }\n\n if (!(source instanceof MethodMatcher)) {\n return null;\n }\n\n return this.convert((MethodMatcher) source);\n }\n\n private Junction convert(MethodMatcher matcher) {\n Junction c = null;\n if (matcher.getName() != null && matcher.getNameMatchType() != null) {\n switch (matcher.getNameMatchType()) {\n case EQUALS:\n if (\"\".equals(matcher.getName())) {\n c = isConstructor();\n } else {\n c = named(matcher.getName());\n }\n break;\n case START_WITH:\n c = nameStartsWith(matcher.getName());\n break;\n case END_WITH:\n c = nameEndsWith(matcher.getName());\n break;\n case CONTAINS:\n c = nameContains(matcher.getName());\n break;\n default:\n return null;\n }\n }\n\n Junction mc = fromModifier(matcher.getModifier(), false);\n if (mc != null) {\n c = c == null ? mc : c.and(mc);\n }\n mc = fromModifier(matcher.getNotModifier(), true);\n if (mc != null) {\n c = c == null ? mc : c.and(mc);\n }\n if (matcher.getReturnType() != null) {\n mc = returns(named(matcher.getReturnType()));\n c = c == null ? mc : c.and(mc);\n }\n if (matcher.getArgsLength() > -1) {\n mc = takesArguments(matcher.getArgsLength());\n c = c == null ? mc : c.and(mc);\n }\n String[] args = matcher.getArgs();\n if (args != null) {\n for (int i = 0; i < args.length; i++) {\n if (args[i] != null) {\n mc = takesArgument(i, named(args[i]));\n c = c == null ? mc : c.and(mc);\n }\n }\n }\n\n if (matcher.getArgsLength() >= 0) {\n mc = takesArguments(matcher.getArgsLength());\n c = c == null ? mc : c.and(mc);\n }\n\n if (matcher.getOverriddenFrom() != null) {\n Junction cls = ClassMatcherConvert.INSTANCE.convert(matcher.getOverriddenFrom());\n mc = isOverriddenFrom(cls);\n c = c == null ? mc : c.and(mc);\n }\n\n return c;\n }\n\n Junction fromModifier(int modifier, boolean not) {\n Junction mc = null;\n if ((modifier & ClassMatcher.MODIFIER_MASK) != 0) {\n if ((modifier & Modifier.ACC_ABSTRACT) != 0) {\n mc = isAbstract();\n }\n if ((modifier & Modifier.ACC_PUBLIC) != 0) {\n if (mc != null) {\n mc = not ? mc.or(isPublic()) : mc.and(isPublic());\n } else {\n mc = isPublic();\n }\n }\n if ((modifier & Modifier.ACC_PRIVATE) != 0) {\n if (mc != null) {\n mc = not ? mc.or(isPrivate()) : mc.and(isPrivate());\n } else {\n mc = isPrivate();\n }\n }\n if ((modifier & Modifier.ACC_PROTECTED) != 0) {\n if (mc != null) {\n mc = not ? mc.or(isProtected()) : mc.and(isProtected());\n } else {\n mc = isProtected();\n }\n }\n if ((modifier & Modifier.ACC_STATIC) != 0) {\n if (mc != null) {\n mc = not ? mc.or(isStatic()) : mc.and(isStatic());\n } else {\n mc = isStatic();\n }\n }\n if (not) {\n mc = new NegatingMatcher<>(mc);\n }\n }\n return mc;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/matcher/MethodTransformation.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.matcher;\n\nimport com.megaease.easeagent.core.plugin.Dispatcher;\nimport com.megaease.easeagent.core.plugin.interceptor.InterceptorPluginDecorator;\nimport com.megaease.easeagent.core.plugin.interceptor.ProviderChain;\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.api.config.ConfigConst;\nimport com.megaease.easeagent.plugin.enums.Order;\nimport com.megaease.easeagent.plugin.interceptor.Interceptor;\nimport com.megaease.easeagent.plugin.Ordered;\nimport com.megaease.easeagent.plugin.interceptor.AgentInterceptorChain;\nimport lombok.Data;\nimport net.bytebuddy.description.method.MethodDescription;\nimport net.bytebuddy.matcher.ElementMatcher.Junction;\n\nimport java.util.Comparator;\nimport java.util.List;\nimport java.util.function.Supplier;\nimport java.util.stream.Collectors;\n\n@Data\n@SuppressWarnings(\"unused\")\npublic class MethodTransformation {\n private static final Logger log = LoggerFactory.getLogger(MethodTransformation.class);\n\n private int index;\n private Junction matcher;\n private ProviderChain.Builder providerBuilder;\n\n public MethodTransformation(int index,\n Junction matcher,\n ProviderChain.Builder chain) {\n this.index = index;\n this.matcher = matcher;\n this.providerBuilder = chain;\n }\n\n public AgentInterceptorChain getAgentInterceptorChain(final int uniqueIndex,\n final String type,\n final String method,\n final String methodDescription) {\n List> suppliers = this.providerBuilder.build()\n .getSupplierChain();\n\n List interceptors = suppliers.stream()\n .map(Supplier::get)\n .sorted(Comparator.comparing(Ordered::order))\n .collect(Collectors.toList());\n\n interceptors.forEach(i -> {\n InterceptorPluginDecorator interceptor;\n if (i instanceof InterceptorPluginDecorator) {\n interceptor = (InterceptorPluginDecorator) i;\n try {\n interceptor.init(interceptor.getConfig(), type, method, methodDescription);\n interceptor.init(interceptor.getConfig(), uniqueIndex);\n } catch (Exception e) {\n log.error(\"Interceptor init fail: {}::{}, {}\", type, method, interceptor.getClass().getSimpleName());\n }\n }\n });\n\n return new AgentInterceptorChain(interceptors);\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/registry/AdviceRegistry.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.registry;\n\nimport com.megaease.easeagent.core.plugin.matcher.MethodTransformation;\nimport com.megaease.easeagent.core.plugin.transformer.advice.AgentAdvice.Dispatcher;\nimport com.megaease.easeagent.core.plugin.transformer.advice.AgentAdvice.OffsetMapping;\nimport com.megaease.easeagent.core.plugin.transformer.advice.AgentJavaConstantValue;\nimport com.megaease.easeagent.core.plugin.transformer.advice.MethodIdentityJavaConstant;\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.interceptor.AgentInterceptorChain;\nimport com.megaease.easeagent.plugin.utils.common.WeakConcurrentMap;\nimport net.bytebuddy.description.method.MethodDescription;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.implementation.bytecode.StackManipulation;\n\nimport java.lang.ref.WeakReference;\nimport java.util.Map;\nimport java.util.concurrent.ConcurrentHashMap;\nimport java.util.concurrent.atomic.AtomicInteger;\nimport java.util.concurrent.locks.ReentrantLock;\n\npublic class AdviceRegistry {\n\n private AdviceRegistry() {\n }\n\n private static final ThreadLocal> CURRENT_CLASS_LOADER = new ThreadLocal<>();\n private static final Logger log = LoggerFactory.getLogger(AdviceRegistry.class);\n static Map methodsSet = new ConcurrentHashMap<>();\n\n public static Integer check(TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n Dispatcher.Resolved.ForMethodEnter methodEnter,\n Dispatcher.Resolved.ForMethodExit methodExit) {\n String clazz = instrumentedType.getName();\n String method = instrumentedMethod.getName();\n String methodDescriptor = instrumentedMethod.getDescriptor();\n String key = clazz + \":\" + method + methodDescriptor;\n PointcutsUniqueId newIdentity = new PointcutsUniqueId();\n PointcutsUniqueId pointcutsUniqueId = methodsSet.putIfAbsent(key, newIdentity);\n\n Integer pointcutIndex;\n boolean merge = false;\n\n // already exist\n if (pointcutsUniqueId != null) {\n newIdentity.tryRelease();\n pointcutIndex = getPointcutIndex(methodEnter);\n // this pointcut's interceptors have injected into chain\n if (pointcutsUniqueId.checkPointcutExist(pointcutIndex)) {\n if (pointcutsUniqueId.checkClassloaderExist()) {\n // don't need to instrumented again.\n return 0;\n } else {\n /*\n * Although the interceptor of the pointcut has been injected,\n * the method of this class owned by current loader has not been instrumented\n */\n updateStackManipulation(methodEnter, pointcutsUniqueId.getUniqueId());\n updateStackManipulation(methodExit, pointcutsUniqueId.getUniqueId());\n return pointcutsUniqueId.getUniqueId();\n }\n } else {\n // Orchestration\n merge = true;\n }\n } else {\n // new\n pointcutsUniqueId = newIdentity;\n pointcutIndex = updateStackManipulation(methodEnter, pointcutsUniqueId.getUniqueId());\n updateStackManipulation(methodExit, pointcutsUniqueId.getUniqueId());\n }\n\n // merge or registry\n MethodTransformation methodTransformation = PluginRegistry.getMethodTransformation(pointcutIndex);\n if (methodTransformation == null) {\n log.error(\"MethodTransformation get fail for {}\", pointcutIndex);\n return 0;\n }\n int uniqueId = pointcutsUniqueId.getUniqueId();\n AgentInterceptorChain chain = methodTransformation\n .getAgentInterceptorChain(uniqueId, clazz, method, methodDescriptor);\n\n try {\n pointcutsUniqueId.lock();\n AgentInterceptorChain previousChain = com.megaease.easeagent.core.plugin.Dispatcher.getChain(uniqueId);\n if (previousChain == null) {\n com.megaease.easeagent.core.plugin.Dispatcher.register(uniqueId, chain);\n } else {\n chain.merge(previousChain);\n com.megaease.easeagent.core.plugin.Dispatcher.updateChain(uniqueId, chain);\n }\n } finally {\n pointcutsUniqueId.unlock();\n }\n\n if (merge) {\n return 0;\n }\n\n return uniqueId;\n }\n\n static Integer getPointcutIndex(Dispatcher.Resolved resolved) {\n int index = 0;\n Map enterMap = resolved.getOffsetMapping();\n for (Map.Entry offset : enterMap.entrySet()) {\n OffsetMapping om = offset.getValue();\n if (!(om instanceof OffsetMapping.ForStackManipulation)) {\n continue;\n }\n OffsetMapping.ForStackManipulation forStackManipulation = (OffsetMapping.ForStackManipulation) om;\n if (!(forStackManipulation.getStackManipulation() instanceof AgentJavaConstantValue)) {\n continue;\n }\n\n AgentJavaConstantValue value = (AgentJavaConstantValue) forStackManipulation.getStackManipulation();\n index = value.getPointcutIndex();\n break;\n }\n return index;\n }\n\n static Integer updateStackManipulation(Dispatcher.Resolved resolved, Integer value) {\n int index = 0;\n Map enterMap = resolved.getOffsetMapping();\n\n for (Map.Entry offset : enterMap.entrySet()) {\n OffsetMapping om = offset.getValue();\n if (!(om instanceof OffsetMapping.ForStackManipulation)) {\n continue;\n }\n OffsetMapping.ForStackManipulation forStackManipulation = (OffsetMapping.ForStackManipulation) om;\n if (!(forStackManipulation.getStackManipulation() instanceof AgentJavaConstantValue)) {\n continue;\n }\n\n AgentJavaConstantValue oldValue = (AgentJavaConstantValue) forStackManipulation.getStackManipulation();\n index = oldValue.getPointcutIndex();\n\n MethodIdentityJavaConstant constant = new MethodIdentityJavaConstant(value);\n StackManipulation stackManipulation = new AgentJavaConstantValue(constant, index);\n enterMap.put(offset.getKey(), forStackManipulation.with(stackManipulation));\n\n return index;\n }\n return index;\n }\n\n public static void setCurrentClassLoader(ClassLoader loader) {\n CURRENT_CLASS_LOADER.set(new WeakReference<>(loader));\n }\n\n public static ClassLoader getCurrentClassLoader() {\n return CURRENT_CLASS_LOADER.get().get();\n }\n\n public static void cleanCurrentClassLoader() {\n CURRENT_CLASS_LOADER.remove();\n }\n\n private static class PointcutsUniqueId {\n static AtomicInteger index = new AtomicInteger(1);\n ReentrantLock lock = new ReentrantLock();\n int uniqueId;\n ConcurrentHashMap pointcutIndexSet = new ConcurrentHashMap<>();\n WeakConcurrentMap cache = new WeakConcurrentMap<>();\n\n public PointcutsUniqueId() {\n this.uniqueId = index.incrementAndGet();\n }\n\n public boolean checkPointcutExist(Integer pointcutIndex) {\n return this.pointcutIndexSet.putIfAbsent(pointcutIndex, pointcutIndex) != null;\n }\n\n public int getUniqueId() {\n return this.uniqueId;\n }\n\n public boolean checkClassloaderExist() {\n ClassLoader loader = getCurrentClassLoader();\n return cache.putIfProbablyAbsent(loader, true) != null;\n }\n\n public void lock() {\n this.lock.lock();\n }\n\n public void unlock() {\n this.lock.unlock();\n }\n\n /**\n * Some empty slots may appear, the effect can be ignored and can be optimized later\n */\n public void tryRelease() {\n int id = this.uniqueId;\n index.compareAndSet(id, id - 1);\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/registry/PluginRegistry.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.registry;\n\nimport com.google.common.base.Strings;\nimport com.megaease.easeagent.core.plugin.interceptor.ProviderChain;\nimport com.megaease.easeagent.core.plugin.interceptor.ProviderChain.Builder;\nimport com.megaease.easeagent.core.plugin.interceptor.ProviderPluginDecorator;\nimport com.megaease.easeagent.core.plugin.matcher.*;\nimport com.megaease.easeagent.core.utils.AgentArray;\nimport com.megaease.easeagent.plugin.AgentPlugin;\nimport com.megaease.easeagent.plugin.Points;\nimport com.megaease.easeagent.plugin.api.logging.Logger;\nimport com.megaease.easeagent.plugin.bridge.EaseAgent;\nimport com.megaease.easeagent.plugin.interceptor.InterceptorProvider;\nimport com.megaease.easeagent.plugin.matcher.IClassMatcher;\nimport com.megaease.easeagent.plugin.matcher.IMethodMatcher;\nimport net.bytebuddy.description.method.MethodDescription;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.matcher.ElementMatcher;\nimport net.bytebuddy.matcher.ElementMatcher.Junction;\n\nimport java.util.*;\nimport java.util.concurrent.ConcurrentHashMap;\nimport java.util.stream.Collectors;\n\npublic class PluginRegistry {\n static Logger log = EaseAgent.getLogger(PluginRegistry.class);\n\n static final ConcurrentHashMap QUALIFIER_TO_PLUGIN = new ConcurrentHashMap<>();\n static final ConcurrentHashMap POINTS_TO_PLUGIN = new ConcurrentHashMap<>();\n static final ConcurrentHashMap PLUGIN_CLASSNAME_TO_PLUGIN = new ConcurrentHashMap<>();\n static final ConcurrentHashMap POINTS_CLASSNAME_TO_POINTS = new ConcurrentHashMap<>();\n\n static final ConcurrentHashMap QUALIFIER_TO_INDEX = new ConcurrentHashMap<>();\n static final ConcurrentHashMap INDEX_TO_METHOD_TRANSFORMATION = new ConcurrentHashMap<>();\n static final AgentArray INTERCEPTOR_PROVIDERS = new AgentArray<>();\n\n private PluginRegistry() {\n }\n\n public static void register(AgentPlugin plugin) {\n PLUGIN_CLASSNAME_TO_PLUGIN.putIfAbsent(plugin.getClass().getCanonicalName(), plugin);\n }\n\n public static void register(Points points) {\n POINTS_CLASSNAME_TO_POINTS.putIfAbsent(points.getClass().getCanonicalName(), points);\n }\n\n public static Collection getPoints() {\n return POINTS_CLASSNAME_TO_POINTS.values();\n }\n\n public static Points getPoints(String pointsClassName) {\n return POINTS_CLASSNAME_TO_POINTS.get(pointsClassName);\n }\n\n private static String getMethodQualifier(String classname, String qualifier) {\n return classname + \":\" + qualifier;\n }\n\n public static ClassTransformation registerClassTransformation(Points points) {\n String pointsClassName = points.getClass().getCanonicalName();\n IClassMatcher classMatcher = points.getClassMatcher();\n boolean hasDynamicField = points.isAddDynamicField();\n Junction innerClassMatcher = ClassMatcherConvert.INSTANCE.convert(classMatcher);\n ElementMatcher loaderMatcher = ClassLoaderMatcherConvert.INSTANCE\n .convert(points.getClassLoaderMatcher());\n\n Set methodMatchers = points.getMethodMatcher();\n\n Set mInfo = methodMatchers.stream().map(matcher -> {\n Junction bMethodMatcher = MethodMatcherConvert.INSTANCE.convert(matcher);\n String qualifier = getMethodQualifier(pointsClassName, matcher.getQualifier());\n Integer index = QUALIFIER_TO_INDEX.get(qualifier);\n if (index == null) {\n // it is unusual for this is a pointcut without interceptor.\n // maybe there is some error in plugin providers configuration\n return null;\n }\n Builder providerBuilder = INTERCEPTOR_PROVIDERS.get(index);\n if (providerBuilder == null) {\n return null;\n }\n MethodTransformation mt = new MethodTransformation(index, bMethodMatcher, providerBuilder);\n if (INDEX_TO_METHOD_TRANSFORMATION.putIfAbsent(index, mt) != null) {\n log.error(\"There are duplicate qualifier in Points:{}!\", qualifier);\n }\n return mt;\n }).filter(Objects::nonNull).collect(Collectors.toSet());\n\n AgentPlugin plugin = POINTS_TO_PLUGIN.get(pointsClassName);\n int order = plugin.order();\n return ClassTransformation.builder().classMatcher(innerClassMatcher)\n .hasDynamicField(hasDynamicField)\n .methodTransformations(mInfo)\n .classloaderMatcher(loaderMatcher)\n .typeFieldAccessor(points.getTypeFieldAccessor())\n .order(order).build();\n }\n\n public static int register(InterceptorProvider provider) {\n String qualifier = provider.getAdviceTo();\n // map interceptor/pointcut to plugin\n\n AgentPlugin plugin = PLUGIN_CLASSNAME_TO_PLUGIN.get(provider.getPluginClassName());\n if (plugin == null) {\n // code autogenerate issues that are unlikely to occur!\n throw new RuntimeException();\n }\n QUALIFIER_TO_PLUGIN.putIfAbsent(qualifier, plugin);\n POINTS_TO_PLUGIN.putIfAbsent(getPointsClassName(qualifier), plugin);\n\n // generate index and supplier chain\n Integer index = QUALIFIER_TO_INDEX.get(provider.getAdviceTo());\n if (index == null) {\n synchronized (QUALIFIER_TO_INDEX) {\n index = QUALIFIER_TO_INDEX.get(provider.getAdviceTo());\n if (index == null) {\n index = INTERCEPTOR_PROVIDERS.add(ProviderChain.builder());\n QUALIFIER_TO_INDEX.putIfAbsent(provider.getAdviceTo(), index);\n }\n }\n }\n INTERCEPTOR_PROVIDERS.get(index)\n .addProvider(new ProviderPluginDecorator(plugin, provider));\n\n return index;\n }\n\n public static String getPointsClassName(String name) {\n int index;\n if (Strings.isNullOrEmpty(name)) {\n return \"unknown\";\n }\n index = name.indexOf(':');\n if (index < 0) {\n return name;\n }\n return name.substring(0, index);\n }\n\n public static MethodTransformation getMethodTransformation(int pointcutIndex) {\n return INDEX_TO_METHOD_TRANSFORMATION.get(pointcutIndex);\n }\n\n public static void addMethodTransformation(int pointcutIndex, MethodTransformation info) {\n INDEX_TO_METHOD_TRANSFORMATION.putIfAbsent(pointcutIndex, info);\n }\n\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/transformer/AnnotationTransformer.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.transformer;\n\nimport com.megaease.easeagent.core.plugin.annotation.EaseAgentInstrumented;\nimport com.megaease.easeagent.core.plugin.matcher.MethodTransformation;\nimport net.bytebuddy.agent.builder.AgentBuilder;\nimport net.bytebuddy.asm.AsmVisitorWrapper;\nimport net.bytebuddy.asm.MemberAttributeExtension;\nimport net.bytebuddy.description.annotation.AnnotationDescription;\nimport net.bytebuddy.description.method.MethodDescription;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.dynamic.DynamicType;\nimport net.bytebuddy.implementation.Implementation;\nimport net.bytebuddy.jar.asm.MethodVisitor;\nimport net.bytebuddy.matcher.ElementMatcher;\nimport net.bytebuddy.pool.TypePool;\nimport net.bytebuddy.utility.JavaModule;\n\npublic class AnnotationTransformer implements AgentBuilder.Transformer {\n private final AsmVisitorWrapper visitor;\n private final MethodTransformation methodTransformInfo;\n private final AnnotationDescription annotation;\n\n public AnnotationTransformer(MethodTransformation info) {\n this.methodTransformInfo = info;\n this.annotation = AnnotationDescription.Builder\n .ofType(EaseAgentInstrumented.class)\n .define(\"value\", info.getIndex())\n .build();\n MemberAttributeExtension.ForMethod mForMethod = new MemberAttributeExtension.ForMethod()\n .annotateMethod(this.annotation);\n this.visitor = new ForMethodDelegate(mForMethod, annotation, methodTransformInfo)\n .on(methodTransformInfo.getMatcher());\n }\n\n @Override\n public DynamicType.Builder transform(DynamicType.Builder builder, TypeDescription typeDescription, ClassLoader classLoader, JavaModule module) {\n return builder.visit(this.visitor);\n }\n\n public static class ForMethodDelegate implements AsmVisitorWrapper.ForDeclaredMethods.MethodVisitorWrapper {\n private final TypeDescription annotation;\n private final MemberAttributeExtension.ForMethod mForMethod;\n private final MethodTransformation info;\n\n ForMethodDelegate(MemberAttributeExtension.ForMethod mForMethod,\n AnnotationDescription annotation,\n MethodTransformation info) {\n this.annotation = annotation.getAnnotationType();\n this.mForMethod = mForMethod;\n this.info = info;\n }\n\n @Override\n public MethodVisitor wrap(TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n TypePool typePool,\n int writerFlags,\n int readerFlags) {\n AnnotationDescription annotation = instrumentedMethod.getDeclaredAnnotations().ofType(this.annotation);\n if (annotation != null) {\n // merge interceptor chain\n Integer index = annotation.getValue(\"value\").resolve(Integer.class);\n }\n return methodVisitor;\n }\n\n public AsmVisitorWrapper on(ElementMatcher matcher) {\n return new AsmVisitorWrapper.ForDeclaredMethods().invokable(matcher, this);\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/transformer/CompoundPluginTransformer.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.transformer;\n\nimport net.bytebuddy.agent.builder.AgentBuilder;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.dynamic.DynamicType;\nimport net.bytebuddy.utility.JavaModule;\n\nimport java.util.ArrayList;\nimport java.util.List;\n\npublic class CompoundPluginTransformer implements AgentBuilder.Transformer {\n private final List transformers;\n\n public CompoundPluginTransformer(List transformers) {\n this.transformers = new ArrayList<>();\n for (AgentBuilder.Transformer transformer : transformers) {\n if (transformer instanceof CompoundPluginTransformer) {\n this.transformers.addAll(((CompoundPluginTransformer) transformer).transformers);\n continue;\n }\n this.transformers.add(transformer);\n }\n }\n\n @Override\n public DynamicType.Builder transform(DynamicType.Builder builder,\n TypeDescription typeDescription,\n ClassLoader classLoader,\n JavaModule module) {\n for (AgentBuilder.Transformer transformer : this.transformers) {\n builder = transformer.transform(builder, typeDescription, classLoader, module);\n }\n\n return builder;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/transformer/DynamicFieldAdvice.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.transformer;\n\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.field.DynamicFieldAccessor;\nimport com.megaease.easeagent.plugin.field.NullObject;\nimport net.bytebuddy.asm.Advice;\n\npublic class DynamicFieldAdvice {\n\n private DynamicFieldAdvice() {\n }\n\n public static final Logger log = LoggerFactory.getLogger(DynamicFieldAdvice.class);\n\n public static class DynamicInstanceInit {\n\n private DynamicInstanceInit() {\n }\n\n @Advice.OnMethodExit\n public static void exit(@Advice.This(optional = true) Object target) {\n if (target instanceof DynamicFieldAccessor) {\n DynamicFieldAccessor accessor = (DynamicFieldAccessor) target;\n if (accessor.getEaseAgent$$DynamicField$$Data() == null) {\n accessor.setEaseAgent$$DynamicField$$Data(NullObject.NULL);\n }\n }\n }\n }\n\n public static class DynamicClassInit {\n\n private DynamicClassInit() {\n }\n\n @Advice.OnMethodExit\n public static void exit(@Advice.Origin(\"#m\") String method) {\n }\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/transformer/DynamicFieldTransformer.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.transformer;\n\nimport com.google.common.cache.Cache;\nimport com.google.common.cache.CacheBuilder;\nimport com.megaease.easeagent.core.plugin.transformer.DynamicFieldAdvice.DynamicInstanceInit;\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.field.DynamicFieldAccessor;\nimport net.bytebuddy.agent.builder.AgentBuilder;\nimport net.bytebuddy.asm.Advice;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.dynamic.DynamicType;\nimport net.bytebuddy.implementation.FieldAccessor;\nimport net.bytebuddy.jar.asm.Opcodes;\nimport net.bytebuddy.matcher.ElementMatchers;\nimport net.bytebuddy.utility.JavaModule;\n\nimport java.util.concurrent.ConcurrentHashMap;\n\npublic class DynamicFieldTransformer implements AgentBuilder.Transformer {\n private static final Logger log = LoggerFactory.getLogger(DynamicFieldTransformer.class);\n private static final ConcurrentHashMap> FIELD_MAP = new ConcurrentHashMap<>();\n\n private final String fieldName;\n private final Class accessor;\n private final AgentBuilder.Transformer.ForAdvice transformer;\n\n public DynamicFieldTransformer(String fieldName) {\n this(fieldName, DynamicFieldAccessor.class);\n }\n\n public DynamicFieldTransformer(String fieldName, Class accessor) {\n this.fieldName = fieldName;\n this.accessor = accessor;\n this.transformer = new AgentBuilder.Transformer\n .ForAdvice(Advice.withCustomMapping())\n .include(getClass().getClassLoader())\n .advice(ElementMatchers.isConstructor(), DynamicInstanceInit.class.getName());\n }\n\n @Override\n public DynamicType.Builder transform(DynamicType.Builder b,\n TypeDescription td, ClassLoader cl, JavaModule m) {\n if (check(td, this.accessor, cl) && this.fieldName != null) {\n try {\n b = b.defineField(this.fieldName, Object.class, Opcodes.ACC_PRIVATE)\n .implement(this.accessor)\n .intercept(FieldAccessor.ofField(this.fieldName));\n } catch (Exception e) {\n log.debug(\"Type:{} add extend field again!\", td.getName());\n }\n return transformer.transform(b, td, cl, m);\n }\n return b;\n }\n\n /**\n * Avoiding add a accessor interface to a class repeatedly\n *\n * @param td represent the class to be enhanced\n * @param accessor access interface class\n * @param cl current classloader\n * @return return true when it is the first time\n */\n private static boolean check(TypeDescription td, Class accessor, ClassLoader cl) {\n String key = td.getCanonicalName() + accessor.getCanonicalName();\n\n Cache checkCache = FIELD_MAP.get(key);\n if (checkCache == null) {\n Cache cache = CacheBuilder.newBuilder().weakKeys().build();\n if (cl == null) {\n cl = Thread.currentThread().getContextClassLoader();\n }\n cache.put(cl, true);\n checkCache = FIELD_MAP.putIfAbsent(key, cache);\n if (checkCache == null) {\n return true;\n }\n }\n\n return checkCache.getIfPresent(cl) == null;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/transformer/ForAdviceTransformer.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.transformer;\n\nimport com.megaease.easeagent.core.plugin.CommonInlineAdvice;\nimport com.megaease.easeagent.core.plugin.annotation.Index;\nimport com.megaease.easeagent.core.plugin.matcher.MethodTransformation;\nimport com.megaease.easeagent.core.plugin.registry.AdviceRegistry;\nimport com.megaease.easeagent.core.plugin.transformer.advice.AgentAdvice;\nimport com.megaease.easeagent.core.plugin.transformer.advice.AgentAdvice.OffsetMapping;\nimport com.megaease.easeagent.core.plugin.transformer.advice.AgentForAdvice;\nimport com.megaease.easeagent.core.plugin.transformer.advice.AgentJavaConstantValue;\nimport com.megaease.easeagent.core.plugin.transformer.advice.MethodIdentityJavaConstant;\nimport com.megaease.easeagent.core.plugin.transformer.classloader.CompoundClassloader;\nimport net.bytebuddy.agent.builder.AgentBuilder;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.dynamic.DynamicType;\nimport net.bytebuddy.implementation.bytecode.StackManipulation;\nimport net.bytebuddy.utility.JavaModule;\n\npublic class ForAdviceTransformer implements AgentBuilder.Transformer {\n\n private final AgentForAdvice transformer;\n private final MethodTransformation methodTransformInfo;\n\n public ForAdviceTransformer(MethodTransformation methodTransformInfo) {\n this.methodTransformInfo = methodTransformInfo;\n\n MethodIdentityJavaConstant value = new MethodIdentityJavaConstant(methodTransformInfo.getIndex());\n StackManipulation stackManipulation = new AgentJavaConstantValue(value, methodTransformInfo.getIndex());\n TypeDescription typeDescription = value.getTypeDescription();\n\n OffsetMapping.Factory factory = new OffsetMapping.ForStackManipulation.Factory<>(Index.class,\n stackManipulation,\n typeDescription.asGenericType());\n\n this.transformer = new AgentForAdvice(AgentAdvice.withCustomMapping()\n .bind(factory))\n .include(getClass().getClassLoader())\n .advice(methodTransformInfo.getMatcher(),\n CommonInlineAdvice.class.getCanonicalName());\n }\n\n @Override\n public DynamicType.Builder transform(DynamicType.Builder b, TypeDescription td, ClassLoader cl, JavaModule m) {\n CompoundClassloader.compound(this.getClass().getClassLoader(), cl);\n\n AdviceRegistry.setCurrentClassLoader(cl);\n DynamicType.Builder bd = transformer.transform(b, td, cl, m);\n AdviceRegistry.cleanCurrentClassLoader();\n\n return bd;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/transformer/TypeFieldTransformer.java", "content": "package com.megaease.easeagent.core.plugin.transformer;\n\nimport com.google.common.cache.Cache;\nimport com.google.common.cache.CacheBuilder;\nimport com.megaease.easeagent.log4j2.Logger;\nimport com.megaease.easeagent.log4j2.LoggerFactory;\nimport com.megaease.easeagent.plugin.field.TypeFieldGetter;\nimport net.bytebuddy.agent.builder.AgentBuilder;\nimport net.bytebuddy.asm.Advice;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.dynamic.DynamicType;\nimport net.bytebuddy.implementation.FieldAccessor;\nimport net.bytebuddy.utility.JavaModule;\n\nimport java.util.concurrent.ConcurrentHashMap;\n\npublic class TypeFieldTransformer implements AgentBuilder.Transformer {\n private static final Logger log = LoggerFactory.getLogger(TypeFieldTransformer.class);\n\n private static final ConcurrentHashMap> FIELD_MAP = new ConcurrentHashMap<>();\n private final String fieldName;\n private final Class accessor;\n private final AgentBuilder.Transformer.ForAdvice transformer;\n public TypeFieldTransformer(String fieldName) {\n this.fieldName = fieldName;\n this.accessor = TypeFieldGetter.class;\n this.transformer = new AgentBuilder.Transformer\n .ForAdvice(Advice.withCustomMapping())\n .include(getClass().getClassLoader());\n }\n\n @Override\n public DynamicType.Builder transform(DynamicType.Builder b,\n TypeDescription td, ClassLoader cl, JavaModule m) {\n if (check(td, this.accessor, cl) && this.fieldName != null) {\n try {\n b = b.implement(this.accessor)\n .intercept(FieldAccessor.ofField(this.fieldName));\n } catch (Exception e) {\n log.debug(\"Type:{} add extend field again!\", td.getName());\n }\n return transformer.transform(b, td, cl, m);\n }\n return b;\n }\n\n /**\n * Avoiding add a accessor interface to a class repeatedly\n *\n * @param td represent the class to be enhanced\n * @param accessor access interface class\n * @param cl current classloader\n * @return return true when it is the first time\n */\n private static boolean check(TypeDescription td, Class accessor, ClassLoader cl) {\n String key = td.getCanonicalName() + accessor.getCanonicalName();\n\n Cache checkCache = FIELD_MAP.get(key);\n if (checkCache == null) {\n Cache cache = CacheBuilder.newBuilder().weakKeys().build();\n if (cl == null) {\n cl = Thread.currentThread().getContextClassLoader();\n }\n cache.put(cl, true);\n checkCache = FIELD_MAP.putIfAbsent(key, cache);\n if (checkCache == null) {\n return true;\n }\n }\n\n return checkCache.getIfPresent(cl) == null;\n }\n}\n" }, { "path": "core/src/main/java/com/megaease/easeagent/core/plugin/transformer/advice/AgentAdvice.java", "content": "/*\n * Copyright (c) 2021, MegaEase\n * All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\npackage com.megaease.easeagent.core.plugin.transformer.advice;\n\nimport com.megaease.easeagent.core.plugin.registry.AdviceRegistry;\nimport net.bytebuddy.ClassFileVersion;\nimport net.bytebuddy.asm.Advice;\nimport net.bytebuddy.asm.AsmVisitorWrapper;\nimport net.bytebuddy.build.HashCodeAndEqualsPlugin;\nimport net.bytebuddy.description.annotation.AnnotationDescription;\nimport net.bytebuddy.description.enumeration.EnumerationDescription;\nimport net.bytebuddy.description.field.FieldDescription;\nimport net.bytebuddy.description.method.MethodDescription;\nimport net.bytebuddy.description.method.MethodList;\nimport net.bytebuddy.description.method.ParameterDescription;\nimport net.bytebuddy.description.method.ParameterList;\nimport net.bytebuddy.description.type.TypeDefinition;\nimport net.bytebuddy.description.type.TypeDescription;\nimport net.bytebuddy.dynamic.ClassFileLocator;\nimport net.bytebuddy.dynamic.TargetType;\nimport net.bytebuddy.dynamic.scaffold.FieldLocator;\nimport net.bytebuddy.implementation.FieldAccessor;\nimport net.bytebuddy.implementation.Implementation;\nimport net.bytebuddy.implementation.SuperMethodCall;\nimport net.bytebuddy.implementation.bytecode.*;\nimport net.bytebuddy.implementation.bytecode.assign.Assigner;\nimport net.bytebuddy.implementation.bytecode.collection.ArrayAccess;\nimport net.bytebuddy.implementation.bytecode.collection.ArrayFactory;\nimport net.bytebuddy.implementation.bytecode.constant.*;\nimport net.bytebuddy.implementation.bytecode.member.FieldAccess;\nimport net.bytebuddy.implementation.bytecode.member.MethodInvocation;\nimport net.bytebuddy.implementation.bytecode.member.MethodVariableAccess;\nimport net.bytebuddy.jar.asm.*;\nimport net.bytebuddy.utility.CompoundList;\nimport net.bytebuddy.utility.JavaConstant;\nimport net.bytebuddy.utility.JavaType;\nimport net.bytebuddy.utility.OpenedClassReader;\nimport net.bytebuddy.utility.visitor.ExceptionTableSensitiveMethodVisitor;\nimport net.bytebuddy.utility.visitor.FramePaddingMethodVisitor;\nimport net.bytebuddy.utility.visitor.LineNumberPrependingMethodVisitor;\nimport net.bytebuddy.utility.visitor.StackAwareMethodVisitor;\n\nimport java.io.IOException;\nimport java.io.Serializable;\nimport java.lang.annotation.*;\nimport java.lang.reflect.Constructor;\nimport java.lang.reflect.Method;\nimport java.util.*;\n\nimport static net.bytebuddy.matcher.ElementMatchers.*;\n\n@SuppressWarnings(\"unused, rawtypes, unchecked\")\npublic class AgentAdvice extends Advice {\n private static final ClassReader UNDEFINED = null;\n\n /**\n * A reference to the {@link OnMethodEnter#skipOn()} method.\n */\n private static final MethodDescription.InDefinedShape SKIP_ON;\n\n /**\n * A reference to the {@link OnMethodEnter#prependLineNumber()} method.\n */\n private static final MethodDescription.InDefinedShape PREPEND_LINE_NUMBER;\n\n /**\n * A reference to the {@link OnMethodEnter#inline()} method.\n */\n private static final MethodDescription.InDefinedShape INLINE_ENTER;\n\n /**\n * A reference to the {@link OnMethodEnter#suppress()} method.\n */\n private static final MethodDescription.InDefinedShape SUPPRESS_ENTER;\n\n /**\n * A reference to the {@link OnMethodExit#repeatOn()} method.\n */\n private static final MethodDescription.InDefinedShape REPEAT_ON;\n\n /**\n * A reference to the {@link OnMethodExit#onThrowable()} method.\n */\n private static final MethodDescription.InDefinedShape ON_THROWABLE;\n\n /**\n * A reference to the {@link OnMethodExit#backupArguments()} method.\n */\n private static final MethodDescription.InDefinedShape BACKUP_ARGUMENTS;\n\n /**\n * A reference to the {@link OnMethodExit#inline()} method.\n */\n private static final MethodDescription.InDefinedShape INLINE_EXIT;\n\n /**\n * A reference to the {@link OnMethodExit#suppress()} method.\n */\n private static final MethodDescription.InDefinedShape SUPPRESS_EXIT;\n\n static {\n MethodList enter = TypeDescription.ForLoadedType.of(OnMethodEnter.class).getDeclaredMethods();\n SKIP_ON = enter.filter(named(\"skipOn\")).getOnly();\n PREPEND_LINE_NUMBER = enter.filter(named(\"prependLineNumber\")).getOnly();\n INLINE_ENTER = enter.filter(named(\"inline\")).getOnly();\n SUPPRESS_ENTER = enter.filter(named(\"suppress\")).getOnly();\n MethodList exit = TypeDescription.ForLoadedType.of(OnMethodExit.class).getDeclaredMethods();\n REPEAT_ON = exit.filter(named(\"repeatOn\")).getOnly();\n ON_THROWABLE = exit.filter(named(\"onThrowable\")).getOnly();\n BACKUP_ARGUMENTS = exit.filter(named(\"backupArguments\")).getOnly();\n INLINE_EXIT = exit.filter(named(\"inline\")).getOnly();\n SUPPRESS_EXIT = exit.filter(named(\"suppress\")).getOnly();\n }\n\n\n /**\n * The dispatcher for instrumenting the instrumented method upon entering.\n */\n private final Dispatcher.Resolved.ForMethodEnter methodEnter;\n\n /**\n * The dispatcher for instrumenting the instrumented method upon exiting.\n */\n private final Dispatcher.Resolved.ForMethodExit methodExit;\n\n private final Dispatcher.Resolved.ForMethodExit methodExitNonThrowable;\n\n /**\n * The assigner to use.\n */\n private final Assigner assigner;\n\n /**\n * The exception handler to apply.\n */\n private final ExceptionHandler exceptionHandler;\n\n /**\n * The delegate implementation to apply if this advice is used as an instrumentation.\n */\n private final Implementation delegate;\n\n /**\n * Creates a new advice.\n *\n * @param methodEnter The dispatcher for instrumenting the instrumented method upon entering.\n * @param methodExit The dispatcher for instrumenting the instrumented method upon exiting.\n */\n protected AgentAdvice(Dispatcher.Resolved.ForMethodEnter methodEnter,\n // Dispatcher.Resolved.ForMethodExit methodExitNonThrowable,\n Dispatcher.Resolved.ForMethodExit methodExit) {\n this(methodEnter, methodExit,\n null,\n Assigner.DEFAULT, ExceptionHandler.Default.SUPPRESSING, SuperMethodCall.INSTANCE);\n }\n\n protected AgentAdvice(Dispatcher.Resolved.ForMethodEnter methodEnter,\n Dispatcher.Resolved.ForMethodExit methodExitNonThrowable,\n Dispatcher.Resolved.ForMethodExit methodExit) {\n this(methodEnter, methodExit,\n methodExitNonThrowable,\n Assigner.DEFAULT,\n ExceptionHandler.Default.SUPPRESSING,\n SuperMethodCall.INSTANCE);\n }\n\n /**\n * Creates a new advice.\n *\n * @param methodEnter The dispatcher for instrumenting the instrumented method upon entering.\n * @param methodExit The dispatcher for instrumenting the instrumented method upon exiting.\n * @param assigner The assigner to use.\n * @param exceptionHandler The exception handler to apply.\n * @param delegate The delegate implementation to apply if this advice is used as an instrumentation.\n */\n private AgentAdvice(Dispatcher.Resolved.ForMethodEnter methodEnter,\n Dispatcher.Resolved.ForMethodExit methodExit,\n Dispatcher.Resolved.ForMethodExit methodExitNonThrowable,\n Assigner assigner,\n ExceptionHandler exceptionHandler,\n Implementation delegate) {\n super(null, null);\n this.methodEnter = methodEnter;\n this.methodExit = methodExit;\n this.methodExitNonThrowable = methodExitNonThrowable;\n this.assigner = assigner;\n this.exceptionHandler = exceptionHandler;\n this.delegate = delegate;\n }\n\n private static boolean isNoExceptionHandler(TypeDescription t) {\n return t.getName().endsWith(\"NoExceptionHandler\");\n // return t.represents(NoExceptionHandler.class) || t.represents()\n }\n\n /**\n * Creates a new advice.\n *\n * @param advice A description of the type declaring the advice.\n * @param postProcessorFactory The post processor factory to use.\n * @param classFileLocator The class file locator for locating the advisory class's class file.\n * @param userFactories A list of custom factories for user generated offset mappings.\n * @param delegator The delegator to use.\n * @return A method visitor wrapper representing the supplied advice.\n */\n protected static AgentAdvice tto(TypeDescription advice,\n PostProcessor.Factory postProcessorFactory,\n ClassFileLocator classFileLocator,\n List> userFactories,\n Delegator delegator) {\n Dispatcher.Unresolved methodEnter = Dispatcher.Inactive.INSTANCE,\n methodExit = Dispatcher.Inactive.INSTANCE,\n methodExitNoException = Dispatcher.Inactive.INSTANCE;\n for (MethodDescription.InDefinedShape methodDescription : advice.getDeclaredMethods()) {\n methodEnter = locate(OnMethodEnter.class, INLINE_ENTER, methodEnter, methodDescription, delegator);\n AnnotationDescription.Loadable al = methodDescription.getDeclaredAnnotations().ofType(OnMethodExit.class);\n if (al != null) {\n TypeDescription throwable = al.getValue(ON_THROWABLE).resolve(TypeDescription.class);\n if (isNoExceptionHandler(throwable)) {\n methodExitNoException = locate(OnMethodExit.class, INLINE_EXIT, methodExitNoException, methodDescription, delegator);\n } else {\n methodExit = locate(OnMethodExit.class, INLINE_EXIT, methodExit, methodDescription, delegator);\n }\n }\n }\n if (methodExit == Dispatcher.Inactive.INSTANCE) {\n methodEnter = methodExitNoException;\n }\n if (!methodEnter.isAlive() && !methodExit.isAlive() && !methodExitNoException.isAlive()) {\n throw new IllegalArgumentException(\"No advice defined by \" + advice);\n }\n try {\n ClassReader classReader = methodEnter.isBinary() || methodExit.isBinary()\n ? OpenedClassReader.of(classFileLocator.locate(advice.getName()).resolve())\n : UNDEFINED;\n return new AgentAdvice(methodEnter.asMethodEnter(userFactories, classReader, methodExit, postProcessorFactory),\n methodExitNoException.asMethodExit(userFactories, classReader, methodEnter, postProcessorFactory),\n methodExit.asMethodExit(userFactories, classReader, methodEnter, postProcessorFactory));\n } catch (IOException exception) {\n throw new IllegalStateException(\"Error reading class file of \" + advice, exception);\n }\n }\n\n public static AgentAdvice.WithCustomMapping withCustomMapping() {\n return new WithCustomMapping();\n }\n\n private static Dispatcher.Unresolved locate(Class type,\n MethodDescription.InDefinedShape property,\n Dispatcher.Unresolved dispatcher,\n MethodDescription.InDefinedShape methodDescription,\n Delegator delegator) {\n AnnotationDescription annotation = methodDescription.getDeclaredAnnotations().ofType(type);\n if (annotation == null) {\n return dispatcher;\n } else if (dispatcher.isAlive()) {\n throw new IllegalStateException(\"Duplicate advice for \" + dispatcher + \" and \" + methodDescription);\n } else if (!methodDescription.isStatic()) {\n throw new IllegalStateException(\"Advice for \" + methodDescription + \" is not static\");\n } else {\n return new Dispatcher.Inlining(methodDescription);\n }\n }\n\n /**\n * Configures this advice to use the specified assigner. Any previous or default assigner is replaced.\n *\n * @param assigner The assigner to use,\n * @return A version of this advice that uses the specified assigner.\n */\n @Override\n public AgentAdvice withAssigner(Assigner assigner) {\n return new AgentAdvice(methodEnter, methodExitNonThrowable,\n methodExit, assigner, exceptionHandler, delegate);\n }\n\n /**\n * Configures this advice to execute the given exception handler upon a suppressed exception. The stack manipulation is executed with a\n * {@link Throwable} instance on the operand stack. The stack must be empty upon completing the exception handler.\n *\n * @param exceptionHandler The exception handler to apply.\n * @return A version of this advice that applies the supplied exception handler.\n */\n @Override\n public AgentAdvice withExceptionHandler(ExceptionHandler exceptionHandler) {\n return new AgentAdvice(methodEnter, methodExitNonThrowable,\n methodExit, assigner, exceptionHandler, delegate);\n }\n\n /**\n * A builder step for creating an {@link Advice} that uses custom mappings of annotations to constant pool values.\n */\n @HashCodeAndEqualsPlugin.Enhance\n public static class WithCustomMapping extends Advice.WithCustomMapping {\n /**\n * The post processor factory to apply.\n */\n private final PostProcessor.Factory postProcessorFactory;\n\n /**\n * The delegator to use.\n */\n private final Delegator delegator;\n\n /**\n * A map containing dynamically computed constant pool values that are mapped by their triggering annotation type.\n */\n private final Map, OffsetMapping.Factory> offsetMappings;\n\n /**\n * Creates a new custom mapping builder step without including any custom mappings.\n */\n public WithCustomMapping() {\n this(PostProcessor.NoOp.INSTANCE, Collections.emptyMap(), Delegator.ForStaticInvocation.INSTANCE);\n }\n\n\n /**\n * Creates a new custom mapping builder step with the given custom mappings.\n *\n * @param postProcessorFactory The post processor factory to apply.\n * @param offsetMappings A map containing dynamically computed constant pool values that are mapped by their triggering annotation type.\n * @param delegator The delegator to use.\n */\n protected WithCustomMapping(PostProcessor.Factory postProcessorFactory,\n Map, OffsetMapping.Factory> offsetMappings,\n Delegator delegator) {\n this.postProcessorFactory = postProcessorFactory;\n this.offsetMappings = offsetMappings;\n this.delegator = delegator;\n }\n\n /**\n * Binds the supplied annotation to a type constant of the supplied value. Constants can be strings, method handles, method types\n * and any primitive or the value {@code null}.\n *\n * @param type The type of the annotation being bound.\n * @param value The value to bind to the annotation.\n * @param The annotation type.\n * @return A new builder for an advice that considers the supplied annotation type during binding.\n */\n @Override\n public WithCustomMapping bind(Class type, Object value) {\n return bind(OffsetMapping.ForStackManipulation.Factory.of(type, value));\n }\n\n /**\n * Binds an annotation to a dynamically computed value. Whenever the {@link Advice} component discovers the given annotation on\n * a parameter of an advice method, the dynamic value is asked to provide a value that is then assigned to the parameter in question.\n *\n * @param offsetMapping The dynamic value that is computed for binding the parameter to a value.\n * @return A new builder for an advice that considers the supplied annotation type during binding.\n */\n public WithCustomMapping bind(OffsetMapping.Factory offsetMapping) {\n Map, OffsetMapping.Factory> offsetMappings = new HashMap<>(this.offsetMappings);\n if (!offsetMapping.getAnnotationType().isAnnotation()) {\n throw new IllegalArgumentException(\"Not an annotation type: \" + offsetMapping.getAnnotationType());\n } else if (offsetMappings.put(offsetMapping.getAnnotationType(), offsetMapping) != null) {\n throw new IllegalArgumentException(\"Annotation type already mapped: \" + offsetMapping.getAnnotationType());\n }\n return new WithCustomMapping(postProcessorFactory, offsetMappings, delegator);\n }\n\n\n /**\n * Implements advice where every matched method is advised by the given type's advisory methods.\n *\n * @param advice A description of the type declaring the advice.\n * @param classFileLocator The class file locator for locating the advisory class's class file.\n * @return A method visitor wrapper representing the supplied advice.\n */\n @Override\n public AgentAdvice to(TypeDescription advice, ClassFileLocator classFileLocator) {\n return AgentAdvice.tto(advice, postProcessorFactory, classFileLocator,\n new ArrayList<>(offsetMappings.values()), delegator);\n }\n }\n\n /**\n * Wraps the method visitor to implement this advice.\n *\n * @param instrumentedType The instrumented type.\n * @param instrumentedMethod The instrumented method.\n * @param methodVisitor The method visitor to write to.\n * @param implementationContext The implementation context to use.\n * @param writerFlags The ASM writer flags to use.\n * @param readerFlags The, plies this advice.\n */\n @Override\n protected MethodVisitor doWrap(TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n int writerFlags,\n int readerFlags) {\n Dispatcher.Resolved.ForMethodExit exit;\n if (instrumentedMethod.isConstructor()) {\n exit = methodExitNonThrowable;\n } else {\n exit = methodExit;\n }\n\n if (AdviceRegistry.check(instrumentedType, instrumentedMethod, methodEnter, exit) == 0) {\n return methodVisitor;\n }\n\n methodVisitor = new FramePaddingMethodVisitor(methodEnter.isPrependLineNumber()\n ? new LineNumberPrependingMethodVisitor(methodVisitor)\n : methodVisitor);\n\n if (instrumentedMethod.isConstructor()) {\n return new WithExitAdvice.WithoutExceptionHandling(methodVisitor,\n implementationContext,\n assigner,\n exceptionHandler.resolve(instrumentedMethod, instrumentedType),\n instrumentedType,\n instrumentedMethod,\n methodEnter,\n methodExitNonThrowable,\n writerFlags,\n readerFlags);\n } else {\n return new WithExitAdvice.WithExceptionHandling(methodVisitor,\n implementationContext,\n assigner,\n exceptionHandler.resolve(instrumentedMethod, instrumentedType),\n instrumentedType,\n instrumentedMethod,\n methodEnter,\n methodExit,\n writerFlags,\n readerFlags,\n methodExit.getThrowable());\n }\n }\n\n /**\n * A method visitor that weaves the advice methods' byte codes.\n */\n protected abstract static class AdviceVisitor\n extends ExceptionTableSensitiveMethodVisitor implements Dispatcher.RelocationHandler.Relocation {\n\n /**\n * The expected index for the {@code this} variable.\n */\n private static final int THIS_VARIABLE_INDEX = 0;\n\n /**\n * The expected name for the {@code this} variable.\n */\n private static final String THIS_VARIABLE_NAME = \"this\";\n\n /**\n * A description of the instrumented method.\n */\n protected final MethodDescription instrumentedMethod;\n\n /**\n * A label that indicates the start of the preparation of a user method execution.\n */\n private final Label preparationStart;\n\n /**\n * The dispatcher to be used for method enter.\n */\n private final Dispatcher.Bound methodEnter;\n\n /**\n * The dispatcher to be used for method exit.\n */\n protected final Dispatcher.Bound methodExit;\n\n /**\n * The handler for accessing arguments of the method's local variable array.\n */\n protected final ArgumentHandler.ForInstrumentedMethod argumentHandler;\n\n /**\n * A handler for computing the method size requirements.\n */\n protected final MethodSizeHandler.ForInstrumentedMethod methodSizeHandler;\n\n /**\n * A handler for translating and injecting stack map frames.\n */\n protected final StackMapFrameHandler.ForInstrumentedMethod stackMapFrameHandler;\n\n /**\n * Creates a new advice visitor.\n *\n * @param methodVisitor The actual method visitor that is underlying this method visitor to which all instructions are written.\n * @param implementationContext The implementation context to use.\n * @param assigner The assigner to use.\n * @param exceptionHandler The stack manipulation to apply within a suppression handler.\n * @param instrumentedType A description of the instrumented type.\n * @param instrumentedMethod The instrumented method.\n * @param methodEnter The method enter advice.\n * @param methodExit The method exit advice.\n * @param postMethodTypes A list of virtual method arguments that are available after the instrumented method has completed.\n * @param writerFlags The ASM writer flags that were set.\n * @param readerFlags The ASM reader flags that were set.\n */\n protected AdviceVisitor(MethodVisitor methodVisitor,\n Context implementationContext,\n Assigner assigner,\n StackManipulation exceptionHandler,\n TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n Dispatcher.Resolved.ForMethodEnter methodEnter,\n Dispatcher.Resolved.ForMethodExit methodExit,\n List postMethodTypes,\n int writerFlags,\n int readerFlags) {\n super(OpenedClassReader.ASM_API, methodVisitor);\n this.instrumentedMethod = instrumentedMethod;\n preparationStart = new Label();\n SortedMap namedTypes = new TreeMap<>();\n namedTypes.putAll(methodEnter.getNamedTypes());\n namedTypes.putAll(methodExit.getNamedTypes());\n argumentHandler = methodExit.getArgumentHandlerFactory().resolve(instrumentedMethod,\n methodEnter.getAdviceType(),\n methodExit.getAdviceType(),\n namedTypes);\n List initialTypes = CompoundList.of(methodExit.getAdviceType().represents(void.class)\n ? Collections.emptyList()\n : Collections.singletonList(methodExit.getAdviceType().asErasure()), argumentHandler.getNamedTypes());\n List latentTypes = methodEnter.getActualAdviceType().represents(void.class)\n ? Collections.emptyList()\n : Collections.singletonList(methodEnter.getActualAdviceType().asErasure());\n List preMethodTypes = methodEnter.getAdviceType().represents(void.class)\n ? Collections.emptyList()\n : Collections.singletonList(methodEnter.getAdviceType().asErasure());\n methodSizeHandler = MethodSizeHandler.Default.of(instrumentedMethod,\n initialTypes,\n preMethodTypes,\n postMethodTypes,\n argumentHandler.isCopyingArguments(),\n writerFlags);\n stackMapFrameHandler = StackMapFrameHandler.Default.of(instrumentedType,\n instrumentedMethod,\n initialTypes,\n latentTypes,\n preMethodTypes,\n postMethodTypes,\n methodExit.isAlive(),\n argumentHandler.isCopyingArguments(),\n implementationContext.getClassFileVersion(),\n writerFlags,\n readerFlags);\n this.methodEnter = methodEnter.bind(instrumentedType,\n instrumentedMethod,\n methodVisitor,\n implementationContext,\n assigner,\n argumentHandler,\n methodSizeHandler,\n stackMapFrameHandler,\n exceptionHandler,\n this);\n this.methodExit = methodExit.bind(instrumentedType,\n instrumentedMethod,\n methodVisitor,\n implementationContext,\n assigner,\n argumentHandler,\n methodSizeHandler,\n stackMapFrameHandler,\n exceptionHandler,\n new ForLabel(preparationStart));\n }\n\n @Override\n protected void onAfterExceptionTable() {\n methodEnter.prepare();\n onUserPrepare();\n methodExit.prepare();\n methodEnter.initialize();\n methodExit.initialize();\n stackMapFrameHandler.injectInitializationFrame(mv);\n methodEnter.apply();\n mv.visitLabel(preparationStart);\n methodSizeHandler.requireStackSize(argumentHandler.prepare(mv));\n stackMapFrameHandler.injectStartFrame(mv);\n onUserStart();\n }\n\n /**\n * Invoked when the user method's exception handler (if any) is supposed to be prepared.\n */\n protected abstract void onUserPrepare();\n\n /**\n * Writes the advice for entering the instrumented method.\n */\n protected abstract void onUserStart();\n\n @Override\n protected void onVisitVarInsn(int opcode, int offset) {\n mv.visitVarInsn(opcode, argumentHandler.argument(offset));\n }\n\n @Override\n protected void onVisitIincInsn(int offset, int increment) {\n mv.visitIincInsn(argumentHandler.argument(offset), increment);\n }\n\n @Override\n public void onVisitFrame(int type, int localVariableLength, Object[] localVariable, int stackSize, Object[] stack) {\n stackMapFrameHandler.translateFrame(mv, type, localVariableLength, localVariable, stackSize, stack);\n }\n\n @Override\n public void visitMaxs(int stackSize, int localVariableLength) {\n onUserEnd();\n mv.visitMaxs(methodSizeHandler.compoundStackSize(stackSize), methodSizeHandler.compoundLocalVariableLength(localVariableLength));\n }\n\n @Override\n public void visitLocalVariable(String name, String descriptor, String signature, Label start, Label end, int index) {\n // The 'this' variable is exempt from remapping as it is assumed immutable and remapping it confuses debuggers to not display the variable.\n mv.visitLocalVariable(name, descriptor, signature, start, end, index == THIS_VARIABLE_INDEX && THIS_VARIABLE_NAME.equals(name)\n ? index\n : argumentHandler.variable(index));\n }\n\n @Override\n public AnnotationVisitor visitLocalVariableAnnotation(int typeReference,\n TypePath typePath,\n Label[] start,\n Label[] end,\n int[] index,\n String descriptor,\n boolean visible) {\n int[] translated = new int[index.length];\n for (int anIndex = 0; anIndex < index.length; anIndex++) {\n translated[anIndex] = argumentHandler.variable(index[anIndex]);\n }\n return mv.visitLocalVariableAnnotation(typeReference, typePath, start, end, translated, descriptor, visible);\n }\n\n /**\n * Writes the advice for completing the instrumented method.\n */\n protected abstract void onUserEnd();\n\n }\n\n /**\n * An advice visitor that does not apply exit advice.\n */\n protected static class WithoutExitAdvice extends AdviceVisitor {\n\n /**\n * Creates an advice visitor that does not apply exit advice.\n *\n * @param methodVisitor The method visitor for the instrumented method.\n * @param implementationContext The implementation context to use.\n * @param assigner The assigner to use.\n * @param exceptionHandler The stack manipulation to apply within a suppression handler.\n * @param instrumentedType A description of the instrumented type.\n * @param instrumentedMethod A description of the instrumented method.\n * @param methodEnter The dispatcher to be used for method enter.\n * @param writerFlags The ASM writer flags that were set.\n * @param readerFlags The ASM reader flags that were set.\n */\n protected WithoutExitAdvice(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n Assigner assigner,\n StackManipulation exceptionHandler,\n TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n Dispatcher.Resolved.ForMethodEnter methodEnter,\n int writerFlags,\n int readerFlags) {\n super(methodVisitor,\n implementationContext,\n assigner,\n exceptionHandler,\n instrumentedType,\n instrumentedMethod,\n methodEnter,\n Dispatcher.Inactive.INSTANCE,\n Collections.emptyList(),\n writerFlags,\n readerFlags);\n }\n\n /**\n * {@inheritDoc}\n */\n public void apply(MethodVisitor methodVisitor) {\n if (instrumentedMethod.getReturnType().represents(boolean.class)\n || instrumentedMethod.getReturnType().represents(byte.class)\n || instrumentedMethod.getReturnType().represents(short.class)\n || instrumentedMethod.getReturnType().represents(char.class)\n || instrumentedMethod.getReturnType().represents(int.class)) {\n methodVisitor.visitInsn(Opcodes.ICONST_0);\n methodVisitor.visitInsn(Opcodes.IRETURN);\n } else if (instrumentedMethod.getReturnType().represents(long.class)) {\n methodVisitor.visitInsn(Opcodes.LCONST_0);\n methodVisitor.visitInsn(Opcodes.LRETURN);\n } else if (instrumentedMethod.getReturnType().represents(float.class)) {\n methodVisitor.visitInsn(Opcodes.FCONST_0);\n methodVisitor.visitInsn(Opcodes.FRETURN);\n } else if (instrumentedMethod.getReturnType().represents(double.class)) {\n methodVisitor.visitInsn(Opcodes.DCONST_0);\n methodVisitor.visitInsn(Opcodes.DRETURN);\n } else if (instrumentedMethod.getReturnType().represents(void.class)) {\n methodVisitor.visitInsn(Opcodes.RETURN);\n } else {\n methodVisitor.visitInsn(Opcodes.ACONST_NULL);\n methodVisitor.visitInsn(Opcodes.ARETURN);\n }\n }\n\n @Override\n protected void onUserPrepare() {\n /* do nothing */\n }\n\n @Override\n protected void onUserStart() {\n /* do nothing */\n }\n\n @Override\n protected void onUserEnd() {\n /* do nothing */\n }\n }\n\n /**\n * An advice visitor that applies exit advice.\n */\n protected abstract static class WithExitAdvice extends AdviceVisitor {\n /**\n * Indicates the handler for the value returned by the advice method.\n */\n protected final Label returnHandler;\n\n /**\n * Creates an advice visitor that applies exit advice.\n *\n * @param methodVisitor The method visitor for the instrumented method.\n * @param implementationContext The implementation context to use.\n * @param assigner The assigner to use.\n * @param exceptionHandler The stack manipulation to apply within a suppression handler.\n * @param instrumentedType A description of the instrumented type.\n * @param instrumentedMethod A description of the instrumented method.\n * @param methodEnter The dispatcher to be used for method enter.\n * @param methodExit The dispatcher to be used for method exit.\n * @param postMethodTypes A list of virtual method arguments that are available after the instrumented method has completed.\n * @param writerFlags The ASM writer flags that were set.\n * @param readerFlags The ASM reader flags that were set.\n */\n protected WithExitAdvice(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n Assigner assigner,\n StackManipulation exceptionHandler,\n TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n Dispatcher.Resolved.ForMethodEnter methodEnter,\n Dispatcher.Resolved.ForMethodExit methodExit,\n List postMethodTypes,\n int writerFlags,\n int readerFlags) {\n super(new StackAwareMethodVisitor(methodVisitor, instrumentedMethod),\n implementationContext,\n assigner,\n exceptionHandler,\n instrumentedType,\n instrumentedMethod,\n methodEnter,\n methodExit,\n postMethodTypes,\n writerFlags,\n readerFlags);\n returnHandler = new Label();\n }\n\n /**\n * {@inheritDoc}\n */\n public void apply(MethodVisitor methodVisitor) {\n if (instrumentedMethod.getReturnType().represents(boolean.class)\n || instrumentedMethod.getReturnType().represents(byte.class)\n || instrumentedMethod.getReturnType().represents(short.class)\n || instrumentedMethod.getReturnType().represents(char.class)\n || instrumentedMethod.getReturnType().represents(int.class)) {\n methodVisitor.visitInsn(Opcodes.ICONST_0);\n } else if (instrumentedMethod.getReturnType().represents(long.class)) {\n methodVisitor.visitInsn(Opcodes.LCONST_0);\n } else if (instrumentedMethod.getReturnType().represents(float.class)) {\n methodVisitor.visitInsn(Opcodes.FCONST_0);\n } else if (instrumentedMethod.getReturnType().represents(double.class)) {\n methodVisitor.visitInsn(Opcodes.DCONST_0);\n } else if (!instrumentedMethod.getReturnType().represents(void.class)) {\n methodVisitor.visitInsn(Opcodes.ACONST_NULL);\n }\n methodVisitor.visitJumpInsn(Opcodes.GOTO, returnHandler);\n }\n\n @Override\n protected void onVisitInsn(int opcode) {\n switch (opcode) {\n case Opcodes.RETURN:\n ((StackAwareMethodVisitor) mv).drainStack();\n break;\n case Opcodes.IRETURN:\n methodSizeHandler.requireLocalVariableLength(((StackAwareMethodVisitor) mv).drainStack(Opcodes.ISTORE, Opcodes.ILOAD, StackSize.SINGLE));\n break;\n case Opcodes.FRETURN:\n methodSizeHandler.requireLocalVariableLength(((StackAwareMethodVisitor) mv).drainStack(Opcodes.FSTORE, Opcodes.FLOAD, StackSize.SINGLE));\n break;\n case Opcodes.DRETURN:\n methodSizeHandler.requireLocalVariableLength(((StackAwareMethodVisitor) mv).drainStack(Opcodes.DSTORE, Opcodes.DLOAD, StackSize.DOUBLE));\n break;\n case Opcodes.LRETURN:\n methodSizeHandler.requireLocalVariableLength((((StackAwareMethodVisitor) mv).drainStack(Opcodes.LSTORE, Opcodes.LLOAD, StackSize.DOUBLE)));\n break;\n case Opcodes.ARETURN:\n methodSizeHandler.requireLocalVariableLength((((StackAwareMethodVisitor) mv).drainStack(Opcodes.ASTORE, Opcodes.ALOAD, StackSize.SINGLE)));\n break;\n default:\n mv.visitInsn(opcode);\n return;\n }\n mv.visitJumpInsn(Opcodes.GOTO, returnHandler);\n }\n\n @Override\n protected void onUserEnd() {\n mv.visitLabel(returnHandler);\n onUserReturn();\n stackMapFrameHandler.injectCompletionFrame(mv);\n methodExit.apply();\n onExitAdviceReturn();\n if (instrumentedMethod.getReturnType().represents(boolean.class)\n || instrumentedMethod.getReturnType().represents(byte.class)\n || instrumentedMethod.getReturnType().represents(short.class)\n || instrumentedMethod.getReturnType().represents(char.class)\n || instrumentedMethod.getReturnType().represents(int.class)) {\n mv.visitVarInsn(Opcodes.ILOAD, argumentHandler.returned());\n mv.visitInsn(Opcodes.IRETURN);\n } else if (instrumentedMethod.getReturnType().represents(long.class)) {\n mv.visitVarInsn(Opcodes.LLOAD, argumentHandler.returned());\n mv.visitInsn(Opcodes.LRETURN);\n } else if (instrumentedMethod.getReturnType().represents(float.class)) {\n mv.visitVarInsn(Opcodes.FLOAD, argumentHandler.returned());\n mv.visitInsn(Opcodes.FRETURN);\n } else if (instrumentedMethod.getReturnType().represents(double.class)) {\n mv.visitVarInsn(Opcodes.DLOAD, argumentHandler.returned());\n mv.visitInsn(Opcodes.DRETURN);\n } else if (!instrumentedMethod.getReturnType().represents(void.class)) {\n mv.visitVarInsn(Opcodes.ALOAD, argumentHandler.returned());\n mv.visitInsn(Opcodes.ARETURN);\n } else {\n mv.visitInsn(Opcodes.RETURN);\n }\n methodSizeHandler.requireStackSize(instrumentedMethod.getReturnType().getStackSize().getSize());\n }\n\n /**\n * Invoked after the user method has returned.\n */\n protected abstract void onUserReturn();\n\n /**\n * Invoked after the exit advice method has returned.\n */\n protected abstract void onExitAdviceReturn();\n\n /**\n * An advice visitor that does not capture exceptions.\n */\n protected static class WithoutExceptionHandling extends WithExitAdvice {\n\n /**\n * Creates a new advice visitor that does not capture exceptions.\n *\n * @param methodVisitor The method visitor for the instrumented method.\n * @param implementationContext The implementation context to use.\n * @param assigner The assigner to use.\n * @param exceptionHandler The stack manipulation to apply within a suppression handler.\n * @param instrumentedType A description of the instrumented type.\n * @param instrumentedMethod A description of the instrumented method.\n * @param methodEnter The dispatcher to be used for method enter.\n * @param methodExit The dispatcher to be used for method exit.\n * @param writerFlags The ASM writer flags that were set.\n * @param readerFlags The ASM reader flags that were set.\n */\n protected WithoutExceptionHandling(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n Assigner assigner,\n StackManipulation exceptionHandler,\n TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n Dispatcher.Resolved.ForMethodEnter methodEnter,\n Dispatcher.Resolved.ForMethodExit methodExit,\n int writerFlags,\n int readerFlags) {\n super(methodVisitor,\n implementationContext,\n assigner,\n exceptionHandler,\n instrumentedType,\n instrumentedMethod,\n methodEnter,\n methodExit,\n instrumentedMethod.getReturnType().represents(void.class)\n ? Collections.emptyList()\n : Collections.singletonList(instrumentedMethod.getReturnType().asErasure()),\n writerFlags,\n readerFlags);\n }\n\n @Override\n protected void onUserPrepare() {\n /* do nothing */\n }\n\n @Override\n protected void onUserStart() {\n /* do nothing */\n }\n\n @Override\n protected void onUserReturn() {\n if (instrumentedMethod.getReturnType().represents(boolean.class)\n || instrumentedMethod.getReturnType().represents(byte.class)\n || instrumentedMethod.getReturnType().represents(short.class)\n || instrumentedMethod.getReturnType().represents(char.class)\n || instrumentedMethod.getReturnType().represents(int.class)) {\n stackMapFrameHandler.injectReturnFrame(mv);\n mv.visitVarInsn(Opcodes.ISTORE, argumentHandler.returned());\n } else if (instrumentedMethod.getReturnType().represents(long.class)) {\n stackMapFrameHandler.injectReturnFrame(mv);\n mv.visitVarInsn(Opcodes.LSTORE, argumentHandler.returned());\n } else if (instrumentedMethod.getReturnType().represents(float.class)) {\n stackMapFrameHandler.injectReturnFrame(mv);\n mv.visitVarInsn(Opcodes.FSTORE, argumentHandler.returned());\n } else if (instrumentedMethod.getReturnType().represents(double.class)) {\n stackMapFrameHandler.injectReturnFrame(mv);\n mv.visitVarInsn(Opcodes.DSTORE, argumentHandler.returned());\n } else if (!instrumentedMethod.getReturnType().represents(void.class)) {\n stackMapFrameHandler.injectReturnFrame(mv);\n mv.visitVarInsn(Opcodes.ASTORE, argumentHandler.returned());\n }\n }\n\n @Override\n protected void onExitAdviceReturn() {\n /* do nothing */\n }\n }\n\n /**\n * An advice visitor that captures exceptions by weaving try-catch blocks around user code.\n */\n protected static class WithExceptionHandling extends WithExitAdvice {\n /**\n * The type of the handled throwable type for which this advice is invoked.\n */\n private final TypeDescription throwable;\n\n /**\n * Indicates the exception handler.\n */\n private final Label exceptionHandler;\n\n /**\n * Indicates the start of the user method.\n */\n protected final Label userStart;\n\n /**\n * Creates a new advice visitor that captures exception by weaving try-catch blocks around user code.\n *\n * @param methodVisitor The method visitor for the instrumented method.\n * @param instrumentedType A description of the instrumented type.\n * @param implementationContext The implementation context to use.\n * @param assigner The assigner to use.\n * @param exceptionHandler The stack manipulation to apply within a suppression handler.\n * @param instrumentedMethod A description of the instrumented method.\n * @param methodEnter The dispatcher to be used for method enter.\n * @param methodExit The dispatcher to be used for method exit.\n * @param writerFlags The ASM writer flags that were set.\n * @param readerFlags The ASM reader flags that were set.\n * @param throwable The type of the handled throwable type for which this advice is invoked.\n */\n protected WithExceptionHandling(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n Assigner assigner,\n StackManipulation exceptionHandler,\n TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n Dispatcher.Resolved.ForMethodEnter methodEnter,\n Dispatcher.Resolved.ForMethodExit methodExit,\n int writerFlags,\n int readerFlags,\n TypeDescription throwable) {\n super(methodVisitor,\n implementationContext,\n assigner,\n exceptionHandler,\n instrumentedType,\n instrumentedMethod,\n methodEnter,\n methodExit,\n instrumentedMethod.getReturnType().represents(void.class)\n ? Collections.singletonList(TypeDescription.THROWABLE)\n : Arrays.asList(instrumentedMethod.getReturnType().asErasure(), TypeDescription.THROWABLE),\n writerFlags,\n readerFlags);\n this.throwable = throwable;\n this.exceptionHandler = new Label();\n userStart = new Label();\n }\n\n @Override\n protected void onUserPrepare() {\n mv.visitTryCatchBlock(userStart, returnHandler, exceptionHandler, throwable.getInternalName());\n }\n\n @Override\n protected void onUserStart() {\n mv.visitLabel(userStart);\n }\n\n @Override\n protected void onUserReturn() {\n stackMapFrameHandler.injectReturnFrame(mv);\n if (instrumentedMethod.getReturnType().represents(boolean.class)\n || instrumentedMethod.getReturnType().represents(byte.class)\n || instrumentedMethod.getReturnType().represents(short.class)\n || instrumentedMethod.getReturnType().represents(char.class)\n || instrumentedMethod.getReturnType().represents(int.class)) {\n mv.visitVarInsn(Opcodes.ISTORE, argumentHandler.returned());\n } else if (instrumentedMethod.getReturnType().represents(long.class)) {\n mv.visitVarInsn(Opcodes.LSTORE, argumentHandler.returned());\n } else if (instrumentedMethod.getReturnType().represents(float.class)) {\n mv.visitVarInsn(Opcodes.FSTORE, argumentHandler.returned());\n } else if (instrumentedMethod.getReturnType().represents(double.class)) {\n mv.visitVarInsn(Opcodes.DSTORE, argumentHandler.returned());\n } else if (!instrumentedMethod.getReturnType().represents(void.class)) {\n mv.visitVarInsn(Opcodes.ASTORE, argumentHandler.returned());\n }\n mv.visitInsn(Opcodes.ACONST_NULL);\n mv.visitVarInsn(Opcodes.ASTORE, argumentHandler.thrown());\n Label endOfHandler = new Label();\n mv.visitJumpInsn(Opcodes.GOTO, endOfHandler);\n mv.visitLabel(exceptionHandler);\n stackMapFrameHandler.injectExceptionFrame(mv);\n mv.visitVarInsn(Opcodes.ASTORE, argumentHandler.thrown());\n if (instrumentedMethod.getReturnType().represents(boolean.class)\n || instrumentedMethod.getReturnType().represents(byte.class)\n || instrumentedMethod.getReturnType().represents(short.class)\n || instrumentedMethod.getReturnType().represents(char.class)\n || instrumentedMethod.getReturnType().represents(int.class)) {\n mv.visitInsn(Opcodes.ICONST_0);\n mv.visitVarInsn(Opcodes.ISTORE, argumentHandler.returned());\n } else if (instrumentedMethod.getReturnType().represents(long.class)) {\n mv.visitInsn(Opcodes.LCONST_0);\n mv.visitVarInsn(Opcodes.LSTORE, argumentHandler.returned());\n } else if (instrumentedMethod.getReturnType().represents(float.class)) {\n mv.visitInsn(Opcodes.FCONST_0);\n mv.visitVarInsn(Opcodes.FSTORE, argumentHandler.returned());\n } else if (instrumentedMethod.getReturnType().represents(double.class)) {\n mv.visitInsn(Opcodes.DCONST_0);\n mv.visitVarInsn(Opcodes.DSTORE, argumentHandler.returned());\n } else if (!instrumentedMethod.getReturnType().represents(void.class)) {\n mv.visitInsn(Opcodes.ACONST_NULL);\n mv.visitVarInsn(Opcodes.ASTORE, argumentHandler.returned());\n }\n mv.visitLabel(endOfHandler);\n methodSizeHandler.requireStackSize(StackSize.SINGLE.getSize());\n }\n\n @Override\n protected void onExitAdviceReturn() {\n mv.visitVarInsn(Opcodes.ALOAD, argumentHandler.thrown());\n Label endOfHandler = new Label();\n mv.visitJumpInsn(Opcodes.IFNULL, endOfHandler);\n mv.visitVarInsn(Opcodes.ALOAD, argumentHandler.thrown());\n mv.visitInsn(Opcodes.ATHROW);\n mv.visitLabel(endOfHandler);\n stackMapFrameHandler.injectPostCompletionFrame(mv);\n }\n }\n }\n\n\n /**\n * A dispatcher for implementing advice.\n */\n public interface Dispatcher {\n\n /**\n * Indicates that a method does not represent advice and does not need to be visited.\n */\n MethodVisitor IGNORE_METHOD = null;\n\n /**\n * Expresses that an annotation should not be visited.\n */\n AnnotationVisitor IGNORE_ANNOTATION = null;\n\n /**\n * Returns {@code true} if this dispatcher is alive.\n *\n * @return {@code true} if this dispatcher is alive.\n */\n boolean isAlive();\n\n /**\n * The type that is produced as a result of executing this advice method.\n *\n * @return A description of the type that is produced by this advice method.\n */\n TypeDefinition getAdviceType();\n\n /**\n * A dispatcher that is not yet resolved.\n */\n interface Unresolved extends Dispatcher {\n\n /**\n * Indicates that this dispatcher requires access to the class file declaring the advice method.\n *\n * @return {@code true} if this dispatcher requires access to the advice method's class file.\n */\n boolean isBinary();\n\n /**\n * Returns the named types declared by this enter advice.\n *\n * @return The named types declared by this enter advice.\n */\n Map getNamedTypes();\n\n /**\n * Resolves this dispatcher as a dispatcher for entering a method.\n *\n * @param userFactories A list of custom factories for binding parameters of an advice method.\n * @param classReader A class reader to query for a class file which might be {@code null} if this dispatcher is not binary.\n * @param methodExit The unresolved dispatcher for the method exit advice.\n * @param postProcessorFactory The post processor factory to use.\n * @return This dispatcher as a dispatcher for entering a method.\n */\n Resolved.ForMethodEnter asMethodEnter(List> userFactories,\n ClassReader classReader,\n Unresolved methodExit,\n PostProcessor.Factory postProcessorFactory);\n\n /**\n * Resolves this dispatcher as a dispatcher for exiting a method.\n *\n * @param userFactories A list of custom factories for binding parameters of an advice method.\n * @param classReader A class reader to query for a class file which might be {@code null} if this dispatcher is not binary.\n * @param methodEnter The unresolved dispatcher for the method enter advice.\n * @param postProcessorFactory The post processor factory to use.\n * @return This dispatcher as a dispatcher for exiting a method.\n */\n Resolved.ForMethodExit asMethodExit(List> userFactories,\n ClassReader classReader,\n Unresolved methodEnter,\n PostProcessor.Factory postProcessorFactory);\n }\n\n\n /**\n * A suppression handler for optionally suppressing exceptions.\n */\n interface SuppressionHandler {\n\n /**\n * Binds the suppression handler for instrumenting a specific method.\n *\n * @param exceptionHandler The stack manipulation to apply within a suppression handler.\n * @return A bound version of the suppression handler.\n */\n Bound bind(StackManipulation exceptionHandler);\n\n /**\n * A bound version of a suppression handler that must not be reused.\n */\n interface Bound {\n\n /**\n * Invoked to prepare the suppression handler, i.e. to write an exception handler entry if appropriate.\n *\n * @param methodVisitor The method visitor to apply the preparation to.\n */\n void onPrepare(MethodVisitor methodVisitor);\n\n /**\n * Invoked at the start of a method.\n *\n * @param methodVisitor The method visitor of the instrumented method.\n */\n void onStart(MethodVisitor methodVisitor);\n\n /**\n * Invoked at the end of a method.\n *\n * @param methodVisitor The method visitor of the instrumented method.\n * @param implementationContext The implementation context to use.\n * @param methodSizeHandler The advice method's method size handler.\n * @param stackMapFrameHandler A handler for translating and injecting stack map frames.\n * @param returnType The return type of the advice method.\n */\n void onEnd(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n MethodSizeHandler.ForAdvice methodSizeHandler,\n StackMapFrameHandler.ForAdvice stackMapFrameHandler,\n TypeDefinition returnType);\n\n /**\n * Invoked at the end of a method if the exception handler should be wrapped in a skipping block.\n *\n * @param methodVisitor The method visitor of the instrumented method.\n * @param implementationContext The implementation context to use.\n * @param methodSizeHandler The advice method's method size handler.\n * @param stackMapFrameHandler A handler for translating and injecting stack map frames.\n * @param returnType The return type of the advice method.\n */\n void onEndWithSkip(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n MethodSizeHandler.ForAdvice methodSizeHandler,\n StackMapFrameHandler.ForAdvice stackMapFrameHandler,\n TypeDefinition returnType);\n }\n\n /**\n * A non-operational suppression handler that does not suppress any method.\n */\n enum NoOp implements SuppressionHandler, Bound {\n\n /**\n * The singleton instance.\n */\n INSTANCE;\n\n /**\n * {@inheritDoc}\n */\n public Bound bind(StackManipulation exceptionHandler) {\n return this;\n }\n\n /**\n * {@inheritDoc}\n */\n public void onPrepare(MethodVisitor methodVisitor) {\n /* do nothing */\n }\n\n /**\n * {@inheritDoc}\n */\n public void onStart(MethodVisitor methodVisitor) {\n /* do nothing */\n }\n\n /**\n * {@inheritDoc}\n */\n public void onEnd(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n MethodSizeHandler.ForAdvice methodSizeHandler,\n StackMapFrameHandler.ForAdvice stackMapFrameHandler,\n TypeDefinition returnType) {\n /* do nothing */\n }\n\n /**\n * {@inheritDoc}\n */\n public void onEndWithSkip(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n MethodSizeHandler.ForAdvice methodSizeHandler,\n StackMapFrameHandler.ForAdvice stackMapFrameHandler,\n TypeDefinition returnType) {\n /* do nothing */\n }\n }\n\n /**\n * A suppression handler that suppresses a given throwable type.\n */\n @HashCodeAndEqualsPlugin.Enhance\n class Suppressing implements SuppressionHandler {\n\n /**\n * The suppressed throwable type.\n */\n private final TypeDescription suppressedType;\n\n /**\n * Creates a new suppressing suppression handler.\n *\n * @param suppressedType The suppressed throwable type.\n */\n protected Suppressing(TypeDescription suppressedType) {\n this.suppressedType = suppressedType;\n }\n\n /**\n * Resolves an appropriate suppression handler.\n *\n * @param suppressedType The suppressed type or {@link NoExceptionHandler} if no type should be suppressed.\n * @return An appropriate suppression handler.\n */\n protected static SuppressionHandler of(TypeDescription suppressedType) {\n return isNoExceptionHandler(suppressedType)\n ? NoOp.INSTANCE\n : new Suppressing(suppressedType);\n }\n\n /**\n * {@inheritDoc}\n */\n public SuppressionHandler.Bound bind(StackManipulation exceptionHandler) {\n return new Bound(suppressedType, exceptionHandler);\n }\n\n /**\n * An active, bound suppression handler.\n */\n protected static class Bound implements SuppressionHandler.Bound {\n\n /**\n * The suppressed throwable type.\n */\n private final TypeDescription suppressedType;\n\n /**\n * The stack manipulation to apply within a suppression handler.\n */\n private final StackManipulation exceptionHandler;\n\n /**\n * A label indicating the start of the method.\n */\n private final Label startOfMethod;\n\n /**\n * A label indicating the end of the method.\n */\n private final Label endOfMethod;\n\n /**\n * Creates a new active, bound suppression handler.\n *\n * @param suppressedType The suppressed throwable type.\n * @param exceptionHandler The stack manipulation to apply within a suppression handler.\n */\n protected Bound(TypeDescription suppressedType, StackManipulation exceptionHandler) {\n this.suppressedType = suppressedType;\n this.exceptionHandler = exceptionHandler;\n startOfMethod = new Label();\n endOfMethod = new Label();\n }\n\n /**\n * {@inheritDoc}\n */\n public void onPrepare(MethodVisitor methodVisitor) {\n methodVisitor.visitTryCatchBlock(startOfMethod, endOfMethod, endOfMethod, suppressedType.getInternalName());\n }\n\n /**\n * {@inheritDoc}\n */\n public void onStart(MethodVisitor methodVisitor) {\n methodVisitor.visitLabel(startOfMethod);\n }\n\n /**\n * {@inheritDoc}\n */\n public void onEnd(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n MethodSizeHandler.ForAdvice methodSizeHandler,\n StackMapFrameHandler.ForAdvice stackMapFrameHandler,\n TypeDefinition returnType) {\n methodVisitor.visitLabel(endOfMethod);\n stackMapFrameHandler.injectExceptionFrame(methodVisitor);\n methodSizeHandler.requireStackSize(1 + exceptionHandler.apply(methodVisitor, implementationContext).getMaximalSize());\n if (returnType.represents(boolean.class)\n || returnType.represents(byte.class)\n || returnType.represents(short.class)\n || returnType.represents(char.class)\n || returnType.represents(int.class)) {\n methodVisitor.visitInsn(Opcodes.ICONST_0);\n } else if (returnType.represents(long.class)) {\n methodVisitor.visitInsn(Opcodes.LCONST_0);\n } else if (returnType.represents(float.class)) {\n methodVisitor.visitInsn(Opcodes.FCONST_0);\n } else if (returnType.represents(double.class)) {\n methodVisitor.visitInsn(Opcodes.DCONST_0);\n } else if (!returnType.represents(void.class)) {\n methodVisitor.visitInsn(Opcodes.ACONST_NULL);\n }\n }\n\n /**\n * {@inheritDoc}\n */\n public void onEndWithSkip(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n MethodSizeHandler.ForAdvice methodSizeHandler,\n StackMapFrameHandler.ForAdvice stackMapFrameHandler,\n TypeDefinition returnType) {\n Label skipExceptionHandler = new Label();\n methodVisitor.visitJumpInsn(Opcodes.GOTO, skipExceptionHandler);\n onEnd(methodVisitor, implementationContext, methodSizeHandler, stackMapFrameHandler, returnType);\n methodVisitor.visitLabel(skipExceptionHandler);\n stackMapFrameHandler.injectReturnFrame(methodVisitor);\n }\n }\n }\n }\n\n /**\n * A relocation handler is responsible for chaining the usual control flow of an instrumented method.\n */\n interface RelocationHandler {\n\n /**\n * Binds this relocation handler to a relocation dispatcher.\n *\n * @param instrumentedMethod The instrumented method.\n * @param relocation The relocation to apply.\n * @return A bound relocation handler.\n */\n Bound bind(MethodDescription instrumentedMethod, Relocation relocation);\n\n /**\n * A re-locator is responsible for triggering a relocation if a relocation handler triggers a relocating condition.\n */\n interface Relocation {\n\n /**\n * Applies this relocator.\n *\n * @param methodVisitor The method visitor to use.\n */\n void apply(MethodVisitor methodVisitor);\n\n /**\n * A relocation that unconditionally jumps to a given label.\n */\n @HashCodeAndEqualsPlugin.Enhance\n class ForLabel implements Relocation {\n\n /**\n * The label to jump to.\n */\n private final Label label;\n\n /**\n * Creates a new relocation for an unconditional jump to a given label.\n *\n * @param label The label to jump to.\n */\n public ForLabel(Label label) {\n this.label = label;\n }\n\n /**\n * {@inheritDoc}\n */\n public void apply(MethodVisitor methodVisitor) {\n methodVisitor.visitJumpInsn(Opcodes.GOTO, label);\n }\n }\n }\n\n /**\n * A bound {@link RelocationHandler}.\n */\n interface Bound {\n\n /**\n * Indicates that this relocation handler does not require a minimal stack size.\n */\n int NO_REQUIRED_SIZE = 0;\n\n /**\n * Applies this relocation handler.\n *\n * @param methodVisitor The method visitor to use.\n * @param offset The offset of the relevant value.\n * @return The minimal required stack size to apply this relocation handler.\n */\n int apply(MethodVisitor methodVisitor, int offset);\n }\n\n /**\n * A disabled relocation handler that does never trigger a relocation.\n */\n enum Disabled implements RelocationHandler, Bound {\n\n /**\n * The singleton instance.\n */\n INSTANCE;\n\n /**\n * {@inheritDoc}\n */\n public Bound bind(MethodDescription instrumentedMethod, Relocation relocation) {\n return this;\n }\n\n /**\n * {@inheritDoc}\n */\n public int apply(MethodVisitor methodVisitor, int offset) {\n return NO_REQUIRED_SIZE;\n }\n }\n\n /**\n * A relocation handler that triggers a relocation for a default or non-default value.\n */\n enum ForValue implements RelocationHandler {\n\n /**\n * A relocation handler for an {@code int} type or any compatible type.\n */\n INTEGER(Opcodes.ILOAD, Opcodes.IFNE, Opcodes.IFEQ, 0) {\n @Override\n protected void convertValue(MethodVisitor methodVisitor) {\n /* do nothing */\n }\n },\n\n /**\n * A relocation handler for a {@code long} type.\n */\n LONG(Opcodes.LLOAD, Opcodes.IFNE, Opcodes.IFEQ, 0) {\n @Override\n protected void convertValue(MethodVisitor methodVisitor) {\n methodVisitor.visitInsn(Opcodes.L2I);\n }\n },\n\n /**\n * A relocation handler for a {@code float} type.\n */\n FLOAT(Opcodes.FLOAD, Opcodes.IFNE, Opcodes.IFEQ, 2) {\n @Override\n protected void convertValue(MethodVisitor methodVisitor) {\n methodVisitor.visitInsn(Opcodes.FCONST_0);\n methodVisitor.visitInsn(Opcodes.FCMPL);\n }\n },\n\n /**\n * A relocation handler for a {@code double} type.\n */\n DOUBLE(Opcodes.DLOAD, Opcodes.IFNE, Opcodes.IFEQ, 4) {\n @Override\n protected void convertValue(MethodVisitor methodVisitor) {\n methodVisitor.visitInsn(Opcodes.DCONST_0);\n methodVisitor.visitInsn(Opcodes.DCMPL);\n }\n },\n\n /**\n * A relocation handler for a reference type.\n */\n REFERENCE(Opcodes.ALOAD, Opcodes.IFNONNULL, Opcodes.IFNULL, 0) {\n @Override\n protected void convertValue(MethodVisitor methodVisitor) {\n /* do nothing */\n }\n };\n\n /**\n * An opcode for loading a value of the represented type from the local variable array.\n */\n private final int load;\n\n /**\n * The opcode to check for a non-default value.\n */\n private final int defaultJump;\n\n /**\n * The opcode to check for a default value.\n */\n private final int nonDefaultJump;\n\n /**\n * The minimal required stack size to apply this relocation handler.\n */\n private final int requiredSize;\n\n /**\n * Creates a new relocation handler for a type's default or non-default value.\n *\n * @param load An opcode for loading a value of the represented type from the local variable array.\n * @param defaultJump The opcode to check for a non-default value.\n * @param nonDefaultJump The opcode to check for a default value.\n * @param requiredSize The minimal required stack size to apply this relocation handler.\n */\n ForValue(int load, int defaultJump, int nonDefaultJump, int requiredSize) {\n this.load = load;\n this.defaultJump = defaultJump;\n this.nonDefaultJump = nonDefaultJump;\n this.requiredSize = requiredSize;\n }\n\n /**\n * Resolves a relocation handler for a given type.\n *\n * @param typeDefinition The type to be resolved for a relocation attempt.\n * @param inverted {@code true} if the relocation should be applied for any non-default value of a type.\n * @return An appropriate relocation handler.\n */\n protected static RelocationHandler of(TypeDefinition typeDefinition, boolean inverted) {\n ForValue skipDispatcher;\n if (typeDefinition.represents(long.class)) {\n skipDispatcher = LONG;\n } else if (typeDefinition.represents(float.class)) {\n skipDispatcher = FLOAT;\n } else if (typeDefinition.represents(double.class)) {\n skipDispatcher = DOUBLE;\n } else if (typeDefinition.represents(void.class)) {\n throw new IllegalStateException(\"Cannot skip on default value for void return type\");\n } else if (typeDefinition.isPrimitive()) { // anyOf(byte, short, char, int)\n skipDispatcher = INTEGER;\n } else {\n skipDispatcher = REFERENCE;\n }\n return inverted\n ? skipDispatcher.new Inverted()\n : skipDispatcher;\n }\n\n /**\n * Applies a value conversion prior to a applying a conditional jump.\n *\n * @param methodVisitor The method visitor to use.\n */\n protected abstract void convertValue(MethodVisitor methodVisitor);\n\n /**\n * {@inheritDoc}\n */\n public RelocationHandler.Bound bind(MethodDescription instrumentedMethod, Relocation relocation) {\n return new Bound(instrumentedMethod, relocation, false);\n }\n\n /**\n * An inverted version of the outer relocation handler.\n */\n @HashCodeAndEqualsPlugin.Enhance(includeSyntheticFields = true)\n protected class Inverted implements RelocationHandler {\n\n /**\n * {@inheritDoc}\n */\n public Bound bind(MethodDescription instrumentedMethod, Relocation relocation) {\n return new ForValue.Bound(instrumentedMethod, relocation, true);\n }\n }\n\n /**\n * A bound relocation handler for {@link ForValue}.\n */\n @HashCodeAndEqualsPlugin.Enhance(includeSyntheticFields = true)\n protected class Bound implements RelocationHandler.Bound {\n\n /**\n * The instrumented method.\n */\n private final MethodDescription instrumentedMethod;\n\n /**\n * The relocation to apply.\n */\n private final Relocation relocation;\n\n /**\n * {@code true} if the relocation should be applied for any non-default value of a type.\n */\n private final boolean inverted;\n\n /**\n * Creates a new bound relocation handler.\n *\n * @param instrumentedMethod The instrumented method.\n * @param relocation The relocation to apply.\n * @param inverted {@code true} if the relocation should be applied for any non-default value of a type.\n */\n protected Bound(MethodDescription instrumentedMethod, Relocation relocation, boolean inverted) {\n this.instrumentedMethod = instrumentedMethod;\n this.relocation = relocation;\n this.inverted = inverted;\n }\n\n /**\n * {@inheritDoc}\n */\n public int apply(MethodVisitor methodVisitor, int offset) {\n if (instrumentedMethod.isConstructor()) {\n throw new IllegalStateException(\"Cannot skip code execution from constructor: \" + instrumentedMethod);\n }\n methodVisitor.visitVarInsn(load, offset);\n convertValue(methodVisitor);\n Label noSkip = new Label();\n methodVisitor.visitJumpInsn(inverted\n ? nonDefaultJump\n : defaultJump, noSkip);\n relocation.apply(methodVisitor);\n methodVisitor.visitLabel(noSkip);\n return requiredSize;\n }\n }\n }\n\n /**\n * A relocation handler that is triggered if the checked value is an instance of a given type.\n */\n @HashCodeAndEqualsPlugin.Enhance\n class ForType implements RelocationHandler {\n\n /**\n * The type that triggers a relocation.\n */\n private final TypeDescription typeDescription;\n\n /**\n * Creates a new relocation handler that triggers a relocation if a value is an instance of a given type.\n *\n * @param typeDescription The type that triggers a relocation.\n */\n protected ForType(TypeDescription typeDescription) {\n this.typeDescription = typeDescription;\n }\n\n /**\n * Resolves a relocation handler that is triggered if the checked instance is of a given type.\n *\n * @param typeDescription The type that triggers a relocation.\n * @param checkedType The type that is carrying the checked value.\n * @return An appropriate relocation handler.\n */\n protected static RelocationHandler of(TypeDescription typeDescription, TypeDefinition checkedType) {\n if (typeDescription.represents(void.class)) {\n return Disabled.INSTANCE;\n } else if (typeDescription.represents(OnDefaultValue.class)) {\n return ForValue.of(checkedType, false);\n } else if (typeDescription.represents(OnNonDefaultValue.class)) {\n return ForValue.of(checkedType, true);\n } else if (typeDescription.isPrimitive() || checkedType.isPrimitive()) {\n throw new IllegalStateException(\"Cannot skip method by instance type for primitive return type \" + checkedType);\n } else {\n return new ForType(typeDescription);\n }\n }\n\n /**\n * {@inheritDoc}\n */\n public RelocationHandler.Bound bind(MethodDescription instrumentedMethod, Relocation relocation) {\n return new Bound(instrumentedMethod, relocation);\n }\n\n /**\n * A bound relocation handler for {@link ForType}.\n */\n @HashCodeAndEqualsPlugin.Enhance(includeSyntheticFields = true)\n protected class Bound implements RelocationHandler.Bound {\n\n /**\n * The instrumented method.\n */\n private final MethodDescription instrumentedMethod;\n\n /**\n * The relocation to use.\n */\n private final Relocation relocation;\n\n /**\n * Creates a new bound relocation handler.\n *\n * @param instrumentedMethod The instrumented method.\n * @param relocation The relocation to apply.\n */\n protected Bound(MethodDescription instrumentedMethod, Relocation relocation) {\n this.instrumentedMethod = instrumentedMethod;\n this.relocation = relocation;\n }\n\n /**\n * {@inheritDoc}\n */\n public int apply(MethodVisitor methodVisitor, int offset) {\n if (instrumentedMethod.isConstructor()) {\n throw new IllegalStateException(\"Cannot skip code execution from constructor: \" + instrumentedMethod);\n }\n methodVisitor.visitVarInsn(Opcodes.ALOAD, offset);\n methodVisitor.visitTypeInsn(Opcodes.INSTANCEOF, typeDescription.getInternalName());\n Label noSkip = new Label();\n methodVisitor.visitJumpInsn(Opcodes.IFEQ, noSkip);\n relocation.apply(methodVisitor);\n methodVisitor.visitLabel(noSkip);\n return NO_REQUIRED_SIZE;\n }\n }\n }\n }\n\n /**\n * Represents a resolved dispatcher.\n */\n interface Resolved extends Dispatcher {\n\n /**\n * Returns the named types defined by this advice.\n *\n * @return The named types defined by this advice.\n */\n Map getNamedTypes();\n\n /**\n * Binds this dispatcher for resolution to a specific method.\n *\n * @param instrumentedType The instrumented type.\n * @param instrumentedMethod The instrumented method.\n * @param methodVisitor The method visitor for writing the instrumented method.\n * @param implementationContext The implementation context to use.\n * @param assigner The assigner to use.\n * @param argumentHandler A handler for accessing values on the local variable array.\n * @param methodSizeHandler A handler for computing the method size requirements.\n * @param stackMapFrameHandler A handler for translating and injecting stack map frames.\n * @param exceptionHandler The stack manipulation to apply within a suppression handler.\n * @param relocation A relocation to use with a relocation handler.\n * @return A dispatcher that is bound to the instrumented method.\n */\n Bound bind(TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n Assigner assigner,\n ArgumentHandler.ForInstrumentedMethod argumentHandler,\n MethodSizeHandler.ForInstrumentedMethod methodSizeHandler,\n StackMapFrameHandler.ForInstrumentedMethod stackMapFrameHandler,\n StackManipulation exceptionHandler,\n RelocationHandler.Relocation relocation);\n\n\n Map getOffsetMapping();\n\n /**\n * Represents a resolved dispatcher for entering a method.\n */\n interface ForMethodEnter extends Resolved {\n\n /**\n * Returns {@code true} if the first discovered line number information should be prepended to the advice code.\n *\n * @return {@code true} if the first discovered line number information should be prepended to the advice code.\n */\n boolean isPrependLineNumber();\n\n /**\n * Returns the actual advice type, even if it is not required post advice processing.\n *\n * @return The actual advice type, even if it is not required post advice processing.\n */\n TypeDefinition getActualAdviceType();\n }\n\n /**\n * Represents a resolved dispatcher for exiting a method.\n */\n interface ForMethodExit extends Resolved {\n\n /**\n * Returns the type of throwable for which this exit advice is supposed to be invoked.\n *\n * @return The {@link Throwable} type for which to invoke this exit advice or a description of {@link NoExceptionHandler}\n * if this exit advice does not expect to be invoked upon any throwable.\n */\n TypeDescription getThrowable();\n\n /**\n * Returns a factory for creating an {@link ArgumentHandler}.\n *\n * @return A factory for creating an {@link ArgumentHandler}.\n */\n ArgumentHandler.Factory getArgumentHandlerFactory();\n }\n\n /**\n * An abstract base implementation of a {@link Resolved} dispatcher.\n */\n @HashCodeAndEqualsPlugin.Enhance\n abstract class AbstractBase implements Resolved {\n\n /**\n * The represented advice method.\n */\n protected final MethodDescription.InDefinedShape adviceMethod;\n\n /**\n * The post processor to apply.\n */\n protected final PostProcessor postProcessor;\n\n /**\n * A mapping from offset to a mapping for this offset with retained iteration order of the method's parameters.\n */\n protected final Map offsetMappings;\n\n /**\n * The suppression handler to use.\n */\n protected final SuppressionHandler suppressionHandler;\n\n /**\n * The relocation handler to use.\n */\n protected final RelocationHandler relocationHandler;\n\n /**\n * Creates a new resolved version of a dispatcher.\n *\n * @param adviceMethod The represented advice method.\n * @param postProcessor The post processor to use.\n * @param factories A list of factories to resolve for the parameters of the advice method.\n * @param throwableType The type to handle by a suppression handler or {@link NoExceptionHandler} to not handle any exceptions.\n * @param relocatableType The type to trigger a relocation of the method's control flow or {@code void} if no relocation should be executed.\n * @param adviceType The applied advice type.\n */\n protected AbstractBase(MethodDescription.InDefinedShape adviceMethod,\n PostProcessor postProcessor,\n List> factories,\n TypeDescription throwableType,\n TypeDescription relocatableType,\n OffsetMapping.Factory.AdviceType adviceType) {\n this.adviceMethod = adviceMethod;\n this.postProcessor = postProcessor;\n Map> offsetMappings = new HashMap<>();\n for (OffsetMapping.Factory factory : factories) {\n offsetMappings.put(TypeDescription.ForLoadedType.of(factory.getAnnotationType()), factory);\n }\n this.offsetMappings = new LinkedHashMap<>();\n for (ParameterDescription.InDefinedShape parameterDescription : adviceMethod.getParameters()) {\n OffsetMapping offsetMapping = null;\n for (AnnotationDescription annotationDescription : parameterDescription.getDeclaredAnnotations()) {\n OffsetMapping.Factory factory = offsetMappings.get(annotationDescription.getAnnotationType());\n if (factory != null) {\n @SuppressWarnings(\"unchecked, rawtypes\")\n OffsetMapping current = factory.make(parameterDescription,\n (AnnotationDescription.Loadable) annotationDescription.prepare(factory.getAnnotationType()),\n adviceType);\n if (offsetMapping == null) {\n offsetMapping = current;\n } else {\n throw new IllegalStateException(parameterDescription + \" is bound to both \" + current + \" and \" + offsetMapping);\n }\n }\n }\n this.offsetMappings.put(parameterDescription.getOffset(), offsetMapping == null\n ? new OffsetMapping.ForArgument.Unresolved(parameterDescription)\n : offsetMapping);\n }\n suppressionHandler = SuppressionHandler.Suppressing.of(throwableType);\n relocationHandler = RelocationHandler.ForType.of(relocatableType, adviceMethod.getReturnType());\n }\n\n /**\n * {@inheritDoc}\n */\n public boolean isAlive() {\n return true;\n }\n\n @Override\n public Map getOffsetMapping() {\n return this.offsetMappings;\n }\n }\n }\n\n /**\n * A bound resolution of an advice method.\n */\n interface Bound {\n\n /**\n * Prepares the advice method's exception handlers.\n */\n void prepare();\n\n /**\n * Initialized the advice's methods local variables.\n */\n void initialize();\n\n /**\n * Applies this dispatcher.\n */\n void apply();\n }\n\n /**\n * An implementation for inactive devise that does not write any byte code.\n */\n enum Inactive implements Dispatcher.Unresolved, Resolved.ForMethodEnter, Resolved.ForMethodExit, Bound {\n\n /**\n * The singleton instance.\n */\n INSTANCE;\n\n /**\n * {@inheritDoc}\n */\n public boolean isAlive() {\n return false;\n }\n\n /**\n * {@inheritDoc}\n */\n public boolean isBinary() {\n return false;\n }\n\n /**\n * {@inheritDoc}\n */\n public TypeDescription getAdviceType() {\n return TypeDescription.VOID;\n }\n\n /**\n * {@inheritDoc}\n */\n public boolean isPrependLineNumber() {\n return false;\n }\n\n /**\n * {@inheritDoc}\n */\n public TypeDefinition getActualAdviceType() {\n return TypeDescription.VOID;\n }\n\n /**\n * {@inheritDoc}\n */\n public Map getNamedTypes() {\n return Collections.emptyMap();\n }\n\n /**\n * {@inheritDoc}\n */\n public TypeDescription getThrowable() {\n return NoExceptionHandler.DESCRIPTION;\n }\n\n /**\n * {@inheritDoc}\n */\n public ArgumentHandler.Factory getArgumentHandlerFactory() {\n return ArgumentHandler.Factory.SIMPLE;\n }\n\n /**\n * {@inheritDoc}\n */\n public Resolved.ForMethodEnter asMethodEnter(List> userFactories,\n ClassReader classReader,\n Unresolved methodExit,\n PostProcessor.Factory postProcessorFactory) {\n return this;\n }\n\n /**\n * {@inheritDoc}\n */\n public Resolved.ForMethodExit asMethodExit(List> userFactories,\n ClassReader classReader,\n Unresolved methodEnter,\n PostProcessor.Factory postProcessorFactory) {\n return this;\n }\n\n /**\n * {@inheritDoc}\n */\n public void prepare() {\n /* do nothing */\n }\n\n /**\n * {@inheritDoc}\n */\n public void initialize() {\n /* do nothing */\n }\n\n /**\n * {@inheritDoc}\n */\n public void apply() {\n /* do nothing */\n }\n\n /**\n * {@inheritDoc}\n */\n public Bound bind(TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n Assigner assigner,\n ArgumentHandler.ForInstrumentedMethod argumentHandler,\n MethodSizeHandler.ForInstrumentedMethod methodSizeHandler,\n StackMapFrameHandler.ForInstrumentedMethod stackMapFrameHandler,\n StackManipulation exceptionHandler,\n RelocationHandler.Relocation relocation) {\n return this;\n }\n\n @Override\n public Map getOffsetMapping() {\n return null;\n }\n }\n\n /**\n * A dispatcher for an advice method that is being inlined into the instrumented method.\n */\n @HashCodeAndEqualsPlugin.Enhance\n class Inlining implements Unresolved {\n\n /**\n * The advice method.\n */\n protected final MethodDescription.InDefinedShape adviceMethod;\n\n /**\n * A mapping of all available local variables by their name to their type.\n */\n private final Map namedTypes;\n\n /**\n * Creates a dispatcher for inlined advice method.\n *\n * @param adviceMethod The advice method.\n */\n protected Inlining(MethodDescription.InDefinedShape adviceMethod) {\n this.adviceMethod = adviceMethod;\n namedTypes = new HashMap<>();\n for (ParameterDescription parameterDescription\n : adviceMethod.getParameters().filter(isAnnotatedWith(Local.class))) {\n String name = parameterDescription.getDeclaredAnnotations()\n .ofType(Local.class).getValue(OffsetMapping.ForLocalValue.Factory.LOCAL_VALUE)\n .resolve(String.class);\n TypeDefinition previous = namedTypes.put(name, parameterDescription.getType());\n if (previous != null && !previous.equals(parameterDescription.getType())) {\n throw new IllegalStateException(\"Local variable for \" + name + \" is defined with inconsistent types\");\n }\n }\n }\n\n /**\n * {@inheritDoc}\n */\n public boolean isAlive() {\n return true;\n }\n\n /**\n * {@inheritDoc}\n */\n public boolean isBinary() {\n return true;\n }\n\n /**\n * {@inheritDoc}\n */\n public TypeDescription getAdviceType() {\n return adviceMethod.getReturnType().asErasure();\n }\n\n /**\n * {@inheritDoc}\n */\n public Map getNamedTypes() {\n return namedTypes;\n }\n\n /**\n * {@inheritDoc}\n */\n public Dispatcher.Resolved.ForMethodEnter asMethodEnter(List> userFactories,\n ClassReader classReader,\n Unresolved methodExit,\n PostProcessor.Factory postProcessorFactory) {\n return Resolved.ForMethodEnter.of(adviceMethod,\n postProcessorFactory.make(adviceMethod, false),\n namedTypes,\n userFactories,\n methodExit.getAdviceType(),\n classReader,\n methodExit.isAlive());\n }\n\n /**\n * {@inheritDoc}\n */\n public Dispatcher.Resolved.ForMethodExit asMethodExit(List> userFactories,\n ClassReader classReader,\n Unresolved methodEnter,\n PostProcessor.Factory postProcessorFactory) {\n Map namedTypes = new HashMap<>(methodEnter.getNamedTypes()), uninitializedNamedTypes = new HashMap<>();\n for (Map.Entry entry : this.namedTypes.entrySet()) {\n TypeDefinition typeDefinition = namedTypes.get(entry.getKey()), uninitializedTypeDefinition = uninitializedNamedTypes.get(entry.getKey());\n if (typeDefinition == null && uninitializedTypeDefinition == null) {\n namedTypes.put(entry.getKey(), entry.getValue());\n uninitializedNamedTypes.put(entry.getKey(), entry.getValue());\n } else if (!(typeDefinition == null ? uninitializedTypeDefinition : typeDefinition).equals(entry.getValue())) {\n throw new IllegalStateException(\"Local variable for \" + entry.getKey() + \" is defined with inconsistent types\");\n }\n }\n return Resolved.ForMethodExit.of(adviceMethod,\n postProcessorFactory.make(adviceMethod, true),\n namedTypes, uninitializedNamedTypes, userFactories,\n classReader, methodEnter.getAdviceType());\n }\n\n public Dispatcher.Resolved.ForMethodExit asMethodExitNonThrowable(\n List> userFactories,\n ClassReader classReader,\n Unresolved methodEnter,\n PostProcessor.Factory postProcessorFactory) {\n Map namedTypes = new HashMap<>(methodEnter.getNamedTypes()), uninitializedNamedTypes = new HashMap<>();\n for (Map.Entry entry : this.namedTypes.entrySet()) {\n TypeDefinition typeDefinition = namedTypes.get(entry.getKey()), uninitializedTypeDefinition = uninitializedNamedTypes.get(entry.getKey());\n if (typeDefinition == null && uninitializedTypeDefinition == null) {\n namedTypes.put(entry.getKey(), entry.getValue());\n uninitializedNamedTypes.put(entry.getKey(), entry.getValue());\n } else if (!(typeDefinition == null ? uninitializedTypeDefinition : typeDefinition).equals(entry.getValue())) {\n throw new IllegalStateException(\"Local variable for \" + entry.getKey() + \" is defined with inconsistent types\");\n }\n }\n return Resolved.ForMethodExit.ofNonThrowable(adviceMethod,\n postProcessorFactory.make(adviceMethod, true),\n namedTypes, uninitializedNamedTypes, userFactories,\n classReader, methodEnter.getAdviceType());\n }\n\n /**\n * A resolved version of a dispatcher.\n */\n protected abstract static class Resolved extends Dispatcher.Resolved.AbstractBase {\n\n /**\n * A class reader to query for the class file of the advice method.\n */\n protected final ClassReader classReader;\n\n /**\n * Creates a new resolved version of a dispatcher.\n *\n * @param adviceMethod The represented advice method.\n * @param postProcessor The post processor to apply.\n * @param factories A list of factories to resolve for the parameters of the advice method.\n * @param throwableType The type to handle by a suppression handler or {@link NoExceptionHandler} to not handle any exceptions.\n * @param relocatableType The type to trigger a relocation of the method's control flow or {@code void} if no relocation should be executed.\n * @param classReader A class reader to query for the class file of the advice method.\n */\n protected Resolved(MethodDescription.InDefinedShape adviceMethod,\n PostProcessor postProcessor,\n List> factories,\n TypeDescription throwableType,\n TypeDescription relocatableType,\n ClassReader classReader) {\n super(adviceMethod, postProcessor, factories, throwableType, relocatableType, OffsetMapping.Factory.AdviceType.INLINING);\n this.classReader = classReader;\n }\n\n /**\n * Resolves the initialization types of this advice method.\n *\n * @param argumentHandler The argument handler to use for resolving the initialization.\n * @return A mapping of parameter offsets to the type to initialize.\n */\n protected abstract Map resolveInitializationTypes(ArgumentHandler argumentHandler);\n\n /**\n * Applies a resolution for a given instrumented method.\n *\n * @param methodVisitor A method visitor for writing byte code to the instrumented method.\n * @param implementationContext The implementation context to use.\n * @param assigner The assigner to use.\n * @param argumentHandler A handler for accessing values on the local variable array.\n * @param methodSizeHandler A handler for computing the method size requirements.\n * @param stackMapFrameHandler A handler for translating and injecting stack map frames.\n * @param instrumentedType A description of the instrumented type.\n * @param instrumentedMethod A description of the instrumented method.\n * @param suppressionHandler A bound suppression handler that is used for suppressing exceptions of this advice method.\n * @param relocationHandler A bound relocation handler that is responsible for considering a non-standard control flow.\n * @param exceptionHandler The exception handler that is resolved for the instrumented method.\n * @return A method visitor for visiting the advice method's byte code.\n */\n protected abstract MethodVisitor apply(MethodVisitor methodVisitor,\n Implementation.Context implementationContext,\n Assigner assigner,\n ArgumentHandler.ForInstrumentedMethod argumentHandler,\n MethodSizeHandler.ForInstrumentedMethod methodSizeHandler,\n StackMapFrameHandler.ForInstrumentedMethod stackMapFrameHandler,\n TypeDescription instrumentedType,\n MethodDescription instrumentedMethod,\n SuppressionHandler.Bound suppressionHandler,\n RelocationHandler.Bound relocationHandler,\n StackManipulation exceptionHandler);\n\n /**\n * A bound advice method that copies the code by first extracting the exception table and later appending the\n * code of the method without copying any meta data.\n */\n protected class AdviceMethodInliner extends ClassVisitor implements Bound {\n\n /**\n * A description of the instrumented type.\n */\n protected final TypeDescription instrumentedType;\n\n /**\n * The instrumented method.\n */\n protected final MethodDescription instrumentedMethod;\n\n /**\n * The method visitor for writing the instrumented method.\n */\n protected final MethodVisitor methodVisitor;\n\n /**\n * The implementation context to use.\n */\n protected final Implementation.Context implementationContext;\n\n /**\n * The assigner to use.\n */\n protected final Assigner assigner;\n\n /**\n * A handler for accessing values on the local variable array.\n */\n protected final ArgumentHandler.ForInstrumentedMethod argumentHandler;\n\n /**\n * A handler for computing the method size requirements.\n */\n protected final MethodSizeHandler.ForInstrumentedMethod methodSizeHandler;\n\n /**\n * A handler for translating and injecting stack map frames.\n */\n protected final StackMapFrameHandler.ForInstrumentedMethod stackMapFrameHandler;\n\n /**\n * A bound suppression handler that is used for suppressing exceptions of this advice method.\n */\n protected final SuppressionHandler.Bound suppressionHandler;\n\n /**\n * A bound relocation handler that is responsible for considering a non-standard control flow.\n */\n protected final RelocationHandler.Bound relocationHandler;\n\n /**\n * The exception handler that is resolved for the instrumented method.\n */\n protected final StackManipulation exceptionHandler;\n\n /**\n * A class reader for parsing the class file containing the represented advice method.\n */\n protected final ClassReader classReader;\n\n /**\n * The labels that were found during parsing the method's exception handler in the order of their discovery.\n */\n protected final List