From abec4b0fa7dbd3d385ad6fdfa154130b09b8edf0 Mon Sep 17 00:00:00 2001
From: tteckster <tteckster@gmail.com>
Date: Sun, 2 Oct 2022 20:46:25 -0400
Subject: [PATCH] Update influxdb-install.sh

---
 setup/influxdb-install.sh | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/setup/influxdb-install.sh b/setup/influxdb-install.sh
index 3faa99b9..ec96f9d1 100644
--- a/setup/influxdb-install.sh
+++ b/setup/influxdb-install.sh
@@ -81,9 +81,21 @@ wget -qO- https://repos.influxdata.com/influxdb.key | sudo apt-key add - &>/dev/
 echo "deb https://repos.influxdata.com/debian $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/influxdb.list &>/dev/null
 msg_ok "Set up InfluxDB Repository"
 
+read -r -p "Which version of InfluxDB to install? (1 or 2) " prompt
+if [[ $prompt == "2" ]]
+then
+INFLUX="2"
+else
+INFLUX="1"
+fi
+
 msg_info "Installing InfluxDB"
 apt-get update &>/dev/null
+if [[ $INFLUX == "2" ]]; then
+apt-get install -y influxdb2 &>/dev/null
+else
 apt-get install -y influxdb &>/dev/null
+fi
 msg_ok "Installed InfluxDB"
 
 read -r -p "Would you like to add Telegraf? <y/N> " prompt