2c3e289509
* Working on refactoring jobs service * Making more consistent with the previous existing code * Remove no longer needed functions * Making a base PeridicScheduler to use it in most of the schedulers implementations * Removing accidental complexity from on of the jobs * Removing accidental complexity from expirynotify * Fixing compilation from previous commit * Remove accidental complexity from the export_delete job * Simplifying the workers by making a reusable worker * Using simple worker for export_delete job * Simpliying export process job * Simpliying extract content job * Simpliying import delete job * Simpliying import process job * Simpliying product noticies job * Simpliying fix crt channel unreads job (only removing the uneeded register function) * Simpliying migrations job (only removing the uneeded register function) * fixup * Simpliying plugins job (only removing the uneeded register function) * Simpliying bleve indexing job (only removing the uneeded register function) * Simpliying resend invitation email job (only removing the uneeded register function) * Fixing tests * Simplifying migration tests infrastructure * Adding missed license to files * Adding an empty file to imports package to ensure this package exist even without enterprise repo * Regenerating einterfaces mocks * Adding missed license to files * Updating i18n/en.json file * help fixing enterprise tests compilation * Adding new DailyScheduler * Fixing typo and changing the waitTime type for periodic sechduler * Making the daily scheduler more generic * Adding comments to clarify not used parameters in interface scheduler interface implementations * Using merror to handle multiple errors in jobs workers * Fixing linter errors * Addressing PR review comments * Reverting go.tools.mod changes * Removing the static check for worker type in the model (moving it to the insertion of new jobs * Moving migrations job to the jobs directory * Fixing (and improving a bit) tests * Apply suggestions from code review Co-authored-by: Doug Lauder <wiggin77@warpmail.net> * Fixing enterprise tests * Removing unneeded InitWorkers/InitSchedulers calls * Fix expirenotify job when error happens * Fixing govet errors Co-authored-by: Mattermod <mattermod@users.noreply.github.com> Co-authored-by: Doug Lauder <wiggin77@warpmail.net>
65 строки
1.6 KiB
Go
65 строки
1.6 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package export_process
|
|
|
|
import (
|
|
"io"
|
|
"path/filepath"
|
|
|
|
"github.com/mattermost/mattermost-server/v6/jobs"
|
|
"github.com/mattermost/mattermost-server/v6/model"
|
|
"github.com/mattermost/mattermost-server/v6/services/configservice"
|
|
"github.com/mattermost/mattermost-server/v6/shared/mlog"
|
|
)
|
|
|
|
const jobName = "ExportProcess"
|
|
|
|
type AppIface interface {
|
|
configservice.ConfigService
|
|
WriteFile(fr io.Reader, path string) (int64, *model.AppError)
|
|
BulkExport(writer io.Writer, outPath string, opts model.BulkExportOpts) *model.AppError
|
|
}
|
|
|
|
func MakeWorker(jobServer *jobs.JobServer, app AppIface) model.Worker {
|
|
isEnabled := func(cfg *model.Config) bool { return true }
|
|
execute := func(job *model.Job) error {
|
|
opts := model.BulkExportOpts{
|
|
CreateArchive: true,
|
|
}
|
|
|
|
includeAttachments, ok := job.Data["include_attachments"]
|
|
if ok && includeAttachments == "true" {
|
|
opts.IncludeAttachments = true
|
|
}
|
|
|
|
outPath := *app.Config().ExportSettings.Directory
|
|
exportFilename := model.NewId() + "_export.zip"
|
|
|
|
rd, wr := io.Pipe()
|
|
|
|
errCh := make(chan *model.AppError, 1)
|
|
go func() {
|
|
defer close(errCh)
|
|
_, appErr := app.WriteFile(rd, filepath.Join(outPath, exportFilename))
|
|
errCh <- appErr
|
|
}()
|
|
|
|
appErr := app.BulkExport(wr, outPath, opts)
|
|
if err := wr.Close(); err != nil {
|
|
mlog.Warn("Worker: error closing writer")
|
|
}
|
|
|
|
if appErr != nil {
|
|
return appErr
|
|
}
|
|
|
|
if appErr := <-errCh; appErr != nil {
|
|
return appErr
|
|
}
|
|
return nil
|
|
}
|
|
worker := jobs.NewSimpleWorker(jobName, jobServer, execute, isEnabled)
|
|
return worker
|
|
}
|