Expose session state on ServiceBusReceivedMessageContext

[EldertGrootenboer]

Expose session state on ServiceBusReceivedMessageContext

Resolves #49207

Summary

Adds getSessionState() and setSessionState(byte[]) to
ServiceBusReceivedMessageContext, so an application can read and write
session state directly from a ServiceBusSessionProcessorClient message
handler without opening a separate session receiver. This brings the Java
session processor to parity with .NET, whose
ProcessSessionMessageEventArgs already exposes
GetSessionStateAsync / SetSessionStateAsync.

What changed

Area	Change
ServiceBusReceivedMessageContext	New public byte[] getSessionState() and void setSessionState(byte[])
SessionsMessagePump.SessionReceiversTracker	Routes a session-processor call to the message's own session receiver; throws IllegalStateException if that session is no longer held
ServiceBusSessionReactorReceiver	Session-state get/set via the session's management node + link name (mirrors the existing beginLockRenew)
ServiceBusReceiverAsyncClient	Package-private message-scoped getSessionState(message) / setSessionState(message, state) for the non-session/V1 path
ServiceBusSessionAcquirer.Session	getSessionManagement() accessor

The only new public surface is the two methods on
ServiceBusReceivedMessageContext; everything else is package-private.

Behavior

ServiceBusReceivedMessageContext is shared by both session and non-session
processors. On a non-session entity the new methods throw a clear
IllegalStateException (the underlying receiver's existing guard). On a
session entity they target the message's own session, so they are correct
for a processor handling multiple concurrent sessions.

Testing

• Unit — ServiceBusReceivedMessageContextTest (10 tests: both delegation
  branches, get/set, empty→null, and non-session / session-not-active error
  asymmetry pairs) and SessionReceiversTrackerTest (6 tests: no-session /
  untracked → IllegalStateException, delegation, case-insensitive lookup).
• Full module suite — 1006 tests, 0 failures.
• Live — verified end-to-end against a real session-enabled namespace:
  a session processor read null initial state, wrote state, and read it back.
• spotless, checkstyle, and revapi clean.

Notes for reviewers

This adds public API surface, so it needs API review / Architecture Board
sign-off. Opened as a draft pending that review. A usage sample snippet
can follow as a small addition if desired.

[azure-pipelines]

Azure Pipelines:
Successfully started running 1 pipeline(s).
34 pipeline(s) were filtered out due to trigger conditions.
There may be pipelines that require an authorized user to comment /azp run to run.

[Copilot]

Pull request overview

This PR adds synchronous session-state access to ServiceBusReceivedMessageContext, enabling session processors to read and write session state from within the message handler without creating a separate session receiver, addressing parity with .NET and resolving #49207.

Changes:

• Added public getSessionState() / setSessionState(byte[]) to ServiceBusReceivedMessageContext, delegating to either the session processor’s tracked receiver or the underlying receiver client.
• Implemented session-processor routing via SessionsMessagePump.SessionReceiversTracker and session-scoped management-node operations in ServiceBusSessionReactorReceiver.
• Added unit tests covering both delegation paths and error cases, and documented the feature in the module changelog.

Reviewed changes

Copilot reviewed 8 out of 8 changed files in this pull request and generated 4 comments.

Show a summary per file

File	Description
sdk/servicebus/azure-messaging-servicebus/src/main/java/com/azure/messaging/servicebus/ServiceBusReceivedMessageContext.java	Adds the new public session-state APIs and delegates to the appropriate underlying component.
sdk/servicebus/azure-messaging-servicebus/src/main/java/com/azure/messaging/servicebus/SessionsMessagePump.java	Adds session-state routing to the message’s session receiver via SessionReceiversTracker.
sdk/servicebus/azure-messaging-servicebus/src/main/java/com/azure/messaging/servicebus/ServiceBusSessionReactorReceiver.java	Implements session-state get/set using the session’s management node and link name.
sdk/servicebus/azure-messaging-servicebus/src/main/java/com/azure/messaging/servicebus/ServiceBusSessionAcquirer.java	Exposes the session management node accessor used for session-scoped operations.
sdk/servicebus/azure-messaging-servicebus/src/main/java/com/azure/messaging/servicebus/ServiceBusReceiverAsyncClient.java	Adds package-private message-scoped get/set session-state helpers used by the context.
sdk/servicebus/azure-messaging-servicebus/src/test/java/com/azure/messaging/servicebus/ServiceBusReceivedMessageContextTest.java	Adds unit tests validating delegation branches and error propagation behavior.
sdk/servicebus/azure-messaging-servicebus/src/test/java/com/azure/messaging/servicebus/SessionReceiversTrackerTest.java	Adds unit tests for tracker lookup, delegation, and error cases (including case-insensitive session-id lookup).
sdk/servicebus/azure-messaging-servicebus/CHANGELOG.md	Documents the new context APIs.

[Copilot]

Pull request overview

Copilot reviewed 8 out of 8 changed files in this pull request and generated 2 comments.

[Copilot]

Pull request overview

Copilot reviewed 8 out of 8 changed files in this pull request and generated 5 comments.

Comments suppressed due to low confidence (1)

sdk/servicebus/azure-messaging-servicebus/src/main/java/com/azure/messaging/servicebus/ServiceBusReceiverAsyncClient.java:1905

• getSessionState(String) has two issues: (1) empty sessionId isn’t validated (it will be rejected later by the management channel with a different exception), and (2) the tracing span name is mistakenly ServiceBus.setSessionState even though this method calls channel.getSessionState(...). Add an empty-string guard and rename the trace operation to ServiceBus.getSessionState.

    private Mono<byte[]> getSessionState(String sessionId) {
        if (isDisposed.get()) {
            return monoError(LOGGER,
                new IllegalStateException(String.format(INVALID_OPERATION_DISPOSED_RECEIVER, "getSessionState")));
        } else if (!isSessionEnabled) {
            return monoError(LOGGER, new IllegalStateException("Cannot get session state on a non-session receiver."));
        } else if (sessionId == null) {
            return monoError(LOGGER,
                new IllegalStateException("Cannot get session state because the session id is null."));
        }
        assert sessionManager != null; // guaranteed to be non-null when isSessionEnabled is true.
        final String linkName = sessionManager.getLinkName(sessionId);

        return tracer
            .traceMono("ServiceBus.setSessionState",
                connectionProcessor.flatMap(connection -> connection.getManagementNode(entityPath, entityType))
                    .flatMap(channel -> channel.getSessionState(sessionId, linkName)))
            .onErrorMap((err) -> mapError(err, ServiceBusErrorSource.RECEIVE));

[Copilot]

Pull request overview

Copilot reviewed 8 out of 8 changed files in this pull request and generated 3 comments.

[Copilot]

Pull request overview

Copilot reviewed 8 out of 8 changed files in this pull request and generated 2 comments.

[Copilot]

Pull request overview

Copilot reviewed 8 out of 8 changed files in this pull request and generated no new comments.