[SQL] MySQL - SET 과 서브쿼리 사용하기

🔊 본 포스팅에서 사용되는 테이블의 자료와 출처는 프로그래머스임을 밝힙니다. 더 다양한 SQL 문제를 풀어보시려면 프로그래머스 사이트를 방문해보세요! 

 

코딩 테스트를 제공하는 프로그래머스 사이트

 

이번 포스팅에서 소개할 SQL 구문은 프로그래머스 SQL 고득점 Kit의 Group by 카테고리의 입양 시각 구하기(2) 문제 풀이에 사용된 구문이다. 문제 풀이에 사용될 테이블과 문제에 대한 자세한 설명은 해당 문제의 링크를 참고하자.

 

문제에서 요구하는 사항은 다음과 같다.

 

보호소에서는 몇 시에 입양이 가장 활발하게 일어나는지 알아보려 합니다. 0시부터 23시까지, 각 시간대별로 입양이 몇 건이나 발생했는지 조회하는 SQL문을 작성해주세요. 이때 결과는 시간대 순으로 정렬해야 합니다.

 

이 때 활용할 방법은 SET 이라는 문법을 이용해 새로운 변수를 할당해주고 서브쿼리를 이용하면 된다. SET이 무엇인지, 왜 사용해야 하는지는 여기를 참고하자.

 

우선은 정답인 전체 SQL 구문을 살펴보고 어떤 의미를 담고 있는지 하나하나씩 뜯어보자.

 

SET @HOUR = -1;
SELECT @HOUR := @HOUR + 1 AS HOUR,
      (SELECT COUNT(*)
       FROM ANIMAL_OUTS
       WHERE @HOUR = HOUR(DATETIME)) AS COUNT
FROM ANIMAL_OUTS
WHERE @HOUR < 23

 

가장 중요하다고 할 수 있는 부분은 SET을 활용해 새로운 칼럼을 만들어주는 것이다. 이렇게 하는 이유는 원본 테이블의 DATETIME이 0시~23시까지 연속적으로 있는 것이 아니라 중간에 몇몇 빠져있는 시간대가 있기 때문이다. 따라서 문제에서 요구하는 연속적인 [0,1,2,3, ..., 23] 시까지의 값을 만들어주어야 했다.

 

다음으로 수행한 구문을 보자.

 

@HOUR := @HOUR + 1

이 구문을 직관적으로 이해하기 위해 Python 문법을 예를 들어 본다면 다음과 같을 수 있겠다.

 

cnt = 0 
for n in range(0,10):
    print(n, end=' ')

 

그래서 만약 다음과 같은 SQL 구문을 실행하게 된다면 ANIMAL_OUTS 라는 테이블의 행 개수(N개)만큼 HOUR 값이 [0, 1, 2, 3, ... , N-1] 로 총 N개의 값이 출력될 것이다.

 

SET @HOUR = -1;
SELECT @HOUR := @HOUR + 1 AS HOUR
FROM ANIMAL_OUTS

 

이제 이렇게 SET을 이용해 새롭게 만든 HOUR이라는 변수와 ANIMAL_OUTS의 DATETIME 변수의 시간(HOUR)과 동일한 값을 갖는 데이터를 COUNT 하게 되면 다음과 같다.

 

SET @HOUR = -1;
SELECT @HOUR := @HOUR + 1 AS HOUR,
      (SELECT COUNT(*)
       FROM ANIMAL_OUTS
       WHERE @HOUR = HOUR(DATETIME))
FROM ANIMAL_OUTS

 

마지막으로 0시 ~ 23시까지의 데이터만 출력해야 하기 때문에 WHERE 조건문을 추가해주자.

 

SET @HOUR = -1;
SELECT @HOUR := @HOUR + 1 AS HOUR,
      (SELECT COUNT(*)
       FROM ANIMAL_OUTS
       WHERE @HOUR = HOUR(DATETIME))
FROM ANIMAL_OUTS
WHERE @HOUR < 23

 

한 가지 주의해야 할 점은 마지막 WHERE 절에서 '@HOUR' 을 사용했다는 점이다. 그냥 'HOUR' 을 사용하면 안될까? 안 된다. 왜냐하면 'HOUR' 이라는 변수는 SELECT 절에서 새롭게 만들어진 새로운 칼럼변수이기 때문이다.(이러한 주의 사항을 기억하고 있으면 다른 문제 풀이에 많은 도움이 된다.)

 

그래서 '@HOUR' 을 사용했기 때문에 '23시 까지'의 값을 출력하기 위해 '@HOUR < 23'이 된다. 왜냐하면 '@HOUR' 은 '@HOUR + 1'로 재할당 되었기 때문이다.

 

@HOUR := @HOUR + 1