diff pkg/imports/ufa.go @ 2203:8d1a945d0c3b

Uploaded fairway availabilty import: Implemented in terms of normal fairway availabilty import.
author Sascha L. Teichmann <sascha.teichmann@intevation.de>
date Tue, 12 Feb 2019 23:24:09 +0100
parents
children 25f73251a6ac
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/pkg/imports/ufa.go	Tue Feb 12 23:24:09 2019 +0100
@@ -0,0 +1,93 @@
+// 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 (
+	"context"
+	"database/sql"
+	"errors"
+	"os"
+	"path/filepath"
+
+	"gemma.intevation.de/gemma/pkg/soap"
+	"gemma.intevation.de/gemma/pkg/soap/ifaf"
+)
+
+type UploadedFairwayAvailability struct {
+	Dir string
+}
+
+const UFAJobKind JobKind = "ufa"
+
+type ufaJobCreator struct{}
+
+func init() {
+	RegisterJobCreator(UFAJobKind, ufaJobCreator{})
+}
+
+func (ufaJobCreator) Description() string {
+	return "uploaded fairway availability"
+}
+
+func (ufaJobCreator) Create() Job { return new(UploadedFairwayAvailability) }
+
+func (ufaJobCreator) Depends() []string {
+	// Same as faJobCreator
+	return faJobCreator{}.Depends()
+}
+
+func (ufaJobCreator) AutoAccept() bool { return true }
+
+func (ufaJobCreator) StageDone(context.Context, *sql.Tx, int64) error {
+	return nil
+}
+
+func (ufa *UploadedFairwayAvailability) CleanUp() error {
+	return os.RemoveAll(ufa.Dir)
+}
+
+// Do executes the actual uploaded fairway availability import.
+func (ufa *UploadedFairwayAvailability) Do(
+	ctx context.Context,
+	importID int64,
+	conn *sql.Conn,
+	feedback Feedback,
+) (interface{}, error) {
+
+	fetch := func(
+		ctx context.Context,
+		tx *sql.Tx,
+		bns bottlenecks,
+	) ([]*ifaf.FairwayAvailability, error) {
+
+		var response ifaf.Get_bottleneck_faResponse
+
+		if err := soap.ValidateFile(
+			filepath.Join(ufa.Dir, "data.xml"),
+			"IFAF.xsd",
+			&response,
+		); err != nil {
+			return nil, err
+		}
+
+		if response.Get_bottleneck_faResult == nil {
+			return nil, errors.New("No bottlenecks found")
+		}
+
+		result := response.Get_bottleneck_faResult
+		return result.FairwayAvailability, nil
+	}
+
+	return storeFairwayAvailability(ctx, conn, feedback, fetch)
+}