chore: Refactor collection export to match import (#3483)
* chore: Refactor collection export to use FileOperations processor and task * Tweak options
This commit is contained in:
@@ -1,5 +1,11 @@
|
||||
import { Transaction } from "sequelize";
|
||||
import { APM } from "@server/logging/tracing";
|
||||
import { Collection, Event, Team, User, FileOperation } from "@server/models";
|
||||
import {
|
||||
FileOperationType,
|
||||
FileOperationState,
|
||||
FileOperationFormat,
|
||||
} from "@server/models/FileOperation";
|
||||
import { getAWSKeyForFileOp } from "@server/utils/s3";
|
||||
|
||||
async function collectionExporter({
|
||||
@@ -7,34 +13,49 @@ async function collectionExporter({
|
||||
team,
|
||||
user,
|
||||
ip,
|
||||
transaction,
|
||||
}: {
|
||||
collection?: Collection;
|
||||
team: Team;
|
||||
user: User;
|
||||
ip: string;
|
||||
transaction: Transaction;
|
||||
}) {
|
||||
const collectionId = collection?.id;
|
||||
const key = getAWSKeyForFileOp(user.teamId, collection?.name || team.name);
|
||||
const fileOperation = await FileOperation.create({
|
||||
type: "export",
|
||||
state: "creating",
|
||||
key,
|
||||
url: null,
|
||||
size: 0,
|
||||
collectionId,
|
||||
userId: user.id,
|
||||
teamId: user.teamId,
|
||||
});
|
||||
const fileOperation = await FileOperation.create(
|
||||
{
|
||||
type: FileOperationType.Export,
|
||||
state: FileOperationState.Creating,
|
||||
format: FileOperationFormat.MarkdownZip,
|
||||
key,
|
||||
url: null,
|
||||
size: 0,
|
||||
collectionId,
|
||||
userId: user.id,
|
||||
teamId: user.teamId,
|
||||
},
|
||||
{
|
||||
transaction,
|
||||
}
|
||||
);
|
||||
|
||||
// Event is consumed on worker in queues/processors/exports
|
||||
await Event.create({
|
||||
name: collection ? "collections.export" : "collections.export_all",
|
||||
collectionId,
|
||||
teamId: user.teamId,
|
||||
actorId: user.id,
|
||||
modelId: fileOperation.id,
|
||||
ip,
|
||||
});
|
||||
await Event.create(
|
||||
{
|
||||
name: "fileOperations.create",
|
||||
teamId: user.teamId,
|
||||
actorId: user.id,
|
||||
modelId: fileOperation.id,
|
||||
collectionId,
|
||||
ip,
|
||||
data: {
|
||||
type: FileOperationType.Import,
|
||||
},
|
||||
},
|
||||
{
|
||||
transaction,
|
||||
}
|
||||
);
|
||||
|
||||
fileOperation.user = user;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user