#6158

src/upgrades/1.7.3/topic_votes.js

@@ -0,0 +1,60 @@
+'use strict';
+
+var async = require('async');
+var batch = require('../../batch');
+var db = require('../../database');
+
+module.exports = {
+	name: 'Add votes to topics',
+	timestamp: Date.UTC(2017, 11, 8),
+	method: function (callback) {
+		var progress = this.progress;
+
+		batch.processSortedSet('topics:tid', function (tids, next) {
+			async.eachLimit(tids, 500, function (tid, _next) {
+				progress.incr();
+				var topicData;
+				async.waterfall([
+					function (next) {
+						db.getObjectFields('topic:' + tid, ['mainPid', 'cid'], next);
+					},
+					function (_topicData, next) {
+						topicData = _topicData;
+						if (!topicData.mainPid || !topicData.cid) {
+							return _next();
+						}
+						db.getObject('post:' + topicData.mainPid, next);
+					},
+					function (postData, next) {
+						if (!postData) {
+							return _next();
+						}
+						var upvotes = parseInt(postData.upvotes, 10) || 0;
+						var downvotes = parseInt(postData.downvotes, 10) || 0;
+						var data = {
+							upvotes: upvotes,
+							downvotes: downvotes,
+						};
+						var votes = upvotes - downvotes;
+						async.parallel([
+							function (next) {
+								db.setObject('topic:' + tid, data, next);
+							},
+							function (next) {
+								db.sortedSetAdd('topics:votes', votes, tid, next);
+							},
+							function (next) {
+								db.sortedSetAdd('cid:' + topicData.cid + ':tids:votes', votes, tid, next);
+							},
+						], function (err) {
+							next(err);
+						});
+					},
+				], _next);
+			}, next);
+		}, {
+			progress: progress,
+			batch: 500,
+		}, callback);
+	},
+};