(appengine) Edit rooms queries for exandra

Signed-off-by: Eddy Babetto <eddy.babetto@secomind.com>

Author: eddbbt

apps/astarte_appengine_api/lib/astarte_appengine_api/groups/queries.ex

@@ -150,21 +150,10 @@ defmodule Astarte.AppEngine.API.Groups.Queries do
     with true <- Map.has_key?(result, group_name) do
       {:ok, true}
     else
-      {:error, :invalid_device_id} ->
-        {:error, :device_not_found}
-
-      {:error, reason} ->
-        _ = Logger.error("Database error: #{inspect(reason)}.", tag: "db_error")
-        {:error, :database_error}
-
-      [] ->
-        {:ok, false}
-
       false ->
         {:ok, false}
 
-      smth ->
-        dbg(smth)
+      _ ->
         {:error, :device_not_found}
     end
   end
@@ -311,24 +300,16 @@ defmodule Astarte.AppEngine.API.Groups.Queries do
     keyspace = DataAccessRealm.keyspace_name(realm_name)
 
     query =
-      from(d in GroupedDevice,
+      from d in GroupedDevice,
+        hints: ["PER PARTITION LIMIT 1"],
         prefix: ^keyspace,
         where: d.group_name == ^group_name,
-        select: d.group_name
-      )
-
-    with false <- Repo.all(query) |> Enum.empty?() do
-      true
-    else
-      {:error, reason} ->
-        _ = Logger.error("Database error: #{inspect(reason)}.", tag: "db_error")
-        false
-
-      [] ->
-        false
+        select: d.group_name,
+        limit: 1
 
-      true ->
-        false
+    case Repo.fetch_one(query) do
+      {:ok, _} -> true
+      _not_found -> false
     end
   end
 
@@ -478,12 +459,6 @@ defmodule Astarte.AppEngine.API.Groups.Queries do
 
   def add_to_grouped_device(realm_name, group_name, decoded_device_ids) do
     keyspace = DataAccessRealm.keyspace_name(realm_name)
-    grouped_device_table = GroupedDevice.__schema__(:source)
-
-    insert_grouped_device_sql = """
-      INSERT INTO #{keyspace}.#{grouped_device_table} (group_name, insertion_uuid, device_id)
-      values (?, ?, ?)
-    """
 
     queries =
       decoded_device_ids
@@ -499,8 +474,14 @@ defmodule Astarte.AppEngine.API.Groups.Queries do
         update_device_groups =
           Repo.to_sql(:update_all, query)
 
-        insert_grouped_device_params = [group_name, insertion_uuid, device_id]
-        insert_grouped_device = {insert_grouped_device_sql, insert_grouped_device_params}
+        grouped_device =
+          %GroupedDevice{
+            group_name: group_name,
+            insertion_uuid: insertion_uuid,
+            device_id: device_id
+          }
+
+        insert_grouped_device = Repo.insert_to_sql(grouped_device, prefix: keyspace)
 
         [update_device_groups, insert_grouped_device]
       end)

apps/astarte_appengine_api/lib/astarte_appengine_api/rooms/room.ex

@@ -22,6 +22,7 @@ defmodule Astarte.AppEngine.API.Rooms.Room do
   alias Astarte.AppEngine.API.Device.DevicesList
   alias Astarte.AppEngine.API.Groups
   alias Astarte.AppEngine.API.Rooms.WatchRequest
+  alias Astarte.AppEngine.API.Rooms.Queries
   alias Astarte.AppEngine.API.RPC.DataUpdaterPlant
   alias Astarte.AppEngine.API.RPC.DataUpdaterPlant.VolatileTrigger
   alias Astarte.AppEngine.API.Utils
@@ -32,7 +33,6 @@ defmodule Astarte.AppEngine.API.Rooms.Room do
   alias Astarte.Core.Triggers.SimpleTriggersProtobuf.SimpleTriggerContainer
   alias Astarte.Core.Triggers.SimpleTriggersProtobuf.TaggedSimpleTrigger
   alias Astarte.Core.Triggers.SimpleTriggersProtobuf.TriggerTargetContainer
-  alias Astarte.AppEngine.API.Rooms.Queries
 
   require Logger