AstarteImport: Add support for db_host_port argument

Updated Astarte.Import to accept db_host_and_port as the argument

Updated CLI.main/3 to receive and handle db_host_and_port

Ensured db_host_and_port is properly passed to populate/4 instead of relying on Application.get_env/1

Improved flexibility in database configuration by allowing direct host and port specification

Added @db_host_and_port to test cases, including "Test import into Cassandra database

Signed-off-by: Jusuf <jusuf.avdic@secomind.com>

Author: Jusuf95

tools/astarte_import/README.md

@@ -11,7 +11,7 @@ docker run -e CASSANDRA_DB_HOST=127.0.0.1 -e CASSANDRA_DB_PORT=9042 \
 Command to run data import:
 
 ```bash
-mix astarte.import <realm> <xml file>
+mix astarte.import <db_host_and_port> <realm> <xml file>
 ```
 
 ```xml

tools/astarte_import/lib/astarte/import/cli.ex

@@ -22,7 +22,7 @@ defmodule Astarte.Import.CLI do
 
   @chunk_size 4096
 
-  def main([realm, file_name]) do
+  def main([db_host_and_port, realm, file_name]) do
     with {:started, {:ok, _}} <- {:started, Application.ensure_all_started(:astarte_import)},
          true <- String.valid?(realm),
          true <- String.valid?(file_name),
@@ -38,7 +38,7 @@ defmodule Astarte.Import.CLI do
       end
 
       # Call the populate function with the necessary arguments
-      PopulateDB.populate(realm, data, more_data)
+      PopulateDB.populate(db_host_and_port, realm, data, more_data)
     else
       error ->
         Logger.error("Failed to start import: #{inspect(error)}")

tools/astarte_import/lib/astarte/import/populatedb.ex

@@ -41,11 +41,10 @@ defmodule Astarte.Import.PopulateDB do
     ]
   end
 
-  def populate(realm, xml, continuation_fun \\ :undefined) do
+  def populate(db_host_and_port, realm, xml, continuation_fun \\ :undefined) do
     Logger.info("Import started.", realm: realm)
 
-    nodes = Application.get_env(:cqerl, :cassandra_nodes)
-    {host, port} = Enum.random(nodes)
+    [host, port] = String.split(db_host_and_port, ":")
     Logger.info("Connecting to #{host}:#{port} cassandra database.", realm: realm)
 
     {:ok, xandra_conn} = Xandra.start_link(nodes: ["#{host}:#{port}"])

tools/astarte_import/lib/mix/tasks/astarte_import.ex

@@ -11,7 +11,7 @@ defmodule Mix.Tasks.Astarte.Import do
 
     # Process the arguments
     case args do
-      [realm, file_name] ->
+      [db_host_and_port, realm, file_name] ->
         Config.validate!()
 
         xandra_options = Config.xandra_options!()
@@ -28,7 +28,7 @@ defmodule Mix.Tasks.Astarte.Import do
         opts = [strategy: :one_for_one, name: AstarteImport.Supervisor]
         Supervisor.start_link(children, opts)
 
-        CLI.main([realm, file_name])
+        CLI.main([db_host_and_port, realm, file_name])
 
       _ ->
         Logger.info("Usage: mix astarte.import <realm> <file_name>")

tools/astarte_import/test/astarte/populatedb_test.exs

@@ -21,6 +21,7 @@ defmodule Astarte.PopulateDBTest do
   alias Astarte.Import.PopulateDB
 
   @realm "test"
+  @db_host_and_port "localhost:9042"
 
   @xml """
   <?xml version="1.0" encoding="UTF-8"?>
@@ -77,6 +78,6 @@ defmodule Astarte.PopulateDBTest do
   """
 
   test "Test import into Cassandra database" do
-    PopulateDB.populate(@realm, @xml)
+    PopulateDB.populate(@db_host_and_port, @realm, @xml)
   end
 end