summaryrefslogtreecommitdiffstats
path: root/cmd/platform/message_export.go
diff options
context:
space:
mode:
authorJonathan <jonfritz@gmail.com>2017-11-30 09:07:04 -0500
committerGitHub <noreply@github.com>2017-11-30 09:07:04 -0500
commit375c0632fab03e3fb54865e320585888499c076d (patch)
treece6cba679337a82370d5a730c428bef21a9964bf /cmd/platform/message_export.go
parentd0d9ba4a7e43301697d1c9f495930e0c0179fdc3 (diff)
downloadchat-375c0632fab03e3fb54865e320585888499c076d.tar.gz
chat-375c0632fab03e3fb54865e320585888499c076d.tar.bz2
chat-375c0632fab03e3fb54865e320585888499c076d.zip
PLT-7503: Create Message Export Scheduled Task and CLI Command (#7612)
* Created message export scheduled task * Added CLI command to immediately kick off an export job * Added email addresses for users joining and leaving the channel to the export * Added support for both MySQL and PostgreSQL * Fixing gofmt error * Added a new ChannelMemberHistory store and associated tests * Updating the ChannelMemberHistory channel as users create/join/leave channels * Added user email to the message export object so it can be included in the actiance export xml * Don't fail to log a leave event if a corresponding join event wasn't logged * Adding copyright notices * Adding message export settings to daily diagnostics report * Added System Console integration for message export * Cleaned up TODOs * Made batch size configurable * Added export from timestamp to CLI command * Made ChannelMemberHistory table updates best effort * Added a context-based timeout option to the message export CLI * Minor PR updates/improvements * Removed unnecessary fields from MessageExport object to reduce query overhead * Removed JSON functions from the message export query in an effort to optimize performance * Changed the way that channel member history queries and purges work to better account for edge cases * Fixing a test I missed with the last refactor * Added file copy functionality to file backend, improved config validation, added default config values * Fixed file copy tests * More concise use of the testing libraries * Fixed context leak error * Changed default export path to correctly place an 'export' directory under the 'data' directory * Can't delete records from a read replica * Fixed copy file tests * Start job workers when license is applied, if configured to do so * Suggestions from the PR * Moar unit tests * Fixed test imports
Diffstat (limited to 'cmd/platform/message_export.go')
-rw-r--r--cmd/platform/message_export.go79
1 files changed, 79 insertions, 0 deletions
diff --git a/cmd/platform/message_export.go b/cmd/platform/message_export.go
new file mode 100644
index 000000000..fb1f4073b
--- /dev/null
+++ b/cmd/platform/message_export.go
@@ -0,0 +1,79 @@
+// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package main
+
+import (
+ "errors"
+
+ "context"
+
+ "time"
+
+ "github.com/mattermost/mattermost-server/model"
+ "github.com/spf13/cobra"
+)
+
+var messageExportCmd = &cobra.Command{
+ Use: "export",
+ Short: "Export data from Mattermost",
+ Long: "Export data from Mattermost in a format suitable for import into a third-party application",
+ Example: "export --format=actiance --exportFrom=12345",
+ RunE: messageExportCmdF,
+}
+
+func init() {
+ messageExportCmd.Flags().String("format", "actiance", "The format to export data in")
+ messageExportCmd.Flags().Int64("exportFrom", -1, "The timestamp of the earliest post to export, expressed in seconds since the unix epoch.")
+ messageExportCmd.Flags().Int("timeoutSeconds", -1, "The maximum number of seconds to wait for the job to complete before timing out.")
+}
+
+func messageExportCmdF(cmd *cobra.Command, args []string) error {
+ a, err := initDBCommandContextCobra(cmd)
+ if err != nil {
+ return err
+ }
+
+ if !*a.Config().MessageExportSettings.EnableExport {
+ return errors.New("ERROR: The message export feature is not enabled")
+ }
+
+ // for now, format is hard-coded to actiance. In time, we'll have to support other formats and inject them into job data
+ if format, err := cmd.Flags().GetString("format"); err != nil {
+ return errors.New("format flag error")
+ } else if format != "actiance" {
+ return errors.New("unsupported export format")
+ }
+
+ startTime, err := cmd.Flags().GetInt64("exportFrom")
+ if err != nil {
+ return errors.New("exportFrom flag error")
+ } else if startTime < 0 {
+ return errors.New("exportFrom must be a positive integer")
+ }
+
+ timeoutSeconds, err := cmd.Flags().GetInt("timeoutSeconds")
+ if err != nil {
+ return errors.New("timeoutSeconds error")
+ } else if timeoutSeconds < 0 {
+ return errors.New("timeoutSeconds must be a positive integer")
+ }
+
+ if messageExportI := a.MessageExport; messageExportI != nil {
+ ctx := context.Background()
+ if timeoutSeconds > 0 {
+ var cancel context.CancelFunc
+ ctx, cancel = context.WithTimeout(ctx, time.Second*time.Duration(timeoutSeconds))
+ defer cancel()
+ }
+
+ job, err := messageExportI.StartSynchronizeJob(ctx, startTime)
+ if err != nil || job.Status == model.JOB_STATUS_ERROR || job.Status == model.JOB_STATUS_CANCELED {
+ CommandPrintErrorln("ERROR: Message export job failed. Please check the server logs")
+ } else {
+ CommandPrettyPrintln("SUCCESS: Message export job complete")
+ }
+ }
+
+ return nil
+}