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

refactor(pgwire): allow sharing of pg stream #20362

Merged
merged 2 commits into from
Feb 11, 2025
Merged

Conversation

BugenZhao
Copy link
Member

I hereby agree to the terms of the RisingWave Labs, Inc. Contributor License Agreement.

What's changed and what's your intention?

Refactor pgwire to allow PgStream to be cloned and used concurrently, so that we can use it to write messages asynchronously, typically for Notices.

Checklist

  • I have written necessary rustdoc comments.
  • I have added necessary unit tests and integration tests.
  • I have added test labels as necessary.
  • I have added fuzzing tests or opened an issue to track them.
  • My PR contains breaking changes.
  • My PR changes performance-critical code, so I will run (micro) benchmarks and present the results.
  • My PR contains critical fixes that are necessary to be merged into the latest release.

Documentation

  • My PR needs documentation updates.
Release note

Copy link
Member Author

BugenZhao commented Feb 3, 2025


/// The logic of Conn is very simple, just a static dispatcher for TcpStream: Unencrypted or Ssl:
/// Encrypted.
pub enum Conn<S> {
Copy link
Member Author

Choose a reason for hiding this comment

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

Conn is removed because the variants are now under the Arc in PgStream, so that we can upgrading the underlying stream for all shared PgStream instances.

pub struct PgStream<S> {
/// The underlying stream.
stream: Option<S>,
stream: Arc<Mutex<PgStreamInner<S>>>,
Copy link
Member Author

Choose a reason for hiding this comment

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

The overhead for this Mutex will be minimal since there won't be any contention during actual use.

Copy link
Contributor

@ZENOTME ZENOTME left a comment

Choose a reason for hiding this comment

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

LGTM

@BugenZhao BugenZhao enabled auto-merge February 11, 2025 07:58
@BugenZhao BugenZhao added this pull request to the merge queue Feb 11, 2025
Merged via the queue into main with commit 99a13a5 Feb 11, 2025
29 checks passed
@BugenZhao BugenZhao deleted the bz/share-pg-stream branch February 11, 2025 10:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants