API ServiceBusErrorSource to represent source of error

sdk/servicebus/azure-messaging-servicebus/src/main/java/com/azure/messaging/servicebus/ServiceBusErrorSource.java

@@ -0,0 +1,17 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.azure.messaging.servicebus;
+
+/**
+ * Represent the scenario in which user was when the error happened.
+ */
+public enum ServiceBusErrorSource {
+    SEND,
+    RECEIVE,
+    APPEND,
+    COMPLETE,
+    DEFER,
+    DEAD_LETTER,
+    PEEK;
+}

sdk/servicebus/azure-messaging-servicebus/src/main/java/com/azure/messaging/servicebus/ServiceBusReceivedMessageContext.java

@@ -11,6 +11,7 @@ public class ServiceBusReceivedMessageContext {
     private final ServiceBusReceivedMessage message;
     private final String sessionId;
     private final Throwable error;
+    private ServiceBusErrorSource errorSource;
 
     /**
      * Creates an instance where a message was successfully received.
@@ -35,6 +36,31 @@ public class ServiceBusReceivedMessageContext {
         this.message = null;
     }
 
+    /**
+     * Creates an instance where an error occurred such as session-lock-lost.
+     *
+     * @param sessionId Session id that the error occurred in.
+     * @param error AMQP exception that occurred in session.
+     * @param errorSource the source of the error.
+     */
+    ServiceBusReceivedMessageContext(String sessionId, Throwable error, ServiceBusErrorSource errorSource) {
+        this(sessionId,error);
+        this.errorSource = Objects.requireNonNull(errorSource, "'errorSource' cannot be null.");
+    }
+
+    /**
+     * Creates an instance where an error occurred such as session-lock-lost.
+     *
+     * @param error AMQP exception that occurred in session.
+     * @param errorSource the source of the error.
+     */
+    ServiceBusReceivedMessageContext(Throwable error, ServiceBusErrorSource errorSource) {
+        this.error = Objects.requireNonNull(error, "'error' cannot be null.");
+        this.errorSource = Objects.requireNonNull(errorSource, "'errorSource' cannot be null.");
+        this.message = null;
+        this.sessionId = null;
+    }
+
     /**
      * Gets the session id of the message or that the error occurred in.
      *
@@ -53,6 +79,27 @@ public Throwable getThrowable() {
         return error;
     }
 
+    /**
+     * Gets the {@link ServiceBusErrorSource} of the error where it occurred.
+     *
+     * @return The {@link ServiceBusErrorSource} of the error or message.
+     */
+    public ServiceBusErrorSource getErrorSource() {
+        return errorSource;
+    }
+
+    /**
+     * Gets the {@link ServiceBusErrorSource} of the error where it occurred.
+     *
+     * @param errorSource {@link ServiceBusErrorSource} where error occurred.
+     *
+     * @return The updated {@link ServiceBusReceivedMessageContext} of the error.
+     */
+    ServiceBusReceivedMessageContext setErrorSource(ServiceBusErrorSource errorSource) {
+        this.errorSource = errorSource;
+        return this;
+    }
+
     /**
      * Gets the message received from Service Bus.
      *