Chocobozzz/PeerTube
1
0
Fork 0
mirror of https://github.com/Chocobozzz/PeerTube.git synced 2024-11-10 12:26:35 +03:00
Code Issues Projects Releases Wiki Activity

Pod model refractoring -> use mongoose api

Browse Source
This commit is contained in:
Chocobozzz 2016-06-30 22:39:08 +02:00
parent d14b3e37a2
commit a3ee6fa22d
8 changed files with 111 additions and 99 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
server/controllers/api/v1/pods.js
View File

@ -7,12 +7,12 @@ const mongoose = require('mongoose')
const logger = require('../../../helpers/logger')
const friends = require('../../../lib/friends')
const middlewares = require('../../../middlewares')
const Pods = require('../../../models/pods')
const oAuth2 = middlewares.oauth2
const reqValidator = middlewares.reqValidators.pods
const signatureValidator = middlewares.reqValidators.remote.signature
const router = express.Router()
const Pod = mongoose.model('Pod')
const Video = mongoose.model('Video')
router.get('/', listPodsUrl)
@ -33,7 +33,11 @@ function addPods (req, res, next) {
async.waterfall([
function addPod (callback) {
Pods.add(informations, callback)
const pod = new Pod(informations)
pod.save(function (err, podCreated) {
// Be sure about the number of parameters for the callback
return callback(err, podCreated)
})
},
function sendMyVideos (podCreated, callback) {
@ -60,7 +64,7 @@ function addPods (req, res, next) {
}
function listPodsUrl (req, res, next) {
Pods.listAllUrls(function (err, podsUrlList) {
Pod.listOnlyUrls(function (err, podsUrlList) {
if (err) return next(err)
res.json(podsUrlList)
@ -79,8 +83,13 @@ function removePods (req, res, next) {
const url = req.body.signature.url
async.waterfall([
function (callback) {
Pods.remove(url, function (err) {
function loadPod (callback) {
Pod.loadByUrl(url, callback)
},
function removePod (pod, callback) {
pod.remove(function (err) {
// Be sure we only return one argument in the callback
return callback(err)
})
},

1
server/helpers/customValidators.js
View File

@ -85,6 +85,7 @@ function isVideoNameValid (value) {
}
function isVideoPodUrlValid (value) {
// TODO: set options (TLD...)
return validator.isURL(value)
}

9
server/initializers/constants.js
View File

@ -4,7 +4,10 @@
const API_VERSION = 'v1'
// Score a pod has when we create it as a friend
let FRIEND_BASE_SCORE = 100
const FRIEND_SCORE = {
BASE: 100,
MAX: 1000
}
// Time to wait between requests to the friends (10 min)
let INTERVAL = 600000
@ -54,7 +57,7 @@ const VIDEOS_CONSTRAINTS_FIELDS = {
// Special constants for a test instance
if (isTestInstance() === true) {
FRIEND_BASE_SCORE = 20
FRIEND_SCORE.BASE = 20
INTERVAL = 10000
VIDEOS_CONSTRAINTS_FIELDS.DURATION.max = 14
}
@ -63,7 +66,7 @@ if (isTestInstance() === true) {
module.exports = {
API_VERSION: API_VERSION,
FRIEND_BASE_SCORE: FRIEND_BASE_SCORE,
FRIEND_SCORE: FRIEND_SCORE,
INTERVAL: INTERVAL,
PAGINATION_COUNT_DEFAULT: PAGINATION_COUNT_DEFAULT,
PODS_SCORE: PODS_SCORE,

1
server/initializers/database.js
View File

@ -6,6 +6,7 @@ const mongoose = require('mongoose')
const logger = require('../helpers/logger')
// Bootstrap models
require('../models/pods')
require('../models/video')
// Request model needs Video model
require('../models/request')

20
server/lib/friends.js
View File

@ -9,16 +9,16 @@ const request = require('request')
const constants = require('../initializers/constants')
const logger = require('../helpers/logger')
const peertubeCrypto = require('../helpers/peertubeCrypto')
const Pods = require('../models/pods')
const requests = require('../helpers/requests')
const http = config.get('webserver.https') ? 'https' : 'http'
const host = config.get('webserver.host')
const port = config.get('webserver.port')
const Pod = mongoose.model('Pod')
const Request = mongoose.model('Request')
const Video = mongoose.model('Video')
const pods = {
const friends = {
addVideoToFriends: addVideoToFriends,
hasFriends: hasFriends,
getMyCertificate: getMyCertificate,
@ -33,7 +33,7 @@ function addVideoToFriends (video) {
}
function hasFriends (callback) {
Pods.count(function (err, count) {
Pod.countAll(function (err, count) {
if (err) return callback(err)
const hasFriends = (count !== 0)
@ -79,7 +79,7 @@ function quitFriends (callback) {
async.waterfall([
function getPodsList (callbackAsync) {
return Pods.list(callbackAsync)
return Pod.list(callbackAsync)
},
function announceIQuitMyFriends (pods, callbackAsync) {
@ -106,7 +106,7 @@ function quitFriends (callback) {
},
function removePodsFromDB (callbackAsync) {
Pods.removeAll(function (err) {
Pod.removeAll(function (err) {
return callbackAsync(err)
})
},
@ -160,7 +160,7 @@ function sendOwnedVideosToPod (podId) {
// ---------------------------------------------------------------------------
module.exports = pods
module.exports = friends
// ---------------------------------------------------------------------------
@ -230,8 +230,12 @@ function makeRequestsToWinningPods (cert, podsList, callback) {
}
if (res.statusCode === 200) {
Pods.add({ url: pod.url, publicKey: body.cert, score: constants.FRIEND_BASE_SCORE }, function (err, podCreated) {
if (err) logger.error('Cannot add friend %s pod.', pod.url)
const podObj = new Pod({ url: pod.url, publicKey: body.cert })
podObj.save(function (err, podCreated) {
if (err) {
logger.error('Cannot add friend %s pod.', pod.url, { error: err })
return callbackEach()
}
// Add our videos to the request scheduler
sendOwnedVideosToPod(podCreated._id)

6
server/middlewares/secure.js
View File

@ -1,8 +1,10 @@
'use strict'
const logger = require('../helpers/logger')
const mongoose = require('mongoose')
const peertubeCrypto = require('../helpers/peertubeCrypto')
const Pods = require('../models/pods')
const Pod = mongoose.model('Pod')
const secureMiddleware = {
decryptBody: decryptBody
@ -10,7 +12,7 @@ const secureMiddleware = {
function decryptBody (req, res, next) {
const url = req.body.signature.url
Pods.findByUrl(url, function (err, pod) {
Pod.loadByUrl(url, function (err, pod) {
if (err) {
logger.error('Cannot get signed url in decryptBody.', { error: err })
return res.sendStatus(500)

106
server/models/pods.js
View File

@ -2,107 +2,89 @@
const mongoose = require('mongoose')
const map = require('lodash/map')
const validator = require('express-validator').validator
const constants = require('../initializers/constants')
const logger = require('../helpers/logger')
// ---------------------------------------------------------------------------
const podsSchema = mongoose.Schema({
const PodSchema = mongoose.Schema({
url: String,
publicKey: String,
score: { type: Number, max: constants.FRIEND_BASE_SCORE }
score: { type: Number, max: constants.FRIEND_SCORE.MAX }
})
const PodsDB = mongoose.model('pods', podsSchema)
// ---------------------------------------------------------------------------
// TODO: set options (TLD...)
PodSchema.path('url').validate(validator.isURL)
PodSchema.path('publicKey').required(true)
PodSchema.path('score').validate(function (value) { return !isNaN(value) })
const Pods = {
add: add,
count: count,
findById: findById,
findByUrl: findByUrl,
findBadPods: findBadPods,
PodSchema.statics = {
countAll: countAll,
incrementScores: incrementScores,
list: list,
listAllIds: listAllIds,
listAllUrls: listAllUrls,
remove: remove,
removeAll: removeAll,
removeAllByIds: removeAllByIds
listOnlyUrls: listOnlyUrls,
listBadPods: listBadPods,
load: load,
loadByUrl: loadByUrl,
removeAll: removeAll
}
// TODO: check if the pod is not already a friend
function add (data, callback) {
if (!callback) callback = function () {}
const params = {
url: data.url,
publicKey: data.publicKey,
score: constants.FRIEND_BASE_SCORE
}
PodSchema.pre('save', function (next) {
const self = this
PodsDB.create(params, callback)
}
Pod.loadByUrl(this.url, function (err, pod) {
if (err) return next(err)
function count (callback) {
return PodsDB.count(callback)
}
if (pod) return next(new Error('Pod already exists.'))
function findBadPods (callback) {
PodsDB.find({ score: 0 }, callback)
}
self.score = constants.FRIEND_SCORE.BASE
return next()
})
})
function findById (id, callback) {
PodsDB.findById(id, callback)
}
const Pod = mongoose.model('Pod', PodSchema)
function findByUrl (url, callback) {
PodsDB.findOne({ url: url }, callback)
// ------------------------------ Statics ------------------------------
function countAll (callback) {
return this.count(callback)
}
function incrementScores (ids, value, callback) {
if (!callback) callback = function () {}
PodsDB.update({ _id: { $in: ids } }, { $inc: { score: value } }, { multi: true }, callback)
return this.update({ _id: { $in: ids } }, { $inc: { score: value } }, { multi: true }, callback)
}
function list (callback) {
PodsDB.find(function (err, podsList) {
if (err) {
logger.error('Cannot get the list of the pods.')
return callback(err)
}
return callback(null, podsList)
})
return this.find(callback)
}
function listAllIds (callback) {
return PodsDB.find({}, { _id: 1 }, function (err, pods) {
return this.find({}, { _id: 1 }, function (err, pods) {
if (err) return callback(err)
return callback(null, map(pods, '_id'))
})
}
function listAllUrls (callback) {
return PodsDB.find({}, { _id: 0, url: 1 }, callback)
function listOnlyUrls (callback) {
return this.find({}, { _id: 0, url: 1 }, callback)
}
function remove (url, callback) {
if (!callback) callback = function () {}
PodsDB.remove({ url: url }, callback)
function listBadPods (callback) {
return this.find({ score: 0 }, callback)
}
function load (id, callback) {
return this.findById(id, callback)
}
function loadByUrl (url, callback) {
return this.findOne({ url: url }, callback)
}
function removeAll (callback) {
if (!callback) callback = function () {}
PodsDB.remove(callback)
return this.remove({}, callback)
}
function removeAllByIds (ids, callback) {
if (!callback) callback = function () {}
PodsDB.remove({ _id: { $in: ids } }, callback)
}
// ---------------------------------------------------------------------------
module.exports = Pods

48
server/models/request.js
View File

@ -6,9 +6,9 @@ const mongoose = require('mongoose')
const constants = require('../initializers/constants')
const logger = require('../helpers/logger')
const Pods = require('../models/pods')
const requests = require('../helpers/requests')
const Pod = mongoose.model('Pod')
const Video = mongoose.model('Video')
let timer = null
@ -31,7 +31,7 @@ RequestSchema.pre('save', function (next) {
const self = this
if (self.to.length === 0) {
Pods.listAllIds(function (err, podIds) {
Pod.listAllIds(function (err, podIds) {
if (err) return next(err)
// No friends
@ -140,7 +140,7 @@ function makeRequests () {
const requestToMake = requestsToMake[toPodId]
// FIXME: mongodb request inside a loop :/
Pods.findById(toPodId, function (err, toPod) {
Pod.load(toPodId, function (err, toPod) {
if (err) {
logger.error('Error finding pod by id.', { err: err })
return callbackEach()
@ -185,7 +185,7 @@ function makeRequests () {
function removeBadPods () {
async.waterfall([
function findBadPods (callback) {
Pods.findBadPods(function (err, pods) {
Pod.listBadPods(function (err, pods) {
if (err) {
logger.error('Cannot find bad pods.', { error: err })
return callback(err)
@ -199,21 +199,23 @@ function removeBadPods () {
if (pods.length === 0) return callback(null)
const urls = map(pods, 'url')
const ids = map(pods, '_id')
Video.listByUrls(urls, function (err, videosList) {
if (err) {
logger.error('Cannot list videos urls.', { error: err, urls: urls })
return callback(null, ids, [])
return callback(null, pods, [])
}
return callback(null, ids, videosList)
return callback(null, pods, videosList)
})
},
function removeVideosOfTheseBadPods (podIds, videosList, callback) {
function removeVideosOfTheseBadPods (pods, videosList, callback) {
// We don't have to remove pods, skip
if (typeof podIds === 'function') return podIds(null)
if (typeof pods === 'function') {
callback = pods
return callback(null)
}
async.each(videosList, function (video, callbackEach) {
video.remove(callbackEach)
@ -224,22 +226,30 @@ function removeBadPods () {
return
}
return callback(null, podIds)
return callback(null, pods)
})
},
function removeBadPodsFromDB (podIds, callback) {
function removeBadPodsFromDB (pods, callback) {
// We don't have to remove pods, skip
if (typeof podIds === 'function') return podIds(null)
if (typeof pods === 'function') {
callback = pods
return callback(null)
}
Pods.removeAllByIds(podIds, callback)
async.each(pods, function (pod, callbackEach) {
pod.remove(callbackEach)
}, function (err) {
if (err) return callback(err)
return callback(null, pods.length)
})
}
], function (err, removeResult) {
], function (err, numberOfPodsRemoved) {
if (err) {
logger.error('Cannot remove bad pods.', { error: err })
} else if (removeResult) {
const podsRemoved = removeResult.result.n
logger.info('Removed %d pods.', podsRemoved)
} else if (numberOfPodsRemoved) {
logger.info('Removed %d pods.', numberOfPodsRemoved)
} else {
logger.info('No need to remove bad pods.')
}
@ -249,11 +259,11 @@ function removeBadPods () {
function updatePodsScore (goodPods, badPods) {
logger.info('Updating %d good pods and %d bad pods scores.', goodPods.length, badPods.length)
Pods.incrementScores(goodPods, constants.PODS_SCORE.BONUS, function (err) {
Pod.incrementScores(goodPods, constants.PODS_SCORE.BONUS, function (err) {
if (err) logger.error('Cannot increment scores of good pods.')
})
Pods.incrementScores(badPods, constants.PODS_SCORE.MALUS, function (err) {
Pod.incrementScores(badPods, constants.PODS_SCORE.MALUS, function (err) {
if (err) logger.error('Cannot decrement scores of bad pods.')
removeBadPods()
})