Mercurial > gemma
view pkg/imports/ugm.go @ 5736:55892008ec96 default tip
Fixed a bunch of corner cases in WG import.
author | Sascha Wilde <wilde@sha-bang.de> |
---|---|
date | Wed, 29 May 2024 19:02:42 +0200 |
parents | 6270951dda28 |
children |
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 ( "context" "database/sql" "errors" "os" "path/filepath" "gemma.intevation.de/gemma/pkg/soap" "gemma.intevation.de/gemma/pkg/soap/nts" ) // UploadedGaugeMeasurement is an Jon to extract gauge measurement data // from an uploaded XML file and stores it into the database. type UploadedGaugeMeasurement struct { Dir string `json:"dir"` } // UGMJobKind is the unique name of this import job type. const UGMJobKind JobKind = "ugm" type ugmJobCreator struct{} func init() { RegisterJobCreator(UGMJobKind, ugmJobCreator{}) } func (ugmJobCreator) Description() string { return "uploaded gauge measurements" } func (ugmJobCreator) Create() Job { return new(UploadedGaugeMeasurement) } func (ugmJobCreator) Depends() [2][]string { return gmJobCreator{}.Depends() } func (ugmJobCreator) AutoAccept() bool { return true } // StageDone is a NOP for gauge measurements imports. func (ugmJobCreator) StageDone(context.Context, *sql.Tx, int64, Feedback) error { return nil } // CleanUp removes the temporary files from the filesystem. func (ugm *UploadedGaugeMeasurement) CleanUp() error { return os.RemoveAll(ugm.Dir) } // Do executes the actual uploaded gauge measurement import. func (ugm *UploadedGaugeMeasurement) Do( ctx context.Context, importID int64, conn *sql.Conn, feedback Feedback, ) (any, error) { fetch := func() ([]*nts.RIS_Message_Type, error) { var dst nts.Get_messages_result if err := soap.ValidateFile( filepath.Join(ugm.Dir, "data.xml"), "NtS.xsd", &dst, ); err != nil { return nil, err } if len(dst.Result_message) == 0 { return nil, errors.New("no gauge measurements found") } return dst.Result_message, nil } return storeGaugeMeasurements( ctx, importID, fetch, conn, feedback, ) }