Update migrate.ts

This commit is contained in:
syuilo 2019-04-11 19:42:35 +09:00
parent 46b78cb4ff
commit d2b75f3501
No known key found for this signature in database
GPG key ID: BDC4C49D06AB9D69

View file

@ -302,13 +302,13 @@ async function main() {
const u = await _User.findOne({ const u = await _User.findOne({
_id: new mongo.ObjectId(user.id) _id: new mongo.ObjectId(user.id)
}); });
const avatar = await DriveFiles.findOne(u.avatarId.toHexString()); const avatar = u.avatarId ? await DriveFiles.findOne(u.avatarId.toHexString()) : null;
const banner = await DriveFiles.findOne(u.bannerId.toHexString()); const banner = u.bannerId ? await DriveFiles.findOne(u.bannerId.toHexString()) : null;
await Users.update(user.id, { await Users.update(user.id, {
avatarId: avatar.id, avatarId: avatar ? avatar.id : null,
bannerId: banner.id, bannerId: banner ? banner.id : null,
avatarUrl: avatar.url, avatarUrl: avatar ? avatar.url : null,
bannerUrl: banner.url bannerUrl: banner ? banner.url : null
}); });
} }
@ -369,10 +369,14 @@ async function main() {
} }
} }
let allDriveFilesCount = await _DriveFile.count(); let allDriveFilesCount = await _DriveFile.count({
'metadata._user.host': null
});
if (test && allDriveFilesCount > limit) allDriveFilesCount = limit; if (test && allDriveFilesCount > limit) allDriveFilesCount = limit;
for (let i = 0; i < allDriveFilesCount; i++) { for (let i = 0; i < allDriveFilesCount; i++) {
const file = await _DriveFile.findOne({}, { const file = await _DriveFile.findOne({
'metadata._user.host': null
}, {
skip: i skip: i
}); });
try { try {