aboutsummaryrefslogtreecommitdiffhomepage
path: root/scripts/mongo-to-postgre.js
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/mongo-to-postgre.js')
-rwxr-xr-xscripts/mongo-to-postgre.js244
1 files changed, 0 insertions, 244 deletions
diff --git a/scripts/mongo-to-postgre.js b/scripts/mongo-to-postgre.js
deleted file mode 100755
index 6cd1ef3b4..000000000
--- a/scripts/mongo-to-postgre.js
+++ /dev/null
@@ -1,244 +0,0 @@
1#!/usr/bin/env node
2
3'use strict'
4
5// TODO: document this script
6
7const program = require('commander')
8const eachSeries = require('async/eachSeries')
9const series = require('async/series')
10const waterfall = require('async/waterfall')
11const fs = require('fs')
12const path = require('path')
13const MongoClient = require('mongodb').MongoClient
14
15const constants = require('../server/initializers/constants')
16
17program
18 .option('-mh, --mongo-host [host]', 'MongoDB host', 'localhost')
19 .option('-mp, --mongo-port [weight]', 'MongoDB port', '27017')
20 .option('-md, --mongo-database [dbname]', 'MongoDB database')
21 .parse(process.argv)
22
23if (!program.mongoDatabase) {
24 console.error('The mongodb database is mandatory.')
25 process.exit(-1)
26}
27
28const mongoUrl = 'mongodb://' + program.mongoHost + ':' + program.mongoPort + '/' + program.mongoDatabase
29const dbSequelize = require('../server/initializers/database')
30
31console.log('Connecting to ' + mongoUrl)
32MongoClient.connect(mongoUrl, function (err, dbMongo) {
33 if (err) throw err
34
35 console.log('Connected to ' + mongoUrl)
36
37 const videoMongo = dbMongo.collection('videos')
38 const userMongo = dbMongo.collection('users')
39 const podMongo = dbMongo.collection('pods')
40
41 podMongo.count(function (err, podsLength) {
42 if (err) throw err
43
44 if (podsLength > 0) {
45 console.error('You need to quit friends first.')
46 process.exit(-1)
47 }
48
49 console.log('Connecting to ' + dbSequelize.sequelize.config.database)
50 dbSequelize.init(true, function (err) {
51 if (err) throw err
52
53 console.log('Connected to SQL database %s.', dbSequelize.sequelize.config.database)
54
55 series([
56 function (next) {
57 dbSequelize.sequelize.sync({ force: true }).asCallback(next)
58 },
59
60 function (next) {
61 migrateVideos(videoMongo, dbSequelize, next)
62 },
63
64 function (next) {
65 migrateUsers(userMongo, dbSequelize, next)
66 }
67 ], function (err) {
68 if (err) console.error(err)
69
70 process.exit(0)
71 })
72 })
73 })
74})
75
76// ---------------------------------------------------------------------------
77
78function migrateUsers (userMongo, dbSequelize, callback) {
79 userMongo.find().toArray(function (err, mongoUsers) {
80 if (err) return callback(err)
81
82 eachSeries(mongoUsers, function (mongoUser, callbackEach) {
83 console.log('Migrating user %s', mongoUser.username)
84
85 const userData = {
86 username: mongoUser.username,
87 password: mongoUser.password,
88 role: mongoUser.role
89 }
90 const options = {
91 hooks: false
92 }
93
94 dbSequelize.User.create(userData, options).asCallback(callbackEach)
95 }, callback)
96 })
97}
98
99function migrateVideos (videoMongo, dbSequelize, finalCallback) {
100 videoMongo.find().toArray(function (err, mongoVideos) {
101 if (err) return finalCallback(err)
102
103 eachSeries(mongoVideos, function (mongoVideo, callbackEach) {
104 console.log('Migrating video %s.', mongoVideo.name)
105
106 waterfall([
107
108 function startTransaction (callback) {
109 dbSequelize.sequelize.transaction().asCallback(function (err, t) {
110 return callback(err, t)
111 })
112 },
113
114 function findOrCreatePod (t, callback) {
115 if (mongoVideo.remoteId === null) return callback(null, t, null)
116
117 const query = {
118 where: {
119 host: mongoVideo.podHost
120 },
121 defaults: {
122 host: mongoVideo.podHost
123 },
124 transaction: t
125 }
126
127 dbSequelize.Pod.findOrCreate(query).asCallback(function (err, result) {
128 // [ instance, wasCreated ]
129 const res = result ? result[0] : null
130 return callback(err, t, res)
131 })
132 },
133
134 function findOrCreateAuthor (t, pod, callback) {
135 const podId = pod ? pod.id : null
136 const username = mongoVideo.author
137
138 const query = {
139 where: {
140 podId,
141 name: username
142 },
143 defaults: {
144 podId,
145 name: username
146 },
147 transaction: t
148 }
149
150 dbSequelize.Author.findOrCreate(query).asCallback(function (err, result) {
151 // [ instance, wasCreated ]
152 const res = result ? result[0] : null
153 return callback(err, t, res)
154 })
155 },
156
157 function findOrCreateTags (t, author, callback) {
158 const tags = mongoVideo.tags
159 const tagInstances = []
160
161 eachSeries(tags, function (tag, callbackEach) {
162 const query = {
163 where: {
164 name: tag
165 },
166 defaults: {
167 name: tag
168 },
169 transaction: t
170 }
171
172 dbSequelize.Tag.findOrCreate(query).asCallback(function (err, res) {
173 if (err) return callbackEach(err)
174
175 // res = [ tag, isCreated ]
176 const tag = res[0]
177 tagInstances.push(tag)
178 return callbackEach()
179 })
180 }, function (err) {
181 return callback(err, t, author, tagInstances)
182 })
183 },
184
185 function createVideoObject (t, author, tagInstances, callback) {
186 const videoData = {
187 name: mongoVideo.name,
188 remoteId: mongoVideo.remoteId,
189 extname: mongoVideo.extname,
190 infoHash: mongoVideo.magnet.infoHash,
191 description: mongoVideo.description,
192 authorId: author.id,
193 duration: mongoVideo.duration,
194 createdAt: mongoVideo.createdDate
195 }
196
197 const video = dbSequelize.Video.build(videoData)
198
199 return callback(null, t, tagInstances, video)
200 },
201
202 function moveVideoFile (t, tagInstances, video, callback) {
203 const basePath = constants.CONFIG.STORAGE.VIDEOS_DIR
204 const src = path.join(basePath, mongoVideo._id.toString()) + video.extname
205 const dst = path.join(basePath, video.id) + video.extname
206 fs.rename(src, dst, function (err) {
207 if (err) return callback(err)
208
209 return callback(null, t, tagInstances, video)
210 })
211 },
212
213 function insertVideoIntoDB (t, tagInstances, video, callback) {
214 const options = {
215 transaction: t
216 }
217
218 video.save(options).asCallback(function (err, videoCreated) {
219 return callback(err, t, tagInstances, videoCreated)
220 })
221 },
222
223 function associateTagsToVideo (t, tagInstances, video, callback) {
224 const options = { transaction: t }
225
226 video.setTags(tagInstances, options).asCallback(function (err) {
227 return callback(err, t)
228 })
229 }
230
231 ], function (err, t) {
232 if (err) {
233 // Abort transaction?
234 if (t) t.rollback()
235
236 return callbackEach(err)
237 }
238
239 // Commit transaction
240 return t.commit().asCallback(callbackEach)
241 })
242 }, finalCallback)
243 })
244}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-30 23:15:33 +0000