Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix the triple header context transfer issue #1467

Open
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

EvenLjj
Copy link
Collaborator

@EvenLjj EvenLjj commented Feb 5, 2025

fix the triple header context transfer issue

Summary by CodeRabbit

  • Chores
    • Updated the project to version 5.13.3-SNAPSHOT with refreshed build and revision details.
  • New Features
    • Enhanced request tracing by streamlining header processing and improving user context tracking for more accurate diagnostics.

@sofastack-cla sofastack-cla bot added bug Something isn't working cla:yes CLA is ok size/M labels Feb 5, 2025
Copy link
Contributor

coderabbitai bot commented Feb 5, 2025

Walkthrough

This pull request updates the project version from 5.13.2 to 5.13.3-SNAPSHOT across multiple Maven files and the Version class. It also refactors the server request header interception by introducing a new helper method to convert headers into context, and adds utilities to extract user identification in tracing components. These changes adjust version constants, streamline context management, and enhance header-based user ID extraction within the remoting module.

Changes

File(s) Change Summary
all/pom.xml, bom/pom.xml, pom.xml Updated version and <revision> properties from 5.13.2 to 5.13.3-SNAPSHOT
core/api/.../Version.java Updated version constants: VERSION from "5.13.2" to "5.13.3", RPC_VERSION from 51302 to 51303, and BUILD_VERSION with new timestamp
remoting/remoting-triple/.../ServerReqHeaderInterceptor.java Refactored interceptCall to remove inline context creation and added a new method convertHeaderToContext
remoting/remoting-triple/.../TracingContextKey.java Added a new public constant KEY_TRIPLE_USER_ID, a private context key, and a public getKeyUserId() method
remoting/remoting-triple/.../TripleTracerAdapter.java Introduced imports for FastJSON and added a new static method getUserId to retrieve and parse user ID from request headers

Sequence Diagram(s)

sequenceDiagram
    participant C as Client
    participant I as ServerReqHeaderInterceptor
    participant H as convertHeaderToContext
    participant T as TripleTracerAdapter
    participant U as getUserId

    C->>I: Send Request (with Metadata)
    I->>H: Call convertHeaderToContext(call, headers, sofaRequest, serviceDef)
    H->>U: Invoke getUserId(Metadata)
    U-->>H: Return User ID
    H-->>I: Return Constructed Context
    I->>T: Call serverReceived(context)
    T-->>I: Acknowledge Tracing
    I->>C: Process Request
Loading

Possibly related PRs

Suggested labels

size/S

Suggested reviewers

  • Lo1nt
  • chuailiwu
  • wangchengming666

Poem

I'm a hopping rabbit, swift and keen,
Updating versions, keeping code clean.
Header flows refined with a graceful twist,
Constants and context now coexist.
I celebrate these changes with a joyful leap,
Bounding through code while secrets I keep! 🐇
Happy coding in fields so deep!

Tip

🌐 Web search-backed reviews and chat
  • We have enabled web search-based reviews and chat for all users. This feature allows CodeRabbit to access the latest documentation and information on the web.
  • You can disable this feature by setting web_search: false in the knowledge_base settings.
  • Please share any feedback in the Discord discussion.
✨ Finishing Touches
  • 📝 Generate Docstrings (Beta)

Thank you for using CodeRabbit. We offer it for free to the OSS community and would appreciate your support in helping us grow. If you find it useful, would you consider giving us a shout-out on your favorite social media?

❤️ Share
🪧 Tips

Chat

There are 3 ways to chat with CodeRabbit:

  • Review comments: Directly reply to a review comment made by CodeRabbit. Example:
    • I pushed a fix in commit <commit_id>, please review it.
    • Generate unit testing code for this file.
    • Open a follow-up GitHub issue for this discussion.
  • Files and specific lines of code (under the "Files changed" tab): Tag @coderabbitai in a new review comment at the desired location with your query. Examples:
    • @coderabbitai generate unit testing code for this file.
    • @coderabbitai modularize this function.
  • PR comments: Tag @coderabbitai in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples:
    • @coderabbitai gather interesting stats about this repository and render them as a table. Additionally, render a pie chart showing the language distribution in the codebase.
    • @coderabbitai read src/utils.ts and generate unit testing code.
    • @coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.
    • @coderabbitai help me debug CodeRabbit configuration file.

