mirror of
https://github.com/hanzoai/team.git
synced 2026-11-01 07:55:31 +00:00
Harshil SharmaandGitHub
a4c9288cc8
* Report POC * Including more error logs * Added localisationj for each reviewer * Optimisations * Minor tweaks * restored go module files * lint fixes * Added back transslations * Added translations * linter and test fixes * restored go module files * e2e lint fix * lint fixes * AI fixes * fixed typo * fixed nil pointer error * Added more tests * Publish report even if deletion fails * Fixed the e2e test * Distinguished between no data and deleted data * lint fixes * fixed tests * e2e test fix * Updated test to also upload actual file * Removed file name tracking * Text updates * fixed e2e test * lint fix
81 lines
2.5 KiB
Go
81 lines
2.5 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package delete_expired_posts
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/mattermost/mattermost/server/public/shared/request"
|
|
"github.com/mattermost/mattermost/server/v8/channels/jobs"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
)
|
|
|
|
const (
|
|
expiredPostsBatchSize = 100
|
|
expiredPostsJobBatchWaitTime = 100 * time.Millisecond
|
|
)
|
|
|
|
type AppIface interface {
|
|
DeletePost(rctx request.CTX, postID, deleteByID string) (*model.Post, *model.AppError)
|
|
PermanentDeletePostDataRetainStub(rctx request.CTX, post *model.Post, deleteByID string) (*model.PostDeletionReport, *model.AppError)
|
|
GetSinglePost(rctx request.CTX, postID string, includeDeleted bool) (*model.Post, *model.AppError)
|
|
GetPostsByIds(postIDs []string) ([]*model.Post, int64, *model.AppError)
|
|
}
|
|
|
|
func MakeWorker(jobServer *jobs.JobServer, store store.Store, app AppIface) *jobs.SimpleWorker {
|
|
const workerName = "DeleteExpiredPosts"
|
|
|
|
isEnabled := func(cfg *model.Config) bool {
|
|
return model.SafeDereference(cfg.ServiceSettings.EnableBurnOnRead)
|
|
}
|
|
execute := func(logger mlog.LoggerIFace, job *model.Job) error {
|
|
if job.Data == nil {
|
|
job.Data = make(model.StringMap)
|
|
}
|
|
|
|
deletedPostIDs := make([]string, 0)
|
|
lastPostId := ""
|
|
for {
|
|
time.Sleep(expiredPostsJobBatchWaitTime)
|
|
postIDs, err := store.TemporaryPost().GetExpiredPosts(request.EmptyContext(logger), lastPostId, expiredPostsBatchSize)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(postIDs) == 0 {
|
|
break
|
|
}
|
|
|
|
lastPostId = postIDs[len(postIDs)-1]
|
|
|
|
expiredPosts, _, appErr := app.GetPostsByIds(postIDs)
|
|
if appErr != nil {
|
|
logger.Error("Failed to get expired posts by IDs", mlog.Err(appErr))
|
|
return appErr
|
|
}
|
|
|
|
for _, post := range expiredPosts {
|
|
_, appErr = app.PermanentDeletePostDataRetainStub(request.EmptyContext(logger), post, "")
|
|
if appErr != nil {
|
|
logger.Error("Failed to delete expired post", mlog.Err(appErr), mlog.String("post_id", post.Id))
|
|
continue
|
|
}
|
|
deletedPostIDs = append(deletedPostIDs, post.Id)
|
|
}
|
|
}
|
|
|
|
deletedPostIDsJSON, err := json.Marshal(deletedPostIDs)
|
|
if err != nil {
|
|
logger.Error("Failed to marshal deleted post IDs", mlog.Err(err))
|
|
return err
|
|
}
|
|
job.Data["deleted_post_ids"] = string(deletedPostIDsJSON)
|
|
return nil
|
|
}
|
|
return jobs.NewSimpleWorker(workerName, jobServer, execute, isEnabled)
|
|
}
|