Mercurial > gemma
view pkg/controllers/geostyling.go @ 5718:3d497077f888 uploadwg
Implemented direct file upload as alternative import method for WG.
For testing and data corrections it is useful to be able to import
waterway gauges data directly by uploading a xml file.
author | Sascha Wilde <wilde@sha-bang.de> |
---|---|
date | Thu, 18 Apr 2024 19:23:19 +0200 |
parents | 5f47eeea988d |
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 controllers import ( "bytes" "fmt" "io" "net/http" "github.com/gorilla/mux" "gemma.intevation.de/gemma/pkg/geoserver" "gemma.intevation.de/gemma/pkg/log" "gemma.intevation.de/gemma/pkg/models" ) const ( maxStyleSize = 5 * 1024 * 1024 styleName = "style" ) func extractStyle(req *http.Request) ([]byte, error) { f, _, err := req.FormFile(styleName) if err != nil { return nil, err } defer f.Close() var buf bytes.Buffer if _, err := io.Copy(&buf, io.LimitReader(f, maxStyleSize)); err != nil { return nil, err } return buf.Bytes(), nil } func supportedWMSFeature(name string) bool { return len(models.InternalServices.Filter( models.IntAnd( models.IntWMS, models.IntByName(name)))) > 0 } func uploadStyle(rw http.ResponseWriter, req *http.Request) { feature := mux.Vars(req)["feature"] // only allow internal WMS features if !supportedWMSFeature(feature) { http.Error(rw, fmt.Sprintf("WMS feature %s is not found.", feature), http.StatusNotFound) return } style, err := extractStyle(req) if err != nil { log.Errorf("%v\n", err) http.Error(rw, "error: "+err.Error(), http.StatusBadRequest) return } log.Infof("uploaded file length: %d\n", len(style)) if err := models.UpdateInternalStyle(req, feature, style); err != nil { log.Errorf("%v\n", err) http.Error(rw, "error: "+err.Error(), http.StatusInternalServerError) return } geoserver.ReconfigureStyle(feature) // Nothing to return rw.WriteHeader(http.StatusNoContent) }