Note: Be mindful of the bot's finite context window. It's strongly recommended to break down tasks such as reading entire modules into smaller chunks. For a focused discussion, use review comments to chat about specific files and their changes, instead of using the PR comments.

CodeRabbit Commands (Invoked using PR comments)

  • @coderabbitai pause to pause the reviews on a PR.
  • @coderabbitai resume to resume the paused reviews.
  • @coderabbitai review to trigger an incremental review. This is useful when automatic reviews are disabled for the repository.
  • @coderabbitai full review to do a full review from scratch and review all the files again.
  • @coderabbitai summary to regenerate the summary of the PR.
  • @coderabbitai generate docstrings to generate docstrings for this PR. (Beta)
  • @coderabbitai resolve resolve all the CodeRabbit review comments.
  • @coderabbitai configuration to show the current CodeRabbit configuration for the repository.
  • @coderabbitai help to get help.

Other keywords and placeholders

  • Add @coderabbitai ignore anywhere in the PR description to prevent this PR from being reviewed.
  • Add @coderabbitai summary to generate the high-level summary at a specific location in the PR description.
  • Add @coderabbitai anywhere in the PR title to generate the title automatically.

CodeRabbit Configuration File (.coderabbit.yaml)

  • You can programmatically configure CodeRabbit by adding a .coderabbit.yaml file to the root of your repository.
  • Please see the configuration documentation for more information.
  • If your editor has YAML language server enabled, you can add the path at the top of this file to enable auto-completion and validation: # yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json

Documentation and Community

  • Visit our Documentation for detailed information on how to use CodeRabbit.
  • Join our Discord Community to get help, request features, and share feedback.
  • Follow us on X/Twitter for updates and announcements.

Copy link
Contributor

@coderabbitai coderabbitai bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

🧹 Nitpick comments (1)
remoting/remoting-triple/src/main/java/com/alipay/sofa/rpc/interceptor/ServerReqHeaderInterceptor.java (1)

165-178: Add method documentation.

Consider adding Javadoc to document the purpose, parameters, and return value of this protected method.

