Mercurial > gemma
view pkg/controllers/imports.go @ 984:3c9ea8ab856a
Fixed small glitches in sounding result importer.
author | Sascha L. Teichmann <teichmann@intevation.de> |
---|---|
date | Fri, 19 Oct 2018 16:05:46 +0200 |
parents | 7934b5c1a910 |
children | a301d240905f |
line wrap: on
line source
package controllers import ( "bufio" "io" "io/ioutil" "log" "net/http" "os" "path/filepath" "gemma.intevation.de/gemma/pkg/auth" "gemma.intevation.de/gemma/pkg/config" "gemma.intevation.de/gemma/pkg/imports" ) const ( maxSoundingResultSize = 25 * 1024 * 1024 soundingResultName = "soundingresult" ) func downloadSoundingResult(req *http.Request) (string, error) { f, _, err := req.FormFile(soundingResultName) if err != nil { return "", err } defer f.Close() dir, err := ioutil.TempDir(config.TmpDir(), soundingResultName) if err != nil { return "", err } o, err := os.Create(filepath.Join(dir, "sr.zip")) if err != nil { os.RemoveAll(dir) return "", err } out := bufio.NewWriter(o) if _, err = io.Copy(out, io.LimitReader(f, maxSoundingResultSize)); err != nil { o.Close() os.RemoveAll(dir) return "", err } if err = out.Flush(); err != nil { o.Close() os.RemoveAll(dir) return "", err } return dir, nil } func importSoundingResult(rw http.ResponseWriter, req *http.Request) { dir, err := downloadSoundingResult(req) if err != nil { log.Printf("error: %v\n", err) http.Error(rw, "error: "+err.Error(), http.StatusInternalServerError) return } session, _ := auth.GetSession(req) sr := &imports.SoundingResult{ Who: session.User, Dir: dir, } jobID := imports.AddJob(sr) log.Printf("Added job %d\n", jobID) result := struct { ID int64 `json:"id"` }{ ID: jobID, } SendJSON(rw, http.StatusCreated, &result) }