Using INTO in COLLECT while also using AGGREGATE. Can it be done? - arangodb

Question

The shape of the documents I'm using look similar to this:
{
"survey_id": 123,
"numerical_answers": [
{
"component_id": 345,
"number": 5,
},
{
"component_id": 346,
"number": 5,
}
]
}
The output of ArangoDB should look like this:
[
{
"component_id": 345,
"distribution": [
{
"component_id": 345,
"score": null,
"total": 42
},
{
"component_id": 345,
"score": 1,
"total": 76
},
{
"component_id": 345,
"score": 2,
"total": 37
},
{
"component_id": 345,
"score": 3,
"total": 40
},
{
"component_id": 345,
"score": 4,
"total": 93
},
{
"component_id": 345,
"score": 5,
"total": 212
}
],
"total": 500,
"avg": 3.404
}
]
My AQL looks like this:
FOR doc IN ##collection
FILTER doc.`survey_id` == #survey_id
LET componentScoreGroup = {
component_id: #component_id,
score: FIRST(doc.numerical_answers[* filter CURRENT.component_id == #component_id return CURRENT.number])
}
COLLECT component_id = componentScoreGroup.component_id, score = componentScoreGroup.score WITH COUNT INTO total
LET distribution = {
component_id: component_id,
score: score,
total: total
}
COLLECT component_id2 = distribution.component_id into groups keep distribution
LET finalTotal = sum(groups[*].distribution.total),
summedScore = sum(groups[* return CURRENT.distribution.score * CURRENT.distribution.total])
RETURN {
component_id: component_id2,
distribution: groups[*].distribution,
total: finalTotal,
avg: summedScore / finalTotal
}
I would like to use the more performant AGGREGATE syntax in the COLLECT but I'm not sure I can because I also want to pull in the distribution content in the final return.
Any help would be greatly appreciated!

Answer 1

is this closer to what you are looking for?
FOR doc IN ##collection
FILTER doc.survey_id == #survey_id
LET componentScoreGroup = {
component_id: #component_id,
score: FIRST(doc.numerical_answers[* filter CURRENT.component_id == #component_id return CURRENT.number])
}
COLLECT id = componentScoreGroup.component_id
AGGREGATE avg = AVG(componentScoreGroup.score),
total = SUM(componentScoreGroup.score)
INTO group
return { "id" : id , "avg" : avg , "total" : total, "number" : LENGTH(group), "group" : group}