summaryrefslogtreecommitdiffstats
path: root/jobs/server.go
blob: 6c857e7dc46b1ed5ed6ccc575406ea69bfd349cf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
// 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/mattermost-server/model"
	"github.com/mattermost/mattermost-server/store"
	"github.com/mattermost/mattermost-server/utils"
)

type JobServer struct {
	Store      store.Store
	Workers    *Workers
	Schedulers *Schedulers
}

var Srv JobServer

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"))
	}
}

func (server *JobServer) StartWorkers() {
	Srv.Workers = InitWorkers().Start()
}

func (server *JobServer) StartSchedulers() {
	Srv.Schedulers = InitSchedulers().Start()
}

func (server *JobServer) StopWorkers() {
	if Srv.Workers != nil {
		Srv.Workers.Stop()
	}
}

func (server *JobServer) StopSchedulers() {
	if Srv.Schedulers != nil {
		Srv.Schedulers.Stop()
	}
}