Search code examples
javascriptarraysobject

Javascript group nested object array into another array


I have an array that is used to track status of certain documents. This looks like the following when outputted from a Mongo Aggregate function. k is the status, and v is the count of documents with this status.

[
    {
        "legal": [
            { "k": "", "v": 6 },
            { "v": 3},
            { "k": "To be uploaded", "v": 5 },
            { "k": "To be reviewed", "v": 96 }
        ],
        "operations": [
            { "v": 1 },
            { "k": "To be uploaded", "v": 3 },
            { "k": "To be reviewed", "v": 24 }
        ],
        "accounting": [
            { "k": "To be reviewed", "v": 137 },
            { "k": "", "v": 3 },
            { "v": 2 },
            { "k": "To be uploaded", "v": 24 },
            { "k": "Not Required", "v": 1 }
        ]
    }
]

I want to graph these on a stacked bar chart, and require the following format. The data array is in the order ["legal", "operations", "accounting"]

 [
    {
      name: "",  // no status specified
      data: [9, 1, 5]
    },
    {
      name: "To be uploaded",
      data: [5, 3, 24]
    },
    {
      name: "To be reviewed",
      data: [96, 24, 137]
    },
    {
      name: "Not Required",
      data: [0, 0, 1]
    },
  ]

Solution

  • One approach would be to use a combination of Array#map and Array#reduce methods as follows. Array#reduce would be great at creating a summary of and cleaning up the data object, while Array#map would help with the final presentation. It can be improved upon, of course:

    const input = [ { "legal": [ { "k": "", "v": 6 }, { "v": 3}, { "k": "To be uploaded", "v": 5 }, { "k": "To be reviewed", "v": 96 } ], "operations": [ { "v": 1 }, { "k": "To be uploaded", "v": 3 }, { "k": "To be reviewed", "v": 24 } ], "accounting": [ { "k": "To be reviewed", "v": 137 }, { "k": "", "v": 3 }, { "v": 2 }, { "k": "To be uploaded", "v": 24 }, { "k": "Not Required", "v": 1 } ] } ],
    
        summary = input.map(
          ({legal,operations,accounting}) =>
          ({
              legal:legal.reduce((acc,{k,v}) => ({...acc,[k || "NONE"]:(acc[k || "NONE"] || 0) + v}),{"Not Required":0}),
              operations:operations.reduce((acc,{k,v}) => ({...acc,[k || "NONE"]:(acc[k || "NONE"] || 0) + v}),{"Not Required":0}),
              accounting:accounting.reduce((acc,{k,v}) => ({...acc,[k || "NONE"]:(acc[k || "NONE"] || 0) + v}),{"Not Required":0})
          })
        ),
        
        output = summary
        .flatMap(
          o => Object.keys(Object.values(o)[0])
          .map(
            z => ({
              name: z === "NONE" ? "" : z,
              data: Object.keys(o).map(y => o[y][z])
            }) 
          ) 
        );
        
    console.log( output );

    Alternatively .....

    You can create summary dynamically as follows:

    const input = [ { "legal": [ { "k": "", "v": 6 }, { "v": 3}, { "k": "To be uploaded", "v": 5 }, { "k": "To be reviewed", "v": 96 } ], "operations": [ { "v": 1 }, { "k": "To be uploaded", "v": 3 }, { "k": "To be reviewed", "v": 24 } ], "accounting": [ { "k": "To be reviewed", "v": 137 }, { "k": "", "v": 3 }, { "v": 2 }, { "k": "To be uploaded", "v": 24 }, { "k": "Not Required", "v": 1 } ] } ],
    
        summary = input.flatMap(
          o =>
          Object.keys(o)
          .reduce(
            (ac,cat) => 
            ({
              ...ac,
              [cat]:o[cat].reduce((acc,{k,v}) => ({...acc,[k || "NONE"]:(acc[k || "NONE"] || 0) + v}),{"Not Required":0})
            }),{}
          )
        ),
        
        output = summary
        .flatMap(
          o => Object.keys(Object.values(o)[0])
          .map(
            z => ({
              name: z === "NONE" ? "" : z,
              data: Object.keys(o).map(y => o[y][z])
            }) 
          )
        );
        
    console.log( output );