Mercurial > gemma
view pkg/imports/ubn.go @ 2750:6446bf6d2a89
search: use global search only in global context
author | Thomas Junk <thomas.junk@intevation.de> |
---|---|
date | Thu, 21 Mar 2019 12:23:52 +0100 |
parents | dc4fae4bdb8f |
children | 4acbee65275d |
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/ifbn" ) type UploadedBottleneck struct { Dir string `json:"dir"` // Tolerance used for axis snapping Tolerance float64 `json:"tolerance"` } // UBNJobKind is the import queue type identifier. const UBNJobKind JobKind = "ubn" type ubnJobCreator struct{} func init() { RegisterJobCreator(UBNJobKind, ubnJobCreator{}) } func (ubnJobCreator) Description() string { return "uploaded bottlenecks" } func (ubnJobCreator) AutoAccept() bool { return false } func (ubnJobCreator) Create() Job { return new(UploadedBottleneck) } func (ubnJobCreator) Depends() []string { // Same as normal bottleneck import. return bnJobCreator{}.Depends() } // StageDone moves the imported bottleneck out of the staging area. func (ubnJobCreator) StageDone( ctx context.Context, tx *sql.Tx, id int64, ) error { // Same as normal bottleneck import. return bnJobCreator{}.StageDone(ctx, tx, id) } // CleanUp of a uploaded bottleneck import removes the temp dir. func (ubn *UploadedBottleneck) CleanUp() error { return os.RemoveAll(ubn.Dir) } // Do executes the actual uploaded bottleneck import. func (ubn *UploadedBottleneck) Do( ctx context.Context, importID int64, conn *sql.Conn, feedback Feedback, ) (interface{}, error) { fetch := func() ([]*ifbn.BottleNeckType, error) { var dst ifbn.Export_bn_by_isrsResponse if err := soap.ValidateFile( filepath.Join(ubn.Dir, "data.xml"), "IFBN.xsd", &dst, ); err != nil { return nil, err } if dst.Export_bn_by_isrsResult == nil { return nil, errors.New("No bottlenecks found") } return dst.Export_bn_by_isrsResult.BottleNeckType, nil } return storeBottlenecks( ctx, fetch, importID, conn, feedback, ubn.Tolerance) }