Add jobserver command to platform binary (#7067)

* Add jobserver command to platform binary

* Fixing jobserver config. Adding jobserver flags.

* Fixing govet in makefile
Этот коммит содержится в:
Christopher Speller
2017-08-01 05:53:42 -07:00
коммит произвёл Harrison Healey
родитель 88f398ffdd
Коммит 0331409488
5 изменённых файлов: 33 добавлений и 31 удалений

62
cmd/platform/jobserver.go Обычный файл
Просмотреть файл

@@ -0,0 +1,62 @@
// 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/spf13/cobra"
)
var jobserverCmd = &cobra.Command{
Use: "jobserver",
Short: "Start the Mattermost job server",
Run: jobserverCmdF,
}
func init() {
jobserverCmd.Flags().Bool("nojobs", false, "Do not run jobs on this jobserver.")
jobserverCmd.Flags().Bool("noschedule", false, "Do not schedule jobs from this jobserver.")
}
func jobserverCmdF(cmd *cobra.Command, args []string) {
// Options
noJobs, _ := cmd.Flags().GetBool("nojobs")
noSchedule, _ := cmd.Flags().GetBool("noschedule")
// Initialize
utils.InitAndLoadConfig("config.json")
defer l4g.Close()
jobs.Srv.Store = store.NewLayeredStore()
defer jobs.Srv.Store.Close()
jobs.Srv.LoadLicense()
// Run jobs
l4g.Info("Starting Mattermost job server")
if !noJobs {
jobs.Srv.StartWorkers()
}
if !noSchedule {
jobs.Srv.StartSchedulers()
}
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")
jobs.Srv.StopSchedulers()
jobs.Srv.StopWorkers()
l4g.Info("Stopped Mattermost job server")
}

Просмотреть файл

@@ -41,7 +41,7 @@ func init() {
resetCmd.Flags().Bool("confirm", false, "Confirm you really want to delete everything and a DB backup has been performed.")
rootCmd.AddCommand(serverCmd, versionCmd, userCmd, teamCmd, licenseCmd, importCmd, resetCmd, channelCmd, rolesCmd, testCmd, ldapCmd, configCmd)
rootCmd.AddCommand(serverCmd, versionCmd, userCmd, teamCmd, licenseCmd, importCmd, resetCmd, channelCmd, rolesCmd, testCmd, ldapCmd, configCmd, jobserverCmd)
}
var rootCmd = &cobra.Command{

Просмотреть файл

@@ -127,8 +127,12 @@ func runServer(configFileLocation string) {
}
jobs.Srv.Store = app.Srv.Store
jobs.Srv.StartWorkers()
jobs.Srv.StartSchedulers()
if *utils.Cfg.JobSettings.RunJobs {
jobs.Srv.StartWorkers()
}
if *utils.Cfg.JobSettings.RunScheduler {
jobs.Srv.StartSchedulers()
}
// wait for kill signal before attempting to gracefully shutdown
// the running service