add missing attribute: partitionKey, viaPartitionKey

Author: kostat

services/azure-servicebus/src/main/java/com/microsoft/windowsazure/services/servicebus/implementation/BrokerProperties.java

@@ -36,6 +36,12 @@ public class BrokerProperties {
     @JsonProperty("SessionId")
     private String sessionId;
 
+    @JsonProperty("PartitionKey")
+    private String partitionKey;
+
+    @JsonProperty("ViaPartitionKey")
+    private String viaPartitionKey;
+
     @JsonProperty("DeliveryCount")
     private Integer deliveryCount;
 
@@ -147,6 +153,24 @@ public void setSessionId(String sessionId) {
         this.sessionId = sessionId;
     }
 
+    @JsonIgnore
+    public String getPartitionKey() {
+        return partitionKey;
+    }
+
+    public void setPartitionKey(String partitionKey) {
+        this.partitionKey = partitionKey;
+    }
+
+    @JsonIgnore
+    public String getViaPartitionKey() {
+        return partitionKey;
+    }
+
+    public void setViaPartitionKey(String viaPartitionKey) {
+        this.viaPartitionKey = viaPartitionKey;
+    }
+
     @JsonIgnore
     public String getLabel() {
         return label;

services/azure-servicebus/src/main/java/com/microsoft/windowsazure/services/servicebus/models/BrokeredMessage.java

@@ -349,6 +349,54 @@ public BrokeredMessage setSessionId(String sessionId) {
         return this;
     }
 
+    /**
+     * Returns the partition key for the message.
+     * 
+     * @return A <code>String</code> object that represents the partition key.
+     * 
+     */
+    public String getPartitionKey() {
+        return brokerProperties.getPartitionKey();
+    }
+
+    /**
+     * Sets the partition key for the message.
+     * 
+     * @param partitionKey
+     *            A <code>String</code> object that represents the partition key.
+     * 
+     * @return A <code>Message</code> object that represents the updated
+     *         message.
+     */
+    public BrokeredMessage setPartitionKey(String partitionKey) {
+        brokerProperties.setPartitionKey(partitionKey);
+        return this;
+    }
+
+    /**
+     * Returns the via partition key for the message.
+     * 
+     * @return A <code>String</code> object that represents the via partition key.
+     * 
+     */
+    public String getViaPartitionKey() {
+        return brokerProperties.getViaPartitionKey();
+    }
+
+    /**
+     * Sets the via partition key for the message.
+     * 
+     * @param viaPartitionKey
+     *            A <code>String</code> object that represents the via partition key.
+     * 
+     * @return A <code>Message</code> object that represents the updated
+     *         message.
+     */
+    public BrokeredMessage setViaPartitionKey(String viaPartitionKey) {
+        brokerProperties.setViaPartitionKey(viaPartitionKey);
+        return this;
+    }
+
     /**
      * Returns the label of the message.
      *