@azure/core-amqp 4.1.1-alpha.20240108.2 → 4.2.0

package/dist/index.js

@@ -105,6 +105,7 @@ const Constants = {
     minDurationValue: -922337203685477,
     // https://github.com/Azure/azure-amqp/blob/master/Microsoft.Azure.Amqp/Amqp/AmqpConstants.cs#L47
     maxAbsoluteExpiryTime: new Date("9999-12-31T07:59:59.000Z").getTime(),
+    maxUint32Value: 4294967295,
     aadTokenValidityMarginInMs: 5000,
     connectionReconnectDelay: 300,
     defaultMaxRetries: 3,
@@ -2127,7 +2128,7 @@ const AmqpAnnotatedMessage = {
      * Takes RheaMessage(`Message` type from "rhea") and returns it in the AmqpAnnotatedMessage format.
      */
     fromRheaMessage(msg) {
-        return {
+        const amqpMsg = {
             header: AmqpMessageHeader.fromRheaMessageHeader(msg),
             footer: msg.footer,
             messageAnnotations: msg.message_annotations,
@@ -2136,13 +2137,37 @@ const AmqpAnnotatedMessage = {
             properties: AmqpMessageProperties.fromRheaMessageProperties(msg),
             body: msg.body,
         };
+        if (msg.absolute_expiry_time) {
+            const absoluteExpiryTime = msg.absolute_expiry_time.getTime();
+            amqpMsg.properties.absoluteExpiryTime = Math.min(absoluteExpiryTime, Constants.maxAbsoluteExpiryTime);
+            // The TTL from the header can be at most approximately 49 days (uint32
+            // max value milliseconds) due to the AMQP spec. In order to allow for
+            // larger TTLs set by the user, we take the difference of the
+            // absolute_expiry_time and the creation_time (if both are set). If either of
+            // those properties is not set, we fall back to the TTL from the header.
+            if (msg.creation_time) {
+                amqpMsg.header.timeToLive =
+                    amqpMsg.properties.absoluteExpiryTime - msg.creation_time.getTime();
+            }
+        }
+        return amqpMsg;
     },
     /**
      * Takes AmqpAnnotatedMessage and returns it in the RheaMessage(`Message` type from "rhea") format.
      */
     toRheaMessage(msg) {
-        const message = Object.assign(Object.assign(Object.assign({}, AmqpMessageProperties.toRheaMessageProperties(msg.properties || {})), AmqpMessageHeader.toRheaMessageHeader(msg.header || {})), { body: msg.body, message_annotations: msg.messageAnnotations, delivery_annotations: msg.deliveryAnnotations, application_properties: msg.applicationProperties, footer: msg.footer });
-        return message;
+        var _a, _b;
+        const rhMsg = Object.assign(Object.assign(Object.assign({}, AmqpMessageProperties.toRheaMessageProperties(msg.properties || {})), AmqpMessageHeader.toRheaMessageHeader(msg.header || {})), { body: msg.body, message_annotations: msg.messageAnnotations, delivery_annotations: msg.deliveryAnnotations, application_properties: msg.applicationProperties, footer: msg.footer });
+        // There is a loss of fidelity in the TTL header if larger than uint32 max value. As a workaround
+        // we set the absolute_expiry_time and creation_time on the message based on the TTL. These
+        // values are then used to reconstruct the accurate TTL for received messages.
+        if ((_a = msg.header) === null || _a === void 0 ? void 0 : _a.timeToLive) {
+            const ttl = msg.header.timeToLive;
+            rhMsg.ttl = Math.min(ttl, Constants.maxUint32Value);
+            rhMsg.creation_time = (_b = rhMsg.creation_time) !== null && _b !== void 0 ? _b : new Date();
+            rhMsg.absolute_expiry_time = new Date(Math.min(rhMsg.creation_time.getTime() + ttl, Constants.maxAbsoluteExpiryTime));
+        }
+        return rhMsg;
     },
 };