mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
zeekay
8feb24ea1f
The server module named github.com/hanzoai/team/server/v8 — a path that resolves to the retired workspace's repository, not this one — while server/public still published github.com/mattermost/mattermost/server/public. Both are github.com/hanzoteam/server/... now, which is where this code actually lives. That second half is what made the binary unbootable. Upstream's public path stayed claimed by upstream, mattermost-plugin-ai imported it, and the build linked public/model twice; public/plugin's gob.Register then ran twice under one name and panicked at init, before main. Reverting the rename traded the brand for a working binary. Rebranding the plugin alongside it needs no such trade: the agents fork is github.com/hanzoteam/agents/v2 against our public module, so no module named mattermost/mattermost is in the graph at all. Measured: `go list -deps ./cmd/mattermost` names exactly one public/model, and it is ours.
141 lines
4.3 KiB
Go
141 lines
4.3 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package jobs
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
"github.com/hanzoteam/server/server/public/shared/mlog"
|
|
"github.com/hanzoteam/server/server/public/shared/request"
|
|
"github.com/hanzoteam/server/server/v8/channels/store"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type BatchReportWorkerAppIFace interface {
|
|
SaveReportChunk(format string, prefix string, count int, reportData []model.ReportableObject) *model.AppError
|
|
CompileReportChunks(format string, prefix string, numberOfChunks int, headers []string) *model.AppError
|
|
SendReportToUser(rctx request.CTX, job *model.Job, format string) *model.AppError
|
|
CleanupReportChunks(format string, prefix string, numberOfChunks int) *model.AppError
|
|
}
|
|
|
|
type BatchReportWorker struct {
|
|
*BatchWorker
|
|
app BatchReportWorkerAppIFace
|
|
reportFormat string
|
|
headers []string
|
|
getData func(jobData model.StringMap) ([]model.ReportableObject, model.StringMap, bool, error)
|
|
}
|
|
|
|
func MakeBatchReportWorker(
|
|
jobServer *JobServer,
|
|
store store.Store,
|
|
app BatchReportWorkerAppIFace,
|
|
timeBetweenBatches time.Duration,
|
|
reportFormat string,
|
|
headers []string,
|
|
getData func(jobData model.StringMap) ([]model.ReportableObject, model.StringMap, bool, error),
|
|
) *BatchReportWorker {
|
|
worker := &BatchReportWorker{
|
|
app: app,
|
|
reportFormat: reportFormat,
|
|
headers: headers,
|
|
getData: getData,
|
|
}
|
|
worker.BatchWorker = MakeBatchWorker(jobServer, store, timeBetweenBatches, worker.doBatch)
|
|
return worker
|
|
}
|
|
|
|
func (worker *BatchReportWorker) doBatch(rctx request.CTX, job *model.Job) bool {
|
|
reportData, nextData, done, err := worker.getData(job.Data)
|
|
if err != nil {
|
|
worker.logger.Error("Worker: Failed to get data for report batch. Exiting", mlog.Err(err))
|
|
worker.setJobError(worker.logger, job, model.NewAppError("doBatch", model.NoTranslation, nil, "", http.StatusInternalServerError).Wrap(err))
|
|
return true
|
|
} else if done {
|
|
if err = worker.complete(rctx, job); err != nil {
|
|
worker.logger.Error("Worker: Failed to finish the batch report. Exiting", mlog.Err(err))
|
|
worker.setJobError(worker.logger, job, model.NewAppError("doBatch", model.NoTranslation, nil, "", http.StatusInternalServerError).Wrap(err))
|
|
} else {
|
|
worker.logger.Info("Worker: Report job complete")
|
|
worker.setJobSuccess(worker.logger, job)
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
err = worker.processChunk(job, reportData)
|
|
if err != nil {
|
|
worker.logger.Error("Worker: Failed to save report batch. Exiting", mlog.Err(err))
|
|
worker.setJobError(worker.logger, job, model.NewAppError("doBatch", model.NoTranslation, nil, "", http.StatusInternalServerError).Wrap(err))
|
|
return true
|
|
}
|
|
|
|
job.Data = nextData
|
|
|
|
// We might be able to add progress for this type of job in the future
|
|
// But for now we can just set to 0
|
|
if err := worker.jobServer.SetJobProgress(job, 0); err != nil {
|
|
worker.logger.Error("Worker: Failed to set job progress", mlog.Err(err))
|
|
return false
|
|
}
|
|
return false
|
|
}
|
|
|
|
func getFileCount(jobData model.StringMap) (int, error) {
|
|
if jobData["file_count"] != "" {
|
|
parsedFileCount, parseErr := strconv.Atoi(jobData["file_count"])
|
|
if parseErr != nil {
|
|
return 0, errors.Wrap(parseErr, "failed to parse file_count")
|
|
}
|
|
return parsedFileCount, nil
|
|
}
|
|
|
|
// Assume it hasn't been set
|
|
return 0, nil
|
|
}
|
|
|
|
func (worker *BatchReportWorker) processChunk(job *model.Job, reportData []model.ReportableObject) error {
|
|
fileCount, err := getFileCount(job.Data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
appErr := worker.app.SaveReportChunk(worker.reportFormat, job.Id, fileCount, reportData)
|
|
if appErr != nil {
|
|
return appErr
|
|
}
|
|
|
|
fileCount++
|
|
job.Data["file_count"] = strconv.Itoa(fileCount)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (worker *BatchReportWorker) complete(rctx request.CTX, job *model.Job) error {
|
|
fileCount, err := getFileCount(job.Data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
appErr := worker.app.CompileReportChunks(worker.reportFormat, job.Id, fileCount, worker.headers)
|
|
if appErr != nil {
|
|
return appErr
|
|
}
|
|
|
|
defer func() {
|
|
if err := worker.app.CleanupReportChunks(worker.reportFormat, job.Id, fileCount); err != nil {
|
|
worker.logger.Error("Worker: Failed to cleanup report chunks", mlog.Err(err))
|
|
}
|
|
}()
|
|
|
|
if appErr = worker.app.SendReportToUser(rctx, job, worker.reportFormat); appErr != nil {
|
|
return appErr
|
|
}
|
|
|
|
return nil
|
|
}
|