feat: 小计/总计功能，支持按维度分组汇总 #2337

packages/s2-core/__tests__/bugs/issue-1715-spec.ts

@@ -81,9 +81,10 @@ describe('Multi Values GrandTotal Height Test', () => {
 
     const grandTotalsNode = s2
       .getColumnNodes()
-      .find((node) => node.isGrandTotals);
+      .find((node) => node.isGrandTotals && node.isTotalRoot);
 
-    expect(s2.facet.layoutResult.colsHierarchy.height).toBe(60);
-    expect(grandTotalsNode.height).toEqual(30);
+    // 有多个 Value 时不允许隐藏度量列
+    expect(s2.facet.layoutResult.colsHierarchy.height).toBe(90);
+    expect(grandTotalsNode.height).toEqual(60);
   });
 });

packages/s2-core/__tests__/spreadsheet/__snapshots__/corner-spec.ts.snap

@@ -18,6 +18,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "province",
     "label": "province",
@@ -47,6 +48,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "city",
     "label": "city",
@@ -81,6 +83,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "series-number-node",
     "label": "序号",
@@ -110,6 +113,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "province",
     "label": "province",
@@ -139,6 +143,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "city",
     "label": "city",
@@ -173,6 +178,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "",
     "label": "province/city/数值",
@@ -207,6 +213,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "series-number-node",
     "label": "序号",
@@ -236,6 +243,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "",
     "label": "province/city/数值",
@@ -270,6 +278,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "province",
     "label": "province",
@@ -299,6 +308,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "city",
     "label": "city",
@@ -333,6 +343,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "series-number-node",
     "label": "序号",
@@ -362,6 +373,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "province",
     "label": "province",
@@ -391,6 +403,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "city",
     "label": "city",
@@ -425,6 +438,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "",
     "label": "province/city",
@@ -459,6 +473,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "series-number-node",
     "label": "序号",
@@ -488,6 +503,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "",
     "label": "province/city",
@@ -522,6 +538,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "province",
     "label": "province",
@@ -551,6 +568,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "city",
     "label": "city",
@@ -585,6 +603,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "series-number-node",
     "label": "序号",
@@ -614,6 +633,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "province",
     "label": "province",
@@ -643,6 +663,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "city",
     "label": "city",
@@ -677,6 +698,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "",
     "label": "province/city",
@@ -711,6 +733,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "series-number-node",
     "label": "序号",
@@ -740,6 +763,7 @@ Array [
     "isPivotMode": true,
     "isSubTotals": undefined,
     "isTotalMeasure": undefined,
+    "isTotalRoot": undefined,
     "isTotals": undefined,
     "key": "",
     "label": "province/city",

packages/s2-core/__tests__/spreadsheet/corner-spec.ts

@@ -49,6 +49,7 @@ describe('PivotSheet Corner Tests', () => {
         fields: {
           ...simpleDataConfig.fields,
           columns: [],
+          values: ['price'],
         },
       });
       s2.setOptions({
@@ -88,6 +89,13 @@ describe('PivotSheet Corner Tests', () => {
           },
         },
       });
+      s2.setDataCfg({
+        ...simpleDataConfig,
+        fields: {
+          ...simpleDataConfig.fields,
+          values: ['price'],
+        },
+      });
       s2.render();
 
       const cornerNodes = s2.facet.getCornerNodes();

packages/s2-core/__tests__/spreadsheet/sort-by-order-spec.ts

@@ -104,4 +104,15 @@ describe('Manual Sort Tests', () => {
       }),
     ).toEqual(['整体访问', '小程序访问', '支付宝访问']);
   });
+
+  test('getTotalDimensionValues should include correct values', () => {
+    const sortedType1 = s2.dataSet.getTotalDimensionValues('type1', {});
+    expect(sortedType1).toEqual(['整体访问', '小程序访问', '支付宝访问']);
+
+    expect(s2.dataSet.getTotalDimensionValues('type2', {})).toEqual([
+      '整体访问',
+      '小程序访问',
+      '支付宝访问',
+    ]);
+  });
 });

packages/s2-core/__tests__/unit/cell/header-cell-spec.ts

@@ -54,21 +54,21 @@
     });
 
     test('should not format pivot col and row total cell', () => {
       const colNode = new Node({
         id: `root[&]总计`,
         key: '',
         value: '总计',
         parent: root,
         label: '总计',
-        isTotals: true,
+        isTotalRoot: true,
       });
       const rowNode = new Node({
         id: `root[&]杭州[&]小计`,
         key: '',
         value: '小计',
         parent: root,
         label: '小计',
-        isTotals: true,
+        isTotalRoot: true,
       });
 
       const formatter: Formatter = (value) => {

packages/s2-core/__tests__/unit/data-set/__snapshots__/pivot-data-set-total-spec.ts.snap

@@ -0,0 +1,121 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`Pivot Dataset Total Test test for total with dimension group get correct MultiData when query need to be processed 1`] = `
+Array [
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 7789,
+    "city": "杭州市",
+    "number": 7789,
+    "province": "浙江省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 2367,
+    "city": "绍兴市",
+    "number": 2367,
+    "province": "浙江省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 3877,
+    "city": "宁波市",
+    "number": 3877,
+    "province": "浙江省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 4342,
+    "city": "舟山市",
+    "number": 4342,
+    "province": "浙江省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+]
+`;
+
+exports[`Pivot Dataset Total Test test for total with dimension group get correct MultiData when query need to be processed 2`] = `Array []`;
+
+exports[`Pivot Dataset Total Test test for total with dimension group get correct MultiData when query need to be processed 3`] = `
+Array [
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 7789,
+    "city": "杭州市",
+    "number": 7789,
+    "province": "浙江省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 2367,
+    "city": "绍兴市",
+    "number": 2367,
+    "province": "浙江省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 3877,
+    "city": "宁波市",
+    "number": 3877,
+    "province": "浙江省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 4342,
+    "city": "舟山市",
+    "number": 4342,
+    "province": "浙江省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 1723,
+    "city": "成都市",
+    "number": 1723,
+    "province": "四川省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 1822,
+    "city": "绵阳市",
+    "number": 1822,
+    "province": "四川省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 1943,
+    "city": "南充市",
+    "number": 1943,
+    "province": "四川省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+  Object {
+    "$$extra$$": "number",
+    "$$value$$": 2330,
+    "city": "乐山市",
+    "number": 2330,
+    "province": "四川省",
+    "sub_type": "桌子",
+    "type": "家具",
+  },
+]
+`;