+    /**
+     * Converts request headers into a Context with server span, request headers, and user ID.
+     *
+     * @param call the server call
+     * @param requestHeaders the request headers
+     * @param sofaRequest the SOFA request
+     * @param serverServiceDefinition the server service definition
+     * @return the Context with server span and tracing information
+     */
     protected <ReqT, RespT> Context convertHeaderToContext(ServerCall<ReqT, RespT> call,
                                                           Metadata requestHeaders, SofaRequest sofaRequest,
                                                           ServerServiceDefinition serverServiceDefinition) {
📜 Review details

Configuration used: CodeRabbit UI
Review profile: CHILL
Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between be93251 and 66d35b1.

📒 Files selected for processing (7)
  • all/pom.xml (1 hunks)
  • bom/pom.xml (1 hunks)
  • core/api/src/main/java/com/alipay/sofa/rpc/common/Version.java (1 hunks)
  • pom.xml (1 hunks)
  • remoting/remoting-triple/src/main/java/com/alipay/sofa/rpc/interceptor/ServerReqHeaderInterceptor.java (2 hunks)
  • remoting/remoting-triple/src/main/java/com/alipay/sofa/rpc/tracer/sofatracer/TracingContextKey.java (2 hunks)
  • remoting/remoting-triple/src/main/java/com/alipay/sofa/rpc/tracer/sofatracer/TripleTracerAdapter.java (4 hunks)
✅ Files skipped from review due to trivial changes (2)
  • all/pom.xml
  • core/api/src/main/java/com/alipay/sofa/rpc/common/Version.java
🔇 Additional comments (5)
pom.xml (1)

80-80: Version Revision Updated Correctly.
The <revision> property has been updated to 5.13.3-SNAPSHOT, which is consistent with the intended versioning strategy for a snapshot release. Ensure that any downstream modules or scripts that depend on this property are updated accordingly.

bom/pom.xml (1)

13-13: Consistent Revision Property Change.
Updating <revision> to 5.13.3-SNAPSHOT here aligns with the changes made in the parent pom.xml. This helps maintain uniform versioning across the project artifacts.

remoting/remoting-triple/src/main/java/com/alipay/sofa/rpc/tracer/sofatracer/TracingContextKey.java (1)

34-35: LGTM! Well-structured user ID tracking implementation.

The implementation follows the established pattern for context key management and maintains consistency with existing code.

Also applies to: 76-78

remoting/remoting-triple/src/main/java/com/alipay/sofa/rpc/interceptor/ServerReqHeaderInterceptor.java (1)

66-66: LGTM! Clean refactoring of context management.

The extraction of header-to-context conversion logic improves code organization and maintainability.

remoting/remoting-triple/src/main/java/com/alipay/sofa/rpc/tracer/sofatracer/TripleTracerAdapter.java (1)

56-56: LGTM! Good use of static import.

The static import improves code readability.

Comment on lines +307 to +316
public static String getUserId(Metadata requestHeaders) {
String unitInfo = requestHeaders.get(HEAD_KEY_UNIT_INFO);
Map<String, String> unitInfoMap = JSON.parseObject(unitInfo,
new TypeReference<Map<String, String>>() {
});
if (unitInfoMap != null) {
return unitInfoMap.get(USERID_KEY);
}
return null;
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🛠️ Refactor suggestion

Add null check and error handling.

The method should handle null input and JSON parsing errors gracefully.

     public static String getUserId(Metadata requestHeaders) {
+        if (requestHeaders == null) {
+            return null;
+        }
         String unitInfo = requestHeaders.get(HEAD_KEY_UNIT_INFO);
+        if (unitInfo == null) {
+            return null;
+        }
+        try {
             Map<String, String> unitInfoMap = JSON.parseObject(unitInfo,
                 new TypeReference<Map<String, String>>() {
                 });
             if (unitInfoMap != null) {
                 return unitInfoMap.get(USERID_KEY);
             }
             return null;
+        } catch (Exception e) {
+            LOGGER.warn("Failed to parse unit info: " + unitInfo, e);
+            return null;
+        }
     }
📝 Committable suggestion

‼️ IMPORTANT
Carefully review the code before committing. Ensure that it accurately replaces the highlighted code, contains no missing lines, and has no issues with indentation. Thoroughly test & benchmark the code to ensure it meets the requirements.

Suggested change
public static String getUserId(Metadata requestHeaders) {
String unitInfo = requestHeaders.get(HEAD_KEY_UNIT_INFO);
Map<String, String> unitInfoMap = JSON.parseObject(unitInfo,
new TypeReference<Map<String, String>>() {
});
if (unitInfoMap != null) {
return unitInfoMap.get(USERID_KEY);
}
return null;
}
public static String getUserId(Metadata requestHeaders) {
if (requestHeaders == null) {
return null;
}
String unitInfo = requestHeaders.get(HEAD_KEY_UNIT_INFO);
if (unitInfo == null) {
return null;
}
try {
Map<String, String> unitInfoMap = JSON.parseObject(unitInfo,
new TypeReference<Map<String, String>>() {
});
if (unitInfoMap != null) {
return unitInfoMap.get(USERID_KEY);
}
return null;
} catch (Exception e) {
LOGGER.warn("Failed to parse unit info: " + unitInfo, e);
return null;
}
}

…rry_pick

# Conflicts:
#	all/pom.xml
#	bom/pom.xml
#	core/api/src/main/java/com/alipay/sofa/rpc/common/Version.java
#	pom.xml
@EvenLjj EvenLjj closed this Feb 7, 2025
@EvenLjj EvenLjj reopened this Feb 7, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working cla:yes CLA is ok size/M
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant