view pkg/imports/scheduled.go @ 1665:da0d1a19ebe6

Fairway availability import: Made schedulable, too.
author Sascha L. Teichmann <sascha.teichmann@intevation.de>
date Sun, 23 Dec 2018 13:30:33 +0100
parents pkg/imports/gmsched.go@51a0ba4ede41
children 56b29406a163
line wrap: on
line source

// This is Free Software under GNU Affero General Public License v >= 3.0
// without warranty, see README.md and license for details.
//
// SPDX-License-Identifier: AGPL-3.0-or-later
// License-Filename: LICENSES/AGPL-3.0.txt
//
// Copyright (C) 2018 by via donau
//   – Österreichische Wasserstraßen-Gesellschaft mbH
// Software engineering by Intevation GmbH
//
// Author(s):
//  * Sascha L. Teichmann <sascha.teichmann@intevation.de>

package imports

import (
	"log"

	"gemma.intevation.de/gemma/pkg/common"
	"gemma.intevation.de/gemma/pkg/scheduler"
)

func init() {
	registerAction(GMJobKind, func(cfg *IDConfig) interface{} {
		log.Println("info: schedule 'gm' import")
		return &GaugeMeasurement{
			URL:      *cfg.URL,
			Insecure: false,
		}
	})
	registerAction(FAJobKind, func(cfg *IDConfig) interface{} {
		log.Println("info: schedule 'fa' import")
		return &FairwayAvailability{
			URL:      *cfg.URL,
			Insecure: false,
		}
	})
}

func registerAction(kind JobKind, setup func(cfg *IDConfig) interface{}) {

	action := func(id int64) {
		cfg, err := loadIDConfig(id)
		if err != nil {
			log.Printf("error: %v\n", err)
			return
		}
		if cfg == nil {
			log.Printf("error: No config found for id %d.\n", id)
			return
		}
		if cfg.URL == nil {
			log.Println("error: No URL specified")
			return
		}

		what := setup(cfg)

		var serialized string
		if serialized, err = common.ToJSONString(what); err != nil {
			log.Printf("error: %v\n", err)
			return
		}

		var jobID int64
		if jobID, err = AddJob(
			kind,
			cfg.User,
			cfg.SendEMail, cfg.AutoAccept,
			serialized,
		); err != nil {
			log.Printf("error: %v\n", err)
			return
		}

		log.Printf("info: added import #%d to queue\n", jobID)
	}

	scheduler.RegisterAction(string(kind), action)
}