Блог

У меня есть SQL-запрос, встроенный в систему, которая до сих пор успешно работала в Athena с движком версии 1. Однако в версии 2 движка это не удается, и я не смог понять, почему.

Вот обобщенная версия SQL. Он суммирует количество людей в 3 группах: взрослые, жители Нью-Йорка и их совпадение. (Взрослые Нью-Йорка).

В версии 1 это работает, но в версии 2 я получаю ошибку «столбец z.id_field не может быть разрешен».

 WITH BASE AS (SELECT person_id, age, state
              FROM people 
              WHERE gender  = 'male'
                 )

    ,group_a as (
SELECT distinct (person_id) as id_field
FROM BASE
WHERE age > 17
),
  
  group_b as (
SELECT distinct (person_id) as id_field
FROM BASE
WHERE state = 'NY'
)


SELECT CASE WHEN z.id_field is null then 'group_b_only' WHEN r.id_field is null then 'group_a_only' ELSE 'Overlap' END as group
      , COUNT (coalesce (z.id_field, r.id_field)) as count
FROM group_a AS z FULL OUTER JOIN group_b as r USING (id_field)
GROUP BY 1;
 

В качестве примечания, в любой базе данных это было бы проще в виде агрегирования и, вероятно, также быстрее:

 SELECT grp, COUNT(*)
FROM (SELECT person_id,
             (CASE WHEN MAX(age) > 17 AND MAX(state) = 'NY' THEN 'Both'
                   WHEN MAX(age) > 17 THEN 'Age Only'
                   ELSE 'State Only'
              END) as grp
      FROM people 
      WHERE gender  = 'male' AND
            (age > 17 OR state = 'NY')
      GROUP BY person_id
     ) x
GROUP BY grp;
 

Вышесказанное предполагает, что person_id это может быть повторено в people . Если это не так, то это можно упростить до:

 SELECT (CASE WHEN age > 17 AND state = 'NY' THEN 'Both'
             WHEN age > 17 THEN 'Age Only'
             ELSE 'State Only'
         END) as grp, COUNT(*)
FROM people 
WHERE gender  = 'male' AND
      (age > 17 OR state = 'NY')
GROUP BY grp;