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

[SPARK-48754] Further refactoring to make the code more stable and accessible #33

Closed
wants to merge 8 commits into from

Conversation

grundprinzip
Copy link
Contributor

@grundprinzip grundprinzip commented Jul 6, 2024

What changes were proposed in this pull request?

Refactor the internal client-related functions and structs so that we no longer expose the GRPC protocol directly to the Spark Session and the DataFrame classes. In addition, this adds additional tests to increase the overall coverage to 60%.

Why are the changes needed?

To be able to have proper, centralized retry logic, we need to be able to decouple the consumers from the underlying GRPC stream.

Does this PR introduce any user-facing change?

No

How was this patch tested?

  • Existing tests.
  • Added additional tests for the new client module
  • Cleaned up the existing tests to have a more usable surface area
  • Added some additional mocks to the client handling

@grundprinzip grundprinzip changed the title Client refactoring [SPARK-48754] Further refactoring to make the code more stable and accessible. Jul 9, 2024
@HyukjinKwon HyukjinKwon changed the title [SPARK-48754] Further refactoring to make the code more stable and accessible. [SPARK-48754][CONNECT] Further refactoring to make the code more stable and accessible Jul 9, 2024
@HyukjinKwon HyukjinKwon changed the title [SPARK-48754][CONNECT] Further refactoring to make the code more stable and accessible [SPARK-48754] Further refactoring to make the code more stable and accessible Jul 9, 2024
@HyukjinKwon
Copy link
Member

Merged to master.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants