[PR #6792] [management] native grpc to reduce goroutine count #27080

Open
opened 2026-08-05 07:08:10 -04:00 by saavagebueno · 0 comments
Owner

📋 Pull Request Information

Original PR: https://github.com/netbirdio/netbird/pull/6792
Author: @pascal-fischer
Created: 7/15/2026
Status: 🔄 Open

Base: mainHead: feature/native-grpc


📝 Commits (5)

  • 5aa2a74 use native grpc + unify token refresh
  • 4d8d0e3 use native grpc + unify token refresh
  • ad31406 improve jobs endpoint
  • 820303c fix interval and job cancel
  • 8fd098d fix linter

📊 Changes

12 files changed (+680 additions, -150 deletions)

View changed files

📝 go.mod (+1 -0)
📝 go.sum (+3 -0)
📝 management/internals/server/boot.go (+7 -2)
📝 management/internals/server/server.go (+120 -8)
management/internals/server/server_multiplex_test.go (+109 -0)
📝 management/internals/shared/grpc/server.go (+19 -43)
📝 management/internals/shared/grpc/token_mgr.go (+54 -46)
📝 management/internals/shared/grpc/token_mgr_test.go (+19 -12)
management/internals/shared/grpc/token_refresh_scheduler.go (+158 -0)
management/internals/shared/grpc/token_refresh_scheduler_test.go (+56 -0)
📝 management/server/job/manager.go (+44 -39)
management/server/job/manager_test.go (+90 -0)

📄 Description

Describe your changes

Stack

Checklist

  • Is it a bug fix
  • Is a typo/documentation fix
  • Is a feature enhancement
  • It is a refactor
  • Created tests that fail without the change (if possible)
  • This change does not modify the public API, gRPC protocols, functionality behavior, CLI / service flags, or introduce a new feature — OR I have discussed it with the NetBird team beforehand (link the issue / Slack thread in the description). See CONTRIBUTING.md.

By submitting this pull request, you confirm that you have read and agree to the terms of the Contributor License Agreement.

Documentation

Select exactly one:

  • I added/updated documentation for this change
  • Documentation is not needed for this change (explain why)

Docs PR URL (required if "docs added" is checked)

Paste the PR link from https://github.com/netbirdio/docs here:

https://github.com/netbirdio/docs/pull/__


View with Codesmith Autofix with Codesmith
Need help on this PR? Tag /codesmith with what you need. Autofix is disabled.

Summary by CodeRabbit

  • New Features

    • Added opt-in native gRPC multiplexing on the main HTTPS endpoint (controlled by NB_MGMT_NATIVE_GRPC).
  • Improvements

    • Enhanced TLS/client protocol steering to keep dual-protocol clients on HTTP/1.1 while gRPC uses HTTP/2.
    • Refined token refresh scheduling for time-based authentication.
    • Updated management job delivery to use per-peer streams for more reliable reconnect behavior.
  • Tests

    • Added coverage for multiplexed gRPC/HTTP routing and for scheduler timing, worker skipping, and cancellation semantics.
    • Added stream-based job manager lifecycle tests.

🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.

## 📋 Pull Request Information **Original PR:** https://github.com/netbirdio/netbird/pull/6792 **Author:** [@pascal-fischer](https://github.com/pascal-fischer) **Created:** 7/15/2026 **Status:** 🔄 Open **Base:** `main` ← **Head:** `feature/native-grpc` --- ### 📝 Commits (5) - [`5aa2a74`](https://github.com/netbirdio/netbird/commit/5aa2a748d7e2dd889733d7fd0ed5e519dbf582c1) use native grpc + unify token refresh - [`4d8d0e3`](https://github.com/netbirdio/netbird/commit/4d8d0e30dbdbaaeae66201a0a72d086792c50e87) use native grpc + unify token refresh - [`ad31406`](https://github.com/netbirdio/netbird/commit/ad314064945f7fcb53cd7d130f154e4db2cb70d5) improve jobs endpoint - [`820303c`](https://github.com/netbirdio/netbird/commit/820303c71dbdecb6141b0a8ad6100012e8338a34) fix interval and job cancel - [`8fd098d`](https://github.com/netbirdio/netbird/commit/8fd098dcc73ef053879c1f38d7031878cf33ead3) fix linter ### 📊 Changes **12 files changed** (+680 additions, -150 deletions) <details> <summary>View changed files</summary> 📝 `go.mod` (+1 -0) 📝 `go.sum` (+3 -0) 📝 `management/internals/server/boot.go` (+7 -2) 📝 `management/internals/server/server.go` (+120 -8) ➕ `management/internals/server/server_multiplex_test.go` (+109 -0) 📝 `management/internals/shared/grpc/server.go` (+19 -43) 📝 `management/internals/shared/grpc/token_mgr.go` (+54 -46) 📝 `management/internals/shared/grpc/token_mgr_test.go` (+19 -12) ➕ `management/internals/shared/grpc/token_refresh_scheduler.go` (+158 -0) ➕ `management/internals/shared/grpc/token_refresh_scheduler_test.go` (+56 -0) 📝 `management/server/job/manager.go` (+44 -39) ➕ `management/server/job/manager_test.go` (+90 -0) </details> ### 📄 Description ## Describe your changes ## Issue ticket number and link ## Stack <!-- branch-stack --> ### Checklist - [ ] Is it a bug fix - [ ] Is a typo/documentation fix - [ ] Is a feature enhancement - [x] It is a refactor - [ ] Created tests that fail without the change (if possible) - [ ] This change does **not** modify the public API, gRPC protocols, functionality behavior, CLI / service flags, or introduce a new feature — **OR** I have discussed it with the NetBird team beforehand (link the issue / Slack thread in the description). See [CONTRIBUTING.md](https://github.com/netbirdio/netbird/blob/main/CONTRIBUTING.md#discuss-changes-with-the-netbird-team-first). > By submitting this pull request, you confirm that you have read and agree to the terms of the [Contributor License Agreement](https://github.com/netbirdio/netbird/blob/main/CONTRIBUTOR_LICENSE_AGREEMENT.md). ## Documentation Select exactly one: - [ ] I added/updated documentation for this change - [x] Documentation is **not needed** for this change (explain why) ### Docs PR URL (required if "docs added" is checked) Paste the PR link from https://github.com/netbirdio/docs here: https://github.com/netbirdio/docs/pull/__ <!-- codesmith:footer --> --- <a href="https://app.blacksmith.sh/netbirdio/codesmith/netbird/pr/6792"><picture><source media="(prefers-color-scheme: dark)" srcset="https://pr-comments-assets.blacksmith.sh/codesmith/view-with-codesmith-dark-v2.svg"><source media="(prefers-color-scheme: light)" srcset="https://pr-comments-assets.blacksmith.sh/codesmith/view-with-codesmith-light-v2.svg"><img alt="View with Codesmith" src="https://pr-comments-assets.blacksmith.sh/codesmith/view-with-codesmith-dark-v2.svg"></picture></a> <a href="https://backend.blacksmith.sh/track/enable-autofix?expires=1786751882&installation_id=146802194&pr_number=6792&repository=netbirdio%2Fnetbird&return_to=https%3A%2F%2Fgithub.com%2Fnetbirdio%2Fnetbird%2Fpull%2F6792&signature=ec2ab7af85ac2f0cc42442c7bdb43e918e13bf14ec724aafbf61900356fba980"><picture><source media="(prefers-color-scheme: dark)" srcset="https://pr-comments-assets.blacksmith.sh/codesmith/autofix-with-codesmith-dark.svg"><source media="(prefers-color-scheme: light)" srcset="https://pr-comments-assets.blacksmith.sh/codesmith/autofix-with-codesmith-light.svg"><img alt="Autofix with Codesmith" src="https://pr-comments-assets.blacksmith.sh/codesmith/autofix-with-codesmith-dark.svg"></picture></a> <sup>Need help on this PR? Tag <code>/codesmith</code> with what you need. Autofix is disabled.</sup> <!-- codesmith:autofix:disabled --> <!-- /codesmith:footer --> <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **New Features** - Added opt-in native gRPC multiplexing on the main HTTPS endpoint (controlled by `NB_MGMT_NATIVE_GRPC`). - **Improvements** - Enhanced TLS/client protocol steering to keep dual-protocol clients on HTTP/1.1 while gRPC uses HTTP/2. - Refined token refresh scheduling for time-based authentication. - Updated management job delivery to use per-peer streams for more reliable reconnect behavior. - **Tests** - Added coverage for multiplexed gRPC/HTTP routing and for scheduler timing, worker skipping, and cancellation semantics. - Added stream-based job manager lifecycle tests. <!-- end of auto-generated comment: release notes by coderabbit.ai --> --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>
saavagebueno added the pull-request label 2026-08-05 07:08:10 -04:00
Sign in to join this conversation.
No Label pull-request
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: DYNR/netbird#27080