3 const each
= require('async/each')
4 const map
= require('lodash/map')
5 const waterfall
= require('async/waterfall')
7 const constants
= require('../initializers/constants')
8 const logger
= require('../helpers/logger')
9 const customPodsValidators
= require('../helpers/custom-validators').pods
11 // ---------------------------------------------------------------------------
13 module
.exports = function (sequelize
, DataTypes
) {
14 const Pod
= sequelize
.define('Pod',
17 type: DataTypes
.STRING
,
20 isHost: function (value
) {
21 const res
= customPodsValidators
.isHostValid(value
)
22 if (res
=== false) throw new Error('Host not valid.')
27 type: DataTypes
.STRING(5000),
31 type: DataTypes
.INTEGER
,
32 defaultValue: constants
.FRIEND_SCORE
.BASE
,
36 max: constants
.FRIEND_SCORE
.MAX
40 type: DataTypes
.STRING(400),
64 listRandomPodIdsWithRequest
,
80 // ------------------------------ METHODS ------------------------------
82 function toFormatedJSON () {
88 createdAt: this.createdAt
94 // ------------------------------ Statics ------------------------------
96 function associate (models
) {
97 this.belongsToMany(models
.Request
, {
99 through: models
.RequestToPod
,
104 function countAll (callback
) {
105 return this.count().asCallback(callback
)
108 function incrementScores (ids
, value
, callback
) {
109 if (!callback
) callback = function () {}
112 score: this.sequelize
.literal('score +' + value
)
121 // In this case score is a literal and not an integer so we do not validate it
125 return this.update(update
, options
).asCallback(callback
)
128 function list (callback
) {
129 return this.findAll().asCallback(callback
)
132 function listAllIds (transaction
, callback
) {
134 callback
= transaction
142 if (transaction
) query
.transaction
= transaction
144 return this.findAll(query
).asCallback(function (err
, pods
) {
145 if (err
) return callback(err
)
147 return callback(null, map(pods
, 'id'))
151 function listRandomPodIdsWithRequest (limit
, tableRequestPod
, callback
) {
154 self
.count().asCallback(function (err
, count
) {
155 if (err
) return callback(err
)
158 if (count
=== 0) return callback(null, [])
160 let start
= Math
.floor(Math
.random() * count
) - limit
161 if (start
< 0) start
= 0
164 attributes: [ 'id' ],
173 this.sequelize
.literal('SELECT "podId" FROM "' + tableRequestPod
+ '"')
179 return this.findAll(query
).asCallback(function (err
, pods
) {
180 if (err
) return callback(err
)
182 return callback(null, map(pods
, 'id'))
187 function listBadPods (callback
) {
194 return this.findAll(query
).asCallback(callback
)
197 function load (id
, callback
) {
198 return this.findById(id
).asCallback(callback
)
201 function loadByHost (host
, callback
) {
208 return this.findOne(query
).asCallback(callback
)
211 function removeAll (callback
) {
212 return this.destroy().asCallback(callback
)
215 function updatePodsScore (goodPods
, badPods
) {
218 logger
.info('Updating %d good pods and %d bad pods scores.', goodPods
.length
, badPods
.length
)
220 if (goodPods
.length
!== 0) {
221 this.incrementScores(goodPods
, constants
.PODS_SCORE
.BONUS
, function (err
) {
222 if (err
) logger
.error('Cannot increment scores of good pods.', { error: err
})
226 if (badPods
.length
!== 0) {
227 this.incrementScores(badPods
, constants
.PODS_SCORE
.MALUS
, function (err
) {
228 if (err
) logger
.error('Cannot decrement scores of bad pods.', { error: err
})
229 removeBadPods
.call(self
)
234 // ---------------------------------------------------------------------------
236 // Remove pods with a score of 0 (too many requests where they were unreachable)
237 function removeBadPods () {
241 function findBadPods (callback
) {
242 self
.sequelize
.models
.Pod
.listBadPods(function (err
, pods
) {
244 logger
.error('Cannot find bad pods.', { error: err
})
248 return callback(null, pods
)
252 function removeTheseBadPods (pods
, callback
) {
253 each(pods
, function (pod
, callbackEach
) {
254 pod
.destroy().asCallback(callbackEach
)
256 return callback(err
, pods
.length
)
259 ], function (err
, numberOfPodsRemoved
) {
261 logger
.error('Cannot remove bad pods.', { error: err
})
262 } else if (numberOfPodsRemoved
) {
263 logger
.info('Removed %d pods.', numberOfPodsRemoved
)
265 logger
.info('No need to remove bad pods.')