MySQL使用变量实现部分分组聚合

用户1148526

发布于 2022-05-07 12:48:21

6630

发布于 2022-05-07 12:48:21

文章被收录于专栏：Hadoop数据仓库

在实际应用中经常有这种需求，按照select的字段中的部分字段分组聚合，比如下面的例子：

create table t1 (a varchar(20),b varchar(20),c int);
insert into t1 values('a','123',1);
insert into t1 values('b','456',2);
insert into t1 values('a','789',3);
insert into t1 values('b','123',4);
commit;

要求按字段a分组求和，查询a,b和聚合值，并按a、b排序。在Oracle里可以使用分析函数，写法如下：

select a,b,sum(c) over (partition by a) sum_c from t1 order by a,b;

查询结果和执行计划如图1所示：

图1

在MySQL中可以用两种方式实现：

-- 写法一，使用内连接，两次扫表
select t1.a, t1.b, t2.sum_c
  from t1,
       (select a, sum(c) sum_c
          from t1
         group by a) t2
 where t1.a = t2.a
 order by t1.a , t1.b;

查询结果和执行计划如图2所示：

图2

-- 写法二，使用变量，内层与外层相反排序
select a, b, if(@a = a1, @c:=@c, @c:=sum_c) sum_c, @a:=a1
  from (select a, b, if(@a = a, @c:=@c + c, @c:=c) sum_c, @a:=a a1
        from t1, 
		     (select @a:='', @c:=0) t2
       order by a, b desc) t
 order by a , b;

查询结果和执行计划如图3所示：