summaryrefslogtreecommitdiffstats
path: root/jobs
diff options
context:
space:
mode:
Diffstat (limited to 'jobs')
-rw-r--r--jobs/jobs.go74
-rw-r--r--jobs/jobserver/jobserver.go46
-rw-r--r--jobs/server.go46
-rw-r--r--jobs/testjob.go54
4 files changed, 220 insertions, 0 deletions
diff --git a/jobs/jobs.go b/jobs/jobs.go
new file mode 100644
index 000000000..8c84f4eea
--- /dev/null
+++ b/jobs/jobs.go
@@ -0,0 +1,74 @@
+// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package jobs
+
+import (
+ "sync"
+
+ l4g "github.com/alecthomas/log4go"
+ ejobs "github.com/mattermost/platform/einterfaces/jobs"
+ "github.com/mattermost/platform/model"
+ "github.com/mattermost/platform/store"
+ "github.com/mattermost/platform/utils"
+)
+
+type Jobs struct {
+ startOnce sync.Once
+
+ DataRetention model.Job
+ // SearchIndexing model.Job
+
+ listenerId string
+}
+
+func InitJobs(s store.Store) *Jobs {
+ jobs := &Jobs{
+ // SearchIndexing: MakeTestJob(s, "SearchIndexing"),
+ }
+
+ if dataRetentionInterface := ejobs.GetDataRetentionInterface(); dataRetentionInterface != nil {
+ jobs.DataRetention = dataRetentionInterface.MakeJob(s)
+ }
+
+ return jobs
+}
+
+func (jobs *Jobs) Start() *Jobs {
+ l4g.Info("Starting jobs")
+
+ jobs.startOnce.Do(func() {
+ if jobs.DataRetention != nil && *utils.Cfg.DataRetentionSettings.Enable {
+ go jobs.DataRetention.Run()
+ }
+
+ // go jobs.SearchIndexing.Run()
+ })
+
+ jobs.listenerId = utils.AddConfigListener(jobs.handleConfigChange)
+
+ return jobs
+}
+
+func (jobs *Jobs) handleConfigChange(oldConfig *model.Config, newConfig *model.Config) {
+ if jobs.DataRetention != nil {
+ if !*oldConfig.DataRetentionSettings.Enable && *newConfig.DataRetentionSettings.Enable {
+ go jobs.DataRetention.Run()
+ } else if *oldConfig.DataRetentionSettings.Enable && !*newConfig.DataRetentionSettings.Enable {
+ jobs.DataRetention.Stop()
+ }
+ }
+}
+
+func (jobs *Jobs) Stop() *Jobs {
+ utils.RemoveConfigListener(jobs.listenerId)
+
+ if jobs.DataRetention != nil && *utils.Cfg.DataRetentionSettings.Enable {
+ jobs.DataRetention.Stop()
+ }
+ // jobs.SearchIndexing.Stop()
+
+ l4g.Info("Stopped jobs")
+
+ return jobs
+}
diff --git a/jobs/jobserver/jobserver.go b/jobs/jobserver/jobserver.go
new file mode 100644
index 000000000..813676b78
--- /dev/null
+++ b/jobs/jobserver/jobserver.go
@@ -0,0 +1,46 @@
+// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package main
+
+import (
+ "os"
+ "os/signal"
+ "syscall"
+
+ l4g "github.com/alecthomas/log4go"
+ "github.com/mattermost/platform/jobs"
+ "github.com/mattermost/platform/store"
+ "github.com/mattermost/platform/utils"
+
+ _ "github.com/mattermost/platform/imports"
+)
+
+var Srv jobs.JobServer
+
+func main() {
+ // Initialize
+ utils.InitAndLoadConfig("config.json")
+ defer l4g.Close()
+
+ Srv.Store = store.NewSqlStore()
+ defer Srv.Store.Close()
+
+ Srv.LoadLicense()
+
+ // Run jobs
+ l4g.Info("Starting Mattermost job server")
+ Srv.Jobs = jobs.InitJobs(Srv.Store)
+ Srv.Jobs.Start()
+
+ var signalChan chan os.Signal = make(chan os.Signal)
+ signal.Notify(signalChan, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
+ <-signalChan
+
+ // Cleanup anything that isn't handled by a defer statement
+ l4g.Info("Stopping Mattermost job server")
+
+ Srv.Jobs.Stop()
+
+ l4g.Info("Stopped Mattermost job server")
+}
diff --git a/jobs/server.go b/jobs/server.go
new file mode 100644
index 000000000..dd3448842
--- /dev/null
+++ b/jobs/server.go
@@ -0,0 +1,46 @@
+// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package jobs
+
+import (
+ l4g "github.com/alecthomas/log4go"
+ "github.com/mattermost/platform/model"
+ "github.com/mattermost/platform/store"
+ "github.com/mattermost/platform/utils"
+)
+
+type JobServer struct {
+ Store store.Store
+ Jobs *Jobs
+}
+
+func (server *JobServer) LoadLicense() {
+ licenseId := ""
+ if result := <-server.Store.System().Get(); result.Err == nil {
+ props := result.Data.(model.StringMap)
+ licenseId = props[model.SYSTEM_ACTIVE_LICENSE_ID]
+ }
+
+ var licenseBytes []byte
+
+ if len(licenseId) != 26 {
+ // Lets attempt to load the file from disk since it was missing from the DB
+ _, licenseBytes = utils.GetAndValidateLicenseFileFromDisk()
+ } else {
+ if result := <-server.Store.License().Get(licenseId); result.Err == nil {
+ record := result.Data.(*model.LicenseRecord)
+ licenseBytes = []byte(record.Bytes)
+ l4g.Info("License key valid unlocking enterprise features.")
+ } else {
+ l4g.Info(utils.T("mattermost.load_license.find.warn"))
+ }
+ }
+
+ if licenseBytes != nil {
+ utils.LoadLicense(licenseBytes)
+ l4g.Info("License key valid unlocking enterprise features.")
+ } else {
+ l4g.Info(utils.T("mattermost.load_license.find.warn"))
+ }
+}
diff --git a/jobs/testjob.go b/jobs/testjob.go
new file mode 100644
index 000000000..59d5274e5
--- /dev/null
+++ b/jobs/testjob.go
@@ -0,0 +1,54 @@
+// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package jobs
+
+import (
+ "time"
+
+ l4g "github.com/alecthomas/log4go"
+ "github.com/mattermost/platform/store"
+)
+
+type TestJob struct {
+ store store.Store
+
+ name string
+ stop chan bool
+ stopped chan bool
+}
+
+func MakeTestJob(s store.Store, name string) *TestJob {
+ return &TestJob{
+ store: s,
+ name: name,
+ stop: make(chan bool, 1),
+ stopped: make(chan bool, 1),
+ }
+}
+
+func (job *TestJob) Run() {
+ l4g.Debug("Job %v: Started", job.name)
+
+ running := true
+ for running {
+ l4g.Debug("Job %v: Tick", job.name)
+
+ select {
+ case <-job.stop:
+ l4g.Debug("Job %v: Received stop signal", job.name)
+ running = false
+ case <-time.After(10 * time.Second):
+ continue
+ }
+ }
+
+ l4g.Debug("Job %v: Finished", job.name)
+ job.stopped <- true
+}
+
+func (job *TestJob) Stop() {
+ l4g.Debug("Job %v: Stopping", job.name)
+ job.stop <- true
+ <-job.stopped
+}