wake-up-neo.net

Bedingter $ sum in MongoDB

Meine Sammlung in Mongodb ähnelt der folgenden Tabelle in SQL:

Sentimente (Unternehmen, Sentiment)

Nun muss ich eine Abfrage wie folgt ausführen:

SELECT
  Company, 
  SUM(CASE WHEN Sentiment >0 THEN Sentiment ELSE 0 END) AS SumPosSenti, 
  SUM(CASE WHEN Sentiment <0 THEN Sentiment ELSE 0 END) AS SumNegSenti
FROM Sentiments
GROUP BY Company

Was muss ich tun, um diese Abfrage in Mongo zu schreiben? Ich stecke bei der folgenden Abfrage fest:

db.Sentiments.aggregate(
{ $project: {_id:0, Company:1, Sentiment: 1} },
{ $group: {_id: "$Company", SumPosSenti: {$sum: ? }, SumNegSenti: {$sum: ? } } }
);
28
Aafreen Sheikh

Wie von Sammaye vorgeschlagen, müssen Sie dazu den Aggregationsprojektionsoperator $cond verwenden:

db.Sentiments.aggregate(
    { $project: {
        _id: 0,
        Company: 1,
        PosSentiment: {$cond: [{$gt: ['$Sentiment', 0]}, '$Sentiment', 0]},
        NegSentiment: {$cond: [{$lt: ['$Sentiment', 0]}, '$Sentiment', 0]}
    }},
    { $group: {
        _id: "$Company",
        SumPosSentiment: {$sum: '$PosSentiment'},
        SumNegSentiment: {$sum: '$NegSentiment'}
    }});
44
JohnnyHK

Ab Version 3.4 können wir den Operator $switch verwenden, der die Verarbeitung logischer Bedingungen in der Phase $group ermöglicht. Natürlich müssen wir noch den $sum -Akkumulator verwenden, um die Summe zurückzugeben.

db.Sentiments.aggregate(
    [
        { "$group": { 
            "_id": "$Company",  
            "SumPosSenti": { 
                "$sum": { 
                    "$switch": { 
                        "branches": [ 
                            { 
                                "case": { "$gt": [ "$Sentiment", 0 ] }, 
                                "then": "$Sentiment"
                            }
                        ], 
                        "default": 0 
                    }
                }
            }, 
            "SumNegSenti": {
                "$sum": { 
                    "$switch": { 
                        "branches": [ 
                            { 
                                "case": { "$lt": [ "$Sentiment", 0 ] }, 
                                "then": "$Sentiment"
                            }
                        ], 
                        "default": 0 
                    } 
                }
            }
        }}
    ]
)

Wenn Sie Ihre mongod noch nicht auf 3.4 oder neuer migriert haben, beachten Sie, dass die Phase $project in dieser answer redundant ist, da der Operator $cond einen numerischen Wert zurückgibt, der die Möglichkeit gibt, $group Ihre Dokumente und wende $sum auf den $cond-Ausdruck an. 

Dies verbessert die Leistung in Ihrer Anwendung, insbesondere für große Sammlungen.

db.Sentiments.aggregate(
    [
        { '$group': {
            '_id': '$Company',
            'PosSentiment': { 
                '$sum': {
                    '$cond': [
                        { '$gt': ['$Sentiment', 0]}, 
                        '$Sentiment', 
                        0
                    ]
                }
            },
            'NegSentiment': { 
                '$sum': {
                    '$cond': [
                        { '$lt': ['$Sentiment', 0]}, 
                        '$Sentiment', 
                        0
                    ]
                }
            }
        }}
    ]
)

Betrachten Sie eine Sammlung Sentiments mit den folgenden Dokumenten:

{ "Company": "a", "Sentiment" : 2 }
{ "Company": "a", "Sentiment" : 3 }
{ "Company": "a", "Sentiment" : -1 }
{ "Company": "a", "Sentiment" : -5 }

Die Aggregationsabfrage erzeugt Folgendes:

{ "_id" : "a", "SumPosSenti" : 5, "SumNegSenti" : -6 }
26
styvane

Erklären Sie die obigen Ausschnitte, die die Array-Syntax verwenden:

PosSentiment: {$cond: [{$gt: ['$Sentiment', 0]}, '$Sentiment', 0]}

entspricht:

PosSentiment: {$cond: { if: {$gt: ['$Sentiment', 0]}, then: '$Sentiment', else: 0} }

Die Arraysyntax fasst die lange Syntax zu { $cond: [if, then, else] } zusammen. 

1
dpolicastro