// Copyright 2019 Yunion // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package service import ( "os" "time" "yunion.io/x/log" "yunion.io/x/pkg/errors" _ "yunion.io/x/sqlchemy/backends" api "yunion.io/x/onecloud/pkg/apis/notify" "yunion.io/x/onecloud/pkg/cloudcommon" "yunion.io/x/onecloud/pkg/cloudcommon/app" "yunion.io/x/onecloud/pkg/cloudcommon/cronman" "yunion.io/x/onecloud/pkg/cloudcommon/db" "yunion.io/x/onecloud/pkg/cloudcommon/db/taskman" common_options "yunion.io/x/onecloud/pkg/cloudcommon/options" "yunion.io/x/onecloud/pkg/notify/models" "yunion.io/x/onecloud/pkg/notify/options" "yunion.io/x/onecloud/pkg/notify/policy" _ "yunion.io/x/onecloud/pkg/notify/sender/smsdriver" _ "yunion.io/x/onecloud/pkg/notify/tasks" ) func StartService() { // parse options opts := &options.Options commonOpts := &options.Options.CommonOptions dbOpts := &options.Options.DBOptions baseOpts := &options.Options.BaseOptions common_options.ParseOptions(opts, os.Args, "notify.conf", api.SERVICE_TYPE) policy.Init() // init auth app.InitAuth(commonOpts, func() { log.Infof("Auth complete!") }) common_options.StartOptionManager(opts, opts.ConfigSyncPeriodSeconds, api.SERVICE_TYPE, api.SERVICE_VERSION, options.OnOptionsChange) // init handler applicaion := app.InitApp(baseOpts, true) cloudcommon.InitDB(dbOpts) InitHandlers(applicaion, opts.IsSlaveNode) // init database db.EnsureAppSyncDB(applicaion, dbOpts, models.InitDB) defer cloudcommon.CloseDB() if !opts.IsSlaveNode { if options.Options.EnableWatchUser { err := models.ReceiverManager.StartWatchUserInKeystone() if err != nil { log.Errorln(errors.Wrap(err, "StartWatchUserInKeystone")) } } err := taskman.TaskManager.InitializeData() if err != nil { log.Fatalf("TaskManager.InitializeData fail %s", err) } cron := cronman.InitCronJobManager(true, opts.CronJobWorkerCount, opts.TimeZone) // update service cron.AddJobAtIntervalsWithStartRun("syncReciverFromKeystone", time.Duration(opts.SyncReceiverIntervalMinutes)*time.Minute, models.ReceiverManager.SyncUserFromKeystone, true) // wrapped func to resend notifications cron.AddJobAtIntervals("ReSendNotifications", time.Duration(opts.ReSendScope)*time.Second, models.NotificationManager.ReSend) cron.AddJobEveryFewHour("AutoPurgeSplitable", 4, 30, 0, db.AutoPurgeSplitable, false) cron.AddJobEveryFewDays("InitReceiverProject", 7, 0, 0, 0, models.InitReceiverProject, true) cron.AddJobAtIntervalsWithStartRun("TaskCleanupJob", time.Duration(options.Options.TaskArchiveIntervalMinutes)*time.Minute, taskman.TaskManager.TaskCleanupJob, true) cron.Start() defer cron.Stop() } app.ServeForever(applicaion, baseOpts) }