profxadke/misskey
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix bug nado

Browse source
This commit is contained in:
syuilo 2018-07-25 08:01:12 +09:00
parent 9c4e0a4ae6
commit b3b82e7595
4 changed files with 55 additions and 64 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
.config/example.yml
View file

@ -58,8 +58,7 @@ drive:
# OR # OR
# storage: 'object-storage' # storage: 'minio'
# service: 'minio'
# bucket: # bucket:
# prefix: # prefix:
# config: # config:

5
src/config/types.ts
View file

@ -51,9 +51,8 @@ export type Source = {
drive?: { drive?: {
storage: string; storage: string;
bucket: string; bucket?: string;
prefix: string; prefix?: string;
service?: string;
config?: any; config?: any;
}; };

49
src/services/drive/add-file.ts
View file

@ -21,36 +21,31 @@ import config from '../../config';
const log = debug('misskey:drive:add-file'); const log = debug('misskey:drive:add-file');
async function save(readable: stream.Readable, name: string, type: string, hash: string, size: number, metadata: any): Promise<IDriveFile> { async function save(readable: stream.Readable, name: string, type: string, hash: string, size: number, metadata: any): Promise<IDriveFile> {
if (config.drive && config.drive.storage == 'object-storage') { if (config.drive && config.drive.storage == 'minio') {
if (config.drive.service == 'minio') { const minio = new Minio.Client(config.drive.config);
const id = uuid.v4();
const obj = `${config.drive.prefix}/${id}`;
await minio.putObject(config.drive.bucket, obj, readable);
const minio = new Minio.Client(config.drive.config); Object.assign(metadata, {
const id = uuid.v4(); withoutChunks: true,
const obj = `${config.drive.prefix}/${id}`; storage: 'minio',
await minio.putObject(config.drive.bucket, obj, readable); storageProps: {
id: id
},
url: `${ config.drive.config.secure ? 'https' : 'http' }://${ config.drive.config.endPoint }${ config.drive.config.port ? ':' + config.drive.config.port : '' }/${ config.drive.bucket }/${ obj }`
});
Object.assign(metadata, { const file = await DriveFile.insert({
withoutChunks: true, length: size,
storage: 'object-storage', uploadDate: new Date(),
storageProps: { md5: hash,
id: id filename: name,
}, metadata: metadata,
url: `${ config.drive.config.secure ? 'https' : 'http' }://${ config.drive.config.endPoint }${ config.drive.config.port ? ':' + config.drive.config.port : '' }/${ config.drive.bucket }/${ obj }` contentType: type
}); });
const file = await DriveFile.insert({ return file;
length: size,
uploadDate: new Date(),
md5: hash,
filename: name,
metadata: metadata,
contentType: type
});
return file;
} else {
throw 'unknown storage type';
}
} else { } else {
// Get MongoDB GridFS bucket // Get MongoDB GridFS bucket
const bucket = await getDriveFileBucket(); const bucket = await getDriveFileBucket();

62
src/services/drive/delete-file.ts
View file

@ -4,37 +4,35 @@ import DriveFileThumbnail, { DriveFileThumbnailChunk } from '../../models/drive-
import config from '../../config'; import config from '../../config';
export default async function(file: IDriveFile, isExpired = false) { export default async function(file: IDriveFile, isExpired = false) {
if (file.metadata.withoutChunks) { if (file.metadata.storage == 'minio') {
if (file.metadata.storage == 'object-storage') { const minio = new Minio.Client(config.drive.config);
const minio = new Minio.Client(config.drive.config); const obj = `${config.drive.prefix}/${file.metadata.storageProps.id}`;
const obj = `${config.drive.prefix}/${file.metadata.storageProps.id}`; await minio.removeObject(config.drive.bucket, obj);
await minio.removeObject(config.drive.bucket, obj);
}
} else {
// チャンクをすべて削除
await DriveFileChunk.remove({
files_id: file._id
});
await DriveFile.update({ _id: file._id }, {
$set: {
'metadata.deletedAt': new Date(),
'metadata.isExpired': isExpired
}
});
//#region サムネイルもあれば削除
const thumbnail = await DriveFileThumbnail.findOne({
'metadata.originalId': file._id
});
if (thumbnail) {
await DriveFileThumbnailChunk.remove({
files_id: thumbnail._id
});
await DriveFileThumbnail.remove({ _id: thumbnail._id });
}
//#endregion
} }
// チャンクをすべて削除
await DriveFileChunk.remove({
files_id: file._id
});
await DriveFile.update({ _id: file._id }, {
$set: {
'metadata.deletedAt': new Date(),
'metadata.isExpired': isExpired
}
});
//#region サムネイルもあれば削除
const thumbnail = await DriveFileThumbnail.findOne({
'metadata.originalId': file._id
});
if (thumbnail) {
await DriveFileThumbnailChunk.remove({
files_id: thumbnail._id
});
await DriveFileThumbnail.remove({ _id: thumbnail._id });
}
//#endregion
} }