From 74a18506d1fa2ffa7139363dd444ce92be9e2c4d Mon Sep 17 00:00:00 2001
From: iotmani <1248411+iotmani@users.noreply.github.com>
Date: Tue, 19 Jul 2022 11:58:30 -0700
Subject: [PATCH] [AZI-62] Make partition count configurable in log forwarder
 automation

---
 azure/deploy-to-azure/event_hub.json       | 11 ++++++++++-
 azure/deploy-to-azure/parent_template.json | 10 ++++++++++
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/azure/deploy-to-azure/event_hub.json b/azure/deploy-to-azure/event_hub.json
index ed462f7bb..552694d3e 100644
--- a/azure/deploy-to-azure/event_hub.json
+++ b/azure/deploy-to-azure/event_hub.json
@@ -15,6 +15,13 @@
         "description": "Name of Event Hub"
       }
     },
+    "partitionCount": {
+      "type": "int",
+      "defaultValue": 32,
+      "metadata": {
+        "description": "The number of event hub partitions"
+      }
+    },
     "location": {
       "type": "string",
       "defaultValue": "[resourceGroup().location]",
@@ -44,7 +51,9 @@
           "dependsOn": [
             "[resourceId('Microsoft.EventHub/namespaces/', parameters('eventHubNamespace'))]"
           ],
-          "properties": {}
+          "properties": {
+            "partitionCount": "[parameters('partitionCount')]"
+          }
         }
       ]
     }
diff --git a/azure/deploy-to-azure/parent_template.json b/azure/deploy-to-azure/parent_template.json
index 3f32f4e69..0331b4dc1 100644
--- a/azure/deploy-to-azure/parent_template.json
+++ b/azure/deploy-to-azure/parent_template.json
@@ -36,6 +36,13 @@
         "description": "Name of Event Hub"
       }
     },
+    "partitionCount": {
+      "type": "int",
+      "defaultValue": 32,
+      "metadata": {
+        "description": "The number of event hub partitions"
+      }
+    },
     "functionAppName": {
       "type": "string",
       "defaultValue": "[concat('datadog-functionapp-', newGuid())]",
@@ -95,6 +102,9 @@
           "eventHubName": {
             "value": "[parameters('eventHubName')]"
           },
+          "partitionCount": {
+            "value": "[parameters('partitionCount')]"
+          },
           "location": {
             "value": "[parameters('resourcesLocation')]"
           }