Mercurial > gemma
view pkg/imports/ufa.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/ifaf" ) // UploadedFairwayAvailability is Job to extract // fairway availability data from an XML file and stores // it into the database. type UploadedFairwayAvailability struct { Dir string } // UFAJobKind is the unique name of this import job type. 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() [2][]string { // Same as faJobCreator return faJobCreator{}.Depends() } func (ufaJobCreator) AutoAccept() bool { return true } // StageDone is a NOP for fairway availability imports. func (ufaJobCreator) StageDone(context.Context, *sql.Tx, int64, Feedback) error { return nil } // CleanUp removes the temporary files from the filesystem. 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, _ int64, conn *sql.Conn, feedback Feedback, ) (any, error) { fetch := func( _ context.Context, _ *sql.Tx, _ 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 } result := response.Get_bottleneck_faResult if result == nil { return nil, errors.New("no bottlenecks found") } return result.FairwayAvailability, nil } return storeFairwayAvailability(ctx, conn, feedback, fetch) }