misskey-awawa/tools/migration/node.1510056272.issue_882.js

48 lines
942 B
JavaScript
Raw Normal View History

2017-11-07 12:04:32 +00:00
// for Node.js interpret
const { default: DriveFile } = require('../../built/api/models/drive-file')
2017-11-07 14:10:38 +00:00
const { default: zip } = require('@prezzemolo/zip')
2017-11-07 12:04:32 +00:00
const migrate = async (doc) => {
const result = await DriveFile.update(doc._id, {
$set: {
contentType: doc.metadata.type
},
$unset: {
'metadata.type': ''
}
})
return result.ok === 1
}
async function main() {
2017-11-07 14:10:38 +00:00
const query = {
'metadata.type': {
$exists: true
2017-11-07 12:04:32 +00:00
}
}
2017-11-07 14:10:38 +00:00
const count = await DriveFile.count(query);
2017-11-07 12:04:32 +00:00
2017-11-07 14:10:38 +00:00
const dop = Number.parseInt(process.argv[2]) || 5
const idop = ((count - (count % dop)) / dop) + 1
return zip(
1,
async (time) => {
console.log(`${time} / ${idop}`)
const doc = await db.get('drive_files').find(query, {
limit: dop, skip: time * dop
})
return Promise.all(doc.map(migrate))
},
idop
).then(a => {
const rv = []
a.forEach(e => rv.push(...e))
return rv
})
2017-11-07 12:04:32 +00:00
}
main().then(console.dir).catch(console.error)